107 lines
5.3 KiB
Python
107 lines
5.3 KiB
Python
"""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 ###
|