From 4208e3f961903ce1bb5e24df7fa50d245c0b975a Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Fri, 10 Nov 2023 10:46:57 -0300
Subject: [PATCH] Add support for None documents in PostgresqlVectorComponent

---
 src/backend/langflow/components/vectorstores/pgvector.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/src/backend/langflow/components/vectorstores/pgvector.py b/src/backend/langflow/components/vectorstores/pgvector.py
index f8b4bae74..eecf4d187 100644
--- a/src/backend/langflow/components/vectorstores/pgvector.py
+++ b/src/backend/langflow/components/vectorstores/pgvector.py
@@ -1,4 +1,4 @@
-from typing import Optional
+from typing import Optional, List
 from langflow import CustomComponent
 
 from langchain.vectorstores.pgvector import PGVector
@@ -59,6 +59,13 @@ class PostgresqlVectorComponent(CustomComponent):
         """
 
         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(
                 embedding=embedding,
                 documents=documents,