refactor: Import migration_runner lazily within the migrate command

Co-authored-by: George Weale <gweale@google.com>
PiperOrigin-RevId: 855496468
This commit is contained in:
George Weale
2026-01-12 20:21:57 -08:00
committed by Copybara-Service
parent a4732a8e50
commit 905604faac
+2 -1
View File
@@ -36,7 +36,6 @@ from . import cli_create
from . import cli_deploy
from .. import version
from ..evaluation.constants import MISSING_EVAL_DEPENDENCIES_MESSAGE
from ..sessions.migration import migration_runner
from .cli import run_cli
from .fast_api import get_fast_api_app
from .utils import envs
@@ -1589,6 +1588,8 @@ def cli_migrate_session(
"""Migrates a session database to the latest schema version."""
logs.setup_adk_logger(getattr(logging, log_level.upper()))
try:
from ..sessions.migration import migration_runner
migration_runner.upgrade(source_db_url, dest_db_url)
click.secho("Migration check and upgrade process finished.", fg="green")
except Exception as e: