From 0d27371467cbb4c32aef350e9751ab420ffee959 Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Wed, 13 Nov 2024 17:51:25 +0100
Subject: [PATCH] Checks the pgvector test issue

---
 .../infrastructure/databases/vector/pgvector/PGVectorAdapter.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/cognee/infrastructure/databases/vector/pgvector/PGVectorAdapter.py b/cognee/infrastructure/databases/vector/pgvector/PGVectorAdapter.py
index 01691714b..deb1e6b18 100644
--- a/cognee/infrastructure/databases/vector/pgvector/PGVectorAdapter.py
+++ b/cognee/infrastructure/databases/vector/pgvector/PGVectorAdapter.py
@@ -33,7 +33,7 @@ class PGVectorAdapter(SQLAlchemyAdapter, VectorDBInterface):
         self.api_key = api_key
         self.embedding_engine = embedding_engine
         self.db_uri: str = connection_string
-        self.engine = create_async_engine(self.db_uri)
+        self.engine = create_async_engine(self.db_uri, connect_args={"timeout": 100})
         self.sessionmaker = async_sessionmaker(bind=self.engine, expire_on_commit=False)
 
     async def embed_data(self, data: list[str]) -> list[list[float]]: