huytofu92 commited on
Commit
3397cf1
·
1 Parent(s): b114e5e

Final modifications before final run

Browse files
Files changed (2) hide show
  1. mini_agents.py +3 -2
  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, tavily_search_tool],
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. It can also use Tavily to search the web for more relevantcontext."
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: