fixed tools list
Browse files
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,
|
| 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,
|
|
|
|
| 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)
|