From e7e53abfbe47f067970ec52120850d3201a903b3 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Nicol=C3=B2=20Boschi?= Date: Tue, 9 Jul 2024 17:23:30 +0200 Subject: [PATCH] fix: astra vectorize auth error (#2606) --- .../base/langflow/components/vectorstores/AstraDB.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/src/backend/base/langflow/components/vectorstores/AstraDB.py b/src/backend/base/langflow/components/vectorstores/AstraDB.py index 7aec7ef30..c884ffbb8 100644 --- a/src/backend/base/langflow/components/vectorstores/AstraDB.py +++ b/src/backend/base/langflow/components/vectorstores/AstraDB.py @@ -195,9 +195,12 @@ class AstraVectorStoreComponent(LCVectorStoreComponent): } dict_options["parameters"] = {k: v for k, v in dict_options.get("parameters", {}).items() if k and v} embedding_dict = { - "collection_vector_service_options": CollectionVectorServiceOptions.from_dict(dict_options), - "collection_embedding_api_key": self.embedding.get("collection_embedding_api_key"), + "collection_vector_service_options": CollectionVectorServiceOptions.from_dict(dict_options) } + collection_embedding_api_key = self.embedding.get("collection_embedding_api_key") + if collection_embedding_api_key: + embedding_dict["collection_embedding_api_key"] = collection_embedding_api_key + vector_store_kwargs = { **embedding_dict, "collection_name": self.collection_name,