570392aa6f
* [add]修改迁移文件新建空白表结构 * Add installation guide and environment setup Added installation instructions and environment requirements for MemoryBear. * [delete]删除api，web的readme.md。只保留唯一readme.md * Fix database connection example in README Update database connection configuration example in README.
55 lines
1.8 KiB
Python
55 lines
1.8 KiB
Python
"""202511291902
|
|
|
|
Revision ID: d1e56ecbf058
|
|
Revises: e225ed10f5cb
|
|
Create Date: 2025-11-29 18:14:38.529466
|
|
|
|
"""
|
|
from typing import Sequence, Union
|
|
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision: str = 'd1e56ecbf058'
|
|
down_revision: Union[str, None] = 'e225ed10f5cb'
|
|
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.execute("DROP VIEW IF EXISTS data_config_sorted CASCADE")
|
|
|
|
# 检查表和列是否存在再进行操作
|
|
op.execute("""
|
|
DO $$
|
|
BEGIN
|
|
IF EXISTS (SELECT FROM information_schema.columns
|
|
WHERE table_name = 'data_config' AND column_name = 'llm_id') THEN
|
|
ALTER TABLE data_config ALTER COLUMN llm_id SET DATA TYPE VARCHAR;
|
|
COMMENT ON COLUMN data_config.llm_id IS 'LLM模型配置ID';
|
|
END IF;
|
|
|
|
IF EXISTS (SELECT FROM information_schema.columns
|
|
WHERE table_name = 'data_config' AND column_name = 'llm') THEN
|
|
ALTER TABLE data_config DROP COLUMN llm;
|
|
END IF;
|
|
END $$;
|
|
""")
|
|
# ### end Alembic commands ###
|
|
|
|
|
|
def downgrade() -> None:
|
|
# ### commands auto generated by Alembic - please adjust! ###
|
|
op.add_column('data_config', sa.Column('llm', sa.VARCHAR(), autoincrement=False, nullable=True, comment='LLM模型配置ID'))
|
|
op.alter_column('data_config', 'llm_id',
|
|
existing_type=sa.VARCHAR(),
|
|
comment='临时',
|
|
existing_comment='LLM模型配置ID',
|
|
existing_nullable=True)
|
|
# ### end Alembic commands ###
|