Source code for autogen_agentchat.messages

from typing import List

from autogen_core.components import Image
from pydantic import BaseModel




class BaseMessage(BaseModel):
    """A base message."""

    source: str
    """The name of the agent that sent this message."""





class TextMessage(BaseMessage):
    """A text message."""

    content: str
    """The content of the message."""





class MultiModalMessage(BaseMessage):
    """A multimodal message."""

    content: List[str | Image]
    """The content of the message."""





class StopMessage(BaseMessage):
    """A message requesting stop of a conversation."""

    content: str
    """The content for the stop message."""





class HandoffMessage(BaseMessage):
    """A message requesting handoff of a conversation to another agent."""

    content: str
    """The agent name to handoff the conversation to."""



ChatMessage = TextMessage | MultiModalMessage | StopMessage | HandoffMessage
"""A message used by agents in a team."""


__all__ = [
    "BaseMessage",
    "TextMessage",
    "MultiModalMessage",
    "StopMessage",
    "HandoffMessage",
    "ChatMessage",
]