Create tool_loader.py
Browse files- tool_loader.py +18 -0
tool_loader.py
ADDED
|
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# tool_loader.py
|
| 2 |
+
|
| 3 |
+
import logging
|
| 4 |
+
from transformers import load_tool
|
| 5 |
+
|
| 6 |
+
class ToolLoader:
|
| 7 |
+
def __init__(self, tool_names):
|
| 8 |
+
self.tools = self.load_tools(tool_names)
|
| 9 |
+
|
| 10 |
+
def load_tools(self, tool_names):
|
| 11 |
+
loaded_tools = []
|
| 12 |
+
for tool_name in tool_names:
|
| 13 |
+
try:
|
| 14 |
+
tool = load_tool(tool_name)
|
| 15 |
+
loaded_tools.append(tool)
|
| 16 |
+
except Exception as e:
|
| 17 |
+
log_response(f"Error loading tool '{tool_name}': {e}")
|
| 18 |
+
return loaded_tools
|