sqlmesh 0.188.2.dev7__py3-none-any.whl → 0.188.2.dev9__py3-none-any.whl

sqlmesh/_version.py

@@ -17,5 +17,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.188.2.dev7'
-__version_tuple__ = version_tuple = (0, 188, 2, 'dev7')
+__version__ = version = '0.188.2.dev9'
+__version_tuple__ = version_tuple = (0, 188, 2, 'dev9')

sqlmesh/lsp/context.py

@@ -5,7 +5,7 @@ import typing as t
 
 from sqlmesh.core.model.definition import SqlModel
 from sqlmesh.core.linter.definition import AnnotatedRuleViolation
-from sqlmesh.lsp.custom import RenderModelEntry
+from sqlmesh.lsp.custom import RenderModelEntry, ModelForRendering
 from sqlmesh.lsp.uri import URI
 
 
@@ -148,3 +148,29 @@ class LSPContext:
         # Store in cache
         self._lint_cache[path] = diagnostics
         return diagnostics
+
+    def list_of_models_for_rendering(self) -> t.List[ModelForRendering]:
+        """Get a list of models for rendering.
+
+        Returns:
+            List of ModelForRendering objects.
+        """
+        return [
+            ModelForRendering(
+                name=model.name,
+                fqn=model.fqn,
+                description=model.description,
+                uri=URI.from_path(model._path).value,
+            )
+            for model in self.context.models.values()
+            if isinstance(model, SqlModel) and model._path is not None
+        ] + [
+            ModelForRendering(
+                name=audit.name,
+                fqn=audit.fqn,
+                description=audit.description,
+                uri=URI.from_path(audit._path).value,
+            )
+            for audit in self.context.standalone_audits.values()
+            if audit._path is not None
+        ]

sqlmesh/lsp/custom.py

@@ -46,3 +46,29 @@ class RenderModelResponse(PydanticModel):
     """
 
     models: t.List[RenderModelEntry]
+
+
+ALL_MODELS_FOR_RENDER_FEATURE = "sqlmesh/all_models_for_render"
+
+
+class ModelForRendering(PydanticModel):
+    """
+    A model that is available for rendering.
+    """
+
+    name: str
+    fqn: str
+    description: t.Optional[str] = None
+    uri: str
+
+
+class AllModelsForRenderRequest(PydanticModel):
+    pass
+
+
+class AllModelsForRenderResponse(PydanticModel):
+    """
+    Response to get all the models that are in the current project for rendering purposes.
+    """
+
+    models: t.List[ModelForRendering]

sqlmesh/lsp/main.py

@@ -27,9 +27,12 @@ from sqlmesh.lsp.context import (
 )
 from sqlmesh.lsp.custom import (
     ALL_MODELS_FEATURE,
+    ALL_MODELS_FOR_RENDER_FEATURE,
     RENDER_MODEL_FEATURE,
     AllModelsRequest,
     AllModelsResponse,
+    AllModelsForRenderRequest,
+    AllModelsForRenderResponse,
     RenderModelRequest,
     RenderModelResponse,
 )
@@ -122,6 +125,19 @@ class SQLMeshLanguageServer:
             context = self._context_get_or_load(uri)
             return RenderModelResponse(models=context.render_model(uri))
 
+        @self.server.feature(ALL_MODELS_FOR_RENDER_FEATURE)
+        def all_models_for_render(
+            ls: LanguageServer, params: AllModelsForRenderRequest
+        ) -> AllModelsForRenderResponse:
+            if self.lsp_context is None:
+                current_path = Path.cwd()
+                self._ensure_context_in_folder(current_path)
+            if self.lsp_context is None:
+                raise RuntimeError("No context found")
+            return AllModelsForRenderResponse(
+                models=self.lsp_context.list_of_models_for_rendering()
+            )
+
         @self.server.feature(API_FEATURE)
         def api(ls: LanguageServer, request: ApiRequest) -> t.Dict[str, t.Any]:
             ls.log_trace(f"API request: {request}")

{sqlmesh-0.188.2.dev7.dist-info → sqlmesh-0.188.2.dev9.dist-info}/METADATA

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

{sqlmesh-0.188.2.dev7.dist-info → sqlmesh-0.188.2.dev9.dist-info}/RECORD

@@ -1,5 +1,5 @@
 sqlmesh/__init__.py,sha256=69xw6bTmi4cW90M-sl3g6wLDgxVcLHcKg1q8GS3lqyA,7466
-sqlmesh/_version.py,sha256=gh4buyj4JrUosh_TPs6jJKlPBFisvoRIsNwj4vpqPek,528
+sqlmesh/_version.py,sha256=FVIYC0V3lf_58nGWAIEWDrqEdEx_D_GaMgUNN2eicgY,528
 sqlmesh/magics.py,sha256=fu7-OeT_0_TAkWZOLoiioE3-3iFpvTRxFCbQUlRnfuw,39944
 sqlmesh/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sqlmesh/cicd/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -157,10 +157,10 @@ sqlmesh/integrations/github/cicd/config.py,sha256=tRhMZaLf1p6AWXvi67n85Q-lptJNGt
 sqlmesh/integrations/github/cicd/controller.py,sha256=FmOO93RF5auSU6aDU2beiP5LblLs38eAUYgHGWB9jYc,41666
 sqlmesh/lsp/api.py,sha256=LJdECSaEZoPbu_Xl4KscVG-WLt7_g4dY-zKGO68dH14,2162
 sqlmesh/lsp/completions.py,sha256=Uo8lUzPBeoB58Eu2642YoAtNI0pgIFWFSvwmuJALTv4,3583
-sqlmesh/lsp/context.py,sha256=mqxUn_RuTZZwmRw6Axi4roUQNMDS5zv6a5X595A6Enc,4764
-sqlmesh/lsp/custom.py,sha256=7VPBK8E6w3KFsYxjiQgpdPkSBGzT69HfPvnS3wYpWNo,920
+sqlmesh/lsp/context.py,sha256=JkHV7Du2V_GPws8EYd6MpLeVwr1w40rQMNFczRI1fkA,5671
+sqlmesh/lsp/custom.py,sha256=Mml6V1nYAHdP4Zt8gxly1ZLKYALtxGeAVzPrrYpc7m8,1426
 sqlmesh/lsp/description.py,sha256=4QIGC9ix8IzmWF6DVUNhY4m1Crk3sNQA6hZ1RbyqU78,827
-sqlmesh/lsp/main.py,sha256=yCy4Hz_0Cqe-QGOb0oCHUNUCwPGY4yIEyO--UgB5y8g,28211
+sqlmesh/lsp/main.py,sha256=TmFJbuLQ_CzQ0EuhKmqQK3F16tOpC_kYIAck2hDqd_s,28884
 sqlmesh/lsp/reference.py,sha256=-HkMMzRzq4ike5mSkoWajanYNMAwhQhlPaqw62451ac,16887
 sqlmesh/lsp/uri.py,sha256=mvSLbZGmMTHPpliThOtVWiFcHnfJzp1w39q9l-xD2SI,802
 sqlmesh/migrations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -270,7 +270,7 @@ sqlmesh/utils/pydantic.py,sha256=I898jAFqa3qU6lG2uPpRebVFncOUo8ZrLWM_R3K3DHU,116
 sqlmesh/utils/rich.py,sha256=xNZyPCqEc4TAlJlsEthg-5eYUWpaizz0QjBzO7a1Yv4,1509
 sqlmesh/utils/windows.py,sha256=r7Yin4WEwbvH7sKA3hOJtRsclmscNuBeCklCfVPvtQs,611
 sqlmesh/utils/yaml.py,sha256=np-WVcwmwsKghCgAj1OBBMsK1SXSRn4wR-H37cGiFr8,2282
-sqlmesh-0.188.2.dev7.dist-info/licenses/LICENSE,sha256=OlMefUjgWJdULtf84BLW0AZZcY8DwdgQqb_1j2862j8,11346
+sqlmesh-0.188.2.dev9.dist-info/licenses/LICENSE,sha256=OlMefUjgWJdULtf84BLW0AZZcY8DwdgQqb_1j2862j8,11346
 web/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 web/client/dist/index.html,sha256=U9DgO3KqjoSuGiKVKR8AAdfXFUa0kaXJgIRAsmZ5RH8,5472
 web/client/dist/assets/Audits-CNkn30Bu.js,sha256=0uC1ibXDFzBjPRshILFo9VzNNKHT3nvIgQhrhVDG-1Y,1041
@@ -388,8 +388,8 @@ web/server/api/endpoints/models.py,sha256=ud-lRwriyI_OYg9tZchrBgeiXELiRTLNEodraf
 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.188.2.dev7.dist-info/METADATA,sha256=I0HuoWQXgOWJhmY5OT-OL49CB4scKiIBzky8qhBELws,26531
-sqlmesh-0.188.2.dev7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-sqlmesh-0.188.2.dev7.dist-info/entry_points.txt,sha256=pIYKtg-u-uO7BK_Y0qrdmfyj0bYrcMc3Xiqr5WE3VFk,121
-sqlmesh-0.188.2.dev7.dist-info/top_level.txt,sha256=bspjo5vjELQkuN0qKuTFIbsrUt3Zeo8D2mlKeHG1LdE,12
-sqlmesh-0.188.2.dev7.dist-info/RECORD,,
+sqlmesh-0.188.2.dev9.dist-info/METADATA,sha256=XV_Ua2Rcrw-2XTT6vQ5rKoYgf4Tx-cnsSMFq11pMM2g,26531
+sqlmesh-0.188.2.dev9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+sqlmesh-0.188.2.dev9.dist-info/entry_points.txt,sha256=pIYKtg-u-uO7BK_Y0qrdmfyj0bYrcMc3Xiqr5WE3VFk,121
+sqlmesh-0.188.2.dev9.dist-info/top_level.txt,sha256=bspjo5vjELQkuN0qKuTFIbsrUt3Zeo8D2mlKeHG1LdE,12
+sqlmesh-0.188.2.dev9.dist-info/RECORD,,