prompt_loader.py 486 B

123456789101112131415161718
  1. # -*- coding: utf-8 -*-
  2. """文档对话提示词加载。"""
  3. from pathlib import Path
  4. from typing import Any, Dict
  5. import yaml
  6. PROJECT_ROOT = Path(__file__).resolve().parents[3]
  7. PROMPT_DIR = PROJECT_ROOT / "config" / "prompt"
  8. def load_prompt_config(file_name: str) -> Dict[str, Any]:
  9. prompt_path = PROMPT_DIR / file_name
  10. if not prompt_path.exists():
  11. return {}
  12. with open(prompt_path, "r", encoding="utf-8") as handle:
  13. return yaml.safe_load(handle) or {}