temp [ci skip]

Author: benfdking

sqlmesh/lsp/api.py

@@ -54,3 +54,10 @@ def __init__(self, **data):
                             setattr(model.details, field, None)
 
         super().__init__(**data)
+
+class ApiResponseGetLineage(ApiResponse):
+    """
+    Response from the SQLMesh API for the get_lineage endpoint.
+    Specifies the data type more precisely as a list of models.
+    """
+    data: t.Dict[str, t.List[str]]

sqlmesh/lsp/main.py

@@ -11,11 +11,12 @@
 from sqlmesh._version import __version__
 from sqlmesh.core.context import Context
 from sqlmesh.core.linter.definition import AnnotatedRuleViolation
-from sqlmesh.lsp.api import API_FEATURE, ApiRequest, ApiResponse, ApiResponseGetModels
+from sqlmesh.lsp.api import API_FEATURE, ApiRequest, ApiResponse, ApiResponseGetLineage, ApiResponseGetModels
 from sqlmesh.lsp.completions import get_sql_completions
 from sqlmesh.lsp.context import LSPContext
 from sqlmesh.lsp.custom import ALL_MODELS_FEATURE, AllModelsRequest, AllModelsResponse
 from sqlmesh.lsp.reference import get_model_definitions_for_a_path
+from web.server.api.endpoints.lineage import model_lineage
 from web.server.api.endpoints.models import get_models
 
 
@@ -60,7 +61,13 @@ def api(ls: LanguageServer, request: ApiRequest) -> ApiResponse:
             if request.url == "/api/models":
                 response = ApiResponseGetModels(data=get_models(self.lsp_context.context))
                 return response 
-            return ApiResponse(data={})
+            elif request.url.startswith("/api/lineage"):
+                name = request.url.split("/")[-1]
+                lineage = model_lineage(name, self.lsp_context.context)
+                non_set_lineage = {k: v for k, v in lineage.items() if v is not None}
+                response = ApiResponseGetLineage(data=non_set_lineage)
+                return response
+            raise NotImplementedError(f"API request not implemented: {request.url}")
 
         @self.server.feature(types.TEXT_DOCUMENT_DID_OPEN)
         def did_open(

vscode/react/src/api/instance.ts

@@ -92,6 +92,8 @@ async function fetchAPIDevelopment<T = any, B extends object = any>(
           requestId,
           url: config.url,
           params: config.params as any,
+          data: config.data,
+          method: config.method,
         },
       }, '*')
       return

web/server/api/endpoints/lineage.py

@@ -146,7 +146,7 @@ async def column_lineage(
 
 
 @router.get("/{model_name:str}")
-async def model_lineage(
+def model_lineage(
     model_name: str,
     context: Context = Depends(get_loaded_context),
 ) -> t.Dict[str, t.Set[str]]: