sqfoo commited on
Commit
0e3f38f
·
verified ·
1 Parent(s): 46b6a52

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +5 -5
agent.py CHANGED
@@ -157,6 +157,10 @@ def divide(a: float, b: float) -> float:
157
  return a / b
158
 
159
  # 3. Multi-Modality
 
 
 
 
160
 
161
  # ("human", f"Question: {question}\nReport to validate: {final_answer}")
162
  class BasicAgent:
@@ -188,12 +192,8 @@ class BasicAgent:
188
  - mp3_listen: listen to the content of the attached mp3 file by passing the file directory as input
189
  - image_caption: understand the visual content of the attached image by passing the image directory as input
190
  - python_tool: run the python code
191
- - multiply: multiply two numbers, A and B
192
- - add: add two numbers, A and B
193
- - subtract: Subtract A by B with passing A as the first argument
194
- - divide: Divide A by B with passing A as the first argument
195
  """
196
- self.tools = [web_search, visit_webpage, wiki_search, youtube_transcript, read_file, excel_read, mp3_listen, image_caption, python_tool, multiply, add, subtract, divide]
197
  self.prompt = ChatPromptTemplate.from_messages([
198
  ("system", self.sys_prompt),
199
  ("human", "{input}")
 
157
  return a / b
158
 
159
  # 3. Multi-Modality
160
+ # - multiply: multiply two numbers, A and B
161
+ # - add: add two numbers, A and B
162
+ # - subtract: Subtract A by B with passing A as the first argument
163
+ # - divide: Divide A by B with passing A as the first argument
164
 
165
  # ("human", f"Question: {question}\nReport to validate: {final_answer}")
166
  class BasicAgent:
 
192
  - mp3_listen: listen to the content of the attached mp3 file by passing the file directory as input
193
  - image_caption: understand the visual content of the attached image by passing the image directory as input
194
  - python_tool: run the python code
 
 
 
 
195
  """
196
+ self.tools = [web_search, visit_webpage, wiki_search, youtube_transcript, read_file, excel_read, mp3_listen, image_caption, python_tool]
197
  self.prompt = ChatPromptTemplate.from_messages([
198
  ("system", self.sys_prompt),
199
  ("human", "{input}")