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

feat : add migration for kuzu 0.8.2 as well

Browse source
This commit is contained in:
Igor Ilic 2025-07-24 13:07:07 +02:00
parent 2fff637cda
commit ed09750598
1 changed files with 9 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -83,7 +83,9 @@ class KuzuAdapter(GraphDBInterface):
import kuzu
kuzu_db_version = read_kuzu_storage_version(self.db_path)
if kuzu_db_version == "0.9.0" and kuzu_db_version != kuzu.__version__:
if (
kuzu_db_version == "0.9.0" or kuzu_db_version == "0.8.2"
) and kuzu_db_version != kuzu.__version__:
# TODO: Write migration script that will handle all user graph databases in multi-user mode
# Try to migrate kuzu database to latest version
from .kuzu_migrate import kuzu_migration
@ -95,11 +97,12 @@ class KuzuAdapter(GraphDBInterface):
old_version=kuzu_db_version,
overwrite=True,
)
self.db = Database(
self.db_path,
buffer_pool_size=2048 * 1024 * 1024, # 2048MB buffer pool
max_db_size=4096 * 1024 * 1024,
)
self.db = Database(
self.db_path,
buffer_pool_size=2048 * 1024 * 1024, # 2048MB buffer pool
max_db_size=4096 * 1024 * 1024,
)
self.db.init_database()
self.connection = Connection(self.db)