githubmirror/AgentPilot

agentpilot/utils/plugin.py

16 строк522 B

jbexta

faster loading, plugins

21 дек 2023, 06:12

21 дек 2023, 06:12fc9ae75

import importlib
import inspect
from agentpilot.agent.base import Agent


def get_plugin_agent_class(plugin_name, kwargs=None):
    if not plugin_name:
        return None  # Agent(**kwargs)
    if kwargs is None:
        kwargs = {}

    return next((AC(**kwargs)
                 for AC in importlib.import_module(
        f"agentpilot.plugins.{plugin_name}.modules.agent_plugin").__dict__.values()
                 if inspect.isclass(AC) and issubclass(AC, Agent) and not AC.__name__ == 'Agent'),
                None)