gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Revert "Checks the pgvector test issue"

Browse source 
This reverts commit 0d27371467.
This commit is contained in:
hajdul88 2024-11-13 17:55:52 +01:00
parent 0d27371467
commit d3fdddaa52
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
cognee/infrastructure/databases/vector/pgvector/PGVectorAdapter.py
View file

@ -33,7 +33,7 @@ class PGVectorAdapter(SQLAlchemyAdapter, VectorDBInterface):
self.api_key = api_key self.api_key = api_key
self.embedding_engine = embedding_engine self.embedding_engine = embedding_engine
self.db_uri: str = connection_string 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) self.sessionmaker = async_sessionmaker(bind=self.engine, expire_on_commit=False)
async def embed_data(self, data: list[str]) -> list[list[float]]: async def embed_data(self, data: list[str]) -> list[list[float]]: