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

headlinevc · Nov 13, 2024 · 2cf3881 · 2cf3881
1 parent 2b00041
commit 2cf3881
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 412 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -96,7 +96,7 @@ dependencies = [
     "langchain-google-genai>=1.0.8",
     "langchain-cohere>=0.1.5",
     "langchain-anthropic>=0.1.23",
-    "langchain-astradb~=0.5.1",
+    "langchain-astradb~=0.5.2",
     "langchain-openai~=0.2.2",
     "langchain-google-vertexai~=2.0.5",
     "langchain-groq>=0.1.9",

diff --git a/src/backend/base/langflow/components/vectorstores/astradb_graph.py b/src/backend/base/langflow/components/vectorstores/astradb_graph.py
@@ -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 @@ def build_vector_store(self):
             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,