final-agent-course / agent_tools.py
jjvelezo's picture
Create agent_tools.py
8e7817a verified
from smolagents import Tool, DuckDuckGoSearchTool, GoogleSearchTool, VisitWebpageTool, WikipediaSearchTool, PythonInterpreterTool, FinalAnswerTool
from tavily import TavilyClient
from dotenv import load_dotenv
import os
load_dotenv()
# Inicialización de herramientas
duck_search = DuckDuckGoSearchTool()
google_search = GoogleSearchTool()
visit_page = VisitWebpageTool()
wiki_search = WikipediaSearchTool()
do_python = PythonInterpreterTool()
final_answer = FinalAnswerTool()
tavily_search = TavilyClient()
speech_to_text_tool = Tool.from_space("hf-audio/whisper-large-v3-turbo",
name="speech_to_text_tool",
description="Convierte audio a texto a partir de un archivo o URL.",
api_name="/predict")
visual_qa_tool = Tool.from_space("sitammeur/PicQ",
name="visual_qa_tool",
description="Responde preguntas sobre una imagen proporcionada.",
api_name="/predict")