sqlmesh 0.189.3.dev13__py3-none-any.whl → 0.189.3.dev15__py3-none-any.whl

sqlmesh/_version.py

@@ -17,5 +17,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.189.3.dev13'
-__version_tuple__ = version_tuple = (0, 189, 3, 'dev13')
+__version__ = version = '0.189.3.dev15'
+__version_tuple__ = version_tuple = (0, 189, 3, 'dev15')

sqlmesh/dbt/model.py

@@ -314,14 +314,6 @@ class ModelConfig(BaseModelConfig):
                     **incremental_by_kind_kwargs,
                 )
 
-            incremental_by_time_str = collection_to_str(INCREMENTAL_BY_TIME_STRATEGIES)
-            incremental_by_unique_key_str = collection_to_str(
-                INCREMENTAL_BY_UNIQUE_KEY_STRATEGIES.union(["none"])
-            )
-            get_console().log_warning(
-                f"Using unmanaged incremental materialization for model '{self.canonical_name(context)}'. "
-                f"Some features might not be available. Consider adding either a time_column ({incremental_by_time_str}) or a unique_key ({incremental_by_unique_key_str}) configuration to mitigate this.",
-            )
             strategy = self.incremental_strategy or target.default_incremental_strategy(
                 IncrementalUnmanagedKind
             )

sqlmesh/lsp/main.py

@@ -44,7 +44,12 @@ from sqlmesh.lsp.custom import (
     CustomMethod,
 )
 from sqlmesh.lsp.hints import get_hints
-from sqlmesh.lsp.reference import get_references, get_cte_references
+from sqlmesh.lsp.reference import (
+    LSPCteReference,
+    LSPModelReference,
+    get_references,
+    get_all_references,
+)
 from sqlmesh.lsp.uri import URI
 from web.server.api.endpoints.lineage import column_lineage, model_lineage
 from web.server.api.endpoints.models import get_models
@@ -368,7 +373,7 @@ class SQLMeshLanguageServer:
                 if not references:
                     return None
                 reference = references[0]
-                if not reference.markdown_description:
+                if isinstance(reference, LSPCteReference) or not reference.markdown_description:
                     return None
                 return types.Hover(
                     contents=types.MarkupContent(
@@ -418,8 +423,8 @@ class SQLMeshLanguageServer:
                 references = get_references(self.lsp_context, uri, params.position)
                 location_links = []
                 for reference in references:
-                    # Use target_range if available (for CTEs), otherwise default to start of file
-                    if reference.target_range:
+                    # Use target_range if available (CTEs, Macros), otherwise default to start of file
+                    if not isinstance(reference, LSPModelReference):
                         target_range = reference.target_range
                         target_selection_range = reference.target_range
                     else:
@@ -449,7 +454,7 @@ class SQLMeshLanguageServer:
         def find_references(
             ls: LanguageServer, params: types.ReferenceParams
         ) -> t.Optional[t.List[types.Location]]:
-            """Find all references of a symbol (currently supporting CTEs)"""
+            """Find all references of a symbol (supporting CTEs, models for now)"""
             try:
                 uri = URI(params.text_document.uri)
                 self._ensure_context_for_document(uri)
@@ -457,10 +462,10 @@ class SQLMeshLanguageServer:
                 if self.lsp_context is None:
                     raise RuntimeError(f"No context found for document: {document.path}")
 
-                cte_references = get_cte_references(self.lsp_context, uri, params.position)
+                all_references = get_all_references(self.lsp_context, uri, params.position)
 
                 # Convert references to Location objects
-                locations = [types.Location(uri=ref.uri, range=ref.range) for ref in cte_references]
+                locations = [types.Location(uri=ref.uri, range=ref.range) for ref in all_references]
 
                 return locations if locations else None
             except Exception as e:

sqlmesh/lsp/reference.py

@@ -1,6 +1,7 @@
 from lsprotocol.types import Range, Position
 import typing as t
 from pathlib import Path
+from pydantic import Field
 
 from sqlmesh.core.audit import StandaloneAudit
 from sqlmesh.core.dialect import normalize_model_name
@@ -23,21 +24,37 @@ from sqlmesh import macro
 import inspect
 
 
-class Reference(PydanticModel):
-    """
-    A reference to a model or CTE.
+class LSPModelReference(PydanticModel):
+    """A LSP reference to a model."""
+
+    type: t.Literal["model"] = "model"
+    uri: str
+    range: Range
+    markdown_description: t.Optional[str] = None
 
-    Attributes:
-        range: The range of the reference in the source file
-        uri: The uri of the referenced model
-        markdown_description: The markdown description of the referenced model
-        target_range: The range of the definition for go-to-definition (optional, used for CTEs)
-    """
 
+class LSPCteReference(PydanticModel):
+    """A LSP reference to a CTE."""
+
+    type: t.Literal["cte"] = "cte"
+    uri: str
     range: Range
+    target_range: Range
+
+
+class LSPMacroReference(PydanticModel):
+    """A LSP reference to a macro."""
+
+    type: t.Literal["macro"] = "macro"
     uri: str
+    range: Range
+    target_range: Range
     markdown_description: t.Optional[str] = None
-    target_range: t.Optional[Range] = None
+
+
+Reference = t.Annotated[
+    t.Union[LSPModelReference, LSPCteReference, LSPMacroReference], Field(discriminator="type")
+]
 
 
 def by_position(position: Position) -> t.Callable[[Reference], bool]:
@@ -136,7 +153,7 @@ def get_model_definitions_for_a_path(
         return []
 
     # Find all possible references
-    references = []
+    references: t.List[Reference] = []
 
     with open(file_path, "r", encoding="utf-8") as file:
         read_file = file.readlines()
@@ -173,7 +190,7 @@ def get_model_definitions_for_a_path(
                             table_range = to_lsp_range(table_range_sqlmesh)
 
                             references.append(
-                                Reference(
+                                LSPCteReference(
                                     uri=document_uri.value,  # Same file
                                     range=table_range,
                                     target_range=target_range,
@@ -227,7 +244,7 @@ def get_model_definitions_for_a_path(
                 description = generate_markdown_description(referenced_model)
 
                 references.append(
-                    Reference(
+                    LSPModelReference(
                         uri=referenced_model_uri.value,
                         range=Range(
                             start=to_lsp_position(start_pos_sqlmesh),
@@ -286,7 +303,7 @@ def get_macro_definitions_for_a_path(
         return []
 
     references = []
-    config_for_model, config_path = lsp_context.context.config_for_path(
+    _, config_path = lsp_context.context.config_for_path(
         file_path,
     )
 
@@ -372,7 +389,7 @@ def get_macro_reference(
         # Create a reference to the macro definition
         macro_uri = URI.from_path(path)
 
-        return Reference(
+        return LSPMacroReference(
             uri=macro_uri.value,
             range=to_lsp_range(macro_range),
             target_range=Range(
@@ -405,7 +422,7 @@ def get_built_in_macro_reference(macro_name: str, macro_range: Range) -> t.Optio
     # Calculate the end line number by counting the number of source lines
     end_line_number = line_number + len(source_lines) - 1
 
-    return Reference(
+    return LSPMacroReference(
         uri=URI.from_path(Path(filename)).value,
         range=macro_range,
         target_range=Range(
@@ -416,9 +433,99 @@ def get_built_in_macro_reference(macro_name: str, macro_range: Range) -> t.Optio
     )
 
 
+def get_model_find_all_references(
+    lint_context: LSPContext, document_uri: URI, position: Position
+) -> t.List[LSPModelReference]:
+    """
+    Get all references to a model across the entire project.
+
+    This function finds all usages of a model in other files by searching through
+    all models in the project and checking their dependencies.
+
+    Args:
+        lint_context: The LSP context
+        document_uri: The URI of the document
+        position: The position to check for model references
+
+    Returns:
+        A list of references to the model across all files
+    """
+    # First, get the references in the current file to determine what model we're looking for
+    current_file_references = [
+        ref
+        for ref in get_model_definitions_for_a_path(lint_context, document_uri)
+        if isinstance(ref, LSPModelReference)
+    ]
+
+    # Find the model reference at the cursor position
+    target_model_uri: t.Optional[str] = None
+    for ref in current_file_references:
+        if _position_within_range(position, ref.range):
+            # This is a model reference, get the target model URI
+            target_model_uri = ref.uri
+            break
+
+    if target_model_uri is None:
+        return []
+
+    # Start with the model definition
+    all_references: t.List[LSPModelReference] = [
+        LSPModelReference(
+            uri=ref.uri,
+            range=Range(
+                start=Position(line=0, character=0),
+                end=Position(line=0, character=0),
+            ),
+            markdown_description=ref.markdown_description,
+        )
+    ]
+
+    # Then add the original reference
+    for ref in current_file_references:
+        if ref.uri == target_model_uri and isinstance(ref, LSPModelReference):
+            all_references.append(
+                LSPModelReference(
+                    uri=document_uri.value,
+                    range=ref.range,
+                    markdown_description=ref.markdown_description,
+                )
+            )
+
+    # Search through the models in the project
+    for path, target in lint_context.map.items():
+        if not isinstance(target, (ModelTarget, AuditTarget)):
+            continue
+
+        file_uri = URI.from_path(path)
+
+        # Skip current file, already processed
+        if file_uri.value == document_uri.value:
+            continue
+
+        # Get model references for this file
+        file_references = [
+            ref
+            for ref in get_model_definitions_for_a_path(lint_context, file_uri)
+            if isinstance(ref, LSPModelReference)
+        ]
+
+        # Add references that point to the target model file
+        for ref in file_references:
+            if ref.uri == target_model_uri and isinstance(ref, LSPModelReference):
+                all_references.append(
+                    LSPModelReference(
+                        uri=file_uri.value,
+                        range=ref.range,
+                        markdown_description=ref.markdown_description,
+                    )
+                )
+
+    return all_references
+
+
 def get_cte_references(
     lint_context: LSPContext, document_uri: URI, position: Position
-) -> t.List[Reference]:
+) -> t.List[LSPCteReference]:
     """
     Get all references to a CTE at a specific position in a document.
 
@@ -432,12 +539,12 @@ def get_cte_references(
     Returns:
         A list of references to the CTE (including its definition and all usages)
     """
-    references = get_model_definitions_for_a_path(lint_context, document_uri)
 
-    # Filter for CTE references (those with target_range set and same URI)
-    # TODO: Consider extending Reference class to explicitly indicate reference type instead
-    cte_references = [
-        ref for ref in references if ref.target_range is not None and ref.uri == document_uri.value
+    # Filter to get the CTE references
+    cte_references: t.List[LSPCteReference] = [
+        ref
+        for ref in get_model_definitions_for_a_path(lint_context, document_uri)
+        if isinstance(ref, LSPCteReference)
     ]
 
     if not cte_references:
@@ -450,7 +557,7 @@ def get_cte_references(
             target_cte_definition_range = ref.target_range
             break
         # Check if cursor is on the CTE definition
-        elif ref.target_range and _position_within_range(position, ref.target_range):
+        elif _position_within_range(position, ref.target_range):
             target_cte_definition_range = ref.target_range
             break
 
@@ -459,10 +566,10 @@ def get_cte_references(
 
     # Add the CTE definition
     matching_references = [
-        Reference(
+        LSPCteReference(
             uri=document_uri.value,
             range=target_cte_definition_range,
-            markdown_description="CTE definition",
+            target_range=target_cte_definition_range,
         )
     ]
 
@@ -470,16 +577,44 @@ def get_cte_references(
     for ref in cte_references:
         if ref.target_range == target_cte_definition_range:
             matching_references.append(
-                Reference(
+                LSPCteReference(
                     uri=document_uri.value,
                     range=ref.range,
-                    markdown_description="CTE usage",
+                    target_range=ref.target_range,
                 )
             )
 
     return matching_references
 
 
+def get_all_references(
+    lint_context: LSPContext, document_uri: URI, position: Position
+) -> t.Sequence[Reference]:
+    """
+    Get all references of a symbol at a specific position in a document.
+
+    This function determines the type of reference (CTE, model for now) at the cursor
+    position and returns all references to that symbol across the project.
+
+    Args:
+        lint_context: The LSP context
+        document_uri: The URI of the document
+        position: The position to check for references
+
+    Returns:
+        A list of references to the symbol at the given position
+    """
+    # First try CTE references (within same file)
+    if cte_references := get_cte_references(lint_context, document_uri, position):
+        return cte_references
+
+    # Then try model references (across files)
+    if model_references := get_model_find_all_references(lint_context, document_uri, position):
+        return model_references
+
+    return []
+
+
 def _position_within_range(position: Position, range: Range) -> bool:
     """Check if a position is within a given range."""
     return (

{sqlmesh-0.189.3.dev13.dist-info → sqlmesh-0.189.3.dev15.dist-info}/METADATA

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

{sqlmesh-0.189.3.dev13.dist-info → sqlmesh-0.189.3.dev15.dist-info}/RECORD

@@ -1,5 +1,5 @@
 sqlmesh/__init__.py,sha256=QjwaamJIdW7voXoS880yAx7tHm-Im4_mqGA-XJgafos,7631
-sqlmesh/_version.py,sha256=oMJeJkFgxAVbWZbcY39JF8TsW7nLs6rFzjWM7mJjlrc,530
+sqlmesh/_version.py,sha256=1eLtQgqb1L-uJQOiBtrfXfj7cI1FPqjuUvkYTjt35xQ,530
 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
@@ -132,7 +132,7 @@ sqlmesh/dbt/common.py,sha256=g4PAL-sEdlnfb_zT-WW3xh-oamyCUoK8dUiDEMVwDWI,7804
 sqlmesh/dbt/context.py,sha256=dxClM-G1IeMbP0v5yxC_va8QNU-uKghW2dBlSZHJOk0,10331
 sqlmesh/dbt/loader.py,sha256=oDFj9OcWJxd72eN9u9gHM_bWj7T5EQuGnLViSdtLbMU,15472
 sqlmesh/dbt/manifest.py,sha256=vHNHIkz3SQgZWoHrCsnSKNBptTCUtgG724_3QKTtnhM,24147
-sqlmesh/dbt/model.py,sha256=-TqeldZAqW-4g84Xi_16DpUD3atImk_9OHMbATIKYb8,26328
+sqlmesh/dbt/model.py,sha256=cd4WKnh7aISvH5djKIF0HVLSom8QoilZsVwAMnjKGCg,25728
 sqlmesh/dbt/package.py,sha256=IBBQueVB4_4C9tIk0-GUL-O2uibNxXu6W2a4AeHHcOU,4322
 sqlmesh/dbt/profile.py,sha256=Fz2c0dJNN0xBgYEogCIKBTByjNvlA2BVBzOLQCfdEfM,3788
 sqlmesh/dbt/project.py,sha256=4PJHPVWIKDd71J5__iICqEp21nFiijKjJkiE5kZIH5Y,3980
@@ -164,8 +164,8 @@ sqlmesh/lsp/context.py,sha256=36bUTiMSyfX91bri_F1aivDvSllq0rKidIM_RGAtirs,6253
 sqlmesh/lsp/custom.py,sha256=f6-s5gJop6YjTmX7jfwTIqjYvinhMpv_T9v0H7g4DuM,2172
 sqlmesh/lsp/description.py,sha256=4QIGC9ix8IzmWF6DVUNhY4m1Crk3sNQA6hZ1RbyqU78,827
 sqlmesh/lsp/hints.py,sha256=j-blccBUil94yRoDjWfkSM0HqMmsKD6hP2TaWdWWVk4,3734
-sqlmesh/lsp/main.py,sha256=mi8N7qKybd1aDUhPAinwYXNSxSeZ6Eh3Gop8-_Zopuw,35279
-sqlmesh/lsp/reference.py,sha256=dyeacD_MHATGRIrl8Ax86nDK9EC_dh3TaaVNAepPce0,17902
+sqlmesh/lsp/main.py,sha256=QojzsVkMxm3ZZmMwbIRkzS-RwxYSpXInp5owow16L_4,35410
+sqlmesh/lsp/reference.py,sha256=GKZw0TSvo6Y1FwV67jIliG1ofwDChyymF4yxVxzrEAQ,22026
 sqlmesh/lsp/uri.py,sha256=mvSLbZGmMTHPpliThOtVWiFcHnfJzp1w39q9l-xD2SI,802
 sqlmesh/migrations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sqlmesh/migrations/v0001_init.py,sha256=h8FwMYJj9_O8yGpk12guaSaGwX5A3we2GiKeHrXi15o,2048
@@ -274,7 +274,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.189.3.dev13.dist-info/licenses/LICENSE,sha256=OlMefUjgWJdULtf84BLW0AZZcY8DwdgQqb_1j2862j8,11346
+sqlmesh-0.189.3.dev15.dist-info/licenses/LICENSE,sha256=OlMefUjgWJdULtf84BLW0AZZcY8DwdgQqb_1j2862j8,11346
 web/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 web/client/dist/index.html,sha256=OVPMTMsqC8zy0qnBcMb7PpVcqTHdIJoCmJLYh7gC1Xs,5472
 web/client/dist/assets/Audits-Bvr7M5Cg.js,sha256=GrigkyEVYbM5zy0PAjOXdqmVV0kL-STP5gFQmx35vhw,1041
@@ -392,8 +392,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.189.3.dev13.dist-info/METADATA,sha256=sP5ocONsxRe3O0QMZd-64-exlWzoUK3a9A3vrwdHG8w,26541
-sqlmesh-0.189.3.dev13.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-sqlmesh-0.189.3.dev13.dist-info/entry_points.txt,sha256=pIYKtg-u-uO7BK_Y0qrdmfyj0bYrcMc3Xiqr5WE3VFk,121
-sqlmesh-0.189.3.dev13.dist-info/top_level.txt,sha256=bspjo5vjELQkuN0qKuTFIbsrUt3Zeo8D2mlKeHG1LdE,12
-sqlmesh-0.189.3.dev13.dist-info/RECORD,,
+sqlmesh-0.189.3.dev15.dist-info/METADATA,sha256=FhiDjFsrLh-IeiBxB5zemuzTieFkVC2210fZVgzTsQg,26541
+sqlmesh-0.189.3.dev15.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+sqlmesh-0.189.3.dev15.dist-info/entry_points.txt,sha256=pIYKtg-u-uO7BK_Y0qrdmfyj0bYrcMc3Xiqr5WE3VFk,121
+sqlmesh-0.189.3.dev15.dist-info/top_level.txt,sha256=bspjo5vjELQkuN0qKuTFIbsrUt3Zeo8D2mlKeHG1LdE,12
+sqlmesh-0.189.3.dev15.dist-info/RECORD,,