[add] migration script

2026-01-05 14:10:45 +08:00
parent 1fc81d1347
commit a07e151c95
1 changed files with 106 additions and 0 deletions
--- a/api/migrations/versions/6064f41faac6_202601051352.py
+++ b/api/migrations/versions/6064f41faac6_202601051352.py
@@ -0,0 +1,106 @@
+"""202601051352
+
+Revision ID: 6064f41faac6
+Revises: a6452760a140
+Create Date: 2026-01-05 13:52:25.739146
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = '6064f41faac6'
+down_revision: Union[str, None] = 'a6452760a140'
+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('decay_constant', sa.Float(), nullable=True, comment='ACT-R衰减常数d，默认0.5'))
+    op.add_column('data_config', sa.Column('forgetting_threshold', sa.Float(), nullable=True, comment='遗忘阈值，默认0.3'))
+    op.add_column('data_config', sa.Column('forgetting_interval_hours', sa.Integer(), nullable=True, comment='遗忘周期间隔（小时），默认24'))
+    op.add_column('data_config', sa.Column('enable_llm_summary', sa.Boolean(), nullable=True, comment='是否使用LLM生成摘要，默认True'))
+    op.add_column('data_config', sa.Column('max_merge_batch_size', sa.Integer(), nullable=True, comment='单次最大融合节点对数，默认100'))
+    op.add_column('data_config', sa.Column('max_history_length', sa.Integer(), nullable=True, comment='访问历史最大长度，默认100'))
+    op.add_column('data_config', sa.Column('min_days_since_access', sa.Integer(), nullable=True, comment='最小未访问天数，默认30'))
+    op.add_column('end_users', sa.Column('personality_traits', sa.Text(), nullable=True, comment='性格特点'))
+    op.add_column('end_users', sa.Column('core_values', sa.Text(), nullable=True, comment='核心价值观'))
+    op.add_column('end_users', sa.Column('one_sentence_summary', sa.Text(), nullable=True, comment='一句话总结'))
+    op.add_column('end_users', sa.Column('behavior_pattern', sa.Text(), nullable=True, comment='行为模式'))
+    op.add_column('end_users', sa.Column('key_findings', sa.Text(), nullable=True, comment='关键发现'))
+    op.add_column('end_users', sa.Column('growth_trajectory', sa.Text(), nullable=True, comment='成长轨迹'))
+    
+    # Convert BIGINT (milliseconds) to DateTime using raw SQL with USING clause
+    op.execute("""
+        ALTER TABLE end_users 
+        ALTER COLUMN hire_date TYPE timestamp without time zone 
+        USING to_timestamp(hire_date / 1000.0)
+    """)
+    op.execute("COMMENT ON COLUMN end_users.hire_date IS '入职日期'")
+    
+    op.execute("""
+        ALTER TABLE end_users 
+        ALTER COLUMN updatetime_profile TYPE timestamp without time zone 
+        USING to_timestamp(updatetime_profile / 1000.0)
+    """)
+    op.execute("COMMENT ON COLUMN end_users.updatetime_profile IS '核心档案信息最后更新时间'")
+    
+    op.alter_column('end_users', 'user_summary',
+               existing_type=sa.TEXT(),
+               comment='缓存的用户摘要（基本介绍）',
+               existing_comment='缓存的用户摘要',
+               existing_nullable=True)
+    op.alter_column('end_users', 'memory_insight',
+               existing_type=sa.TEXT(),
+               comment='缓存的记忆洞察报告（总体概述）',
+               existing_comment='缓存的记忆洞察报告',
+               existing_nullable=True)
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.alter_column('end_users', 'memory_insight',
+               existing_type=sa.TEXT(),
+               comment='缓存的记忆洞察报告',
+               existing_comment='缓存的记忆洞察报告（总体概述）',
+               existing_nullable=True)
+    op.alter_column('end_users', 'user_summary',
+               existing_type=sa.TEXT(),
+               comment='缓存的用户摘要',
+               existing_comment='缓存的用户摘要（基本介绍）',
+               existing_nullable=True)
+    
+    # Convert DateTime back to BIGINT (milliseconds) using raw SQL
+    op.execute("""
+        ALTER TABLE end_users 
+        ALTER COLUMN updatetime_profile TYPE bigint 
+        USING (EXTRACT(EPOCH FROM updatetime_profile) * 1000)::bigint
+    """)
+    op.execute("COMMENT ON COLUMN end_users.updatetime_profile IS '核心档案信息最后更新时间（时间戳，毫秒）'")
+    
+    op.execute("""
+        ALTER TABLE end_users 
+        ALTER COLUMN hire_date TYPE bigint 
+        USING (EXTRACT(EPOCH FROM hire_date) * 1000)::bigint
+    """)
+    op.execute("COMMENT ON COLUMN end_users.hire_date IS '入职日期（时间戳，毫秒）'")
+    
+    op.drop_column('end_users', 'growth_trajectory')
+    op.drop_column('end_users', 'key_findings')
+    op.drop_column('end_users', 'behavior_pattern')
+    op.drop_column('end_users', 'one_sentence_summary')
+    op.drop_column('end_users', 'core_values')
+    op.drop_column('end_users', 'personality_traits')
+    op.drop_column('data_config', 'min_days_since_access')
+    op.drop_column('data_config', 'max_history_length')
+    op.drop_column('data_config', 'max_merge_batch_size')
+    op.drop_column('data_config', 'enable_llm_summary')
+    op.drop_column('data_config', 'forgetting_interval_hours')
+    op.drop_column('data_config', 'forgetting_threshold')
+    op.drop_column('data_config', 'decay_constant')
+    # ### end Alembic commands ###