77 lines
2.7 KiB
Python
77 lines
2.7 KiB
Python
"""Phase 5 垂直切片运行脚本（CLI）。
|
||
|
||
用法:
|
||
python scripts/run_phase5_slice.py --template <PATH> --output <PATH> [--fake]
|
||
|
||
- --fake: 使用内置 FakeEngine（无需 LLM 客户端/API Key），用于验证管线打通。
|
||
- 不带 --fake: 使用真实 InferenceEngine()，需先配置 LLM 客户端（人工质量门禁）。
|
||
|
||
管线: WordTemplateParser → StructuredSource → WriteOrchestrator.generate → docx。
|
||
"""
|
||
from __future__ import annotations
|
||
|
||
import argparse
|
||
from types import SimpleNamespace
|
||
|
||
from docx import Document
|
||
|
||
from genesis.data_models import StructuredSource
|
||
from genesis.parsers.word_template_parser import WordTemplateParser
|
||
from genesis.writer.orchestrator import WriteOrchestrator
|
||
|
||
|
||
class FakeEngine:
|
||
"""切片验证用假引擎：返回固定章节内容，避免真实 LLM 调用。"""
|
||
|
||
def chat_structured(self, *, session_id, prompt, variables, schema, retry_count=2):
|
||
return SimpleNamespace(
|
||
data={
|
||
"title": variables.get("title", "章"),
|
||
"blocks": [{"type": "paragraph", "text": f"【{variables.get('title','')}】自动生成的内容（fake 模式）"}],
|
||
},
|
||
status="ok",
|
||
)
|
||
|
||
|
||
def _build_source(template_path: str) -> StructuredSource:
|
||
parsed = WordTemplateParser().parse(template_path)
|
||
return StructuredSource(
|
||
tables=[],
|
||
template=parsed,
|
||
rule_docs=[],
|
||
image_analyses=[],
|
||
existing_system=None,
|
||
comments=[],
|
||
)
|
||
|
||
|
||
def _main() -> None:
|
||
parser = argparse.ArgumentParser(description="Phase 5 垂直切片运行脚本")
|
||
parser.add_argument("--template", required=True, help="概要设计模板 docx 路径")
|
||
parser.add_argument("--output", required=True, help="输出 docx 路径")
|
||
parser.add_argument("--fake", action="store_true", help="使用 FakeEngine（无 LLM）")
|
||
parser.add_argument("--samples-dir", default="samples", help="RAG 样本目录")
|
||
args = parser.parse_args()
|
||
|
||
source = _build_source(args.template)
|
||
engine = FakeEngine() if args.fake else None # None → 真实 InferenceEngine()（需客户端配置）
|
||
|
||
orch = WriteOrchestrator()
|
||
contents = orch.generate(
|
||
source,
|
||
args.output,
|
||
samples_dir=args.samples_dir,
|
||
engine=engine,
|
||
template_path=args.template,
|
||
)
|
||
loaded = Document(args.output)
|
||
joined = "\n".join(p.text for p in loaded.paragraphs)
|
||
print(f"[slice] 生成章节数: {len(contents)}")
|
||
print(f"[slice] 输出路径: {args.output}")
|
||
print(f"[slice] 注入校验: {'OK' if any('自动生成' in p.text for p in loaded.paragraphs) else 'EMPTY'}")
|
||
print(f"[slice] 文本内容预览:\n{joined[:200]}")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
_main()
|