aaf728f3f4
- llm_settings.agnes_api_key TEXT (DB key 优先,.env 兜底) - llm_settings.agnes_base_url_override VARCHAR (留空 = 用 .env) - alembic 0005_agnes_key 迁移 - LlmSettingOut.agnes_api_key_set (bool) 替代直接回传 key - LlmSettingUpdate 加 agnes_api_key / agnes_base_url_override(可空可清空) - providers.get_angel_client 改用 DB key 优先 - enrichment.py 改为 get_angel_client() 工厂调用(热改 key 不需重启) - /admin/llm/settings/test 走 get_angel_client(测的是 DB 里的 key) - 前端 AdminLlmSettings 在'总开关 + 模型'卡里加 Angel api_key 输入框 + base_url 覆盖 + 已配置/未配置指示灯 + 清空按钮 - 顶部'测连接'按钮复用(测的就是 Angel)
82 lines
3.3 KiB
Python
82 lines
3.3 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) # 留空用默认
|
|
|
|
# === 时间 ===
|
|
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}>"
|