admin/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

part of add type to orm (#26262)

Browse Source 
Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
This commit is contained in:
Asuka Minato
2025-10-11 00:40:54 +09:00
committed by GitHub
parent fdb53fdeb1
commit 3922ad876f
4 changed files with 94 additions and 93 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
api/models/task.py
View File

@@ -8,8 +8,6 @@ from sqlalchemy.orm import Mapped, mapped_column
from libs.datetime_utils import naive_utc_now
from models.base import Base
from .engine import db
class CeleryTask(Base):
"""Task result/status."""
@@ -19,7 +17,7 @@ class CeleryTask(Base):
id = mapped_column(sa.Integer, sa.Sequence("task_id_sequence"), primary_key=True, autoincrement=True)
task_id = mapped_column(String(155), unique=True)
status = mapped_column(String(50), default=states.PENDING)
result = mapped_column(db.PickleType, nullable=True)
result = mapped_column(sa.PickleType, nullable=True)
date_done = mapped_column(
DateTime,
default=lambda: naive_utc_now(),
@@ -44,5 +42,5 @@ class CeleryTaskSet(Base):
sa.Integer, sa.Sequence("taskset_id_sequence"), autoincrement=True, primary_key=True
)
taskset_id = mapped_column(String(155), unique=True)
result = mapped_column(db.PickleType, nullable=True)
result = mapped_column(sa.PickleType, nullable=True)
date_done: Mapped[datetime | None] = mapped_column(DateTime, default=lambda: naive_utc_now(), nullable=True)