Associate agent container and message types to group chat behavior protocol (#651)

* Agent container and message types is part of a group chat team behavior protocol.

* lint
This commit is contained in:
Eric Zhu 2024-09-26 09:04:42 -07:00 committed by GitHub
parent 90352f142d
commit 00289912da
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 6 additions and 5 deletions

View File

@ -3,13 +3,14 @@ from typing import List
from autogen_core.base import MessageContext
from autogen_core.components import DefaultTopicId, RoutedAgent, event
from ..agents import BaseChatAgent, ChatMessage
from ...agents import BaseChatAgent, ChatMessage
from ._messages import ContentPublishEvent, ContentRequestEvent
class BaseChatAgentContainer(RoutedAgent):
"""A core agent class that delegates message handling to an
:class:`autogen_agentchat.agents.BaseChatAgent` so that it can be used in a team.
:class:`autogen_agentchat.agents.BaseChatAgent` so that it can be used in a
group chat team.
Args:
parent_topic_type (str): The topic type of the parent orchestrator.

View File

@ -5,7 +5,7 @@ from autogen_core.base import MessageContext, TopicId
from autogen_core.components import RoutedAgent, event
from autogen_core.components.models import AssistantMessage, UserMessage
from .._messages import ContentPublishEvent, ContentRequestEvent
from ._messages import ContentPublishEvent, ContentRequestEvent
class BaseGroupChatManager(RoutedAgent):

View File

@ -8,9 +8,9 @@ from autogen_core.components import ClosureAgent, TypeSubscription
from autogen_core.components.models import UserMessage
from ...agents import BaseChatAgent
from .._base_chat_agent_container import BaseChatAgentContainer
from .._base_team import BaseTeam, TeamRunResult
from .._messages import ContentPublishEvent, ContentRequestEvent
from ._base_chat_agent_container import BaseChatAgentContainer
from ._messages import ContentPublishEvent, ContentRequestEvent
from ._round_robin_group_chat_manager import RoundRobinGroupChatManager