|
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=['unicodedata', 'stat', 'pandas', 'statistics', 'math', 'time', 'queue', 'collections', 'itertools', 'datetime', 'random', 're'], |
|
prompts_path='./prompts.yaml' |
|
) |
|
|
|
GradioUI(agent).launch() |
|
|