Update app.py
Browse files
app.py
CHANGED
@@ -75,12 +75,14 @@ custom_role_conversions=None,
|
|
75 |
# Import tool from Hub
|
76 |
image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
|
77 |
|
|
|
|
|
78 |
with open("prompts.yaml", 'r') as stream:
|
79 |
prompt_templates = yaml.safe_load(stream)
|
80 |
|
81 |
agent = CodeAgent(
|
82 |
model=model,
|
83 |
-
tools=[final_answer, my_custom_tool, get_current_time_in_timezone, image_generation_tool,
|
84 |
max_steps=6,
|
85 |
verbosity_level=1,
|
86 |
grammar=None,
|
|
|
75 |
# Import tool from Hub
|
76 |
image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
|
77 |
|
78 |
+
search_tool = DuckDuckGoSearchTool()
|
79 |
+
|
80 |
with open("prompts.yaml", 'r') as stream:
|
81 |
prompt_templates = yaml.safe_load(stream)
|
82 |
|
83 |
agent = CodeAgent(
|
84 |
model=model,
|
85 |
+
tools=[final_answer, my_custom_tool, get_current_time_in_timezone, image_generation_tool, search_tool], ## add your tools here (don't remove final answer)
|
86 |
max_steps=6,
|
87 |
verbosity_level=1,
|
88 |
grammar=None,
|