ai-agent-book 精选快照（<2MB 代码与文档，来自 github.com/bojieli/ai-agent-book）

chapter9/ai-style-skill/rewrite_demo.py

@@ -0,0 +1,85 @@
+"""把 active 规则连同原文交给 LLM 改写，返回 before/after 与证据回执。"""
+
+from __future__ import annotations
+
+import json
+import re
+from typing import Any, Dict, List, Tuple
+
+from llm_client import chat
+
+# 演示用原文：刻意集齐八类 AI 味。
+SAMPLE_TEXT = (
+    "在这个效率至上的时代，让我们一起重新认识这款笔记工具——它不是简单的记录软件——"
+    "而是你的第二大脑。首先，它能自动整理灵感；其次，它让检索快如闪电；"
+    "最后，它让分享毫无门槛——仿佛一位永不疲倦的管家。总而言之，让我们从现在开始，"
+    "把每一条灵感都安顿好 🚀✨💡。"
+)
+
+_REWRITE_PROMPT = """你是中文文案改写助手。请按下面的写作规则改写给出的文案，消除所有违规之处，
+保持原意，不要增加新事实。只返回 JSON：{{"rewritten": "改写后的全文",
+"applied_rules": ["实际应用的规则 id"]}}
+
+写作规则：
+{rules}
+
+待改写文案：
+{text}
+"""
+
+
+def rewrite_with_llm(
+    text: str,
+    rules: List[Dict[str, Any]],
+    *,
+    provider: str,
+    model: str | None = None,
+    seed: int = 8901,
+) -> Tuple[Dict[str, Any], Dict[str, Any]]:
+    """真实 LLM 改写路径，返回 (改写结果, 证据回执)。"""
+    brief_rules = [
+        {"id": r["id"], "name": r["name"], "definition": r["definition"],
+         "bad_example": r.get("bad_example", ""), "good_example": r.get("good_example", "")}
+        for r in rules
+    ]
+    content, receipt = chat(
+        [{"role": "user", "content": _REWRITE_PROMPT.format(
+            rules=json.dumps(brief_rules, ensure_ascii=False, indent=2), text=text)}],
+        provider=provider, model=model, seed=seed,
+    )
+    payload = json.loads(re.sub(r"^```(?:json)?\s*|\s*```$", "", content.strip(), flags=re.I))
+    return {
+        "mode": "real_llm",
+        "original": text,
+        "rewritten": payload.get("rewritten", ""),
+        "applied_rules": payload.get("applied_rules", []),
+    }, receipt
+
+
+def _load_active_rules() -> List[Dict[str, Any]]:
+    import json as _json
+    from pathlib import Path
+
+    rules_path = Path(__file__).resolve().parent / "skill" / "rules.json"
+    if not rules_path.exists():
+        raise SystemExit("请先生成 Skill：python demo.py 或 python run_ai_style_skill.py")
+    return _json.loads(rules_path.read_text(encoding="utf-8"))
+
+
+def main() -> int:
+    import argparse
+
+    parser = argparse.ArgumentParser(description=__doc__)
+    parser.add_argument("--provider", choices=("ark", "openrouter", "openai"), default="openai")
+    parser.add_argument("--model", default=None)
+    args = parser.parse_args()
+    rules = _load_active_rules()
+    result, _ = rewrite_with_llm(
+        SAMPLE_TEXT, rules, provider=args.provider, model=args.model
+    )
+    print(json.dumps(result, ensure_ascii=False, indent=2))
+    return 0
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())