3218485270
f493bde8a9
- 缓存热点批量流程（有采集缓存不触发 Google） - 简报不足直接从采集缓存生成（轻量补齐） - 三图合成（模特/印花/底图）+ 底图压缩 <2MB - 热点去重→风格去重自动切换 + 不适合类目 review 兜底 - 透明背景（background=transparent）+ 提示词清洗（敏感词/背景描述） - 任务前 basemap 校验 + 模板国家校验 + 模特任务级分配
58 lines
2.2 KiB
Python
58 lines
2.2 KiB
Python
"""配置与提示词加载工具。
|
|
|
|
职责：
|
|
- 读取 configs/countries/<country>.yaml（该国专属种子词/权重/limit 等覆盖）
|
|
- 读取 prompts/<country>/aesthetics.yaml（该国审美 hint、风格-配色 extra 规则、额外黑名单）
|
|
- 读取 prompts/<country>/system_prompt.md（该国 LLM 系统提示覆盖）
|
|
- 把上述合并进 country_config，供节点使用
|
|
"""
|
|
from pathlib import Path
|
|
from typing import Any, Dict
|
|
|
|
import yaml
|
|
|
|
|
|
def load_yaml_safe(path: Path) -> Dict[str, Any]:
|
|
if not path.exists():
|
|
return {}
|
|
try:
|
|
return yaml.safe_load(path.read_text(encoding="utf-8")) or {}
|
|
except Exception as e: # noqa: BLE001
|
|
print(f"[loader] 解析失败 {path}: {e}")
|
|
return {}
|
|
|
|
|
|
def load_text_safe(path: Path) -> str:
|
|
if not path.exists():
|
|
return ""
|
|
try:
|
|
return path.read_text(encoding="utf-8").strip()
|
|
except Exception: # noqa: BLE001
|
|
return ""
|
|
|
|
|
|
def build_country_config(global_config: Dict[str, Any], country: str, project_root: Path) -> Dict[str, Any]:
|
|
"""合并：全局配置 + 国家专属 yaml + 国家审美 yaml。"""
|
|
cc: Dict[str, Any] = load_yaml_safe(project_root / "configs" / "countries" / f"{country}.yaml")
|
|
|
|
aesthetics = load_yaml_safe(project_root / "prompts" / country / "aesthetics.yaml")
|
|
cc.setdefault("style_hint", aesthetics.get("style_hint", ""))
|
|
cc.setdefault("extra_style_rules", aesthetics.get("extra_style_rules", []) or [])
|
|
cc.setdefault("extra_blacklist", aesthetics.get("extra_blacklist", []) or [])
|
|
# 国家专属趋势/风格/行业种子（若 aesthetics 里有也并入 cc 顶层，便于 source 读取）
|
|
for k in ("trending", "style", "related", "timeframe"):
|
|
if k in aesthetics and k not in cc:
|
|
cc[k] = aesthetics[k]
|
|
return cc
|
|
|
|
|
|
def load_system_prompt(prompts_dir: Path, default: str) -> str:
|
|
"""该国 prompts/<country>/system_prompt.md 作为「补充段」叠加到默认规则之后。
|
|
|
|
这样既能按国家定制口吻/合规重点，又不丢失内置核心合规规则。若该国未提供文件则用默认。
|
|
"""
|
|
text = load_text_safe(prompts_dir / "system_prompt.md")
|
|
if not text:
|
|
return default
|
|
return default + "\n\n# 该国专属补充指令\n" + text
|