databricks-tellr-app 0.3.10.dev1__tar.gz → 0.3.10.dev2__tar.gz

{databricks_tellr_app-0.3.10.dev1 → databricks_tellr_app-0.3.10.dev2}/PKG-INFO

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

{databricks_tellr_app-0.3.10.dev1 → databricks_tellr_app-0.3.10.dev2}/databricks_tellr_app.egg-info/PKG-INFO

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

{databricks_tellr_app-0.3.10.dev1 → databricks_tellr_app-0.3.10.dev2}/pyproject.toml

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

{databricks_tellr_app-0.3.10.dev1 → databricks_tellr_app-0.3.10.dev2}/src/core/database.py

@@ -474,16 +474,10 @@ def _run_migrations(engine, schema: str | None = None):
                 f"ALTER TABLE {qualified_table} ADD COLUMN llm_judge_backend VARCHAR(32) "
                 "DEFAULT 'mlflow' NOT NULL"
             ))
-        elif not is_sqlite:
-            try:
-                conn.execute(text(
-                    f"ALTER TABLE {qualified_table} ALTER COLUMN llm_judge_backend "
-                    "SET DEFAULT 'mlflow'"
-                ))
-            except Exception as ex:
-                logger.debug(
-                    "Migration: skip llm_judge_backend server default alter: %s", ex
-                )
+        else:
+            _ensure_llm_judge_backend_default(
+                conn, table_name, qualified_table, schema, is_sqlite
+            )
         # --- config_profiles: migrate is_global bool to global_permission ---
         if "is_global" in columns:
             logger.info(f"Migration: migrating is_global to global_permission")
@@ -528,6 +522,57 @@ def _run_migrations(engine, schema: str | None = None):
         _migrate_image_assets_tags_json_to_jsonb(conn, schema, _qual, is_sqlite)
 
 
+def _ensure_llm_judge_backend_default(
+    conn, table_name, qualified_table, schema, is_sqlite
+) -> None:
+    """Ensure config_profiles.llm_judge_backend has server default 'mlflow'.
+
+    Idempotent and fork-safe. SQLite sets the default at column creation, so
+    there is nothing to do there. On PostgreSQL, only issue the ALTER when the
+    current default is NOT already 'mlflow' — re-issuing it on every startup is
+    needless ownership-gated DDL that fails on a copy-on-write branch where the
+    app's service principal does not own the inherited table, and (because the
+    whole migration runs in one transaction) a swallowed failure would poison
+    every later migration statement. Skipping when already-correct makes a
+    same-schema branch a true no-op; the rare real backfill runs inside a
+    SAVEPOINT so its failure is contained rather than aborting the outer
+    transaction.
+    """
+    from sqlalchemy import text
+
+    if is_sqlite:
+        return
+
+    current_default = None
+    try:
+        sql = (
+            "SELECT column_default FROM information_schema.columns "
+            "WHERE table_name = :t AND column_name = 'llm_judge_backend'"
+        )
+        params = {"t": table_name}
+        if schema:
+            sql += " AND table_schema = :s"
+            params["s"] = schema
+        current_default = conn.execute(text(sql), params).scalar()
+    except Exception:
+        current_default = None
+
+    if current_default is not None and "mlflow" in str(current_default):
+        return  # already correct — no DDL needed
+
+    logger.info("Migration: setting llm_judge_backend server default to 'mlflow'")
+    try:
+        with conn.begin_nested():
+            conn.execute(text(
+                f"ALTER TABLE {qualified_table} ALTER COLUMN llm_judge_backend "
+                "SET DEFAULT 'mlflow'"
+            ))
+    except Exception as ex:
+        logger.debug(
+            "Migration: skip llm_judge_backend server default alter: %s", ex
+        )
+
+
 def _migrate_image_assets_tags_json_to_jsonb(conn, schema, _qual, is_sqlite) -> None:
     """Alter image_assets.tags from json to jsonb on PostgreSQL (idempotent).