From 3371d35ec7490aa3450f6f717377d9431ba23b16 Mon Sep 17 00:00:00 2001 From: Igor Ilic Date: Fri, 25 Jul 2025 16:39:35 +0200 Subject: [PATCH] fix: Add fix for migration of kuzu --- .../b9274c27a25a_kuzu_11_migration.py | 25 ++++++++++--------- 1 file changed, 13 insertions(+), 12 deletions(-) diff --git a/alembic/versions/b9274c27a25a_kuzu_11_migration.py b/alembic/versions/b9274c27a25a_kuzu_11_migration.py index 7602cb4b5..34e3b1dbe 100644 --- a/alembic/versions/b9274c27a25a_kuzu_11_migration.py +++ b/alembic/versions/b9274c27a25a_kuzu_11_migration.py @@ -53,18 +53,19 @@ def upgrade() -> None: graph_config = get_graph_config() if graph_config.graph_database_provider.lower() == "kuzu": - kuzu_db_version = read_kuzu_storage_version(graph_config.graph_file_path) - if ( - kuzu_db_version == "0.9.0" or kuzu_db_version == "0.8.2" - ) and kuzu_db_version != kuzu.__version__: - # Try to migrate kuzu database to latest version - kuzu_migration( - new_db=graph_config.graph_file_path + "_new", - old_db=graph_config.graph_file_path, - new_version=kuzu.__version__, - old_version=kuzu_db_version, - overwrite=True, - ) + if os.path.exists(graph_config.graph_file_path): + kuzu_db_version = read_kuzu_storage_version(graph_config.graph_file_path) + if ( + kuzu_db_version == "0.9.0" or kuzu_db_version == "0.8.2" + ) and kuzu_db_version != kuzu.__version__: + # Try to migrate kuzu database to latest version + kuzu_migration( + new_db=graph_config.graph_file_path + "_new", + old_db=graph_config.graph_file_path, + new_version=kuzu.__version__, + old_version=kuzu_db_version, + overwrite=True, + ) def downgrade() -> None: