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

Migrate to DeclarativeBaseModel

Browse Source
This commit is contained in:
Yeuoly
2024-10-21 20:38:27 +08:00
parent 53e1b45d40
commit 11270a7ef2
9 changed files with 138 additions and 78 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
api/models/account.py
View File

@ -4,6 +4,7 @@ import json
from flask_login import UserMixin
from extensions.ext_database import db
from models.base import Base
from .types import StringUUID
@ -16,7 +17,7 @@ class AccountStatus(str, enum.Enum):
CLOSED = "closed"
class Account(UserMixin, db.Model):
class Account(UserMixin, Base):
__tablename__ = "accounts"
__table_args__ = (db.PrimaryKeyConstraint("id", name="account_pkey"), db.Index("account_email_idx", "email"))

2
api/models/model.py
View File

@ -38,7 +38,7 @@ class FileUploadConfig(BaseModel):
number_limits: int = Field(default=0, gt=0, le=10)
class DifySetup(db.Model):
class DifySetup(BaseModel):
__tablename__ = "dify_setups"
__table_args__ = (db.PrimaryKeyConstraint("version", name="dify_setup_pkey"),)