From 185e262db87fbb67023a234d0de98565566cb5bf Mon Sep 17 00:00:00 2001
From: Mark <zhuwenhui5566@163.com>
Date: Fri, 19 Dec 2025 18:06:49 +0800
Subject: [PATCH] [add] migration script

---
 api/app/models/workspace_model.py             |  2 +-
 .../versions/f96a53af914c_202512191805.py     | 36 +++++++++++++++++++
 2 files changed, 37 insertions(+), 1 deletion(-)
 create mode 100644 api/migrations/versions/f96a53af914c_202512191805.py

diff --git a/api/app/models/workspace_model.py b/api/app/models/workspace_model.py
index abb5adeb..4d42ed32 100644
--- a/api/app/models/workspace_model.py
+++ b/api/app/models/workspace_model.py
@@ -1,7 +1,7 @@
 import datetime
 from enum import StrEnum
 import uuid
-from sqlalchemy import Column, Integer, String, DateTime, ForeignKey, Boolean
+from sqlalchemy import Column, String, DateTime, ForeignKey, Boolean
 from sqlalchemy.dialects.postgresql import UUID
 from sqlalchemy.orm import relationship
 from app.db import Base
diff --git a/api/migrations/versions/f96a53af914c_202512191805.py b/api/migrations/versions/f96a53af914c_202512191805.py
new file mode 100644
index 00000000..9c3d34b5
--- /dev/null
+++ b/api/migrations/versions/f96a53af914c_202512191805.py
@@ -0,0 +1,36 @@
+"""202512191805
+
+Revision ID: f96a53af914c
+Revises: 87a6537b4074
+Create Date: 2025-12-19 18:05:14.964454
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = 'f96a53af914c'
+down_revision: Union[str, None] = '87a6537b4074'
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.add_column('data_config', sa.Column('reflection_model_id', sa.String(), nullable=True, comment='反思模型ID'))
+    op.add_column('data_config', sa.Column('memory_verify', sa.Boolean(), nullable=True, comment='记忆验证'))
+    op.add_column('data_config', sa.Column('quality_assessment', sa.Boolean(), nullable=True, comment='质量评估'))
+    op.add_column('end_users', sa.Column('reflection_time', sa.DateTime(), nullable=True))
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_column('end_users', 'reflection_time')
+    op.drop_column('data_config', 'quality_assessment')
+    op.drop_column('data_config', 'memory_verify')
+    op.drop_column('data_config', 'reflection_model_id')
+    # ### end Alembic commands ###