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