File size: 614 Bytes
873e601 b8b0b89 873e601 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
# tool_loader.py
import logging
from transformers import load_tool
from utils.logger import log_response # Import the logger
from utils.tool_config import tool_names
class ToolLoader:
def __init__(self, tool_names):
self.tools = self.load_tools(tool_names)
def load_tools(self, tool_names):
loaded_tools = []
for tool_name in tool_names:
try:
tool = load_tool(tool_name)
loaded_tools.append(tool)
except Exception as e:
log_response(f"Error loading tool '{tool_name}': {e}")
return loaded_tools
|