chore(vscode): reworking lsp and reloading on save

Author: benfdking

sqlmesh/lsp/main.py

@@ -199,6 +199,17 @@ def did_change(ls: LanguageServer, params: types.DidChangeTextDocumentParams) ->
         @self.server.feature(types.TEXT_DOCUMENT_DID_SAVE)
         def did_save(ls: LanguageServer, params: types.DidSaveTextDocumentParams) -> None:
             uri = URI(params.text_document.uri)
+
+            # Reload the entire context and create a new LSPContext
+            if self.lsp_context is not None:
+                try:
+                    new_context = Context(paths=list(self.lsp_context.context.configs.keys()))
+                    new_full_cntext = LSPContext(new_context)
+                    self.lsp_context = new_full_cntext
+                    return
+                except Exception as e:
+                    pass
+
             context = self._context_get_or_load(uri)
             models = context.map[uri.to_path()]
             if models is None or not isinstance(models, ModelTarget):