76 lines
1.9 KiB
Python
76 lines
1.9 KiB
Python
"""
|
||
Database Migration: User Memories Table.
|
||
[AC-IDMP-14] 用户级记忆滚动摘要表
|
||
|
||
创建时间: 2025-03-08
|
||
变更说明:
|
||
- 新增 user_memories 表用于存储滚动摘要与事实/偏好/未解决问题
|
||
|
||
执行方式:
|
||
- SQLModel 会自动创建表（通过 init_db）
|
||
- 此脚本用于手动迁移或回滚
|
||
|
||
SQL DDL:
|
||
```sql
|
||
CREATE TABLE user_memories (
|
||
id UUID PRIMARY KEY,
|
||
tenant_id VARCHAR NOT NULL,
|
||
user_id VARCHAR NOT NULL,
|
||
summary TEXT,
|
||
facts JSON,
|
||
preferences JSON,
|
||
open_issues JSON,
|
||
summary_version INTEGER NOT NULL DEFAULT 1,
|
||
last_turn_id VARCHAR,
|
||
expires_at TIMESTAMP,
|
||
created_at TIMESTAMP NOT NULL DEFAULT NOW(),
|
||
updated_at TIMESTAMP NOT NULL DEFAULT NOW()
|
||
);
|
||
|
||
CREATE INDEX ix_user_memories_tenant_user ON user_memories(tenant_id, user_id);
|
||
CREATE INDEX ix_user_memories_tenant_user_updated ON user_memories(tenant_id, user_id, updated_at);
|
||
```
|
||
|
||
回滚 SQL:
|
||
```sql
|
||
DROP TABLE IF EXISTS user_memories;
|
||
```
|
||
"""
|
||
|
||
USER_MEMORIES_DDL = """
|
||
CREATE TABLE IF NOT EXISTS user_memories (
|
||
id UUID PRIMARY KEY,
|
||
tenant_id VARCHAR NOT NULL,
|
||
user_id VARCHAR NOT NULL,
|
||
summary TEXT,
|
||
facts JSON,
|
||
preferences JSON,
|
||
open_issues JSON,
|
||
summary_version INTEGER NOT NULL DEFAULT 1,
|
||
last_turn_id VARCHAR,
|
||
expires_at TIMESTAMP,
|
||
created_at TIMESTAMP NOT NULL DEFAULT NOW(),
|
||
updated_at TIMESTAMP NOT NULL DEFAULT NOW()
|
||
);
|
||
"""
|
||
|
||
USER_MEMORIES_INDEXES = """
|
||
CREATE INDEX IF NOT EXISTS ix_user_memories_tenant_user ON user_memories(tenant_id, user_id);
|
||
CREATE INDEX IF NOT EXISTS ix_user_memories_tenant_user_updated ON user_memories(tenant_id, user_id, updated_at);
|
||
"""
|
||
|
||
USER_MEMORIES_ROLLBACK = """
|
||
DROP TABLE IF EXISTS user_memories;
|
||
"""
|
||
|
||
|
||
async def upgrade(conn):
|
||
"""执行迁移"""
|
||
await conn.execute(USER_MEMORIES_DDL)
|
||
await conn.execute(USER_MEMORIES_INDEXES)
|
||
|
||
|
||
async def downgrade(conn):
|
||
"""回滚迁移"""
|
||
await conn.execute(USER_MEMORIES_ROLLBACK)
|