sqlmesh 0.195.1.dev2__tar.gz → 0.195.1.dev4__tar.gz

{sqlmesh-0.195.1.dev2 → sqlmesh-0.195.1.dev4}/PKG-INFO

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

{sqlmesh-0.195.1.dev2 → sqlmesh-0.195.1.dev4}/sqlmesh/_version.py

@@ -17,5 +17,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.195.1.dev2'
-__version_tuple__ = version_tuple = (0, 195, 1, 'dev2')
+__version__ = version = '0.195.1.dev4'
+__version_tuple__ = version_tuple = (0, 195, 1, 'dev4')

{sqlmesh-0.195.1.dev2 → sqlmesh-0.195.1.dev4}/sqlmesh/lsp/main.py

@@ -56,6 +56,33 @@ from sqlmesh.lsp.rename import prepare_rename, rename_symbol, get_document_highl
 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
+from typing import Union
+from dataclasses import dataclass
+
+
+@dataclass
+class NoContext:
+    """State when no context has been attempted to load."""
+
+    pass
+
+
+@dataclass
+class ContextLoaded:
+    """State when context has been successfully loaded."""
+
+    lsp_context: LSPContext
+
+
+@dataclass
+class ContextFailed:
+    """State when context failed to load with an error message."""
+
+    error_message: str
+    context: t.Optional[Context] = None
+
+
+ContextState = Union[NoContext, ContextLoaded, ContextFailed]
 
 
 class SQLMeshLanguageServer:
@@ -72,7 +99,7 @@ class SQLMeshLanguageServer:
         """
         self.server = LanguageServer(server_name, version)
         self.context_class = context_class
-        self.lsp_context: t.Optional[LSPContext] = None
+        self.context_state: ContextState = NoContext()
         self.workspace_folders: t.List[Path] = []
 
         self.has_raised_loading_error: bool = False
@@ -257,16 +284,53 @@ class SQLMeshLanguageServer:
         @self.server.feature(types.TEXT_DOCUMENT_DID_SAVE)
         def did_save(ls: LanguageServer, params: types.DidSaveTextDocumentParams) -> None:
             uri = URI(params.text_document.uri)
-            if self.lsp_context is None:
+            if isinstance(self.context_state, NoContext):
                 return
 
-            context = self.lsp_context.context
-            context.load()
-            self.lsp_context = LSPContext(context)
+            if isinstance(self.context_state, ContextFailed):
+                if self.context_state.context:
+                    try:
+                        self.context_state.context.load()
+                        self.context_state = ContextLoaded(
+                            lsp_context=LSPContext(self.context_state.context)
+                        )
+                    except Exception as e:
+                        ls.log_trace(f"Error loading context: {e}")
+                        if not isinstance(self.context_state, ContextFailed):
+                            raise Exception("Context state should be failed")
+                        self.context_state = ContextFailed(
+                            error_message=str(e), context=self.context_state.context
+                        )
+                        return
+                else:
+                    # If there's no context, try to create one from scratch
+                    try:
+                        self._ensure_context_for_document(uri)
+                        # If successful, context_state will be ContextLoaded
+                        if isinstance(self.context_state, ContextLoaded):
+                            ls.show_message(
+                                "Successfully loaded SQLMesh context",
+                                types.MessageType.Info,
+                            )
+                    except Exception as e:
+                        ls.log_trace(f"Still cannot load context: {e}")
+                    return
+
+            # Reload the context if was successfully
+            try:
+                context = self.context_state.lsp_context.context
+                context.load()
+                self.context_state = ContextLoaded(lsp_context=LSPContext(context))
+            except Exception as e:
+                ls.log_trace(f"Error loading context: {e}")
+                self.context_state = ContextFailed(
+                    error_message=str(e), context=self.context_state.lsp_context.context
+                )
+                return
 
             # Only publish diagnostics if client doesn't support pull diagnostics
             if not self.client_supports_pull_diagnostics:
-                diagnostics = self.lsp_context.lint_model(uri)
+                diagnostics = self.context_state.lsp_context.lint_model(uri)
                 ls.publish_diagnostics(
                     params.text_document.uri,
                     SQLMeshLanguageServer._diagnostics_to_lsp_diagnostics(diagnostics),
@@ -443,11 +507,8 @@ class SQLMeshLanguageServer:
             """Prepare for rename operation by checking if the symbol can be renamed."""
             try:
                 uri = URI(params.text_document.uri)
-                self._ensure_context_for_document(uri)
-                if self.lsp_context is None:
-                    raise RuntimeError(f"No context found for document: {uri}")
-
-                result = prepare_rename(self.lsp_context, uri, params.position)
+                context = self._context_get_or_load(uri)
+                result = prepare_rename(context, uri, params.position)
                 return result
             except Exception as e:
                 ls.log_trace(f"Error preparing rename: {e}")
@@ -460,13 +521,8 @@ class SQLMeshLanguageServer:
             """Perform rename operation on the symbol at the given position."""
             try:
                 uri = URI(params.text_document.uri)
-                self._ensure_context_for_document(uri)
-                if self.lsp_context is None:
-                    raise RuntimeError(f"No context found for document: {uri}")
-
-                workspace_edit = rename_symbol(
-                    self.lsp_context, uri, params.position, params.new_name
-                )
+                context = self._context_get_or_load(uri)
+                workspace_edit = rename_symbol(context, uri, params.position, params.new_name)
                 return workspace_edit
             except Exception as e:
                 ls.show_message(f"Error performing rename: {e}", types.MessageType.Error)
@@ -479,11 +535,8 @@ class SQLMeshLanguageServer:
             """Highlight all occurrences of the symbol at the given position."""
             try:
                 uri = URI(params.text_document.uri)
-                self._ensure_context_for_document(uri)
-                if self.lsp_context is None:
-                    raise RuntimeError(f"No context found for document: {uri}")
-
-                highlights = get_document_highlights(self.lsp_context, uri, params.position)
+                context = self._context_get_or_load(uri)
+                highlights = get_document_highlights(context, uri, params.position)
                 return highlights
             except Exception as e:
                 ls.log_trace(f"Error getting document highlights: {e}")
@@ -670,11 +723,13 @@ class SQLMeshLanguageServer:
             return [], 0
 
     def _context_get_or_load(self, document_uri: t.Optional[URI] = None) -> LSPContext:
-        if self.lsp_context is None:
+        if isinstance(self.context_state, ContextFailed):
+            raise RuntimeError(self.context_state.error_message)
+        if isinstance(self.context_state, NoContext):
             self._ensure_context_for_document(document_uri)
-        if self.lsp_context is None:
-            raise RuntimeError("No context found able to get or load")
-        return self.lsp_context
+        if not isinstance(self.context_state, ContextLoaded):
+            raise RuntimeError("Context is not loaded")
+        return self.context_state.lsp_context
 
     def _ensure_context_for_document(
         self,
@@ -692,10 +747,10 @@ class SQLMeshLanguageServer:
                     self._ensure_context_in_folder(document_folder)
                     return
 
-        return self._ensure_context_in_folder()
+        self._ensure_context_in_folder()
 
     def _ensure_context_in_folder(self, folder_path: t.Optional[Path] = None) -> None:
-        if self.lsp_context is not None:
+        if not isinstance(self.context_state, NoContext):
             return
 
         # If not found in the provided folder, search through all workspace folders
@@ -729,7 +784,7 @@ class SQLMeshLanguageServer:
     def _create_lsp_context(self, paths: t.List[Path]) -> t.Optional[LSPContext]:
         """Create a new LSPContext instance using the configured context class.
 
-        On success, sets self.lsp_context and returns the created context.
+        On success, sets self.context_state to ContextLoaded and returns the created context.
 
         Args:
             paths: List of paths to pass to the context constructor
@@ -738,14 +793,22 @@ class SQLMeshLanguageServer:
             A new LSPContext instance wrapping the created context, or None if creation fails
         """
         try:
-            if self.lsp_context is None:
+            if isinstance(self.context_state, NoContext):
                 context = self.context_class(paths=paths)
                 loaded_sqlmesh_message(self.server, paths[0])
+            elif isinstance(self.context_state, ContextFailed):
+                if self.context_state.context:
+                    context = self.context_state.context
+                    context.load()
+                else:
+                    # If there's no context (initial creation failed), try creating again
+                    context = self.context_class(paths=paths)
+                    loaded_sqlmesh_message(self.server, paths[0])
             else:
-                self.lsp_context.context.load()
-                context = self.lsp_context.context
-            self.lsp_context = LSPContext(context)
-            return self.lsp_context
+                context = self.context_state.lsp_context.context
+                context.load()
+            self.context_state = ContextLoaded(lsp_context=LSPContext(context))
+            return self.context_state.lsp_context
         except Exception as e:
             # Only show the error message once
             if not self.has_raised_loading_error:
@@ -756,6 +819,14 @@ class SQLMeshLanguageServer:
                 self.has_raised_loading_error = True
 
             self.server.log_trace(f"Error creating context: {e}")
+            # Store the error in context state so subsequent requests show the actual error
+            # Try to preserve any partially loaded context if it exists
+            context = None
+            if isinstance(self.context_state, ContextLoaded):
+                context = self.context_state.lsp_context.context
+            elif isinstance(self.context_state, ContextFailed) and self.context_state.context:
+                context = self.context_state.context
+            self.context_state = ContextFailed(error_message=str(e), context=context)
             return None
 
     @staticmethod

{sqlmesh-0.195.1.dev2 → sqlmesh-0.195.1.dev4}/sqlmesh.egg-info/PKG-INFO

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

{sqlmesh-0.195.1.dev2 → sqlmesh-0.195.1.dev4}/sqlmesh.egg-info/SOURCES.txt

@@ -871,50 +871,50 @@ web/client/tailwind.config.cjs
 web/client/tsconfig.json
 web/client/vite.config.ts
 web/client/dist/index.html
-web/client/dist/assets/Audits-sYYNn2gO.js
-web/client/dist/assets/Banner-CIUVdc-z.js
-web/client/dist/assets/ChevronDownIcon-uQGmwNOL.js
-web/client/dist/assets/ChevronRightIcon-B65l6fcy.js
-web/client/dist/assets/Content-CAmZChZo.js
-web/client/dist/assets/Content-CPscdsPQ.js
-web/client/dist/assets/Data-CexnpfbT.js
-web/client/dist/assets/DataCatalog-zqv8aDld.js
-web/client/dist/assets/Editor-BPOAPXLv.js
-web/client/dist/assets/Editor-EsVsPMR8.js
-web/client/dist/assets/Errors-0iATm2JG.js
-web/client/dist/assets/FileExplorer-PTjWgl3B.js
-web/client/dist/assets/Footer-MSdMRGiG.js
-web/client/dist/assets/Header-SKoNqU5i.js
-web/client/dist/assets/Input-Cl0rlyUj.js
-web/client/dist/assets/Lineage-VxUfloHd.js
-web/client/dist/assets/ListboxShow-mv_m9Xqu.js
-web/client/dist/assets/ModelLineage-JG9oPMA2.js
-web/client/dist/assets/Models-Bq6zaOiV.js
-web/client/dist/assets/Page-D8EIuqN3.js
-web/client/dist/assets/Plan-Cr31O86c.js
-web/client/dist/assets/PlusCircleIcon-BA3W9uiU.js
-web/client/dist/assets/ReportErrors-CGhLWJSn.js
-web/client/dist/assets/Root-Cm9f5yLR.js
-web/client/dist/assets/SearchList-KwGRUK3b.js
-web/client/dist/assets/SelectEnvironment-Fdveg7Dn.js
-web/client/dist/assets/SourceList-CrxTdTD0.js
-web/client/dist/assets/SourceListItem-CbL1dN69.js
-web/client/dist/assets/SplitPane-Di60F7Gk.js
-web/client/dist/assets/Tests-Cjw3dIQ9.js
-web/client/dist/assets/Welcome-Cwxy-j_l.js
+web/client/dist/assets/Audits-DrNQaTQA.js
+web/client/dist/assets/Banner-DsG8aGIL.js
+web/client/dist/assets/ChevronDownIcon-CpQjfO01.js
+web/client/dist/assets/ChevronRightIcon-B-9lnqGr.js
+web/client/dist/assets/Content-CSMzf1cU.js
+web/client/dist/assets/Content-vfltDlhj.js
+web/client/dist/assets/Data-k1Dz6qdS.js
+web/client/dist/assets/DataCatalog-DlbWZyBp.js
+web/client/dist/assets/Editor--iBTA9xH.js
+web/client/dist/assets/Editor-C2pcRDlN.js
+web/client/dist/assets/Errors-CQW3lhsM.js
+web/client/dist/assets/FileExplorer-IAIPoTxM.js
+web/client/dist/assets/Footer-D77vCxKH.js
+web/client/dist/assets/Header-Ccb81kaP.js
+web/client/dist/assets/Input-D_Fp4W_B.js
+web/client/dist/assets/Lineage-BHLfUAmC.js
+web/client/dist/assets/ListboxShow-BIPQ-cqE.js
+web/client/dist/assets/ModelLineage-BrTZ2lm3.js
+web/client/dist/assets/Models-B6cHeIlA.js
+web/client/dist/assets/Page-S1ykkm5T.js
+web/client/dist/assets/Plan-DCD4m0nD.js
+web/client/dist/assets/PlusCircleIcon-C40sAC-m.js
+web/client/dist/assets/ReportErrors-CxqgDcYY.js
+web/client/dist/assets/Root-DtUrFfE_.js
+web/client/dist/assets/SearchList-DoRflby9.js
+web/client/dist/assets/SelectEnvironment-equ86mF1.js
+web/client/dist/assets/SourceList-DT-sBCAQ.js
+web/client/dist/assets/SourceListItem-MFS9vpg9.js
+web/client/dist/assets/SplitPane-D2cB3Qkv.js
+web/client/dist/assets/Tests-Duzk1K8j.js
+web/client/dist/assets/Welcome-BgAHQe0m.js
 web/client/dist/assets/_commonjs-dynamic-modules-TDtrdbi3.js
-web/client/dist/assets/context-DKh7qNl9.js
-web/client/dist/assets/context-MffevtK7.js
-web/client/dist/assets/editor-BYglYCad.js
-web/client/dist/assets/file-BATQOjrJ.js
-web/client/dist/assets/floating-ui.react-dom-DHc129jH.js
+web/client/dist/assets/context-CXV8eAaM.js
+web/client/dist/assets/context-g_ECqgnp.js
+web/client/dist/assets/editor-BddPKujl.js
+web/client/dist/assets/file-DUr_Gx7F.js
+web/client/dist/assets/floating-ui.react-dom-DTKAUIBI.js
 web/client/dist/assets/help-6Q7f7fjd.js
-web/client/dist/assets/help-iFVHtKo0.js
-web/client/dist/assets/index-DA7iL93d.js
-web/client/dist/assets/index-DZ2vKQ1l.js
-web/client/dist/assets/plan-8UAMXVjx.js
-web/client/dist/assets/popover-DE6_QtGZ.js
-web/client/dist/assets/project-CfZrmqZ7.js
+web/client/dist/assets/help-LQUMs6nO.js
+web/client/dist/assets/index-BsS6H8XQ.js
+web/client/dist/assets/index-DvxjegO0.js
+web/client/dist/assets/plan-DS237yuz.js
+web/client/dist/assets/popover-CiiY17g7.js
+web/client/dist/assets/project-DUHOhQT2.js
 web/client/dist/css/base.css
 web/client/dist/css/design.css
 web/client/dist/favicons/favicon.svg

{sqlmesh-0.195.1.dev2 → sqlmesh-0.195.1.dev4}/vscode/extension/src/lsp/custom.ts

@@ -14,7 +14,7 @@ interface RenderModelRequest {
   textDocumentUri: string
 }
 
-interface RenderModelResponse {
+interface RenderModelResponse extends BaseResponse {
   models: RenderModelEntry[]
 }
 
@@ -25,7 +25,6 @@ export interface RenderModelEntry {
   rendered_query: string
 }
 
-// @eslint-disable-next-line  @typescript-eslint/consistent-type-definition
 export type CustomLSPMethods =
   | AllModelsMethod
   | AbstractAPICall
@@ -40,7 +39,7 @@ interface AllModelsRequest {
   }
 }
 
-interface AllModelsResponse {
+interface AllModelsResponse extends BaseResponse {
   models: string[]
   keywords: string[]
 }
@@ -55,9 +54,11 @@ export interface AbstractAPICallRequest {
 export interface AbstractAPICall {
   method: 'sqlmesh/api'
   request: AbstractAPICallRequest
-  response: object
+  response: AbstractAPICallResponse
 }
 
+type AbstractAPICallResponse = object & BaseResponse
+
 export interface AllModelsForRenderMethod {
   method: 'sqlmesh/all_models_for_render'
   request: AllModelsForRenderRequest
@@ -67,7 +68,7 @@ export interface AllModelsForRenderMethod {
 // eslint-disable-next-line @typescript-eslint/no-empty-object-type
 interface AllModelsForRenderRequest {}
 
-interface AllModelsForRenderResponse {
+interface AllModelsForRenderResponse extends BaseResponse {
   models: ModelForRendering[]
 }
 
@@ -87,7 +88,7 @@ export interface SupportedMethodsMethod {
 // eslint-disable-next-line @typescript-eslint/no-empty-object-type
 interface SupportedMethodsRequest {}
 
-interface SupportedMethodsResponse {
+interface SupportedMethodsResponse extends BaseResponse {
   methods: CustomMethod[]
 }
 
@@ -105,4 +106,8 @@ export interface FormatProjectMethod {
 interface FormatProjectRequest {}
 
 // eslint-disable-next-line @typescript-eslint/no-empty-object-type
-interface FormatProjectResponse {}
+interface FormatProjectResponse extends BaseResponse {}
+
+interface BaseResponse {
+  response_error?: string
+}

{sqlmesh-0.195.1.dev2 → sqlmesh-0.195.1.dev4}/vscode/extension/src/lsp/lsp.ts

@@ -256,6 +256,9 @@ export class LSPClient implements Disposable {
 
     try {
       const result = await this.client.sendRequest<Response>(method, request)
+      if (result.response_error) {
+        return err(result.response_error)
+      }
       return ok(result)
     } catch (error) {
       traceError(

{sqlmesh-0.195.1.dev2 → sqlmesh-0.195.1.dev4}/vscode/extension/src/webviews/lineagePanel.ts

@@ -10,6 +10,7 @@ import {
 } from 'vscode'
 import { getWorkspaceFolders } from '../utilities/common/vscodeapi'
 import { LSPClient } from '../lsp/lsp'
+import { isErr } from '@bus/result'
 
 export class LineagePanel implements WebviewViewProvider, Disposable {
   public static readonly viewType = 'sqlmesh.lineage'
@@ -97,12 +98,40 @@ export class LineagePanel implements WebviewViewProvider, Disposable {
                   'sqlmesh/api',
                   payload.params,
                 )
-                const responseCallback: CallbackEvent = {
-                  key: 'rpcResponse',
-                  payload: {
-                    requestId,
-                    result: response,
-                  },
+                let responseCallback: CallbackEvent
+                if (isErr(response)) {
+                  let errorMessage: string
+                  switch (response.error.type) {
+                    case 'generic':
+                      errorMessage = response.error.message
+                      break
+                    case 'invalid_state':
+                      errorMessage = `Invalid state: ${response.error.message}`
+                      break
+                    case 'sqlmesh_outdated':
+                      errorMessage = `SQLMesh version issue: ${response.error.message}`
+                      break
+                    default:
+                      errorMessage = 'Unknown error'
+                  }
+                  responseCallback = {
+                    key: 'rpcResponse',
+                    payload: {
+                      requestId,
+                      result: {
+                        ok: false,
+                        error: errorMessage,
+                      },
+                    },
+                  }
+                } else {
+                  responseCallback = {
+                    key: 'rpcResponse',
+                    payload: {
+                      requestId,
+                      result: response,
+                    },
+                  }
                 }
                 await webviewView.webview.postMessage(responseCallback)
                 break

sqlmesh-0.195.1.dev4/vscode/extension/tests/broken_project.spec.ts

@@ -0,0 +1,190 @@
+import { test } from '@playwright/test'
+import fs from 'fs-extra'
+import os from 'os'
+import path from 'path'
+import { openLineageView, startVSCode, SUSHI_SOURCE_PATH } from './utils'
+
+test('bad project, double model', async ({}) => {
+  const tempDir = await fs.mkdtemp(
+    path.join(os.tmpdir(), 'vscode-test-tcloud-'),
+  )
+  await fs.copy(SUSHI_SOURCE_PATH, tempDir)
+
+  // Read the customers.sql file
+  const customersSql = await fs.readFile(
+    path.join(tempDir, 'models', 'customers.sql'),
+    'utf8',
+  )
+
+  // Write the customers.sql file with a double model
+  await fs.writeFile(
+    path.join(tempDir, 'models', 'customers_duplicated.sql'),
+    customersSql,
+  )
+
+  const { window, close } = await startVSCode(tempDir)
+  try {
+    await window.waitForSelector('text=models')
+
+    await window
+      .getByRole('treeitem', { name: 'models', exact: true })
+      .locator('a')
+      .click()
+
+    await window
+      .getByRole('treeitem', { name: 'customers.sql', exact: true })
+      .locator('a')
+      .click()
+
+    await window.waitForSelector('text=Error creating context')
+
+    await window.waitForTimeout(1000)
+  } finally {
+    await close()
+    await fs.remove(tempDir)
+  }
+})
+
+test('working project, then broken through adding double model, then refixed', async ({}) => {
+  const tempDir = await fs.mkdtemp(
+    path.join(os.tmpdir(), 'vscode-test-tcloud-'),
+  )
+  await fs.copy(SUSHI_SOURCE_PATH, tempDir)
+
+  const { window, close } = await startVSCode(tempDir)
+  try {
+    // First, verify the project is working correctly
+    await window.waitForSelector('text=models')
+
+    // Open the lineage view to confirm it loads properly
+    await openLineageView(window)
+    await window.waitForSelector('text=Loaded SQLMesh context')
+
+    // Read the customers.sql file
+    const customersSql = await fs.readFile(
+      path.join(tempDir, 'models', 'customers.sql'),
+      'utf8',
+    )
+
+    // Add a duplicate model to break the project
+    await fs.writeFile(
+      path.join(tempDir, 'models', 'customers_duplicated.sql'),
+      customersSql,
+    )
+
+    // Open the customers model to trigger the error
+    await window
+      .getByRole('treeitem', { name: 'models', exact: true })
+      .locator('a')
+      .click()
+    await window
+      .getByRole('treeitem', { name: 'customers.sql', exact: true })
+      .locator('a')
+      .click()
+    // Save to refresh the context
+    await window.keyboard.press('Control+S')
+    await window.keyboard.press('Meta+S')
+
+    // Wait for the error to appear
+    // TODO: Selector doesn't work in the linage view
+    // await window.waitForSelector('text=Error')
+
+    // Remove the duplicated model to fix the project
+    await fs.remove(path.join(tempDir, 'models', 'customers_duplicated.sql'))
+
+    // Save again to refresh the context
+    await window.keyboard.press('Control+S')
+    await window.keyboard.press('Meta+S')
+
+    // Wait for the error to go away and context to reload
+    // TODO: Selector doesn't work in the linage view
+    // await window.waitForSelector('text=raw.demographics')
+  } finally {
+    await close()
+    await fs.remove(tempDir)
+  }
+})
+
+test('bad project, double model, then fixed', async ({}) => {
+  const tempDir = await fs.mkdtemp(
+    path.join(os.tmpdir(), 'vscode-test-tcloud-'),
+  )
+  await fs.copy(SUSHI_SOURCE_PATH, tempDir)
+
+  // Read the customers.sql file
+  const customersSql = await fs.readFile(
+    path.join(tempDir, 'models', 'customers.sql'),
+    'utf8',
+  )
+
+  // Write the customers.sql file with a double model
+  await fs.writeFile(
+    path.join(tempDir, 'models', 'customers_duplicated.sql'),
+    customersSql,
+  )
+
+  const { window, close } = await startVSCode(tempDir)
+  try {
+    await window.waitForSelector('text=models')
+
+    await window
+      .getByRole('treeitem', { name: 'models', exact: true })
+      .locator('a')
+      .click()
+
+    await window
+      .getByRole('treeitem', { name: 'customers.sql', exact: true })
+      .locator('a')
+      .click()
+
+    await window.waitForSelector('text=Error creating context')
+
+    // Remove the duplicated model
+    await fs.remove(path.join(tempDir, 'models', 'customers_duplicated.sql'))
+
+    // Open the linage view
+    await openLineageView(window)
+
+    // Wait for the error to go away
+    // TODO: Selector doesn't work in the linage view
+    // await window.waitForSelector('text=raw.demographics')
+  } finally {
+    await close()
+    await fs.remove(tempDir)
+  }
+})
+
+test('bad project, double model, check lineage', async ({}) => {
+  const tempDir = await fs.mkdtemp(
+    path.join(os.tmpdir(), 'vscode-test-tcloud-'),
+  )
+  await fs.copy(SUSHI_SOURCE_PATH, tempDir)
+
+  // Read the customers.sql file
+  const customersSql = await fs.readFile(
+    path.join(tempDir, 'models', 'customers.sql'),
+    'utf8',
+  )
+
+  // Write the customers.sql file with a double model
+  await fs.writeFile(
+    path.join(tempDir, 'models', 'customers_duplicated.sql'),
+    customersSql,
+  )
+
+  const { window, close } = await startVSCode(tempDir)
+  try {
+    await window.waitForSelector('text=models')
+
+    // Open the lineage view
+    await openLineageView(window)
+
+    await window.waitForSelector('text=Error creating context')
+    await window.waitForSelector('text=Error:')
+
+    await window.waitForTimeout(1000)
+  } finally {
+    await close()
+    await fs.remove(tempDir)
+  }
+})