"""
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_travel_info_system_prompt() -> str:
    """构建差旅信息提取系统提示词。"""
    return _load_prompt("travel_info_system.md")


def build_normal_info_system_prompt() -> str:
    """构建普通发票信息提取系统提示词。"""
    return _load_prompt("normal_info_system.md")
