fix: astra vectorize auth error (#2606)

This commit is contained in:
Nicolò Boschi 2024-07-09 17:23:30 +02:00 committed by GitHub
commit e7e53abfbe
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -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,