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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: databricks-tellr-app
-Version: 0.4.3.dev8
+Version: 0.4.3.dev9
 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.dev9}/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.dev9}/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.dev9
 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.dev9}/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.dev9"
 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.dev9}/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")