feat(db): 0003 migration + LlmSetting/Source 模型加 blocklist_tags

- 新增 alembic 0003: sources.blocklist_tags + llm_settings.blocklist_tags(JSONB)
- 两层配置:全局(llm_settings) + per-source(sources),合并去重后注入 classify prompt
- 默认空数组,不影响存量数据;admin API 在下个 commit 暴露编辑入口
This commit is contained in:
Mavis
2026-06-09 14:30:38 +08:00
parent 6da59da934
commit d0d1014505
3 changed files with 69 additions and 0 deletions

View File

@@ -0,0 +1,56 @@
"""Sources / llm_settings 加 blocklist_tags(屏蔽分类标签)
设计:
- sources.blocklist_tags: 当前源级屏蔽词(per-source)
- llm_settings.blocklist_tags: 全局屏蔽词(对所有源生效)
- 合并去重后注入到 classify prompt,LLM 判断文章类别是否命中 → drop
注意:
- 这两个字段在 classify 阶段之前是空数组(默认 '[]'::jsonb),
不会影响存量数据;admin_llm/admin.py 会暴露编辑入口。
Revision ID: 0003
Revises: 0002
Create Date: 2026-06-09
"""
from __future__ import annotations
from typing import Sequence, Union
import sqlalchemy as sa
from alembic import op
from sqlalchemy.dialects.postgresql import JSONB
revision: str = "0003"
down_revision: Union[str, None] = "0002"
branch_labels = None
depends_on = None
def upgrade() -> None:
# === sources 加 per-source 屏蔽词 ===
op.add_column(
"sources",
sa.Column(
"blocklist_tags",
JSONB,
nullable=False,
server_default=sa.text("'[]'::jsonb"),
),
)
# === llm_settings 加全局屏蔽词 ===
op.add_column(
"llm_settings",
sa.Column(
"blocklist_tags",
JSONB,
nullable=False,
server_default=sa.text("'[]'::jsonb"),
),
)
def downgrade() -> None:
op.drop_column("llm_settings", "blocklist_tags")
op.drop_column("sources", "blocklist_tags")