code: def _run_alembic_upgrade(connection):
      alembic_cfg = config.Config()
      alembic_cfg.set_main_option(
          "script_location", settings.ALEMBIC_MIGRATIONS_LOCATION
      )
      alembic_cfg.attributes["connection"] = connection
      command.upgrade(alembic_cfg, "head")
  
file_path: dstack/src/dstack/_internal/server/db.py
imports: from contextlib import asynccontextmanager
  from alembic import command, config
  from sqlalchemy import event
  from sqlalchemy.engine.interfaces import DBAPIConnection
  from sqlalchemy.ext.asyncio import AsyncSession, create_async_engine
  from sqlalchemy.orm import sessionmaker
  from sqlalchemy.pool import ConnectionPoolEntry
  from dstack._internal.server import settings
  from dstack._internal.server.settings import DATABASE_URL
label_desc: Use image node_id 3 for calling a global function with example usage: _run_alembic_upgrade(connection) without return types