From 96ebc115f3610d1a0ebc95054bfc16aef1d6d198 Mon Sep 17 00:00:00 2001 From: coolgo0811 Date: Sun, 7 Jan 2024 12:12:06 +0800 Subject: [PATCH] Rename the component's output. --- src/backend/langflow/components/vectorstores/pgvector.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/backend/langflow/components/vectorstores/pgvector.py b/src/backend/langflow/components/vectorstores/pgvector.py index 337ea2271..22e16e460 100644 --- a/src/backend/langflow/components/vectorstores/pgvector.py +++ b/src/backend/langflow/components/vectorstores/pgvector.py @@ -56,13 +56,13 @@ class PGVectorComponent(CustomComponent): try: if documents is None: - embedding = PGVector.from_existing_index( + vector_store = PGVector.from_existing_index( embedding=embedding, collection_name=collection_name, connection_string=pg_server_url, ) - embedding = PGVector.from_documents( + vector_store = PGVector.from_documents( embedding=embedding, documents=documents, collection_name=collection_name, @@ -70,4 +70,4 @@ class PGVectorComponent(CustomComponent): ) except Exception as e: raise RuntimeError(f"Failed to build PGVector: {e}") - return embedding + return vector_store