27 lines
677 B
Python
27 lines
677 B
Python
"""
|
|
LLM 提示词模板
|
|
|
|
从 src/prompts/ 目录加载 .md 文件作为提示词模板。
|
|
"""
|
|
|
|
import os
|
|
|
|
_PROMPTS_DIR = os.path.join(os.path.dirname(__file__), "prompts")
|
|
|
|
|
|
def _load_prompt(filename: str) -> str:
|
|
"""从 prompts 目录加载提示词文件内容。"""
|
|
path = os.path.join(_PROMPTS_DIR, filename)
|
|
with open(path, encoding="utf-8") as f:
|
|
return f.read()
|
|
|
|
|
|
def build_invoice_system_prompt() -> str:
|
|
"""构建发票提取系统提示词。"""
|
|
return _load_prompt("invoice_system.md")
|
|
|
|
|
|
def build_card_info_system_prompt() -> str:
|
|
"""构建支付截图信息提取系统提示词。"""
|
|
return _load_prompt("card_info_system.md")
|