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

Fix: Solves the issue of Neo4j concurrent sessions

Browse source
This commit is contained in:
hajdul88 2024-11-11 16:31:50 +01:00
parent 64424bd3a9
commit 14868ea932
2 changed files with 2 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
cognee/infrastructure/databases/graph/neo4j_driver/adapter.py
View file

@ -18,7 +18,7 @@ class Neo4jAdapter(GraphDBInterface):
graph_database_url: str, graph_database_url: str,
graph_database_username: str, graph_database_username: str,
graph_database_password: str, graph_database_password: str,
driver: Optional[Any] = None, driver: Optional[Any] = None
): ):
self.driver = driver or AsyncGraphDatabase.driver( self.driver = driver or AsyncGraphDatabase.driver(
graph_database_url, graph_database_url,
@ -26,9 +26,6 @@ class Neo4jAdapter(GraphDBInterface):
max_connection_lifetime = 120 max_connection_lifetime = 120
) )
async def close(self) -> None:
await self.driver.close()
@asynccontextmanager @asynccontextmanager
async def get_session(self) -> AsyncSession: async def get_session(self) -> AsyncSession:
async with self.driver.session() as session: async with self.driver.session() as session:
@ -43,7 +40,6 @@ class Neo4jAdapter(GraphDBInterface):
async with self.get_session() as session: async with self.get_session() as session:
result = await session.run(query, parameters = params) result = await session.run(query, parameters = params)
data = await result.data() data = await result.data()
await self.close()
return data return data
except Neo4jError as error: except Neo4jError as error:
logger.error("Neo4j query error: %s", error, exc_info = True) logger.error("Neo4j query error: %s", error, exc_info = True)

2
cognee/tasks/graph/query_graph_connections.py
View file

@ -37,7 +37,7 @@ async def query_graph_connections(query: str, exploration_levels = 1) -> list[(s
return [] return []
node_connections_results = await asyncio.gather( node_connections_results = await asyncio.gather(
*[graph_engine.get_connections(result.id) for result in relevant_results] *[graph_engine.get_connections(str(result.id)) for result in relevant_results]
) )
node_connections = [] node_connections = []