16536fe3a0
- llm_settings 加 meituan_blocked_topics / blocked_keywords / no_comment_text - alembic 0006 迁移,默认 topics=[时政/国际/军事/政治/战争/冲突/制裁/选举], 默认文案='无可奉告' - enrichment._is_meituan_blocked 预检:category 命中 topic 或 关键词 → 直接写'无可奉告',不调美团 API - 命中后 commentary_meituan_model='policy-block' 标识非真实生成 - enrichment_loop Semaphore(3)→(1),Agnes 免费 plan 不再 429 - 前端 AdminLlmSettings 美团卡片加 3 字段 UI(主题/关键词/固定文案)
96 lines
4.0 KiB
Python
96 lines
4.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)
|
|
|
|
# === Angel(Agnes)provider 凭据 — 在 settings 表里存,优先于 .env ===
|
|
# 留空 = 用 .env 里的 agnes_api_key(向后兼容,生产部署常用 .env 注入)
|
|
# 设值 = 走数据库(更便于在 UI 改 key,不用重启)
|
|
# 安全:API 返回 agnes_api_key_set bool,不回传明文
|
|
agnes_api_key: Mapped[str] = mapped_column(Text, default="", nullable=False)
|
|
agnes_base_url_override: Mapped[str] = mapped_column(
|
|
String(255), default="", nullable=False
|
|
) # 留空 = 用 .env
|
|
|
|
# === 美团大模型(LongCat,OpenAI 兼容)===
|
|
# 双 provider 评论架构:Angel + 美团并列,各跑各的 prompt,结果存到 articles 各自的列
|
|
# api_key 留空 = 不启用该 provider
|
|
meituan_api_key: Mapped[str] = mapped_column(Text, default="", nullable=False)
|
|
meituan_base_url: Mapped[str] = mapped_column(
|
|
String(255), default="https://api.longcat.chat/openai/v1", nullable=False
|
|
)
|
|
meituan_chat_model: Mapped[str] = mapped_column(
|
|
String(64), default="LongCat-2.0-Preview", nullable=False
|
|
)
|
|
meituan_interval_sec: Mapped[float] = mapped_column(default=2.0, nullable=False)
|
|
meituan_enabled: Mapped[bool] = mapped_column(Boolean, default=True, nullable=False)
|
|
meituan_commentary_prompt: Mapped[str | None] = mapped_column(Text) # 留空用默认
|
|
# 美团"无可奉告"主题清单 — 命中这些 topic 的文章,直接写"无可奉告" + 不调美团 API
|
|
# (避免 LongCat security_audit_fail 400 循环重试,污染评论失败率)
|
|
# 匹配规则:article.category 含任一项 OR 标题/正文命中关键词(宽松匹配)
|
|
meituan_blocked_topics: Mapped[list[str]] = mapped_column(
|
|
JSONB, nullable=False, default=list, server_default="[]"
|
|
)
|
|
meituan_blocked_keywords: Mapped[list[str]] = mapped_column(
|
|
JSONB, nullable=False, default=list, server_default="[]"
|
|
)
|
|
meituan_no_comment_text: Mapped[str] = mapped_column(
|
|
Text,
|
|
default="无可奉告",
|
|
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}>"
|