File size: 923 Bytes
5e4863e
e1ce2ee
bf4cd3c
e1ce2ee
 
 
 
 
 
0f51d62
 
730a18a
0f51d62
ce9ffcf
0f51d62
e1ce2ee
 
 
060bcfa
 
 
0f51d62
edeb4b3
0689bf3
 
e1ce2ee
ce9ffcf
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28

#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]