Skip to content

Commit 0411d07

Browse files
committed
fix(pii): update type hints to use LLMProvider instead of LLMClient
1 parent 7a8ec81 commit 0411d07

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

chuck_data/commands/pii_tools.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -11,13 +11,13 @@
1111
from typing import Dict, Any, Optional
1212

1313
from chuck_data.clients.databricks import DatabricksAPIClient
14-
from chuck_data.llm.client import LLMClient
14+
from chuck_data.llm.provider import LLMProvider
1515
from chuck_data.ui.tui import get_console
1616

1717

1818
def _helper_tag_pii_columns_logic(
1919
databricks_client: DatabricksAPIClient,
20-
llm_client_instance: LLMClient,
20+
llm_client_instance: LLMProvider,
2121
table_name_param: str,
2222
catalog_name_context: Optional[str] = None,
2323
schema_name_context: Optional[str] = None,
@@ -160,7 +160,7 @@ def _helper_tag_pii_columns_logic(
160160

161161
def _helper_scan_schema_for_pii_logic(
162162
client: DatabricksAPIClient,
163-
llm_client_instance: LLMClient,
163+
llm_client_instance: LLMProvider,
164164
catalog_name: str,
165165
schema_name: str,
166166
show_progress: bool = True,

0 commit comments

Comments
 (0)