Spaces:
Sleeping
Sleeping
Update tools/tools.py
Browse files- 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]
|