ContentAgent / tools /tools.py
yetessam's picture
Update tools/tools.py
1b9ee0e verified
raw
history blame
729 Bytes
from .final_answer import FinalAnswerTool
from .polite_guard import PoliteGuardTool
from .web_search import DuckDuckGoSearchTool
from .time_tool import get_the_current_time_in_timezone
# Initialize tools
#final_answer = FinalAnswerTool() in the current version of smolagents, we don't need to load the final answer tool
polite_guard = PoliteGuardTool()
web_search = DuckDuckGoSearchTool()
# Import tool from Hub
#image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
# Define all tools
def load_tools():
# Removing final answer from tools during build process
#return [final_answer, polite_guard, web_search, get_the_current_time_in_timezone]
return [polite_guard, web_search]