Fix lsp test

Author: treysp

sqlmesh/cli/project_init.py

@@ -36,7 +36,10 @@ def _gen_config(
     start: t.Optional[str],
     template: ProjectTemplate,
     cli_mode: InitCliMode,
+    dialect: t.Optional[str] = None,
 ) -> str:
+    project_dialect = dialect or DIALECT_TO_TYPE.get(engine_type)
+
     connection_settings = (
         settings
         or """      type: duckdb
@@ -96,7 +99,7 @@ def _gen_config(
 # https://sqlmesh.readthedocs.io/en/stable/reference/model_configuration/#model-defaults
 
 model_defaults:
-  dialect: {DIALECT_TO_TYPE.get(engine_type)}
+  dialect: {project_dialect}
   start: {start or yesterday_ds()} # Start date for backfill history
   cron: '@daily'    # Run models daily at 12am UTC (can override per model)
 
@@ -274,6 +277,7 @@ def _gen_example_objects(schema_name: str) -> ExampleObjects:
 def init_example_project(
     path: t.Union[str, Path],
     engine_type: t.Optional[str],
+    dialect: t.Optional[str] = None,
     template: ProjectTemplate = ProjectTemplate.DEFAULT,
     pipeline: t.Optional[str] = None,
     dlt_path: t.Optional[str] = None,
@@ -314,17 +318,17 @@ def init_example_project(
     settings = None
     start = None
     if engine_type and template == ProjectTemplate.DLT:
-        dialect = DIALECT_TO_TYPE.get(engine_type)
-        if pipeline and dialect:
+        project_dialect = dialect or DIALECT_TO_TYPE.get(engine_type)
+        if pipeline and project_dialect:
             dlt_models, settings, start = generate_dlt_models_and_settings(
-                pipeline_name=pipeline, dialect=dialect, dlt_path=dlt_path
+                pipeline_name=pipeline, dialect=project_dialect, dlt_path=dlt_path
             )
         else:
             raise SQLMeshError(
                 "Please provide a DLT pipeline with the `--dlt-pipeline` flag to generate a SQLMesh project from DLT."
             )
 
-    _create_config(config_path, engine_type, settings, start, template, cli_mode)
+    _create_config(config_path, engine_type, dialect, settings, start, template, cli_mode)
     if template == ProjectTemplate.DBT:
         return config_path
 
@@ -359,12 +363,13 @@ def _create_folders(target_folders: t.Sequence[Path]) -> None:
 def _create_config(
     config_path: Path,
     engine_type: t.Optional[str],
+    dialect: t.Optional[str],
     settings: t.Optional[str],
     start: t.Optional[str],
     template: ProjectTemplate,
     cli_mode: InitCliMode,
 ) -> None:
-    project_config = _gen_config(engine_type, settings, start, template, cli_mode)
+    project_config = _gen_config(engine_type, settings, start, template, cli_mode, dialect)
 
     _write_file(
         config_path,

sqlmesh/magics.py

@@ -195,6 +195,7 @@ def init(self, line: str) -> None:
         init_example_project(
             path=args.path,
             engine_type=args.engine,
+            dialect=None,
             template=project_template,
             pipeline=args.dlt_pipeline,
             dlt_path=args.dlt_path,

tests/lsp/test_reference_model_column_prefix.py

@@ -1,7 +1,7 @@
 from pathlib import Path
 
 from lsprotocol.types import Position
-from sqlmesh.cli.example_project import init_example_project
+from sqlmesh.cli.project_init import init_example_project
 from sqlmesh.core.context import Context
 from sqlmesh.lsp.context import LSPContext, ModelTarget
 from sqlmesh.lsp.reference import get_all_references
@@ -84,7 +84,9 @@ def test_column_prefix_references_are_found():
 
 def test_quoted_uppercase_table_and_column_references(tmp_path: Path):
     # Initialize example project in temporary directory with case sensitive normalization
-    init_example_project(tmp_path, dialect="duckdb,normalization_strategy=case_sensitive")
+    init_example_project(
+        tmp_path, engine_type="duckdb", dialect="duckdb,normalization_strategy=case_sensitive"
+    )
 
     # Create a model with quoted uppercase schema and table names
     models_dir = tmp_path / "models"