From e81faebf69bf25117b2965da323a1c52c95ef994 Mon Sep 17 00:00:00 2001
From: Mark <zhuwenhui5566@163.com>
Date: Tue, 10 Mar 2026 14:51:48 +0800
Subject: [PATCH] [add] migration script

---
 api/app/models/end_user_model.py              |  2 +-
 .../versions/fb834419b18f_202603101453.py     | 34 +++++++++++++++++++
 2 files changed, 35 insertions(+), 1 deletion(-)
 create mode 100644 api/migrations/versions/fb834419b18f_202603101453.py

diff --git a/api/app/models/end_user_model.py b/api/app/models/end_user_model.py
index 048d42ea..28a44f1f 100644
--- a/api/app/models/end_user_model.py
+++ b/api/app/models/end_user_model.py
@@ -52,7 +52,7 @@ class EndUser(Base):
     memory_insight_updated_at = Column(DateTime, nullable=True, comment="洞察报告最后更新时间")
 
     # RAG存储模式专用字段 - RAG Storage Mode Fields
-    storage_type = Column(String, nullable=True, default="neo4j", comment="存储模式类型: neo4j / rag")
+    # storage_type = Column(String, nullable=True, default="neo4j", comment="存储模式类型: neo4j / rag")
     rag_tags = Column(Text, nullable=True, comment="RAG模式下提取的标签列表（JSON格式）")
     rag_personas = Column(Text, nullable=True, comment="RAG模式下提取的人物形象列表（JSON格式）")
     rag_summary_updated_at = Column(DateTime, nullable=True, comment="RAG摘要/标签/人物形象最后更新时间")
diff --git a/api/migrations/versions/fb834419b18f_202603101453.py b/api/migrations/versions/fb834419b18f_202603101453.py
new file mode 100644
index 00000000..8b17e2e0
--- /dev/null
+++ b/api/migrations/versions/fb834419b18f_202603101453.py
@@ -0,0 +1,34 @@
+"""202603101453
+
+Revision ID: fb834419b18f
+Revises: 1ac07dc7366f
+Create Date: 2026-03-10 14:46:48.038643
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = 'fb834419b18f'
+down_revision: Union[str, None] = '1ac07dc7366f'
+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('end_users', sa.Column('rag_tags', sa.Text(), nullable=True, comment='RAG模式下提取的标签列表（JSON格式）'))
+    op.add_column('end_users', sa.Column('rag_personas', sa.Text(), nullable=True, comment='RAG模式下提取的人物形象列表（JSON格式）'))
+    op.add_column('end_users', sa.Column('rag_summary_updated_at', sa.DateTime(), nullable=True, comment='RAG摘要/标签/人物形象最后更新时间'))
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_column('end_users', 'rag_summary_updated_at')
+    op.drop_column('end_users', 'rag_personas')
+    op.drop_column('end_users', 'rag_tags')
+    # ### end Alembic commands ###