busya/ss-tools
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: add missing schema columns and Alembic model imports

Browse Source 
- Added 4 missing columns to _ensure_translation_jobs_columns() inline
  migration: target_language_column, target_source_column,
  target_source_language_column, disable_reasoning — these were in the
  SQLAlchemy model but had no Alembic migration or _ensure_*() coverage
- Added missing api_key, assistant, clean_release model imports to
  alembic/env.py so autogenerate can see all tables
This commit is contained in:
busya
2026-05-26 19:00:04 +03:00
parent 81fc7e0902
commit 9337a4cecd
2 changed files with 68 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

65
backend/src/core/database.py
View File

@@ -527,6 +527,71 @@ def _ensure_translation_jobs_columns(bind_engine):
extra={"error": str(migration_error)},
)
# Columns added to model AFTER initial Alembic migration — no Alembic migration exists
if "target_language_column" not in existing_columns:
try:
with bind_engine.begin() as connection:
connection.execute(
text(
"ALTER TABLE translation_jobs "
"ADD COLUMN target_language_column VARCHAR"
)
)
logger.reflect("Added target_language_column to translation_jobs")
except Exception as migration_error:
logger.explore(
"Failed to add target_language_column to translation_jobs",
extra={"error": str(migration_error)},
)
if "target_source_column" not in existing_columns:
try:
with bind_engine.begin() as connection:
connection.execute(
text(
"ALTER TABLE translation_jobs "
"ADD COLUMN target_source_column VARCHAR"
)
)
logger.reflect("Added target_source_column to translation_jobs")
except Exception as migration_error:
logger.explore(
"Failed to add target_source_column to translation_jobs",
extra={"error": str(migration_error)},
)
if "target_source_language_column" not in existing_columns:
try:
with bind_engine.begin() as connection:
connection.execute(
text(
"ALTER TABLE translation_jobs "
"ADD COLUMN target_source_language_column VARCHAR"
)
)
logger.reflect("Added target_source_language_column to translation_jobs")
except Exception as migration_error:
logger.explore(
"Failed to add target_source_language_column to translation_jobs",
extra={"error": str(migration_error)},
)
if "disable_reasoning" not in existing_columns:
try:
with bind_engine.begin() as connection:
connection.execute(
text(
"ALTER TABLE translation_jobs "
"ADD COLUMN disable_reasoning BOOLEAN NOT NULL DEFAULT FALSE"
)
)
logger.reflect("Added disable_reasoning column to translation_jobs")
except Exception as migration_error:
logger.explore(
"Failed to add disable_reasoning to translation_jobs",
extra={"error": str(migration_error)},
)
def _ensure_dataset_review_session_columns(bind_engine):
with belief_scope("_ensure_dataset_review_session_columns"):