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

fix: Add fix for migration of kuzu

Browse source
This commit is contained in:
Igor Ilic 2025-07-25 16:39:35 +02:00
parent a81f1a4551
commit 3371d35ec7
1 changed files with 13 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
alembic/versions/b9274c27a25a_kuzu_11_migration.py
View file

@ -53,18 +53,19 @@ def upgrade() -> None:
graph_config = get_graph_config() graph_config = get_graph_config()
if graph_config.graph_database_provider.lower() == "kuzu": if graph_config.graph_database_provider.lower() == "kuzu":
kuzu_db_version = read_kuzu_storage_version(graph_config.graph_file_path) if os.path.exists(graph_config.graph_file_path):
if ( kuzu_db_version = read_kuzu_storage_version(graph_config.graph_file_path)
kuzu_db_version == "0.9.0" or kuzu_db_version == "0.8.2" if (
) and kuzu_db_version != kuzu.__version__: kuzu_db_version == "0.9.0" or kuzu_db_version == "0.8.2"
# Try to migrate kuzu database to latest version ) and kuzu_db_version != kuzu.__version__:
kuzu_migration( # Try to migrate kuzu database to latest version
new_db=graph_config.graph_file_path + "_new", kuzu_migration(
old_db=graph_config.graph_file_path, new_db=graph_config.graph_file_path + "_new",
new_version=kuzu.__version__, old_db=graph_config.graph_file_path,
old_version=kuzu_db_version, new_version=kuzu.__version__,
overwrite=True, old_version=kuzu_db_version,
) overwrite=True,
)
def downgrade() -> None: def downgrade() -> None: