yetessam commited on
Commit
e1ce2ee
·
verified ·
1 Parent(s): 9ee432e

Update tools/tools.py

Browse files
Files changed (1) hide show
  1. tools/tools.py +11 -3
tools/tools.py CHANGED
@@ -1,14 +1,21 @@
1
 
2
- from .final_answer import FinalAnswerTool
3
  from .polite_guard import PoliteGuardTool
4
- from .web_search import DuckDuckGoSearchTool
5
- from .time_tool import get_the_current_time_in_timezone
 
 
 
 
6
 
7
  # Initialize tools
8
  #final_answer = FinalAnswerTool() in the current version of smolagents, we don't need to load the final answer tool
9
  polite_guard = PoliteGuardTool()
10
  #web_search = DuckDuckGoSearchTool()
11
 
 
 
 
12
  # Import tool from Hub
13
  #image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
14
 
@@ -16,4 +23,5 @@ polite_guard = PoliteGuardTool()
16
  def load_tools():
17
  # Removing final answer from tools during build process
18
  #return [final_answer, polite_guard, web_search, get_the_current_time_in_timezone]
 
19
  return [polite_guard]
 
1
 
2
+ #from .final_answer import FinalAnswerTool
3
  from .polite_guard import PoliteGuardTool
4
+ #from .web_search import DuckDuckGoSearchTool
5
+ #from .time_tool import get_the_current_time_in_timezone
6
+
7
+ #from .profanity_guard import ProfanityGuardTool
8
+
9
+
10
 
11
  # Initialize tools
12
  #final_answer = FinalAnswerTool() in the current version of smolagents, we don't need to load the final answer tool
13
  polite_guard = PoliteGuardTool()
14
  #web_search = DuckDuckGoSearchTool()
15
 
16
+ #profanity_tool = ProfanityGuardTool() # or ProfanityGuardTool(model_name="tarekziade/pardonmyai-tiny")
17
+
18
+
19
  # Import tool from Hub
20
  #image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
21
 
 
23
  def load_tools():
24
  # Removing final answer from tools during build process
25
  #return [final_answer, polite_guard, web_search, get_the_current_time_in_timezone]
26
+ #return [polite_guard, profanity_tool]
27
  return [polite_guard]