Nioi commited on
Commit
050ce04
·
1 Parent(s): 77fac25

fixed tools list

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
agent.py CHANGED
@@ -1,6 +1,6 @@
1
  import os
2
  from smolagents import CodeAgent, DuckDuckGoSearchTool, OpenAIServerModel, VisitWebpageTool, LiteLLMModel
3
- from tools.attached_files import csv_reader, download_file_from_url, excel_reader, transcribe_audio
4
  from tools.basic_math import *
5
  from tools.browser import arvix_search, wiki_search
6
 
@@ -26,4 +26,4 @@ def get_agent(provider:str) -> CodeAgent:
26
  api_base="https://codestral.mistral.ai/v1/chat/completions",
27
  api_key=codestral_key)
28
 
29
- return CodeAgent(tools=[search_tool, web_page_tool, add, subtract, multiply, divide, modulus, rounder, power, square_root, download_file_from_url, csv_reader, excel_reader, transcribe_audio, wiki_search, arvix_search], model=model, additional_authorized_imports=['random', 'time'], add_base_tools=True)
 
1
  import os
2
  from smolagents import CodeAgent, DuckDuckGoSearchTool, OpenAIServerModel, VisitWebpageTool, LiteLLMModel
3
+ from tools.attached_files import csv_reader, excel_reader, transcribe_audio
4
  from tools.basic_math import *
5
  from tools.browser import arvix_search, wiki_search
6
 
 
26
  api_base="https://codestral.mistral.ai/v1/chat/completions",
27
  api_key=codestral_key)
28
 
29
+ return CodeAgent(tools=[search_tool, web_page_tool, add, subtract, multiply, divide, modulus, rounder, power, square_root, csv_reader, excel_reader, transcribe_audio, wiki_search, arvix_search], model=model, additional_authorized_imports=['random', 'time'], add_base_tools=True)