nianjie/backend/prompt.py

13 lines
361 B
Python

from .config import PROMPT_PATH, DEFAULT_SYSTEM_PROMPT_TEXT
def load_system_prompt_text() -> str:
try:
text = PROMPT_PATH.read_text(encoding="utf-8").strip()
return text or DEFAULT_SYSTEM_PROMPT_TEXT
except Exception:
return DEFAULT_SYSTEM_PROMPT_TEXT
SYSTEM_PROMPT = {"role": "system", "content": load_system_prompt_text()}