lucasnseq commited on
Commit
a5d264f
·
verified ·
1 Parent(s): 2118268

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -1
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(