autogen_agentchat.state._states 源代码

from typing import Any, List, Mapping, Optional

from pydantic import BaseModel, Field



[文档]
class BaseState(BaseModel):
    """所有可保存状态的基础类"""

    type: str = Field(default="BaseState")
    version: str = Field(default="1.0.0")




[文档]
class AssistantAgentState(BaseState):
    """助手代理的状态。"""

    llm_context: Mapping[str, Any] = Field(default_factory=lambda: dict([("messages", [])]))
    type: str = Field(default="AssistantAgentState")




[文档]
class TeamState(BaseState):
    """代理团队的状态。"""

    agent_states: Mapping[str, Any] = Field(default_factory=dict)
    type: str = Field(default="TeamState")




[文档]
class BaseGroupChatManagerState(BaseState):
    """所有群聊管理器的基类状态。"""

    message_thread: List[Mapping[str, Any]] = Field(default_factory=list)
    current_turn: int = Field(default=0)
    type: str = Field(default="BaseGroupChatManagerState")




[文档]
class ChatAgentContainerState(BaseState):
    """聊天代理容器的状态。"""

    agent_state: Mapping[str, Any] = Field(default_factory=dict)
    message_buffer: List[Mapping[str, Any]] = Field(default_factory=list)
    type: str = Field(default="ChatAgentContainerState")




[文档]
class RoundRobinManagerState(BaseGroupChatManagerState):
    """:class:`~autogen_agentchat.teams.RoundRobinGroupChat` 管理器的状态。"""

    next_speaker_index: int = Field(default=0)
    type: str = Field(default="RoundRobinManagerState")




[文档]
class SelectorManagerState(BaseGroupChatManagerState):
    """用于 :class:`~autogen_agentchat.teams.SelectorGroupChat` 管理器的状态。"""

    previous_speaker: Optional[str] = Field(default=None)
    type: str = Field(default="SelectorManagerState")




[文档]
class SwarmManagerState(BaseGroupChatManagerState):
    """用于 :class:`~autogen_agentchat.teams.Swarm` 管理器的状态。"""

    current_speaker: str = Field(default="")
    type: str = Field(default="SwarmManagerState")




[文档]
class MagenticOneOrchestratorState(BaseGroupChatManagerState):
    """用于 :class:`~autogen_agentchat.teams.MagneticOneGroupChat` 编排器的状态。"""

    task: str = Field(default="")
    facts: str = Field(default="")
    plan: str = Field(default="")
    n_rounds: int = Field(default=0)
    n_stalls: int = Field(default=0)
    type: str = Field(default="MagenticOneOrchestratorState")




[文档]
class SocietyOfMindAgentState(BaseState):
    """思维社会代理的状态。"""

    inner_team_state: Mapping[str, Any] = Field(default_factory=dict)
    type: str = Field(default="SocietyOfMindAgentState")