|
|
from smolagents import ( |
|
|
DuckDuckGoSearchTool, |
|
|
VisitWebpageTool, |
|
|
WikipediaSearchTool, |
|
|
) |
|
|
from tools.text_search import TextSearch |
|
|
from tools.text_splitter import text_splitter |
|
|
from tools.webpage_parser import WebpageParser |
|
|
from tools.parse_wikipedia_table import WikipediaParser |
|
|
from tools.open_files import OpenFilesTool |
|
|
|
|
|
DEFAULT_ARGS = myagent_args = { |
|
|
"provider": "litellm", |
|
|
"model_id": "gemini/gemini-2.0-flash-lite", |
|
|
|
|
|
"planning_interval": 3, |
|
|
"add_base_tools": True, |
|
|
"tools": [ |
|
|
DuckDuckGoSearchTool(), |
|
|
WikipediaParser(), |
|
|
VisitWebpageTool(), |
|
|
TextSearch(), |
|
|
text_splitter, |
|
|
WikipediaSearchTool( |
|
|
content_type="text", |
|
|
extract_format="HTML" |
|
|
), |
|
|
WebpageParser(), |
|
|
OpenFilesTool(), |
|
|
], |
|
|
"additional_authorized_imports": [ |
|
|
"pandas", |
|
|
"numpy", |
|
|
"datetime", |
|
|
"json", |
|
|
"re", |
|
|
"math", |
|
|
"os", |
|
|
"requests", |
|
|
"csv", |
|
|
"urllib", |
|
|
], |
|
|
"num_ctx": 128_000, |
|
|
"temperature": 0.2, |
|
|
} |
|
|
|
