diff --git a/src/datasight/cli.py b/src/datasight/cli.py index b062451..6aa4bc4 100644 --- a/src/datasight/cli.py +++ b/src/datasight/cli.py @@ -130,7 +130,7 @@ async def run_ask_pipeline( api_key=settings.llm.api_key, base_url=settings.llm.base_url, timeout=settings.llm.timeout, - model=settings.llm.model, + model=resolved_model, ) sql_runner = create_sql_runner_from_settings(settings.database, project_dir) diff --git a/src/datasight/cli_commands/generate.py b/src/datasight/cli_commands/generate.py index 0db5afd..4b5cfaa 100644 --- a/src/datasight/cli_commands/generate.py +++ b/src/datasight/cli_commands/generate.py @@ -309,7 +309,7 @@ async def _run(): # noqa: C901 api_key=settings.llm.api_key, base_url=settings.llm.base_url, timeout=settings.llm.timeout, - model=settings.llm.model, + model=resolved_model, ) if sqlite_source_path is not None: diff --git a/src/datasight/cli_commands/verify.py b/src/datasight/cli_commands/verify.py index e81ded3..67686cc 100644 --- a/src/datasight/cli_commands/verify.py +++ b/src/datasight/cli_commands/verify.py @@ -97,7 +97,7 @@ async def _run(): api_key=settings.llm.api_key, base_url=settings.llm.base_url, timeout=settings.llm.timeout, - model=settings.llm.model, + model=resolved_model, ) sql_runner = create_sql_runner_from_settings(settings.database, project_dir)