from jiuwen.agent.config.base import AgentConfig
from jiuwen.core.context.controller_context.message_manager import MessageMgr
from jiuwen.core.context.controller_context.model_manager import ModelMgr
from jiuwen.core.context.controller_context.tool_manager import ToolMgr
from jiuwen.core.context.controller_context.workflow_manager import WorkflowMgr


class ControllerContextMgr:
    """
    Agent上下文管理器：
    """
    def __init__(self, agent_config: AgentConfig):
        self.workflow_mgr = WorkflowMgr(agent_config)
        self.tool_mgr = ToolMgr()
        self.model_mgr = ModelMgr()
        self.message_mgr = MessageMgr()
