hzli commited on
Commit
e0f69e3
·
verified ·
1 Parent(s): 1559da8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -52,7 +52,11 @@ image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_co
52
 
53
  with open("prompts.yaml", 'r') as stream:
54
  prompt_templates = yaml.safe_load(stream)
55
- prompt_templates["final_answer"] = "最终答案是: {answer}"
 
 
 
 
56
  print(prompt_templates)
57
 
58
  agent = CodeAgent(
 
52
 
53
  with open("prompts.yaml", 'r') as stream:
54
  prompt_templates = yaml.safe_load(stream)
55
+ # 直接在代码中补充 final_answer 下的 pre_messages 模板
56
+ if 'final_answer' not in prompt_templates:
57
+ prompt_templates['final_answer'] = {}
58
+ prompt_templates['final_answer']['pre_messages'] = "这是最终答案生成前的提示信息,你可以根据实际情况修改内容。"
59
+
60
  print(prompt_templates)
61
 
62
  agent = CodeAgent(