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(
|