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

Rename the component's output.

Browse source
This commit is contained in:
coolgo0811 2024-01-07 12:12:06 +08:00
commit 96ebc115f3
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
src/backend/langflow/components/vectorstores/pgvector.py
View file

@ -56,13 +56,13 @@ class PGVectorComponent(CustomComponent):
try: try:
if documents is None: if documents is None:
embedding = PGVector.from_existing_index( vector_store = PGVector.from_existing_index(
embedding=embedding, embedding=embedding,
collection_name=collection_name, collection_name=collection_name,
connection_string=pg_server_url, connection_string=pg_server_url,
) )
embedding = PGVector.from_documents( vector_store = PGVector.from_documents(
embedding=embedding, embedding=embedding,
documents=documents, documents=documents,
collection_name=collection_name, collection_name=collection_name,
@ -70,4 +70,4 @@ class PGVectorComponent(CustomComponent):
) )
except Exception as e: except Exception as e:
raise RuntimeError(f"Failed to build PGVector: {e}") raise RuntimeError(f"Failed to build PGVector: {e}")
return embedding return vector_store