sqlmesh 0.184.2.dev1__py3-none-any.whl → 0.184.2.dev2__py3-none-any.whl

sqlmesh/_version.py

@@ -17,5 +17,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.184.2.dev1'
-__version_tuple__ = version_tuple = (0, 184, 2, 'dev1')
+__version__ = version = '0.184.2.dev2'
+__version_tuple__ = version_tuple = (0, 184, 2, 'dev2')

sqlmesh/lsp/api.py

@@ -10,7 +10,7 @@ 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
+from web.server.models import LineageColumn, Model
 
 API_FEATURE = "sqlmesh/api"
 
@@ -59,3 +59,11 @@ class ApiResponseGetLineage(PydanticModel):
     """
 
     data: t.Dict[str, t.List[str]]
+
+
+class ApiResponseGetColumnLineage(PydanticModel):
+    """
+    Response from the SQLMesh API for the get_column_lineage endpoint.
+    """
+
+    data: t.Dict[str, t.Dict[str, LineageColumn]]

sqlmesh/lsp/main.py

@@ -4,6 +4,7 @@
 import logging
 import typing as t
 from pathlib import Path
+import urllib.parse
 
 from lsprotocol import types
 from pygls.server import LanguageServer
@@ -14,6 +15,7 @@ from sqlmesh.core.linter.definition import AnnotatedRuleViolation
 from sqlmesh.lsp.api import (
     API_FEATURE,
     ApiRequest,
+    ApiResponseGetColumnLineage,
     ApiResponseGetLineage,
     ApiResponseGetModels,
 )
@@ -24,7 +26,7 @@ from sqlmesh.lsp.reference import (
     get_references,
 )
 from sqlmesh.lsp.uri import URI
-from web.server.api.endpoints.lineage import model_lineage
+from web.server.api.endpoints.lineage import column_lineage, model_lineage
 from web.server.api.endpoints.models import get_models
 
 
@@ -92,18 +94,37 @@ class SQLMeshLanguageServer:
         @self.server.feature(API_FEATURE)
         def api(
             ls: LanguageServer, request: ApiRequest
-        ) -> t.Union[ApiResponseGetModels, ApiResponseGetLineage]:
+        ) -> t.Union[ApiResponseGetModels, ApiResponseGetLineage, ApiResponseGetColumnLineage]:
             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)
+
+            parsed_url = urllib.parse.urlparse(request.url)
+            path_parts = parsed_url.path.strip("/").split("/")
+
+            if request.method == "GET":
+                if path_parts == ["api", "models"]:
+                    # /api/models
+                    return ApiResponseGetModels(data=get_models(self.lsp_context.context))
+
+                if path_parts[:2] == ["api", "lineage"]:
+                    if len(path_parts) == 3:
+                        # /api/lineage/{model}
+                        model_name = urllib.parse.unquote(path_parts[2])
+                        lineage = model_lineage(model_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)
+
+                    if len(path_parts) == 4:
+                        # /api/lineage/{model}/{column}
+                        model_name = urllib.parse.unquote(path_parts[2])
+                        column = urllib.parse.unquote(path_parts[3])
+                        logging.info(f"Column lineage request: {model_name} {column}")
+                        column_lineage_response = column_lineage(
+                            model_name, column, False, self.lsp_context.context
+                        )
+                        return ApiResponseGetColumnLineage(data=column_lineage_response)
+
             raise NotImplementedError(f"API request not implemented: {request.url}")
 
         @self.server.feature(types.TEXT_DOCUMENT_DID_OPEN)

{sqlmesh-0.184.2.dev1.dist-info → sqlmesh-0.184.2.dev2.dist-info}/METADATA

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

{sqlmesh-0.184.2.dev1.dist-info → sqlmesh-0.184.2.dev2.dist-info}/RECORD

@@ -1,5 +1,5 @@
 sqlmesh/__init__.py,sha256=DRovDXI-9D9xeYDP7t8Fq7mI31S2btlN-LHPOK3y7PA,7240
-sqlmesh/_version.py,sha256=J69cUN88BHtqUGrpQeggDRwXVan4Pdnn3EHekxw9Vm4,528
+sqlmesh/_version.py,sha256=K1tcpWduhkaFG48YpByt7_MBd5gAWPAjrzyTLfzW7Wc,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,11 +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/api.py,sha256=LJdECSaEZoPbu_Xl4KscVG-WLt7_g4dY-zKGO68dH14,2162
 sqlmesh/lsp/completions.py,sha256=Uo8lUzPBeoB58Eu2642YoAtNI0pgIFWFSvwmuJALTv4,3583
 sqlmesh/lsp/context.py,sha256=Ff09vrsawdYS-7YcXbGZW_RU-tj_-k8WV7maCghpYp0,1438
 sqlmesh/lsp/custom.py,sha256=GvNAiW7imPI8xmIx3qLWN5FljGPHFJKh0N5b6aZnwRo,489
-sqlmesh/lsp/main.py,sha256=cGh2AtCvmHPYV4uCBAVVfxZgtq50W6UmAaUCsMyYYoc,16011
+sqlmesh/lsp/main.py,sha256=EgYQRk8s4W4PaIzUkiEKBKkXxkeWNvGLLQHDxJIPDSs,17024
 sqlmesh/lsp/reference.py,sha256=MiTnNmaGrVs1Scms3jHkZ89Cu8_ZLJGAHcs1LRywFdU,8129
 sqlmesh/lsp/uri.py,sha256=mvSLbZGmMTHPpliThOtVWiFcHnfJzp1w39q9l-xD2SI,802
 sqlmesh/migrations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -267,7 +267,7 @@ sqlmesh/utils/pandas.py,sha256=uKI-ygEtMcGDRzXdBu0RkGmMBkYxwu5O2Wabho841Ls,2196
 sqlmesh/utils/pydantic.py,sha256=-u69PDY-4FLR7ODNKBoN7PsPNmZ0wdhsJcCAG2P6vyY,11061
 sqlmesh/utils/rich.py,sha256=xNZyPCqEc4TAlJlsEthg-5eYUWpaizz0QjBzO7a1Yv4,1509
 sqlmesh/utils/yaml.py,sha256=np-WVcwmwsKghCgAj1OBBMsK1SXSRn4wR-H37cGiFr8,2282
-sqlmesh-0.184.2.dev1.dist-info/licenses/LICENSE,sha256=OlMefUjgWJdULtf84BLW0AZZcY8DwdgQqb_1j2862j8,11346
+sqlmesh-0.184.2.dev2.dist-info/licenses/LICENSE,sha256=OlMefUjgWJdULtf84BLW0AZZcY8DwdgQqb_1j2862j8,11346
 web/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 web/client/dist/index.html,sha256=f4Pg8Cex2Yzuh8nwt5XduYLO41HKza9RIL0rW1d9wKw,5472
 web/client/dist/assets/Audits-DF7amJxU.js,sha256=VxuJ7bKHmPaqK8SvZaqdeSB_m9-oI7fi6HND54mJ4Y0,1041
@@ -379,14 +379,14 @@ web/server/api/endpoints/directories.py,sha256=Aa8U4pTs3Saj5DnJ9TDnX9mZois-EXgpc
 web/server/api/endpoints/environments.py,sha256=g3YC6j4LLlG_wjJqn-Hrz78BWmgS7vhUhzaycsiez00,2124
 web/server/api/endpoints/events.py,sha256=tjhOhGe9jt7L1mOLLapl5qTvicDRnP5lPZHSq9OnSVI,932
 web/server/api/endpoints/files.py,sha256=UnZPScF-HANNNZhNFhxfAPihcU3p85mB7P2rqFc2wG4,6698
-web/server/api/endpoints/lineage.py,sha256=RdK1QmNhgVU6dkfEcgdSLnko1SkGQhJP_IgeDWRJ4vM,5775
+web/server/api/endpoints/lineage.py,sha256=re2mNADlUQsBZVeA7jKw857mQvMxwijynP5iVB7YAnY,5769
 web/server/api/endpoints/meta.py,sha256=QKWWrHewGh5JS8mCSWXTKCIQwXw-TQfaT9tWt_SsNY4,1032
 web/server/api/endpoints/models.py,sha256=SzS5bshzdXyM6hb948sZ_3k-xhpGCHkxkb7pHzPjM_o,4675
 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.184.2.dev1.dist-info/METADATA,sha256=wrEfzrsZSWnACw61vDx3t02IDyKkOmP6wB-9_UvkJDw,25498
-sqlmesh-0.184.2.dev1.dist-info/WHEEL,sha256=Nw36Djuh_5VDukK0H78QzOX-_FQEo6V37m3nkm96gtU,91
-sqlmesh-0.184.2.dev1.dist-info/entry_points.txt,sha256=pIYKtg-u-uO7BK_Y0qrdmfyj0bYrcMc3Xiqr5WE3VFk,121
-sqlmesh-0.184.2.dev1.dist-info/top_level.txt,sha256=bspjo5vjELQkuN0qKuTFIbsrUt3Zeo8D2mlKeHG1LdE,12
-sqlmesh-0.184.2.dev1.dist-info/RECORD,,
+sqlmesh-0.184.2.dev2.dist-info/METADATA,sha256=_LtGgD1Kkkn6kxUjsFdqjqOw8wS8BO92ePAgPYbabAo,25498
+sqlmesh-0.184.2.dev2.dist-info/WHEEL,sha256=Nw36Djuh_5VDukK0H78QzOX-_FQEo6V37m3nkm96gtU,91
+sqlmesh-0.184.2.dev2.dist-info/entry_points.txt,sha256=pIYKtg-u-uO7BK_Y0qrdmfyj0bYrcMc3Xiqr5WE3VFk,121
+sqlmesh-0.184.2.dev2.dist-info/top_level.txt,sha256=bspjo5vjELQkuN0qKuTFIbsrUt3Zeo8D2mlKeHG1LdE,12
+sqlmesh-0.184.2.dev2.dist-info/RECORD,,

web/server/api/endpoints/lineage.py

@@ -126,7 +126,7 @@ def create_models_only_lineage_adjacency_list(
 
 
 @router.get("/{model_name:str}/{column_name:str}")
-async def column_lineage(
+def column_lineage(
     model_name: str,
     column_name: str,
     models_only: bool = False,