From ce4a55a4741e1bb2a9e162da8a3e4f4a8e5bee23 Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Mon, 6 Oct 2025 18:04:39 +0200
Subject: [PATCH] Changes reader writer example to test redis lock

---
 working_dir_error_replication/reader.py | 4 +---
 working_dir_error_replication/writer.py | 5 +++--
 2 files changed, 4 insertions(+), 5 deletions(-)

diff --git a/working_dir_error_replication/reader.py b/working_dir_error_replication/reader.py
index b81857ade..caf8407c1 100644
--- a/working_dir_error_replication/reader.py
+++ b/working_dir_error_replication/reader.py
@@ -6,9 +6,7 @@ from cognee.infrastructure.databases.graph.kuzu.adapter import KuzuAdapter
 
 
 async def main():
-    print("Reader: Waiting 2 seconds...")
-
-    time.sleep(5)
+    time.sleep(20)
     adapter = KuzuAdapter("test.db")
     result = await adapter.query("MATCH (n:Node) RETURN COUNT(n)")
 
diff --git a/working_dir_error_replication/writer.py b/working_dir_error_replication/writer.py
index 2e0ce1728..928ba4d2a 100644
--- a/working_dir_error_replication/writer.py
+++ b/working_dir_error_replication/writer.py
@@ -19,13 +19,14 @@ def create_node(name):
 
 async def main():
     adapter = KuzuAdapter("test.db")
+    nodes = [create_node(f"Node{i}") for i in range(200000)]
+
     print("Writer: Starting...")
-    nodes = [create_node(f"Node{i}") for i in range(100)]
     await adapter.add_nodes(nodes)
 
     print("writer finished...")
 
-    time.sleep(100)
+    time.sleep(10)
 
 if __name__ == "__main__":
     asyncio.run(main())
\ No newline at end of file