From 059fd9403aed6012b428ba747dfceae58e810a75 Mon Sep 17 00:00:00 2001 From: Jordan Frazier <122494242+jordanrfrazier@users.noreply.github.com> Date: Fri, 27 Sep 2024 14:36:09 -0700 Subject: [PATCH] fix: vectorize authentication params (#3954) Fix vectorize authentication params --- .../base/langflow/components/vectorstores/AstraDB.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/src/backend/base/langflow/components/vectorstores/AstraDB.py b/src/backend/base/langflow/components/vectorstores/AstraDB.py index 2e151c8b8..a61a52f7d 100644 --- a/src/backend/base/langflow/components/vectorstores/AstraDB.py +++ b/src/backend/base/langflow/components/vectorstores/AstraDB.py @@ -339,9 +339,7 @@ class AstraVectorStoreComponent(LCVectorStoreComponent): authentication = {**(self.z_04_authentication or kwargs.get("z_04_authentication", {}))} api_key_name = self.z_02_api_key_name or kwargs.get("z_02_api_key_name") - provider_key_name = self.z_03_provider_api_key or kwargs.get("z_03_provider_api_key") - if provider_key_name: - authentication["providerKey"] = provider_key_name + provider_key = self.z_03_provider_api_key or kwargs.get("z_03_provider_api_key") if api_key_name: authentication["providerKey"] = api_key_name @@ -353,7 +351,7 @@ class AstraVectorStoreComponent(LCVectorStoreComponent): "authentication": authentication, "parameters": self.z_01_model_parameters or kwargs.get("z_01_model_parameters", {}), }, - "collection_embedding_api_key": self.z_03_provider_api_key or kwargs.get("z_03_provider_api_key"), + "collection_embedding_api_key": provider_key, } @check_cached_vector_store