Add support for None documents in
PostgresqlVectorComponent
This commit is contained in:
parent
682e51bf6a
commit
4208e3f961
1 changed files with 8 additions and 1 deletions
|
|
@ -1,4 +1,4 @@
|
||||||
from typing import Optional
|
from typing import Optional, List
|
||||||
from langflow import CustomComponent
|
from langflow import CustomComponent
|
||||||
|
|
||||||
from langchain.vectorstores.pgvector import PGVector
|
from langchain.vectorstores.pgvector import PGVector
|
||||||
|
|
@ -59,6 +59,13 @@ class PostgresqlVectorComponent(CustomComponent):
|
||||||
"""
|
"""
|
||||||
|
|
||||||
try:
|
try:
|
||||||
|
if documents is None:
|
||||||
|
return PGVector.from_existing_index(
|
||||||
|
embedding=embedding,
|
||||||
|
collection_name=collection_name,
|
||||||
|
connection_string=pg_server_url,
|
||||||
|
)
|
||||||
|
|
||||||
return PGVector.from_documents(
|
return PGVector.from_documents(
|
||||||
embedding=embedding,
|
embedding=embedding,
|
||||||
documents=documents,
|
documents=documents,
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue