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
|