Files
diary-news/backend/app/models/llm_setting.py
Mavis 380e8b124e feat(llm): 排版容器固定CSS + 插图用正文第一段 + 适中尺寸
- enrichment._enrich_format:把排版好的段落包到带固定 CSS 的 <div class=article-body> 里
  (font: system-ui / 17px / line-height 1.7 / color #3e3e3e / p margin-bottom 1.5em)
  CSS 同时内联到 style 属性,前端 .article-body 全局类做兑底
- enrichment._enrich_image:prompt 改用 body_zh_text 的第一段(原为 title);
  新增 {body} 占位符,image_prompt_template 默认模板同步改写
- 插图尺寸写死为 768x512(适中);image_size 字段保留供用户手改但默认行为不依赖它
- 分类明确多标签(2-5 个),提示词加 {body} 变量,容错读 categories/tags 两种 key
- AdminLlmSettings.vue:placeholder / 变量说明同步更新
2026-06-08 20:53:21 +08:00

51 lines
1.7 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.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)
# === 插图参数 ===
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}>"