From 466707b00300ececbf63f87a84fffa6ddc62ced7 Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Tue, 7 Oct 2025 19:45:48 +0200
Subject: [PATCH] feat: adds redis locks

---
 cognee/infrastructure/databases/graph/kuzu/adapter.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

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