Update integrations/langchain/src/databricks_langchain/vectorstores.py

Co-authored-by: Prithvi Kannan <[email protected]>
databricks · Dec 18, 2024 · 81220c4 · 81220c4
1 parent f5094cc
commit 81220c4
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/integrations/langchain/src/databricks_langchain/vectorstores.py b/integrations/langchain/src/databricks_langchain/vectorstores.py
@@ -235,7 +235,9 @@ def __init__(
             ) from e
 
         try:
-            self.index = VectorSearchClient(**client_args).get_index(
+            self.index = VectorSearchClient(**(client_args or {})).get_index(
+                endpoint_name=endpoint, index_name=index_name
+            )
                 endpoint_name=endpoint, index_name=index_name
             )
         except Exception as e: