код написан

backend/src/core/database.py

@@ -157,6 +157,88 @@ def _ensure_user_dashboard_preferences_columns(bind_engine):
 # [/DEF:_ensure_user_dashboard_preferences_columns:Function]
 
 
+# [DEF:_ensure_user_dashboard_preferences_health_columns:Function]
+# @PURPOSE: Applies additive schema upgrades for user_dashboard_preferences table (health fields).
+def _ensure_user_dashboard_preferences_health_columns(bind_engine):
+    with belief_scope("_ensure_user_dashboard_preferences_health_columns"):
+        table_name = "user_dashboard_preferences"
+        inspector = inspect(bind_engine)
+        if table_name not in inspector.get_table_names():
+            return
+
+        existing_columns = {
+            str(column.get("name") or "").strip()
+            for column in inspector.get_columns(table_name)
+        }
+
+        alter_statements = []
+        if "telegram_id" not in existing_columns:
+            alter_statements.append(
+                "ALTER TABLE user_dashboard_preferences ADD COLUMN telegram_id VARCHAR"
+            )
+        if "email_address" not in existing_columns:
+            alter_statements.append(
+                "ALTER TABLE user_dashboard_preferences ADD COLUMN email_address VARCHAR"
+            )
+        if "notify_on_fail" not in existing_columns:
+            alter_statements.append(
+                "ALTER TABLE user_dashboard_preferences ADD COLUMN notify_on_fail BOOLEAN NOT NULL DEFAULT TRUE"
+            )
+
+        if not alter_statements:
+            return
+
+        try:
+            with bind_engine.begin() as connection:
+                for statement in alter_statements:
+                    connection.execute(text(statement))
+        except Exception as migration_error:
+            logger.warning(
+                "[database][EXPLORE] Profile health preference additive migration failed: %s",
+                migration_error,
+            )
+# [/DEF:_ensure_user_dashboard_preferences_health_columns:Function]
+
+
+# [DEF:_ensure_llm_validation_results_columns:Function]
+# @PURPOSE: Applies additive schema upgrades for llm_validation_results table.
+def _ensure_llm_validation_results_columns(bind_engine):
+    with belief_scope("_ensure_llm_validation_results_columns"):
+        table_name = "llm_validation_results"
+        inspector = inspect(bind_engine)
+        if table_name not in inspector.get_table_names():
+            return
+
+        existing_columns = {
+            str(column.get("name") or "").strip()
+            for column in inspector.get_columns(table_name)
+        }
+
+        alter_statements = []
+        if "task_id" not in existing_columns:
+            alter_statements.append(
+                "ALTER TABLE llm_validation_results ADD COLUMN task_id VARCHAR"
+            )
+        if "environment_id" not in existing_columns:
+            alter_statements.append(
+                "ALTER TABLE llm_validation_results ADD COLUMN environment_id VARCHAR"
+            )
+
+        if not alter_statements:
+            return
+
+        try:
+            with bind_engine.begin() as connection:
+                for statement in alter_statements:
+                    connection.execute(text(statement))
+        except Exception as migration_error:
+            logger.warning(
+                "[database][EXPLORE] ValidationRecord additive migration failed: %s",
+                migration_error,
+            )
+# [/DEF:_ensure_llm_validation_results_columns:Function]
+
+
 # [DEF:_ensure_git_server_configs_columns:Function]
 # @PURPOSE: Applies additive schema upgrades for git_server_configs table.
 # @PRE: bind_engine points to application database.
@@ -205,6 +287,8 @@ def init_db():
         Base.metadata.create_all(bind=tasks_engine)
         Base.metadata.create_all(bind=auth_engine)
         _ensure_user_dashboard_preferences_columns(engine)
+        _ensure_llm_validation_results_columns(engine)
+        _ensure_user_dashboard_preferences_health_columns(engine)
         _ensure_git_server_configs_columns(engine)
 # [/DEF:init_db:Function]