databricks-tellr-app 0.4.3.dev8__tar.gz → 0.4.3.dev10__tar.gz

{databricks_tellr_app-0.4.3.dev8 → databricks_tellr_app-0.4.3.dev10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: databricks-tellr-app
-Version: 0.4.3.dev8
+Version: 0.4.3.dev10
 Summary: Tellr application package for Databricks Apps
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown

{databricks_tellr_app-0.4.3.dev8 → databricks_tellr_app-0.4.3.dev10}/databricks_tellr_app/run.py

@@ -51,6 +51,29 @@ def init_database(seed_databricks_defaults: bool = False) -> None:
         logger.error(f"Failed to initialize database tables: {e}\n{tb}")
         raise SystemExit(1) from e
 
+    # Data migrations/backfills that used to run in the FastAPI lifespan now run
+    # HERE, once, before the server forks its workers — the uvicorn workers must
+    # never execute migration code (4 of them racing the migration chain on boot
+    # wedged startup). init_db() above runs the schema/data migrations; these two
+    # convert legacy profile/session rows to the agent_config shape.
+    logger.info("Migrating profiles/sessions to agent_config...")
+    try:
+        from src.core.database import get_session_local
+        from src.core.migrate_profiles_to_agent_config import (
+            backfill_sessions,
+            migrate_profiles,
+        )
+        migrated = migrate_profiles(get_session_local())
+        if migrated:
+            logger.info(f"Migrated {migrated} profiles to agent_config")
+        backfilled = backfill_sessions(get_session_local())
+        if backfilled:
+            logger.info(f"Backfilled {backfilled} sessions with agent_config")
+    except Exception as e:
+        tb = traceback.format_exc()
+        logger.error(f"Failed to migrate profiles/sessions to agent_config: {e}\n{tb}")
+        raise SystemExit(1) from e
+
     # Seed default content
     logger.info(f"Seeding defaults (include_databricks={seed_databricks_defaults})...")
     try:

{databricks_tellr_app-0.4.3.dev8 → databricks_tellr_app-0.4.3.dev10}/databricks_tellr_app.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: databricks-tellr-app
-Version: 0.4.3.dev8
+Version: 0.4.3.dev10
 Summary: Tellr application package for Databricks Apps
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown

{databricks_tellr_app-0.4.3.dev8 → databricks_tellr_app-0.4.3.dev10}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "databricks-tellr-app"
-version = "0.4.3.dev8"
+version = "0.4.3.dev10"
 description = "Tellr application package for Databricks Apps"
 readme = "README.md"
 requires-python = ">=3.10"

{databricks_tellr_app-0.4.3.dev8 → databricks_tellr_app-0.4.3.dev10}/src/api/main.py

@@ -34,13 +34,10 @@ from src.api.routes.settings import (
 from src.api.services.export_job_queue import start_export_worker
 from src.api.services.job_queue import recover_stuck_requests, start_worker
 from src.core.database import (
-    get_session_local,
-    init_db,
     is_lakebase_environment,
     start_token_refresh,
     stop_token_refresh,
 )
-from src.core.migrate_profiles_to_agent_config import migrate_profiles, backfill_sessions
 
 logger = logging.getLogger(__name__)
 
@@ -92,8 +89,8 @@ async def lifespan(app: FastAPI):
     else:
         logger.info("Pytest detected: skipping MCP session manager startup")
 
-    # Start Lakebase token refresh if running in Databricks Apps
-    # Must happen before init_db() so OAuth token is ready for database connections
+    # Start Lakebase token refresh if running in Databricks Apps. Workers serve
+    # DB-backed requests, so they need fresh OAuth tokens.
     if is_lakebase_environment():
         try:
             await start_token_refresh()
@@ -102,24 +99,11 @@ async def lifespan(app: FastAPI):
             logger.error(f"Failed to start Lakebase token refresh: {e}")
             raise
 
-    # Initialize database tables (idempotent - only creates tables that don't exist)
-    is_pytest = os.getenv("PYTEST_CURRENT_TEST") is not None
-    if not is_pytest:
-        try:
-            init_db()
-            logger.info("Database tables initialized")
-
-            migrated = migrate_profiles(get_session_local())
-            if migrated:
-                logger.info(f"Migrated {migrated} profiles to agent_config")
-            backfilled = backfill_sessions(get_session_local())
-            if backfilled:
-                logger.info(f"Backfilled {backfilled} sessions with agent_config")
-        except Exception as e:
-            logger.error(f"Failed to initialize database: {e}")
-            raise
-    else:
-        logger.info("Pytest detected: skipping database initialization")
+    # NOTE: database migrations run ONCE before the server starts, in
+    # run.py::init_database (invoked as its own step by the Databricks Apps boot
+    # command, and by scripts/init_database.py locally) — NOT here. This lifespan
+    # runs in every uvicorn worker, and having 4 workers race the migration chain
+    # on boot wedged startup. Workers must never run migration code.
 
     if IS_PRODUCTION:
         logger.info("Production mode: serving frontend from package assets")

{databricks_tellr_app-0.4.3.dev8 → databricks_tellr_app-0.4.3.dev10}/src/core/database.py

@@ -1204,12 +1204,22 @@ def _migrate_rewrite_deck_image_placeholders(conn, inspector, schema, _qual, is_
     A no-op when ``image_assets`` (or its ``token`` column) or every target table is
     absent on a given deploy.
     """
-    from sqlalchemy import text
+    from sqlalchemy import inspect, text
+
+    # Reflect the LIVE schema with a FRESH inspector, NOT the shared one that
+    # _run_migrations threads through. _migrate_image_assets_add_token adds the
+    # token column in this SAME transaction, and the shared inspector's reflection
+    # cache predates that ALTER — so a cached read reports "no token column" and
+    # this migration would return early, silently skipping the rewrite on every
+    # fresh DB and every prod fork (where the column is added during this run).
+    # This is the same reflection-cache staleness the design-system migrations in
+    # this file avoid the same way (see _migrate_uncap_brand_text_columns).
+    insp = inspect(conn)
 
     # image_assets + its token column must exist (guaranteed by the prior migration
     # on any live deploy; guarded so this can be exercised standalone).
     try:
-        image_cols = {c["name"] for c in inspector.get_columns("image_assets", schema=schema)}
+        image_cols = {c["name"] for c in insp.get_columns("image_assets", schema=schema)}
     except Exception:
         return
     if "token" not in image_cols:
@@ -1219,7 +1229,7 @@ def _migrate_rewrite_deck_image_placeholders(conn, inspector, schema, _qual, is_
     live_targets: list[tuple[str, str]] = []
     for table_name, column_name in _DECK_PLACEHOLDER_COLUMNS:
         try:
-            cols = {c["name"] for c in inspector.get_columns(table_name, schema=schema)}
+            cols = {c["name"] for c in insp.get_columns(table_name, schema=schema)}
         except Exception:
             continue
         if column_name in cols: