Update agent.py
Browse files
agent.py
CHANGED
@@ -14,7 +14,7 @@ from smolagents import (
|
|
14 |
)
|
15 |
|
16 |
# Local
|
17 |
-
from tools import GetTaskFileTool, LoadXlsxFileTool, LoadTextFileTool
|
18 |
|
19 |
sys_instruction = (
|
20 |
"You are a general-purpose AI assistant. For each question, first reason through your answer, "
|
@@ -107,6 +107,7 @@ def get_agent(
|
|
107 |
SpeechToTextTool(),
|
108 |
LoadXlsxFileTool(),
|
109 |
LoadTextFileTool(),
|
|
|
110 |
FinalAnswerTool(),
|
111 |
],
|
112 |
model=get_model(
|
|
|
14 |
)
|
15 |
|
16 |
# Local
|
17 |
+
from tools import GetTaskFileTool, LoadXlsxFileTool, LoadTextFileTool, AnalyzeImageTool
|
18 |
|
19 |
sys_instruction = (
|
20 |
"You are a general-purpose AI assistant. For each question, first reason through your answer, "
|
|
|
107 |
SpeechToTextTool(),
|
108 |
LoadXlsxFileTool(),
|
109 |
LoadTextFileTool(),
|
110 |
+
AnalyzeImageTool(model_id=model_id),
|
111 |
FinalAnswerTool(),
|
112 |
],
|
113 |
model=get_model(
|