|
from smolagents import GradioUI, CodeAgent, HfApiModel |
|
from scripts.web_search import DuckDuckGoSearchTool |
|
from scripts.visit_webpage import VisitWebpageTool |
|
from scripts.final_answer import FinalAnswerTool |
|
|
|
model = HfApiModel() |
|
web_search = DuckDuckGoSearchTool() |
|
visit_webpage = VisitWebpageTool() |
|
final_answer = FinalAnswerTool() |
|
|
|
agent = CodeAgent( |
|
model=model, |
|
tools= [web_search, visit_webpage, final_answer], |
|
max_steps=6, |
|
verbosity_level=1, |
|
grammar=None, |
|
planning_interval=None, |
|
name=None, |
|
description=None, |
|
authorized_imports=['datetime', 'queue', 'stat', 'itertools', 'pandas', 'random', 'collections', 'math', 'statistics', 'unicodedata', 're', 'time'], |
|
prompts_path='./prompts.yaml' |
|
) |
|
|
|
GradioUI(agent).launch() |
|
|