AI Analysis and Content Strategy fixes. Enhanced Strategy Routes refactoring.

backend/services/subscription/schema_utils.py

@@ -6,6 +6,7 @@ from loguru import logger
 
 _checked_subscription_plan_columns: bool = False
 _checked_usage_summaries_columns: bool = False
+_checked_api_usage_logs_columns: bool = False
 
 
 def ensure_subscription_plan_columns(db: Session) -> None:
@@ -114,9 +115,58 @@ def ensure_usage_summaries_columns(db: Session) -> None:
         raise
 
 
+def ensure_api_usage_logs_columns(db: Session) -> None:
+    """Ensure required columns exist on api_usage_logs for runtime safety.
+    
+    This is a defensive guard for environments where migrations have not yet
+    been applied. If columns are missing (e.g., actual_provider_name), we add them
+    with a safe default so ORM queries do not fail.
+    """
+    global _checked_api_usage_logs_columns
+    if _checked_api_usage_logs_columns:
+        return
+    
+    try:
+        # Discover existing columns using PRAGMA
+        result = db.execute(text("PRAGMA table_info(api_usage_logs)"))
+        cols: Set[str] = {row[1] for row in result}
+        
+        logger.debug(f"Schema check: Found {len(cols)} columns in api_usage_logs table")
+        
+        # Columns we may reference in models but might be missing in older DBs
+        required_columns = {
+            "actual_provider_name": "VARCHAR(50) NULL",
+        }
+        
+        for col_name, ddl in required_columns.items():
+            if col_name not in cols:
+                logger.info(f"Adding missing column {col_name} to api_usage_logs table")
+                try:
+                    db.execute(text(f"ALTER TABLE api_usage_logs ADD COLUMN {col_name} {ddl}"))
+                    db.commit()
+                    logger.info(f"Successfully added column {col_name}")
+                except Exception as alter_err:
+                    logger.error(f"Failed to add column {col_name}: {alter_err}")
+                    db.rollback()
+                    # Don't set flag on error - allow retry
+                    raise
+            else:
+                logger.debug(f"Column {col_name} already exists")
+        
+        # Only set flag if we successfully completed the check
+        _checked_api_usage_logs_columns = True
+    except Exception as e:
+        logger.error(f"Error ensuring api_usage_logs columns: {e}", exc_info=True)
+        db.rollback()
+        # Don't set the flag if there was an error, so we retry next time
+        _checked_api_usage_logs_columns = False
+        raise
+
+
 def ensure_all_schema_columns(db: Session) -> None:
     """Ensure all required columns exist in subscription-related tables."""
     ensure_subscription_plan_columns(db)
     ensure_usage_summaries_columns(db)
+    ensure_api_usage_logs_columns(db)