diff --git a/cognee/infrastructure/databases/graph/kuzu/adapter.py b/cognee/infrastructure/databases/graph/kuzu/adapter.py index eb2dd9987..995edf9ed 100644 --- a/cognee/infrastructure/databases/graph/kuzu/adapter.py +++ b/cognee/infrastructure/databases/graph/kuzu/adapter.py @@ -225,22 +225,24 @@ class KuzuAdapter(GraphDBInterface): logger.error(f"Query execution failed: {str(e)}") raise - if not cache_config.caching: + if cache_config.caching: async with self._connection_change_lock: self.open_connections += 1 if self._is_closed: + self.redis_lock.acquire() self.reopen() logger.info(f"Open connections after open: {self.open_connections}") result = await loop.run_in_executor(self.executor, blocking_query) - if not cache_config.caching: + if cache_config.caching: async with self._connection_change_lock: self.open_connections -= 1 logger.info(f"Opened connections after closing {self.open_connections}") if self.open_connections == 0: self.connection.execute("CHECKPOINT;") self.close() + self.redis_lock.release() return result