sqlmesh 0.181.1.dev2__py3-none-any.whl → 0.181.1.dev4__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.1.dev2'
-__version_tuple__ = version_tuple = (0, 181, 1, 'dev2')
+__version__ = version = '0.181.1.dev4'
+__version_tuple__ = version_tuple = (0, 181, 1, 'dev4')

sqlmesh/core/engine_adapter/spark.py

@@ -105,6 +105,7 @@ class SparkEngineAdapter(
             exp.DataType.Type.BINARY: spark_types.BinaryType,
             exp.DataType.Type.BOOLEAN: spark_types.BooleanType,
             exp.DataType.Type.DATE: spark_types.DateType,
+            exp.DataType.Type.TIMESTAMPNTZ: spark_types.TimestampNTZType,
             exp.DataType.Type.DATETIME: spark_types.TimestampNTZType,
             exp.DataType.Type.TIMESTAMPLTZ: spark_types.TimestampType,
             exp.DataType.Type.TIMESTAMP: spark_types.TimestampType,

sqlmesh/lsp/main.py

@@ -14,7 +14,10 @@ from sqlmesh.core.linter.definition import AnnotatedRuleViolation
 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_model_definitions_for_a_path
+from sqlmesh.lsp.reference import (
+    get_model_definitions_for_a_path,
+    filter_references_by_position,
+)
 
 
 class SQLMeshLanguageServer:
@@ -189,9 +192,7 @@ class SQLMeshLanguageServer:
                 references = get_model_definitions_for_a_path(
                     self.lsp_context, params.text_document.uri
                 )
-                if not references:
-                    return []
-
+                filtered_references = filter_references_by_position(references, params.position)
                 return [
                     types.LocationLink(
                         target_uri=reference.uri,
@@ -205,7 +206,7 @@ class SQLMeshLanguageServer:
                         ),
                         origin_selection_range=reference.range,
                     )
-                    for reference in references
+                    for reference in filtered_references
                 ]
 
             except Exception as e:

sqlmesh/lsp/reference.py

@@ -14,6 +14,39 @@ class Reference(PydanticModel):
     uri: str
 
 
+def filter_references_by_position(
+    references: t.List[Reference], position: Position
+) -> t.List[Reference]:
+    """
+    Filter references to only include those that contain the given position.
+
+    Args:
+        references: List of Reference objects
+        position: The cursor position to check
+
+    Returns:
+        List of Reference objects that contain the position
+    """
+    filtered_references = []
+
+    for reference in references:
+        # Check if position is within the reference range
+        range_start = reference.range.start
+        range_end = reference.range.end
+
+        # Position is within range if it's after or at start and before or at end
+        if (
+            range_start.line < position.line
+            or (range_start.line == position.line and range_start.character <= position.character)
+        ) and (
+            range_end.line > position.line
+            or (range_end.line == position.line and range_end.character >= position.character)
+        ):
+            filtered_references.append(reference)
+
+    return filtered_references
+
+
 def get_model_definitions_for_a_path(
     lint_context: LSPContext, document_uri: str
 ) -> t.List[Reference]:

{sqlmesh-0.181.1.dev2.dist-info → sqlmesh-0.181.1.dev4.dist-info}/METADATA

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

{sqlmesh-0.181.1.dev2.dist-info → sqlmesh-0.181.1.dev4.dist-info}/RECORD

@@ -1,5 +1,5 @@
 sqlmesh/__init__.py,sha256=uPXld_xTZ2NTruSfVJ_vR_w-uAxR3T-gMXAb-EphTe8,7167
-sqlmesh/_version.py,sha256=CuXexU7YUj3BaQIkGUDfnLYhBS7j71XtwIazBc2rtmw,528
+sqlmesh/_version.py,sha256=80O7jnPXUqXsb3x7IIiWwiONodIV1mQ4Y1E8PEkglTo,528
 sqlmesh/magics.py,sha256=7BU9ec9zE4tjvZ3hdEqwFX2VVV-CI0h06TGZo_Ssi8M,39070
 sqlmesh/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sqlmesh/cicd/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -73,7 +73,7 @@ sqlmesh/core/engine_adapter/redshift.py,sha256=_rZWv4rchDAzB6m6idoV1NLHFBKK_gM4n
 sqlmesh/core/engine_adapter/risingwave.py,sha256=NXXuiPfbB5ZSp7igMqZpaX2c-5TYreOlIouPU7iCVJ8,936
 sqlmesh/core/engine_adapter/shared.py,sha256=NyF2osSJDWPp7KlJUBR4GZcS_cJFr4daUPvhGyp4XcQ,12909
 sqlmesh/core/engine_adapter/snowflake.py,sha256=LKiQq43JNukLcDq5OnKALpiZ0xjFK7fVJYJvlXHAWxo,23361
-sqlmesh/core/engine_adapter/spark.py,sha256=0wTMRs3emclwSThp05qvWPobu-Zor9Oc4pDCGMwQlZg,22310
+sqlmesh/core/engine_adapter/spark.py,sha256=VOXStkBBZIzebPo_ivdQFvZtRMiWJHHo7h118Lh5Yt8,22384
 sqlmesh/core/engine_adapter/trino.py,sha256=pZ2E1owyC6DPmkU3RkibR3CIE1gE-8fNd016YHh52II,14915
 sqlmesh/core/linter/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sqlmesh/core/linter/definition.py,sha256=UL05OjFBfoLSYGWXNf4zjZYzIKr-OyFIMIcIh7RhR8s,5237
@@ -158,8 +158,8 @@ sqlmesh/integrations/github/cicd/controller.py,sha256=AiEMsRW4WSVxXP_7oQSIdzBzmY
 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=oOZXCWLYJ722_2zLg6lTXGfw5UXo8xt78qRO6IIKZh4,13588
-sqlmesh/lsp/reference.py,sha256=qLy6ovxcR_q7qcqDYqVbQ_5p8cpPYeBynfL7b12_PNU,6288
+sqlmesh/lsp/main.py,sha256=Xt7cwoocmmefsFO4UvBPlIsiXWs7YEGRp6jgT2tKn60,13672
+sqlmesh/lsp/reference.py,sha256=veet4JZpVmUOIMiRPOvMtzu4DQEPT9Vk9c2bWAsFuQA,7366
 sqlmesh/migrations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sqlmesh/migrations/v0001_init.py,sha256=h8FwMYJj9_O8yGpk12guaSaGwX5A3we2GiKeHrXi15o,2048
 sqlmesh/migrations/v0002_remove_identify.py,sha256=ZeHuodZcQzRRZX0B0Y7SSkK3qqlDuz316kmvzojeyoU,113
@@ -265,7 +265,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.1.dev2.dist-info/licenses/LICENSE,sha256=OlMefUjgWJdULtf84BLW0AZZcY8DwdgQqb_1j2862j8,11346
+sqlmesh-0.181.1.dev4.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 +382,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.1.dev2.dist-info/METADATA,sha256=HcHQUyKlJOMJeTDuPUly3kRPi6hj71donkUcTb-d1rY,25307
-sqlmesh-0.181.1.dev2.dist-info/WHEEL,sha256=0CuiUZ_p9E4cD6NyLD6UG80LBXYyiSYZOKDm5lp32xk,91
-sqlmesh-0.181.1.dev2.dist-info/entry_points.txt,sha256=pIYKtg-u-uO7BK_Y0qrdmfyj0bYrcMc3Xiqr5WE3VFk,121
-sqlmesh-0.181.1.dev2.dist-info/top_level.txt,sha256=bspjo5vjELQkuN0qKuTFIbsrUt3Zeo8D2mlKeHG1LdE,12
-sqlmesh-0.181.1.dev2.dist-info/RECORD,,
+sqlmesh-0.181.1.dev4.dist-info/METADATA,sha256=zgisGGtUmcTd7z3aXs9RzYCOCD7egYJamNkk9SfLId0,25307
+sqlmesh-0.181.1.dev4.dist-info/WHEEL,sha256=0CuiUZ_p9E4cD6NyLD6UG80LBXYyiSYZOKDm5lp32xk,91
+sqlmesh-0.181.1.dev4.dist-info/entry_points.txt,sha256=pIYKtg-u-uO7BK_Y0qrdmfyj0bYrcMc3Xiqr5WE3VFk,121
+sqlmesh-0.181.1.dev4.dist-info/top_level.txt,sha256=bspjo5vjELQkuN0qKuTFIbsrUt3Zeo8D2mlKeHG1LdE,12
+sqlmesh-0.181.1.dev4.dist-info/RECORD,,