From 31280471c94d4663005abffb54bc1e02a72340ef Mon Sep 17 00:00:00 2001 From: hajdul88 <52442977+hajdul88@users.noreply.github.com> Date: Tue, 7 Oct 2025 19:20:36 +0200 Subject: [PATCH] ruff --- cognee/infrastructure/databases/graph/kuzu/adapter.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/cognee/infrastructure/databases/graph/kuzu/adapter.py b/cognee/infrastructure/databases/graph/kuzu/adapter.py index 34300f7d5..eb2dd9987 100644 --- a/cognee/infrastructure/databases/graph/kuzu/adapter.py +++ b/cognee/infrastructure/databases/graph/kuzu/adapter.py @@ -51,7 +51,9 @@ class KuzuAdapter(GraphDBInterface): self.connection: Optional[Connection] = None self.executor = ThreadPoolExecutor() if cache_config.caching: - self.redis_lock = get_cache_engine(lock_key="kuzu-lock-" + str(uuid5(NAMESPACE_OID, db_path))) + self.redis_lock = get_cache_engine( + lock_key="kuzu-lock-" + str(uuid5(NAMESPACE_OID, db_path)) + ) else: self._initialize_connection() self.KUZU_ASYNC_LOCK = asyncio.Lock() @@ -230,7 +232,6 @@ class KuzuAdapter(GraphDBInterface): 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: @@ -238,7 +239,7 @@ class KuzuAdapter(GraphDBInterface): self.open_connections -= 1 logger.info(f"Opened connections after closing {self.open_connections}") if self.open_connections == 0: - self.connection.execute('CHECKPOINT;') + self.connection.execute("CHECKPOINT;") self.close() return result @@ -1905,4 +1906,4 @@ class KuzuAdapter(GraphDBInterface): time_nodes = await self.query(cypher) time_ids_list = [item[0] for item in time_nodes] - return ", ".join(f"'{uid}'" for uid in time_ids_list) \ No newline at end of file + return ", ".join(f"'{uid}'" for uid in time_ids_list)