loader.py

import importlib
import os
from typing import TypeVar

from framework import Assistant
from logger import logger

# from langchain_core.tools.structured import StructuredTool


# def load_agents() -> dict[str, Agent]:
#     return load_objects(Agent, "agents")


def load_assistants() -> dict[str, Assistant]:
    return load_objects(Assistant, "assistants")


# def load_tools() -> dict[str, StructuredTool]:
#     return load_objects(StructuredTool, "tools")


T = TypeVar("T")


def load_objects(object_type: type[T], dirname: str) -> dict[str, T]:
    objects = {}
    objects_dir = os.path.join(os.path.dirname(__file__), dirname)
    logger.info("Loading objects from directory: %s", objects_dir)
    for object_file in os.listdir(objects_dir):
        object_file = os.path.basename(object_file)
        module_name, extension = os.path.splitext(object_file)
        if extension != ".py":
            continue
        logger.info("Loading file: %s", object_file)
        module_name = f"{dirname}.{module_name}"
        module = importlib.import_module(module_name)
        for key, value in vars(module).items():
            if not isinstance(value, object_type):
                continue
            objects[key] = value
            logger.info("Object loaded: %s", key)
    return objects