From 3e6e6b44b55d105d42d5b9f711811d957ad43575 Mon Sep 17 00:00:00 2001
From: Lanznx <109305092@g.nccu.edu.tw>
Date: Fri, 10 Nov 2023 18:39:24 +0800
Subject: [PATCH] put try/except block

---
 .../components/vectorstores/pgvector.py       | 21 +++++++++++--------
 1 file changed, 12 insertions(+), 9 deletions(-)

diff --git a/src/backend/langflow/components/vectorstores/pgvector.py b/src/backend/langflow/components/vectorstores/pgvector.py
index 607e66968..d539fe289 100644
--- a/src/backend/langflow/components/vectorstores/pgvector.py
+++ b/src/backend/langflow/components/vectorstores/pgvector.py
@@ -14,7 +14,9 @@ class PostgresqlVectorComponent(CustomComponent):
 
     display_name: str = "PGVector"
     description: str = "Implementation of Vector Store using PostgreSQL"
-    documentation = "https://python.langchain.com/docs/integrations/vectorstores/pgvector"
+    documentation = (
+        "https://python.langchain.com/docs/integrations/vectorstores/pgvector"
+    )
     beta = True
 
     def build_config(self):
@@ -56,11 +58,12 @@ class PostgresqlVectorComponent(CustomComponent):
         - VectorStore: The Vector Store object.
         """
 
-        return PGVector.from_documents(
-            embedding=embedding,
-            documents=documents,
-            collection_name=collection_name,
-            connection_string=pg_server_url,
-        )
-
-
+        try:
+            return PGVector.from_documents(
+                embedding=embedding,
+                documents=documents,
+                collection_name=collection_name,
+                connection_string=pg_server_url,
+            )
+        except Exception as e:
+            raise RuntimeError(f"Failed to build Vector Store: {str(e)}")