File size: 1,107 Bytes
0d91eab |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 |
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",
# "api_base": OLLAMA_API_BASE,
"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,
}
|