Commit
·
7c53c52
1
Parent(s):
741f470
Update tools list in MainAgent
Browse files
agent.py
CHANGED
|
@@ -3,13 +3,14 @@ import os
|
|
| 3 |
from smolagents import OpenAIServerModel, ToolCallingAgent
|
| 4 |
|
| 5 |
from tools import (
|
| 6 |
-
|
| 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 |
-
|
| 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,
|