From d3fdddaa5221c53e178639f275ebdc7b005f3c7e Mon Sep 17 00:00:00 2001 From: hajdul88 <52442977+hajdul88@users.noreply.github.com> Date: Wed, 13 Nov 2024 17:55:52 +0100 Subject: [PATCH] Revert "Checks the pgvector test issue" This reverts commit 0d27371467cbb4c32aef350e9751ab420ffee959. --- .../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 deb1e6b18..01691714b 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, connect_args={"timeout": 100}) + self.engine = create_async_engine(self.db_uri) self.sessionmaker = async_sessionmaker(bind=self.engine, expire_on_commit=False) async def embed_data(self, data: list[str]) -> list[list[float]]: