@@ -5,10 +5,10 @@ from unstructured_ingest.v2.pipeline.pipeline import Pipeline
55from unstructured_ingest.v2.interfaces import ProcessorConfig
66
77from unstructured_ingest.v2.processes.connectors.sql.databricks_delta_tables import (
8- DatabrickDeltaTablesConnectionConfig ,
9- DatabrickDeltaTablesAccessConfig ,
10- DatabrickDeltaTablesUploadStagerConfig ,
11- DatabrickDeltaTablesUploaderConfig
8+ DatabricksDeltaTablesConnectionConfig ,
9+ DatabricksDeltaTablesAccessConfig ,
10+ DatabricksDeltaTablesUploadStagerConfig ,
11+ DatabricksDeltaTablesUploaderConfig
1212)
1313
1414from unstructured_ingest.v2.processes.connectors.local import (
@@ -44,26 +44,26 @@ if __name__ == "__main__":
4444 embedder_config = EmbedderConfig(embedding_provider = " huggingface" ),
4545
4646 # For authenticating with Databricks personal access tokens.
47- destination_connection_config = DatabrickDeltaTablesConnectionConfig (
48- access_config = DatabrickDeltaTablesAccessConfig (
47+ destination_connection_config = DatabricksDeltaTablesConnectionConfig (
48+ access_config = DatabricksDeltaTablesAccessConfig (
4949 token = os.getenv(" DATABRICKS_TOKEN" ),
5050 ),
5151 server_hostname = os.getenv(" DATABRICKS_HOST" ),
5252 http_path = os.getenv(" DATABRICKS_HTTP_PATH" )
5353 ),
5454
5555 # For authenticating with Databricks managed service principals.
56- # destination_connection_config=DatabrickDeltaTablesConnectionConfig (
57- # access_config=DatabrickDeltaTablesAccessConfig (
56+ # destination_connection_config=DatabricksDeltaTablesConnectionConfig (
57+ # access_config=DatabricksDeltaTablesAccessConfig (
5858 # client_id=os.getenv("DATABRICKS_CLIENT_ID"),
5959 # client_secret=os.getenv("DATABRICKS_CLIENT_SECRET")
6060 # ),
6161 # server_hostname=os.getenv("DATABRICKS_HOST"),
6262 # http_path=os.getenv("DATABRICKS_HTTP_PATH")
6363 # ),
6464
65- stager_config = DatabrickDeltaTablesUploadStagerConfig (),
66- uploader_config = DatabrickDeltaTablesUploaderConfig (
65+ stager_config = DatabricksDeltaTablesUploadStagerConfig (),
66+ uploader_config = DatabricksDeltaTablesUploaderConfig (
6767 catalog = os.getenv(" DATABRICKS_CATALOG" ),
6868 database = os.getenv(" DATABRICKS_DATABASE" ),
6969 table_name = os.getenv(" DATABRICKS_TABLE" ),
0 commit comments