From 81220c44efffdd583737c21813958133fea81315 Mon Sep 17 00:00:00 2001 From: Cameron Hutchison Date: Tue, 17 Dec 2024 18:46:15 -0600 Subject: [PATCH] Update integrations/langchain/src/databricks_langchain/vectorstores.py Co-authored-by: Prithvi Kannan <46332835+prithvikannan@users.noreply.github.com> --- .../langchain/src/databricks_langchain/vectorstores.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/integrations/langchain/src/databricks_langchain/vectorstores.py b/integrations/langchain/src/databricks_langchain/vectorstores.py index 3fbf31f..1af654b 100644 --- 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: