test/dify
1
0
Fork 0
mirror of https://gitee.com/dify_ai/dify.git synced 2024-12-03 11:48:08 +08:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

refactor(migration/model): update column types for workflow schema (#10160)

Browse Source
This commit is contained in:
-LAN- 2024-11-01 16:10:55 +08:00 committed by GitHub
parent 9ac2bb30f4
commit bf048b8d7c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/migrations/versions/2024_11_01_0623-09a8d1878d9b_update_workflows_graph_features_and_.py
View File

@ -39,7 +39,6 @@ def upgrade():
nullable=False) nullable=False)
batch_op.alter_column('features', batch_op.alter_column('features',
existing_type=sa.TEXT(), existing_type=sa.TEXT(),
type_=sa.String(),
nullable=False) nullable=False)
batch_op.alter_column('updated_at', batch_op.alter_column('updated_at',
existing_type=postgresql.TIMESTAMP(), existing_type=postgresql.TIMESTAMP(),
@ -55,8 +54,7 @@ def downgrade():
existing_type=postgresql.TIMESTAMP(), existing_type=postgresql.TIMESTAMP(),
nullable=True) nullable=True)
batch_op.alter_column('features', batch_op.alter_column('features',
existing_type=sa.String(), existing_type=sa.TEXT(),
type_=sa.TEXT(),
nullable=True) nullable=True)
batch_op.alter_column('graph', batch_op.alter_column('graph',
existing_type=sa.TEXT(), existing_type=sa.TEXT(),

5
api/models/workflow.py
View File

@ -4,6 +4,7 @@ from datetime import datetime
from enum import Enum from enum import Enum
from typing import Any, Optional, Union from typing import Any, Optional, Union
import sqlalchemy as sa
from sqlalchemy import func from sqlalchemy import func
from sqlalchemy.orm import Mapped, mapped_column from sqlalchemy.orm import Mapped, mapped_column
@ -99,8 +100,8 @@ class Workflow(db.Model):
app_id: Mapped[str] = mapped_column(StringUUID, nullable=False) app_id: Mapped[str] = mapped_column(StringUUID, nullable=False)
type: Mapped[str] = mapped_column(db.String(255), nullable=False) type: Mapped[str] = mapped_column(db.String(255), nullable=False)
version: Mapped[str] = mapped_column(db.String(255), nullable=False) version: Mapped[str] = mapped_column(db.String(255), nullable=False)
graph: Mapped[str] = mapped_column(db.Text) graph: Mapped[str] = mapped_column(sa.Text)
_features: Mapped[str] = mapped_column("features") _features: Mapped[str] = mapped_column("features", sa.TEXT)
created_by: Mapped[str] = mapped_column(StringUUID, nullable=False) created_by: Mapped[str] = mapped_column(StringUUID, nullable=False)
created_at: Mapped[datetime] = mapped_column( created_at: Mapped[datetime] = mapped_column(
db.DateTime, nullable=False, server_default=db.text("CURRENT_TIMESTAMP(0)") db.DateTime, nullable=False, server_default=db.text("CURRENT_TIMESTAMP(0)")