gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: ensure connection check before executing checkpoint in KuzuAdapter

Browse source
This commit is contained in:
Daulet Amirkhanov 2025-08-19 11:27:30 +01:00
parent 4d021ec970
commit e912b6f37e
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
cognee/infrastructure/databases/graph/kuzu/adapter.py
View file

@ -138,8 +138,9 @@ class KuzuAdapter(GraphDBInterface):
s3_file_storage = S3FileStorage("") s3_file_storage = S3FileStorage("")
async with self.KUZU_ASYNC_LOCK: if self.connection:
self.connection.execute("CHECKPOINT;") async with self.KUZU_ASYNC_LOCK:
self.connection.execute("CHECKPOINT;")
s3_file_storage.s3.put(self.temp_graph_file, self.db_path, recursive=True) s3_file_storage.s3.put(self.temp_graph_file, self.db_path, recursive=True)