chore(lsp): move autocomplete to the lsp

Author: benfdking

sqlmesh/lsp/main.py

@@ -444,6 +444,50 @@ def workspace_diagnostic(
                 )
                 return types.WorkspaceDiagnosticReport(items=[])
 
+        @self.server.feature(types.TEXT_DOCUMENT_COMPLETION)
+        def completion(
+            ls: LanguageServer, params: types.CompletionParams
+        ) -> t.Optional[types.CompletionList]:
+            """Handle completion requests from the client."""
+            try:
+                uri = URI(params.text_document.uri)
+                context = self._context_get_or_load(uri)
+                
+                # Get completions using the existing completions module
+                completion_response = get_sql_completions(context, uri)
+                
+                # Convert to LSP completion items
+                completion_items = []
+                
+                # Add model completions
+                for model in completion_response.models:
+                    completion_items.append(
+                        types.CompletionItem(
+                            label=model,
+                            kind=types.CompletionItemKind.Reference,
+                            detail="SQLMesh Model",
+                        )
+                    )
+                
+                # Add keyword completions
+                for keyword in completion_response.keywords:
+                    completion_items.append(
+                        types.CompletionItem(
+                            label=keyword,
+                            kind=types.CompletionItemKind.Keyword,
+                            detail="SQL Keyword",
+                        )
+                    )
+                
+                return types.CompletionList(
+                    is_incomplete=False,
+                    items=completion_items,
+                )
+                
+            except Exception as e:
+                ls.show_message(f"Error getting completions: {e}", types.MessageType.Error)
+                return None
+
     def _get_diagnostics_for_uri(self, uri: URI) -> t.Tuple[t.List[types.Diagnostic], int]:
         """Get diagnostics for a specific URI, returning (diagnostics, result_id)."""
         # Check if we have cached diagnostics

vscode/extension/src/extension.ts

@@ -83,6 +83,9 @@ export async function activate(context: vscode.ExtensionContext) {
     ),
   )
 
+  // Register custom completion provider as a fallback
+  // The VS Code language client will automatically use LSP completions if the server supports them
+  // This custom provider will only be used if the server doesn't support native completions
   context.subscriptions.push(
     vscode.languages.registerCompletionItemProvider(
       selector,