d0d1014505
- 新增 alembic 0003: sources.blocklist_tags + llm_settings.blocklist_tags(JSONB) - 两层配置:全局(llm_settings) + per-source(sources),合并去重后注入 classify prompt - 默认空数组,不影响存量数据;admin API 在下个 commit 暴露编辑入口
59 lines
2.0 KiB
Python
59 lines
2.0 KiB
Python
"""LLM 设置(单行,owner 可编辑)。
|
|
|
|
字段对应:
|
|
- 排版/分类/点评提示词(用户可改)
|
|
- 插图尺寸 + prompt 模板(用户可改)
|
|
- 总开关 enabled
|
|
- 模型名(默认指向 Agnes,但可改成任意 OpenAI 兼容端点)
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
from datetime import datetime
|
|
|
|
from sqlalchemy import Boolean, DateTime, Integer, String, Text, func
|
|
from sqlalchemy.dialects.postgresql import JSONB
|
|
from sqlalchemy.orm import Mapped, mapped_column
|
|
|
|
from app.database import Base
|
|
|
|
|
|
class LlmSetting(Base):
|
|
__tablename__ = "llm_settings"
|
|
|
|
# 永远只有一行:id=1
|
|
id: Mapped[int] = mapped_column(Integer, primary_key=True, default=1)
|
|
|
|
# === 提示词 ===
|
|
format_prompt: Mapped[str | None] = mapped_column(Text)
|
|
classify_prompt: Mapped[str | None] = mapped_column(Text)
|
|
commentary_prompt: Mapped[str | None] = mapped_column(Text)
|
|
image_prompt_template: Mapped[str | None] = mapped_column(Text)
|
|
|
|
# === 全局屏蔽分类标签(如 ["体育", "娱乐"])===
|
|
# 与 sources.blocklist_tags 合并去重后注入 classify prompt;
|
|
# 命中则删文章(drop)
|
|
blocklist_tags: Mapped[list[str]] = mapped_column(
|
|
JSONB, nullable=False, default=list, server_default="[]"
|
|
)
|
|
|
|
# === 插图参数 ===
|
|
image_size: Mapped[str] = mapped_column(String(16), default="768x512", nullable=False)
|
|
|
|
# === 模型 ===
|
|
chat_model: Mapped[str] = mapped_column(String(64), default="agnes-2.0-flash", nullable=False)
|
|
image_model: Mapped[str] = mapped_column(String(64), default="agnes-image-2.1-flash", nullable=False)
|
|
|
|
# === 限速 ===
|
|
interval_sec: Mapped[float] = mapped_column(default=2.0, nullable=False)
|
|
|
|
# === 总开关 ===
|
|
enabled: Mapped[bool] = mapped_column(Boolean, default=True, nullable=False)
|
|
|
|
# === 时间 ===
|
|
updated_at: Mapped[datetime] = mapped_column(
|
|
DateTime(timezone=True), server_default=func.now(), onupdate=func.now(), nullable=False
|
|
)
|
|
|
|
def __repr__(self) -> str:
|
|
return f"<LlmSetting id={self.id} enabled={self.enabled}>"
|