mirror of
https://github.com/xinnan-tech/xiaozhi-esp32-server.git
synced 2026-07-27 09:33:55 +08:00
Merge pull request #2399 from xinnan-tech/main-agent-base-prompt
常量可配置:允许自定义agent-base-prompt.txt文件
This commit is contained in:
@@ -179,6 +179,9 @@ prompt: |
|
|||||||
- 长篇大论,叽叽歪歪
|
- 长篇大论,叽叽歪歪
|
||||||
- 长时间严肃对话
|
- 长时间严肃对话
|
||||||
|
|
||||||
|
# 默认系统提示词模板文件
|
||||||
|
prompt_template: agent-base-prompt.txt
|
||||||
|
|
||||||
# 结束语prompt
|
# 结束语prompt
|
||||||
end_prompt:
|
end_prompt:
|
||||||
enable: true # 是否开启结束语
|
enable: true # 是否开启结束语
|
||||||
|
|||||||
@@ -66,7 +66,7 @@ class PromptManager:
|
|||||||
def _load_base_template(self):
|
def _load_base_template(self):
|
||||||
"""加载基础提示词模板"""
|
"""加载基础提示词模板"""
|
||||||
try:
|
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}"
|
cache_key = f"prompt_template:{template_path}"
|
||||||
|
|
||||||
# 先从缓存获取
|
# 先从缓存获取
|
||||||
@@ -88,7 +88,7 @@ class PromptManager:
|
|||||||
self.base_prompt_template = template_content
|
self.base_prompt_template = template_content
|
||||||
self.logger.bind(tag=TAG).debug("成功加载基础提示词模板并缓存")
|
self.logger.bind(tag=TAG).debug("成功加载基础提示词模板并缓存")
|
||||||
else:
|
else:
|
||||||
self.logger.bind(tag=TAG).warning("未找到agent-base-prompt.txt文件")
|
self.logger.bind(tag=TAG).warning(f"未找到{template_path}文件")
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
self.logger.bind(tag=TAG).error(f"加载提示词模板失败: {e}")
|
self.logger.bind(tag=TAG).error(f"加载提示词模板失败: {e}")
|
||||||
|
|
||||||
|
|||||||
@@ -38,7 +38,7 @@ class LLMPerformanceTester:
|
|||||||
"""加载系统提示词"""
|
"""加载系统提示词"""
|
||||||
try:
|
try:
|
||||||
prompt_file = os.path.join(
|
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:
|
with open(prompt_file, "r", encoding="utf-8") as f:
|
||||||
content = f.read()
|
content = f.read()
|
||||||
|
|||||||
Reference in New Issue
Block a user