Spaces:
Sleeping
Sleeping
Final modifications before final run
Browse files- mini_agents.py +3 -2
- prompts/prompts.yaml +1 -1
mini_agents.py
CHANGED
@@ -113,13 +113,13 @@ context_search_model = InferenceClientModel(
|
|
113 |
|
114 |
context_search_agent = CodeAgent(
|
115 |
model=context_search_model,
|
116 |
-
tools=[*search_tools
|
117 |
max_steps=4,
|
118 |
additional_authorized_imports=AUTHORIZED_IMPORTS,
|
119 |
planning_interval=4,
|
120 |
name="context_search_agent",
|
121 |
prompt_templates=PROMPT_TEMPLATE["context_search_agent"],
|
122 |
-
description="This agent is responsible for searching the web for context using wikipedia for general information and arxiv for scientific information.
|
123 |
)
|
124 |
|
125 |
master_model = InferenceClientModel(
|
@@ -137,6 +137,7 @@ class MasterAgentWrapper:
|
|
137 |
read_python_file_from_path,
|
138 |
PythonInterpreterTool(),
|
139 |
DuckDuckGoSearchTool(),
|
|
|
140 |
*community_tools,
|
141 |
]
|
142 |
|
|
|
113 |
|
114 |
context_search_agent = CodeAgent(
|
115 |
model=context_search_model,
|
116 |
+
tools=[*search_tools],
|
117 |
max_steps=4,
|
118 |
additional_authorized_imports=AUTHORIZED_IMPORTS,
|
119 |
planning_interval=4,
|
120 |
name="context_search_agent",
|
121 |
prompt_templates=PROMPT_TEMPLATE["context_search_agent"],
|
122 |
+
description="This agent is responsible for searching the web for context using wikipedia for general information and arxiv for scientific information."
|
123 |
)
|
124 |
|
125 |
master_model = InferenceClientModel(
|
|
|
137 |
read_python_file_from_path,
|
138 |
PythonInterpreterTool(),
|
139 |
DuckDuckGoSearchTool(),
|
140 |
+
tavily_search_tool,
|
141 |
*community_tools,
|
142 |
]
|
143 |
|
prompts/prompts.yaml
CHANGED
@@ -157,7 +157,7 @@ system_prompt: |-
|
|
157 |
```
|
158 |
|
159 |
{%- if managed_agents and managed_agents.values() | list %}
|
160 |
-
You should also give tasks to team members whenever possible. They are very useful to solve tasks that require multiple modalities.
|
161 |
Calling a team member works the same as for calling a tool: you can simply pass the task description as a string to the member without specifying any argument.
|
162 |
Given that this team member is a real human, you should be very verbose in your task, it should be a long string providing informations as detailed as necessary.
|
163 |
Here is a list of the team members that you can call:
|
|
|
157 |
```
|
158 |
|
159 |
{%- if managed_agents and managed_agents.values() | list %}
|
160 |
+
You should also give tasks to team members whenever possible. They are very useful to solve tasks that require multiple modalities. One member is also adept at searching wikipedia and arxiv for information.
|
161 |
Calling a team member works the same as for calling a tool: you can simply pass the task description as a string to the member without specifying any argument.
|
162 |
Given that this team member is a real human, you should be very verbose in your task, it should be a long string providing informations as detailed as necessary.
|
163 |
Here is a list of the team members that you can call:
|