From 7a8ec815c54aee25547fcb21e91e4fa59f262a59 Mon Sep 17 00:00:00 2001 From: Pragyan Tripathi Date: Tue, 11 Nov 2025 09:19:59 +0530 Subject: [PATCH 1/2] fix(stitch): update type hints to use LLMProvider instead of LLMClient --- chuck_data/commands/stitch_tools.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/chuck_data/commands/stitch_tools.py b/chuck_data/commands/stitch_tools.py index 4fe31b9..78343c9 100644 --- a/chuck_data/commands/stitch_tools.py +++ b/chuck_data/commands/stitch_tools.py @@ -11,7 +11,7 @@ from typing import Dict, Any from chuck_data.clients.databricks import DatabricksAPIClient -from chuck_data.llm.client import LLMClient +from chuck_data.llm.provider import LLMProvider from chuck_data.config import get_amperity_token from .pii_tools import _helper_scan_schema_for_pii_logic from .cluster_init_tools import _helper_upload_cluster_init_logic @@ -31,7 +31,7 @@ def _helper_setup_stitch_logic( client: DatabricksAPIClient, - llm_client_instance: LLMClient, + llm_client_instance: LLMProvider, target_catalog: str, target_schema: str, ) -> Dict[str, Any]: @@ -54,7 +54,7 @@ def _helper_setup_stitch_logic( def _helper_prepare_stitch_config( client: DatabricksAPIClient, - llm_client_instance: LLMClient, + llm_client_instance: LLMProvider, target_catalog: str, target_schema: str, ) -> Dict[str, Any]: @@ -228,7 +228,7 @@ def _helper_prepare_stitch_config( def _helper_modify_stitch_config( current_config: Dict[str, Any], modification_request: str, - llm_client_instance: LLMClient, + llm_client_instance: LLMProvider, metadata: Dict[str, Any], ) -> Dict[str, Any]: """Phase 2: Modify Stitch configuration based on user request using LLM.""" From 0411d070c9eb581b119e5b9914ddb49797771e80 Mon Sep 17 00:00:00 2001 From: Pragyan Tripathi Date: Tue, 11 Nov 2025 09:20:04 +0530 Subject: [PATCH 2/2] fix(pii): update type hints to use LLMProvider instead of LLMClient --- chuck_data/commands/pii_tools.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/chuck_data/commands/pii_tools.py b/chuck_data/commands/pii_tools.py index cbdc06c..5395892 100644 --- a/chuck_data/commands/pii_tools.py +++ b/chuck_data/commands/pii_tools.py @@ -11,13 +11,13 @@ from typing import Dict, Any, Optional from chuck_data.clients.databricks import DatabricksAPIClient -from chuck_data.llm.client import LLMClient +from chuck_data.llm.provider import LLMProvider from chuck_data.ui.tui import get_console def _helper_tag_pii_columns_logic( databricks_client: DatabricksAPIClient, - llm_client_instance: LLMClient, + llm_client_instance: LLMProvider, table_name_param: str, catalog_name_context: Optional[str] = None, schema_name_context: Optional[str] = None, @@ -160,7 +160,7 @@ def _helper_tag_pii_columns_logic( def _helper_scan_schema_for_pii_logic( client: DatabricksAPIClient, - llm_client_instance: LLMClient, + llm_client_instance: LLMProvider, catalog_name: str, schema_name: str, show_progress: bool = True,