RafaelAgreda commited on
Commit
76ea624
·
verified ·
1 Parent(s): b7105bc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -5
app.py CHANGED
@@ -103,7 +103,9 @@ image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_co
103
 
104
  with open("prompts.yaml", 'r') as stream:
105
  prompt_templates = yaml.safe_load(stream)
106
-
 
 
107
  agent = CodeAgent(
108
  model=model,
109
  tools=[final_answer,search_audiobooks,get_current_time_in_timezone,filter_by_time], ## add your tools here (don't remove final answer)
@@ -113,10 +115,7 @@ agent = CodeAgent(
113
  planning_interval=None,
114
  name=None,
115
  description=None,
116
- #prompt_templates=prompt_templates
117
- prompt_templates={
118
- "final_answer": "Final answer: {answer}",prompt_templates
119
- }
120
  )
121
 
122
 
 
103
 
104
  with open("prompts.yaml", 'r') as stream:
105
  prompt_templates = yaml.safe_load(stream)
106
+
107
+ prompt_templates["final_answer"] = "Final answer: {answer}"
108
+
109
  agent = CodeAgent(
110
  model=model,
111
  tools=[final_answer,search_audiobooks,get_current_time_in_timezone,filter_by_time], ## add your tools here (don't remove final answer)
 
115
  planning_interval=None,
116
  name=None,
117
  description=None,
118
+ prompt_templates=prompt_templates
 
 
 
119
  )
120
 
121