liqiang
b119135836
Build latest book artifacts / build (push) Canceled after 0s
dependency resolution / resolve (3.11) (push) Canceled after 0s
dependency resolution / resolve (3.13) (push) Canceled after 0s
deploy-pages / build (push) Canceled after 0s
deploy-pages / deploy (push) Canceled after 0s
i18n consistency check / check (push) Canceled after 0s
provider adoption tests / test (chapter2/context-compression) (push) Canceled after 0s
provider adoption tests / test (chapter2/prompt-injection) (push) Canceled after 0s
provider adoption tests / test (chapter2/system-hint) (push) Canceled after 0s
provider adoption tests / test (chapter3/log-sanitization) (push) Canceled after 0s
web-search-agent tests / test (push) Canceled after 0s
web-search-agent tests / agentbook (push) Canceled after 0s
86 lines
2.9 KiB
Python
86 lines
2.9 KiB
Python
"""把 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())
|