sqlmesh 0.181.2.dev7__py3-none-any.whl → 0.181.2.dev8__py3-none-any.whl

sqlmesh/_version.py

@@ -17,5 +17,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.181.2.dev7'
-__version_tuple__ = version_tuple = (0, 181, 2, 'dev7')
+__version__ = version = '0.181.2.dev8'
+__version_tuple__ = version_tuple = (0, 181, 2, 'dev8')

sqlmesh/lsp/api.py

@@ -0,0 +1,61 @@
+"""
+This module maps the LSP custom API calls to the SQLMesh web api.
+
+Allowing the LSP to call the web api without having to know the details of the web api
+and thus passing through the details of the web api to the LSP, so that both the LSP
+and the web api can communicate with the same process, avoiding the need to have a
+separate process for the web api.
+"""
+
+import typing as t
+from pydantic import field_validator
+from sqlmesh.utils.pydantic import PydanticModel
+from web.server.models import Model
+
+API_FEATURE = "sqlmesh/api"
+
+
+class ApiRequest(PydanticModel):
+    """
+    Request to call the SQLMesh API.
+    This is a generic request that can be used to call any API endpoint.
+    """
+
+    requestId: str
+    url: str
+    method: t.Optional[str] = "GET"
+    params: t.Optional[t.Dict[str, t.Any]] = None
+    body: t.Optional[t.Dict[str, t.Any]] = None
+
+
+class ApiResponseGetModels(PydanticModel):
+    """
+    Response from the SQLMesh API for the get_models endpoint.
+    """
+
+    data: t.List[Model]
+
+    @field_validator("data", mode="before")
+    def sanitize_datetime_fields(cls, data: t.List[Model]) -> t.List[Model]:
+        """
+        Convert datetime objects to None to avoid serialization issues.
+        """
+        if isinstance(data, list):
+            for model in data:
+                if hasattr(model, "details") and model.details:
+                    # Convert datetime fields to None to avoid serialization issues
+                    for field in ["stamp", "start", "cron_prev", "cron_next"]:
+                        if (
+                            hasattr(model.details, field)
+                            and getattr(model.details, field) is not None
+                        ):
+                            setattr(model.details, field, None)
+        return data
+
+
+class ApiResponseGetLineage(PydanticModel):
+    """
+    Response from the SQLMesh API for the get_lineage endpoint.
+    """
+
+    data: t.Dict[str, t.List[str]]

sqlmesh/lsp/main.py

@@ -11,12 +11,20 @@ from pygls.server import LanguageServer
 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,
+    ApiResponseGetLineage,
+    ApiResponseGetModels,
+)
 from sqlmesh.lsp.completions import get_sql_completions
 from sqlmesh.lsp.context import LSPContext, ModelTarget
 from sqlmesh.lsp.custom import ALL_MODELS_FEATURE, AllModelsRequest, AllModelsResponse
 from sqlmesh.lsp.reference import (
     get_references,
 )
+from web.server.api.endpoints.lineage import model_lineage
+from web.server.api.endpoints.models import get_models
 
 
 class SQLMeshLanguageServer:
@@ -79,6 +87,23 @@ class SQLMeshLanguageServer:
             except Exception as e:
                 return get_sql_completions(None, params.textDocument.uri)
 
+        @self.server.feature(API_FEATURE)
+        def api(
+            ls: LanguageServer, request: ApiRequest
+        ) -> t.Union[ApiResponseGetModels, ApiResponseGetLineage]:
+            ls.log_trace(f"API request: {request}")
+            if self.lsp_context is None:
+                raise RuntimeError("No context found")
+            if request.url == "/api/models":
+                response = ApiResponseGetModels(data=get_models(self.lsp_context.context))
+                return response
+            if 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}
+                return ApiResponseGetLineage(data=non_set_lineage)
+            raise NotImplementedError(f"API request not implemented: {request.url}")
+
         @self.server.feature(types.TEXT_DOCUMENT_DID_OPEN)
         def did_open(ls: LanguageServer, params: types.DidOpenTextDocumentParams) -> None:
             context = self._context_get_or_load(params.text_document.uri)

{sqlmesh-0.181.2.dev7.dist-info → sqlmesh-0.181.2.dev8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sqlmesh
-Version: 0.181.2.dev7
+Version: 0.181.2.dev8
 Summary: Next-generation data transformation framework
 Author-email: "TobikoData Inc." <engineering@tobikodata.com>
 License:                                  Apache License

{sqlmesh-0.181.2.dev7.dist-info → sqlmesh-0.181.2.dev8.dist-info}/RECORD

@@ -1,5 +1,5 @@
 sqlmesh/__init__.py,sha256=DRovDXI-9D9xeYDP7t8Fq7mI31S2btlN-LHPOK3y7PA,7240
-sqlmesh/_version.py,sha256=8r2JEsifNarsOVqbo9kwcaq4X7phTueVMYWzJc5SnMk,528
+sqlmesh/_version.py,sha256=m-E-GGmR7GxW8S19G7ZHHmWmKoUBoqjIUDAYPE33SYI,528
 sqlmesh/magics.py,sha256=9oY1Uz8uwsuTtPRg9HTYDxnGiJ7-OVRCpmCgAhbKHUU,39431
 sqlmesh/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sqlmesh/cicd/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -155,10 +155,11 @@ sqlmesh/integrations/github/cicd/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRk
 sqlmesh/integrations/github/cicd/command.py,sha256=a48qtSeRM7RD3IeydJwHLAZBHRbmFdA7uHh7Yhe5a74,11301
 sqlmesh/integrations/github/cicd/config.py,sha256=qEy0mRfPYJw1pm68UK0v1_4kZBj5QodnpIsOZ8JjZro,1838
 sqlmesh/integrations/github/cicd/controller.py,sha256=AiEMsRW4WSVxXP_7oQSIdzBzmYPxfYDgYeOYRk9sfJA,41388
+sqlmesh/lsp/api.py,sha256=Ls7BjrPSovvOu6VU_eFR0_nHcbsQl020eJvGbdWi7s8,1957
 sqlmesh/lsp/completions.py,sha256=ScW0H9couhfW6QckEX2djVYTClRPXpT2Hdcg1qLfUUY,3479
 sqlmesh/lsp/context.py,sha256=vUpbWYjKYuwDvU3zkzhAgw9An-TeVYe_5KTk-ec_UwM,1569
 sqlmesh/lsp/custom.py,sha256=GvNAiW7imPI8xmIx3qLWN5FljGPHFJKh0N5b6aZnwRo,489
-sqlmesh/lsp/main.py,sha256=HBbJMtWF98yZiTHspdx0hfyGnsMTh2WSirgYvvOwRs4,15204
+sqlmesh/lsp/main.py,sha256=AHVkfOvmDh3NqgfGF41uAnlxUDZdFyx5HEpGgCA_F1o,16346
 sqlmesh/lsp/reference.py,sha256=3vmLpEDbsJ740DfaOc2_ESZnXsUxWLS4lUunoO9km9s,8121
 sqlmesh/migrations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sqlmesh/migrations/v0001_init.py,sha256=h8FwMYJj9_O8yGpk12guaSaGwX5A3we2GiKeHrXi15o,2048
@@ -265,7 +266,7 @@ sqlmesh/utils/pandas.py,sha256=j_tYhuPcWpzuPceZZ6IhdTVbmxYyGJW7l4FHr4bQqjM,1824
 sqlmesh/utils/pydantic.py,sha256=ioaAzYF42PutEJRyAS11-AHguiCG3yWI6VRcTVa2G-g,10470
 sqlmesh/utils/rich.py,sha256=xNZyPCqEc4TAlJlsEthg-5eYUWpaizz0QjBzO7a1Yv4,1509
 sqlmesh/utils/yaml.py,sha256=np-WVcwmwsKghCgAj1OBBMsK1SXSRn4wR-H37cGiFr8,2282
-sqlmesh-0.181.2.dev7.dist-info/licenses/LICENSE,sha256=OlMefUjgWJdULtf84BLW0AZZcY8DwdgQqb_1j2862j8,11346
+sqlmesh-0.181.2.dev8.dist-info/licenses/LICENSE,sha256=OlMefUjgWJdULtf84BLW0AZZcY8DwdgQqb_1j2862j8,11346
 web/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 web/client/dist/index.html,sha256=cQRIlDgkl6OgeWuHI9hM8bJhKUE9-Ji2NpQa2hlgNM8,5472
 web/client/dist/assets/Audits-4U9LnWyN.js,sha256=reXu-sKJ76kiaAZIGkCPvgO68Km9tj1PN_9mx6X6WAY,997
@@ -382,8 +383,8 @@ web/server/api/endpoints/models.py,sha256=SzS5bshzdXyM6hb948sZ_3k-xhpGCHkxkb7pHz
 web/server/api/endpoints/modules.py,sha256=8hqqgonGay_mJmpCw0IdbjsPhWlQH2VLdKAqha-myac,468
 web/server/api/endpoints/plan.py,sha256=OYuQ_4dgbhG9ibTcX1a8mvYgH3ANxw9Fhbb4wg6QWqc,9222
 web/server/api/endpoints/table_diff.py,sha256=Wd5YxeKbZB7AkrDvqFgvV9ZQlnKYOF1n5W_w2EHE_Zw,2085
-sqlmesh-0.181.2.dev7.dist-info/METADATA,sha256=uAqPKGwBv0r6245w3anejNksYKPIkqOPpUwtrvx1G58,25307
-sqlmesh-0.181.2.dev7.dist-info/WHEEL,sha256=DnLRTWE75wApRYVsjgc6wsVswC54sMSJhAEd4xhDpBk,91
-sqlmesh-0.181.2.dev7.dist-info/entry_points.txt,sha256=pIYKtg-u-uO7BK_Y0qrdmfyj0bYrcMc3Xiqr5WE3VFk,121
-sqlmesh-0.181.2.dev7.dist-info/top_level.txt,sha256=bspjo5vjELQkuN0qKuTFIbsrUt3Zeo8D2mlKeHG1LdE,12
-sqlmesh-0.181.2.dev7.dist-info/RECORD,,
+sqlmesh-0.181.2.dev8.dist-info/METADATA,sha256=oXV47yFT5VeGOg2gTitD8hp1TmJRiAIsPKFaQ3U_wc0,25307
+sqlmesh-0.181.2.dev8.dist-info/WHEEL,sha256=DnLRTWE75wApRYVsjgc6wsVswC54sMSJhAEd4xhDpBk,91
+sqlmesh-0.181.2.dev8.dist-info/entry_points.txt,sha256=pIYKtg-u-uO7BK_Y0qrdmfyj0bYrcMc3Xiqr5WE3VFk,121
+sqlmesh-0.181.2.dev8.dist-info/top_level.txt,sha256=bspjo5vjELQkuN0qKuTFIbsrUt3Zeo8D2mlKeHG1LdE,12
+sqlmesh-0.181.2.dev8.dist-info/RECORD,,