youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-04-24 04:45:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor: using DeclarativeBase as parent class of models, refactored tools

Browse Source
This commit is contained in:
Yeuoly
2024-09-29 17:00:58 +08:00
parent c8bc3892b3
commit e9e5c8806a
17 changed files with 225 additions and 120 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
api/models/task.py
View File

@ -3,9 +3,10 @@ from datetime import datetime, timezone
from celery import states
from extensions.ext_database import db
from models.base import Base
class CeleryTask(db.Model):
class CeleryTask(Base):
"""Task result/status."""
__tablename__ = "celery_taskmeta"
@ -29,7 +30,7 @@ class CeleryTask(db.Model):
queue = db.Column(db.String(155), nullable=True)
class CeleryTaskSet(db.Model):
class CeleryTaskSet(Base):
"""TaskSet result."""
__tablename__ = "celery_tasksetmeta"