常量可配置:允许自定义agent-base-prompt.txt文件

This commit is contained in:
Chingfeng Li
2025-10-22 14:53:05 +08:00
parent 4b2d7da4e5
commit 3ba437c937
3 changed files with 6 additions and 3 deletions
+3
View File
@@ -179,6 +179,9 @@ prompt: |
- 长篇大论,叽叽歪歪
- 长时间严肃对话
# 默认系统提示词模板文件
prompt_template: agent-base-prompt.txt
# 结束语prompt
end_prompt:
enable: true # 是否开启结束语
@@ -66,7 +66,7 @@ class PromptManager:
def _load_base_template(self):
"""加载基础提示词模板"""
try:
template_path = "agent-base-prompt.txt"
template_path = self.config.get("prompt_template", "agent-base-prompt.txt")
cache_key = f"prompt_template:{template_path}"
# 先从缓存获取
@@ -88,7 +88,7 @@ class PromptManager:
self.base_prompt_template = template_content
self.logger.bind(tag=TAG).debug("成功加载基础提示词模板并缓存")
else:
self.logger.bind(tag=TAG).warning("未找到agent-base-prompt.txt文件")
self.logger.bind(tag=TAG).warning(f"未找到{template_path}文件")
except Exception as e:
self.logger.bind(tag=TAG).error(f"加载提示词模板失败: {e}")
@@ -38,7 +38,7 @@ class LLMPerformanceTester:
"""加载系统提示词"""
try:
prompt_file = os.path.join(
os.path.dirname(os.path.dirname(__file__)), "agent-base-prompt.txt"
os.path.dirname(os.path.dirname(__file__)), self.config.get("prompt_template", "agent-base-prompt.txt")
)
with open(prompt_file, "r", encoding="utf-8") as f:
content = f.read()