fix(vscode): fix link [ci skip]

Author: benfdking

sqlmesh/lsp/main.py

@@ -119,9 +119,12 @@ def api(ls: LanguageServer, request: ApiRequest) -> t.Dict[str, t.Any]:
                         # /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}")
+                        logging.info(
+                            f"Column lineage request: {model_name} {column} with params {request.params}"
+                        )
+                        models_only = bool(getattr(request.params, "models_only", False))
                         column_lineage_response = column_lineage(
-                            model_name, column, False, self.lsp_context.context
+                            model_name, column, models_only, self.lsp_context.context
                         )
                         return ApiResponseGetColumnLineage(data=column_lineage_response).model_dump(
                             mode="json"

vscode/bus/src/nominal.ts

@@ -0,0 +1,8 @@
+/**
+ * Nominal type
+ *
+ * This is a nominal type that is used to create a new type that is a brand of the original type.
+ * This is useful for creating interfaces that are not compatible with the original type and require
+ * a type check to ensure that the type is correct.
+ */
+export type Nominal<K, T> = K & { __brand: T }

vscode/react/src/components/graph/ModelColumns.tsx

@@ -402,7 +402,7 @@ function ModelColumn({
     refetch: getColumnLineage,
     isFetching,
     isError,
-  } = useApiColumnLineage(nodeId, column.name)
+  } = useApiColumnLineage(nodeId, column.name, { models_only: true })
 
   useEffect(() => {
     if (isNil(selectManually)) return

vscode/react/src/components/graph/ModelNode.tsx

@@ -53,11 +53,10 @@ export default function ModelNode({
     const modelsArray = Object.values(models)
     const decodedId = decodeURIComponent(id)
     const model = modelsArray.find((m: Model) => m.fqn === decodedId)
-    if (!model) {
-      throw new Error(`Model not found: ${id}`)
-    }
     const modelColumns = model?.columns ?? []
 
+    console.log('lineage', lineage)
+
     Object.keys(lineage[decodedId]?.columns ?? {}).forEach((column: string) => {
       const found = modelColumns.find(({ name }: any) => {
         try {
@@ -158,6 +157,20 @@ export default function ModelNode({
   // isFalse(isModelUnknown)
   const shouldDisableColumns = isFalse(isModelSQL)
 
+  const nodeLabel: string = nodeData.label.includes(':')
+    ? (nodeData.label.split(':').pop() ?? nodeData.label)
+    : nodeData.label
+  console.log(
+    'id',
+    id,
+    isCTE,
+    columns,
+    nodeData.label,
+    'does node label include :',
+    nodeData.label.includes(':'),
+    nodeLabel,
+  )
+
   return (
     <div
       onMouseEnter={() => setIsMouseOver(true)}
@@ -200,7 +213,7 @@ export default function ModelNode({
       <ModelNodeHeaderHandles
         id={id}
         type={nodeType}
-        label={nodeData.label}
+        label={nodeLabel}
         isSelected={isSelected}
         isDraggable={true}
         className={clsx(

web/server/api/endpoints/lineage.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import logging
 import typing as t
 from collections import defaultdict
 
@@ -19,6 +20,8 @@
 
 router = APIRouter()
 
+logger = logging.getLogger(__name__)
+
 
 def quote_column(column: str, dialect: str) -> str:
     return exp.to_identifier(column, quoted=True).sql(dialect=dialect)
@@ -134,6 +137,9 @@ def column_lineage(
 ) -> t.Dict[str, t.Dict[str, LineageColumn]]:
     """Get a column's lineage"""
     try:
+        logger.info(
+            f"Getting column lineage for {model_name}.{column_name}, models only: {models_only}"
+        )
         model_name = context.get_model(model_name).fqn
         if models_only:
             return create_models_only_lineage_adjacency_list(model_name, column_name, context)