ContentAgent / tools /tools.py
yetessam's picture
Update tools/tools.py
e1ce2ee verified
raw
history blame contribute delete
923 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
#from .profanity_guard import ProfanityGuardTool
# 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()
#profanity_tool = ProfanityGuardTool() # or ProfanityGuardTool(model_name="tarekziade/pardonmyai-tiny")
# 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, profanity_tool]
return [polite_guard]