wizofavalon commited on
Commit
3cd7ccc
·
verified ·
1 Parent(s): b21e069

added to prompt_template

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -53,6 +53,7 @@ image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_co
53
  with open("prompts.yaml", 'r') as stream:
54
  prompt_templates = yaml.safe_load(stream)
55
  print("TOOLS:", [final_answer, my_custom_tool, get_current_time_in_timezone, image_generation_tool])
 
56
  agent = CodeAgent(
57
  model=model,
58
  tools=[final_answer,get_current_time_in_timezone,my_custom_tool,image_generation_tool], ## add your tools here (don't remove final answer)
 
53
  with open("prompts.yaml", 'r') as stream:
54
  prompt_templates = yaml.safe_load(stream)
55
  print("TOOLS:", [final_answer, my_custom_tool, get_current_time_in_timezone, image_generation_tool])
56
+ prompt_templates["final_answer"] = "This is the final answer to the user's query: {output}"
57
  agent = CodeAgent(
58
  model=model,
59
  tools=[final_answer,get_current_time_in_timezone,my_custom_tool,image_generation_tool], ## add your tools here (don't remove final answer)