Simon
67 lines
3.1 KiB
Python
67 lines
3.1 KiB
Python
"""add knowledge iteration tables: conversation_annotations and knowledge_suggestions
|
|
|
|
Revision ID: 043_knowledge_iteration
|
|
Revises: 042_add_employee_avatar_updated_at
|
|
Create Date: 2026-07-06
|
|
|
|
"""
|
|
from typing import Sequence, Union
|
|
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision: str = '043_knowledge_iteration'
|
|
down_revision: Union[str, None] = '042_add_employee_avatar_updated_at'
|
|
branch_labels: Union[str, Sequence[str], None] = None
|
|
depends_on: Union[str, Sequence[str], None] = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
# --------------------------------------------------------------------------
|
|
# 1. 创建会话标注表 conversation_annotations
|
|
# --------------------------------------------------------------------------
|
|
op.create_table(
|
|
'conversation_annotations',
|
|
sa.Column('id', sa.String(36), primary_key=True),
|
|
sa.Column('conversation_id', sa.String(36), nullable=False, index=True),
|
|
sa.Column('agent_id', sa.String(36), nullable=False),
|
|
sa.Column('message_id', sa.String(36), nullable=False),
|
|
sa.Column('feedback', sa.String(20), nullable=False),
|
|
sa.Column('comment', sa.Text(), nullable=True),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), nullable=False),
|
|
)
|
|
op.create_index('idx_annotation_conversation', 'conversation_annotations', ['conversation_id'])
|
|
op.create_index('idx_annotation_message', 'conversation_annotations', ['message_id'])
|
|
|
|
# --------------------------------------------------------------------------
|
|
# 2. 创建知识库优化建议表 knowledge_suggestions
|
|
# --------------------------------------------------------------------------
|
|
op.create_table(
|
|
'knowledge_suggestions',
|
|
sa.Column('id', sa.String(36), primary_key=True),
|
|
sa.Column('suggestion_type', sa.String(20), nullable=False, server_default='new_faq'),
|
|
sa.Column('status', sa.String(20), nullable=False, server_default='pending', index=True),
|
|
sa.Column('title', sa.String(256), nullable=False),
|
|
sa.Column('content', sa.Text(), nullable=False),
|
|
sa.Column('category', sa.String(64), nullable=False, server_default='其他'),
|
|
sa.Column('tags', sa.JSON(), nullable=False, server_default='[]'),
|
|
sa.Column('source_type', sa.String(30), nullable=False),
|
|
sa.Column('source_data', sa.JSON(), nullable=True),
|
|
sa.Column('reason', sa.Text(), nullable=True),
|
|
sa.Column('reject_reason', sa.Text(), nullable=True),
|
|
sa.Column('reviewer_id', sa.String(36), nullable=True),
|
|
sa.Column('reviewed_at', sa.DateTime(timezone=True), nullable=True),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), nullable=False),
|
|
sa.Column('updated_at', sa.DateTime(timezone=True), nullable=False),
|
|
)
|
|
op.create_index('idx_suggestion_status', 'knowledge_suggestions', ['status'])
|
|
op.create_index('idx_suggestion_type', 'knowledge_suggestions', ['suggestion_type'])
|
|
op.create_index('idx_suggestion_created', 'knowledge_suggestions', ['created_at'])
|
|
|
|
|
|
def downgrade() -> None:
|
|
op.drop_table('knowledge_suggestions')
|
|
op.drop_table('conversation_annotations')
|