File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change 1111from typing import Dict , Any , Optional
1212
1313from chuck_data .clients .databricks import DatabricksAPIClient
14- from chuck_data .llm .client import LLMClient
14+ from chuck_data .llm .provider import LLMProvider
1515from chuck_data .ui .tui import get_console
1616
1717
1818def _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
161161def _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 ,
You can’t perform that action at this time.
0 commit comments