sqlmesh 0.190.1.dev4__py3-none-any.whl → 0.190.1.dev6__py3-none-any.whl

sqlmesh/_version.py

@@ -17,5 +17,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.190.1.dev4'
-__version_tuple__ = version_tuple = (0, 190, 1, 'dev4')
+__version__ = version = '0.190.1.dev6'
+__version_tuple__ = version_tuple = (0, 190, 1, 'dev6')

sqlmesh/lsp/completions.py

@@ -1,9 +1,14 @@
 from functools import lru_cache
 from sqlglot import Dialect, Tokenizer
-from sqlmesh.lsp.custom import AllModelsResponse, MacroCompletion
+from sqlmesh.lsp.custom import (
+    AllModelsResponse,
+    MacroCompletion,
+    ModelCompletion,
+)
 from sqlmesh import macro
 import typing as t
 from sqlmesh.lsp.context import AuditTarget, LSPContext, ModelTarget
+from sqlmesh.lsp.description import generate_markdown_description
 from sqlmesh.lsp.uri import URI
 
 
@@ -26,14 +31,18 @@ def get_sql_completions(
     # Combine keywords - SQL keywords first, then file keywords
     all_keywords = list(sql_keywords) + list(file_keywords - sql_keywords)
 
+    models = list(get_models(context, file_uri))
     return AllModelsResponse(
-        models=list(get_models(context, file_uri)),
+        models=[m.name for m in models],
+        model_completions=models,
         keywords=all_keywords,
         macros=list(get_macros(context, file_uri)),
     )
 
 
-def get_models(context: t.Optional[LSPContext], file_uri: t.Optional[URI]) -> t.Set[str]:
+def get_models(
+    context: t.Optional[LSPContext], file_uri: t.Optional[URI]
+) -> t.List[ModelCompletion]:
     """
     Return a list of models for a given file.
 
@@ -41,23 +50,23 @@ def get_models(context: t.Optional[LSPContext], file_uri: t.Optional[URI]) -> t.
     If there is a context, return a list of all models bar the ones the file itself defines.
     """
     if context is None:
-        return set()
+        return []
+
+    current_path = file_uri.to_path() if file_uri is not None else None
+
+    completions: t.List[ModelCompletion] = []
+    for model in context.context.models.values():
+        if current_path is not None and model._path == current_path:
+            continue
+        description = None
+        try:
+            description = generate_markdown_description(model)
+        except Exception:
+            description = getattr(model, "description", None)
+
+        completions.append(ModelCompletion(name=model.name, description=description))
 
-    all_models = set()
-    # Extract model names from ModelInfo objects
-    for file_info in context.map.values():
-        if isinstance(file_info, ModelTarget):
-            all_models.update(file_info.names)
-
-    # Remove models from the current file
-    path = file_uri.to_path() if file_uri is not None else None
-    if path is not None and path in context.map:
-        file_info = context.map[path]
-        if isinstance(file_info, ModelTarget):
-            for model in file_info.names:
-                all_models.discard(model)
-
-    return all_models
+    return completions
 
 
 def get_macros(

sqlmesh/lsp/custom.py

@@ -30,12 +30,19 @@ class MacroCompletion(PydanticModel):
     description: t.Optional[str] = None
 
 
+class ModelCompletion(PydanticModel):
+    """Information about a model for autocompletion."""
+
+    name: str
+    description: t.Optional[str] = None
+
+
 class AllModelsResponse(CustomMethodResponseBaseClass):
-    """
-    Response to get all the models that are in the current project.
-    """
+    """Response to get all models that are in the current project."""
 
+    #: Deprecated: use ``model_completions`` instead
     models: t.List[str]
+    model_completions: t.List[ModelCompletion]
     keywords: t.List[str]
     macros: t.List[MacroCompletion]
 

sqlmesh/lsp/main.py

@@ -627,12 +627,18 @@ class SQLMeshLanguageServer:
 
                 completion_items = []
                 # Add model completions
-                for model in completion_response.models:
+                for model in completion_response.model_completions:
                     completion_items.append(
                         types.CompletionItem(
-                            label=model,
+                            label=model.name,
                             kind=types.CompletionItemKind.Reference,
                             detail="SQLMesh Model",
+                            documentation=types.MarkupContent(
+                                kind=types.MarkupKind.Markdown,
+                                value=model.description or "No description available",
+                            )
+                            if model.description
+                            else None,
                         )
                     )
                 # Add macro completions

{sqlmesh-0.190.1.dev4.dist-info → sqlmesh-0.190.1.dev6.dist-info}/METADATA

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

{sqlmesh-0.190.1.dev4.dist-info → sqlmesh-0.190.1.dev6.dist-info}/RECORD

@@ -1,5 +1,5 @@
 sqlmesh/__init__.py,sha256=QjwaamJIdW7voXoS880yAx7tHm-Im4_mqGA-XJgafos,7631
-sqlmesh/_version.py,sha256=OjumcFEhGImTtaTOswRvZTfmFxaVyJYjmU7vkQ1YQCA,528
+sqlmesh/_version.py,sha256=-U_QhWMFCSI0QtG8CpandTmK6UBqPUt_MQe1hlymRF0,528
 sqlmesh/magics.py,sha256=gjzoxREb1XT8Yho_PQMlMqhhkLq_sEfBM8S3PTgxvWE,40337
 sqlmesh/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sqlmesh/cicd/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -159,13 +159,13 @@ sqlmesh/integrations/github/cicd/command.py,sha256=rNWb-aMOA9R23YumD5vPwR0PgPGGj
 sqlmesh/integrations/github/cicd/config.py,sha256=tRhMZaLf1p6AWXvi67n85Q-lptJNGt1Wk_Jr61ytpiM,2104
 sqlmesh/integrations/github/cicd/controller.py,sha256=IgUWK9orYSMiElO3GaL6Ja-osFgA7je3RDcLChWUi5s,43769
 sqlmesh/lsp/api.py,sha256=_lA_buSPZFIoLJuINbS53LbNOOaHp3orb0uFTE3aCqI,2328
-sqlmesh/lsp/completions.py,sha256=OTI7Ga6NNtaC_k80W1wI9Tft0gMBmaTJOt6gW_DF6GE,6484
+sqlmesh/lsp/completions.py,sha256=Nt1XpFL_-QAIRM8BBBC6Ia6YG0hMZriiNJra7nGfD4Q,6676
 sqlmesh/lsp/context.py,sha256=QGbNL0sifxzVJ7T1U5oWcJ5FXrOrLrVKt1jtH1qM_50,6099
-sqlmesh/lsp/custom.py,sha256=3eNuwP4Q5vT7l9RxpPYUiY-91c2w5FSTEKSHdGJf7CA,2705
+sqlmesh/lsp/custom.py,sha256=IJ70li8d5RbOFDrjHfsI2UNOiOp-WB22lrgpdJ7lVSg,2942
 sqlmesh/lsp/description.py,sha256=4QIGC9ix8IzmWF6DVUNhY4m1Crk3sNQA6hZ1RbyqU78,827
 sqlmesh/lsp/helpers.py,sha256=EFc1u3-b7kSv5-tNwmKUDxId72RCLDBnN2lLTgRSzzQ,1020
 sqlmesh/lsp/hints.py,sha256=j-blccBUil94yRoDjWfkSM0HqMmsKD6hP2TaWdWWVk4,3734
-sqlmesh/lsp/main.py,sha256=4fpL59afwuI4iPBHQedWEarM9Wa5e5IWQ_KMF8eW6NU,36664
+sqlmesh/lsp/main.py,sha256=w06r1KpMwxILMTTWYOSc7Cc9dXq97TVPrembj_T32DY,37012
 sqlmesh/lsp/reference.py,sha256=8hZT3wH8zmWiFEySS1_nVhhLg-CKCGncf9wG3LGBef8,26432
 sqlmesh/lsp/uri.py,sha256=mvSLbZGmMTHPpliThOtVWiFcHnfJzp1w39q9l-xD2SI,802
 sqlmesh/migrations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -275,7 +275,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.190.1.dev4.dist-info/licenses/LICENSE,sha256=OlMefUjgWJdULtf84BLW0AZZcY8DwdgQqb_1j2862j8,11346
+sqlmesh-0.190.1.dev6.dist-info/licenses/LICENSE,sha256=OlMefUjgWJdULtf84BLW0AZZcY8DwdgQqb_1j2862j8,11346
 web/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 web/client/dist/index.html,sha256=k5bAEoPDJ0BhYQ4hBcKWObITCxIgqOUC28nGXvEDrt0,5472
 web/client/dist/assets/Audits-Pgs8N6IE.js,sha256=BQuc9kiznZSqTnzNrJSDNqAAdOTGjNQJt32M1sHZFok,1041
@@ -393,8 +393,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=SCy3-pr5x6bEEiOaZGCDbiHQhpgxwmnz3Acqa2Ie6jI,2090
-sqlmesh-0.190.1.dev4.dist-info/METADATA,sha256=FZHJfxSn1wpx1MUxMo33Oz7WPs5aY1Q9A1S8BAEuETo,26540
-sqlmesh-0.190.1.dev4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-sqlmesh-0.190.1.dev4.dist-info/entry_points.txt,sha256=pIYKtg-u-uO7BK_Y0qrdmfyj0bYrcMc3Xiqr5WE3VFk,121
-sqlmesh-0.190.1.dev4.dist-info/top_level.txt,sha256=bspjo5vjELQkuN0qKuTFIbsrUt3Zeo8D2mlKeHG1LdE,12
-sqlmesh-0.190.1.dev4.dist-info/RECORD,,
+sqlmesh-0.190.1.dev6.dist-info/METADATA,sha256=gcHiiN-JLRAwIsm_tpRUMYDWtUEpDviJ5PZwuAk3QFc,26540
+sqlmesh-0.190.1.dev6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+sqlmesh-0.190.1.dev6.dist-info/entry_points.txt,sha256=pIYKtg-u-uO7BK_Y0qrdmfyj0bYrcMc3Xiqr5WE3VFk,121
+sqlmesh-0.190.1.dev6.dist-info/top_level.txt,sha256=bspjo5vjELQkuN0qKuTFIbsrUt3Zeo8D2mlKeHG1LdE,12
+sqlmesh-0.190.1.dev6.dist-info/RECORD,,