File size: 602 Bytes
873e601 5b63d01 ec1cbe8 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 logger import log_response # Import the logger
from 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
|