zaldivards commited on
Commit
7c53c52
·
1 Parent(s): 741f470

Update tools list in MainAgent

Browse files
Files changed (1) hide show
  1. agent.py +8 -2
agent.py CHANGED
@@ -3,13 +3,14 @@ import os
3
  from smolagents import OpenAIServerModel, ToolCallingAgent
4
 
5
  from tools import (
6
- AudioTranscriber,
7
  txt_reader,
8
  pdf_reader,
9
  excel_reader,
10
  math_calculator,
11
  search_engine,
12
  image_transcriber,
 
13
  )
14
  from utils import OPENAI_MODEL_ID
15
 
@@ -24,6 +25,10 @@ If you are asked for a string, don't use articles, neither abbreviations (e.g. f
24
  If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.
25
 
26
  Each question will contain the question, the task ID, and if available, a file name.
 
 
 
 
27
  """
28
 
29
 
@@ -38,11 +43,12 @@ class MainAgent:
38
  tools=[
39
  search_engine,
40
  math_calculator,
41
- AudioTranscriber(),
42
  txt_reader,
43
  pdf_reader,
44
  excel_reader,
45
  image_transcriber,
 
46
  ],
47
  max_steps=5,
48
  model=model,
 
3
  from smolagents import OpenAIServerModel, ToolCallingAgent
4
 
5
  from tools import (
6
+ AudioTranscriberTool,
7
  txt_reader,
8
  pdf_reader,
9
  excel_reader,
10
  math_calculator,
11
  search_engine,
12
  image_transcriber,
13
+ YoutubeTranscriberTool,
14
  )
15
  from utils import OPENAI_MODEL_ID
16
 
 
25
  If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.
26
 
27
  Each question will contain the question, the task ID, and if available, a file name.
28
+
29
+ Key points:
30
+ - Do not add any prefix or suffix to YOUR FINAL ANSWER.
31
+ - Answer shortly and concisely, following the template strictly.
32
  """
33
 
34
 
 
43
  tools=[
44
  search_engine,
45
  math_calculator,
46
+ AudioTranscriberTool(),
47
  txt_reader,
48
  pdf_reader,
49
  excel_reader,
50
  image_transcriber,
51
+ YoutubeTranscriberTool(),
52
  ],
53
  max_steps=5,
54
  model=model,