Spaces:
Sleeping
Sleeping
added to prompt_template
Browse files
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)
|