diff --git a/packages/adapters/sqlalchemy_impl/generation_task_repository.py b/packages/adapters/sqlalchemy_impl/generation_task_repository.py index 433aaa829..1c3a65ba8 100755 --- a/packages/adapters/sqlalchemy_impl/generation_task_repository.py +++ b/packages/adapters/sqlalchemy_impl/generation_task_repository.py @@ -43,6 +43,7 @@ def _to_domain(model: GenerationTaskModel) -> GenerationTask: output_height=getattr(model, "output_height", 720) or 720, cover_url=getattr(model, "cover_url", "") or "", title_config=dict(getattr(model, "title_config", {}) or {}), + extra_meta=dict(getattr(model, "extra_meta", {}) or {}), logs=model.logs or "[]", created_at=model.created_at, updated_at=model.updated_at, @@ -88,6 +89,7 @@ class SQLAlchemyGenerationTaskRepository: output_height=task.output_height, cover_url=task.cover_url or "", title_config=dict(task.title_config) if task.title_config else {}, + extra_meta=dict(task.extra_meta) if task.extra_meta else {}, logs=task.logs, created_at=task.created_at, updated_at=task.updated_at, @@ -322,6 +324,7 @@ class SQLAlchemyGenerationTaskRepository: model.output_height = task.output_height model.cover_url = task.cover_url or "" model.title_config = dict(task.title_config) if task.title_config else {} + model.extra_meta = dict(task.extra_meta) if task.extra_meta else {} model.logs = task.logs self.session.commit() return task