jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

fix: Upgrade astradb version and graph db component (#4580)

Browse source
This commit is contained in:
Eric Hare 2024-11-13 12:24:25 -08:00 • committed by GitHub
commit 2cf3881dc6
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 10 additions and 412 deletions
Download patch file Download diff file Expand all files Collapse all files

13
src/backend/base/langflow/components/vectorstores/astradb_graph.py
View file

@ -50,14 +50,8 @@ class AstraGraphVectorStoreComponent(LCVectorStoreComponent):
required=True,
),
StrInput(
name="link_to_metadata_key",
display_name="Outgoing links metadata key",
info="Metadata key used for outgoing links.",
advanced=True,
),
StrInput(
name="link_from_metadata_key",
display_name="Incoming links metadata key",
name="metadata_incoming_links_key",
display_name="Metadata incoming links key",
info="Metadata key used for incoming links.",
advanced=True,
),
@ -214,8 +208,7 @@ class AstraGraphVectorStoreComponent(LCVectorStoreComponent):
vector_store = AstraDBGraphVectorStore(
embedding=self.embedding,
collection_name=self.collection_name,
link_to_metadata_key=self.link_to_metadata_key or "links_to",
link_from_metadata_key=self.link_from_metadata_key or "links_from",
metadata_incoming_links_key=self.metadata_incoming_links_key or "incoming_links",
token=self.token,
api_endpoint=self.api_endpoint,
namespace=self.namespace or None,