From 68fc59e5bec06919e11d51394ac29481f2e3ec81 Mon Sep 17 00:00:00 2001 From: biturd Date: Mon, 8 Sep 2025 03:55:01 +0800 Subject: [PATCH] fix: graphiti search only connect with default db 'neo4j' and custom database config invalid --- graphiti_core/driver/neo4j_driver.py | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) diff --git a/graphiti_core/driver/neo4j_driver.py b/graphiti_core/driver/neo4j_driver.py index 7ac9a5a8..2f421693 100644 --- a/graphiti_core/driver/neo4j_driver.py +++ b/graphiti_core/driver/neo4j_driver.py @@ -38,17 +38,20 @@ class Neo4jDriver(GraphDriver): self._database = database async def execute_query(self, cypher_query_: LiteralString, **kwargs: Any) -> EagerResult: - # Check if database_ is provided in kwargs. - # If not populated, set the value to retain backwards compatibility + # Extract database from kwargs or use default + database = kwargs.pop('database_', self._database) params = kwargs.pop('params', None) if params is None: params = {} - params.setdefault('database_', self._database) try: - result = await self.client.execute_query(cypher_query_, parameters_=params, **kwargs) + result = await self.client.execute_query( + cypher_query_, parameters_=params, database_=database, **kwargs + ) except Exception as e: - logger.error(f'Error executing Neo4j query: {e}\n{cypher_query_}\n{params}') + logger.error( + f'Error executing Neo4j query: {e}\n{cypher_query_}\nDatabase: {database}\nParams: {params}' + ) raise return result