githubmirror/AgentPilot

src/system/base.py
38 строк1 KB
jbexta
0.3.2
12 сен 2024, 00:40
12 сен 2024, 00:4060cc9b9
from src.system.apis import APIManager
from src.system.config import ConfigManager
from src.system.blocks import BlockManager
from src.system.files import FileManager
from src.system.providers import ProviderManager
from src.system.roles import RoleManager
from src.system.environments import EnvironmentManager
from src.system.plugins import PluginManager
from src.system.tools import ToolManager
from src.system.vectordbs import VectorDBManager
from src.system.venvs import VenvManager
# from src.system.workspaces import WorkspaceManager


class SystemManager:
    def __init__(self):
        self.apis = APIManager()
        self.blocks = BlockManager(parent=self)
        self.config = ConfigManager()
        self.files = FileManager()
        self.providers = ProviderManager()
        # self.models = ModelManager()
        self.plugins = PluginManager()
        self.roles = RoleManager()
        self.environments = EnvironmentManager()
        self.tools = ToolManager(parent=self)
        self.vectordbs = VectorDBManager(parent=self)
        self.venvs = VenvManager(parent=self)
        # self.workspaces = WorkspaceManager(parent=self)
        # self.load()

    def load(self):
        for mgr in self.__dict__.values():
            if hasattr(mgr, 'load'):
                mgr.load()


manager = SystemManager()