chore(vscode): reworking lsp and reloading on save

[ci skip]

Author: benfdking

sqlmesh/lsp/main.py

@@ -199,6 +199,13 @@ 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:
+                old_context = self.lsp_context.context
+                old_context.load()  # Reload the context
+                self.lsp_context = LSPContext(old_context)
+
             context = self._context_get_or_load(uri)
             models = context.map[uri.to_path()]
             if models is None or not isinstance(models, ModelTarget):