tool_loader.py

# 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