Skip to content

Commit f00f7d2

Browse files
jorge-wonoloekzhu
andauthored
Avoid re-registering a message type already registered (#6354)
This change avoid re-registering a structured message already registered to the team by a previous agent also included in the team. This issue occurs when agents share Pydantic models as output format ## Related issue number Closes #6353 --------- Co-authored-by: Eric Zhu <ekzhu@users.noreply.github.com>
1 parent 0015315 commit f00f7d2

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

python/packages/autogen-agentchat/src/autogen_agentchat/teams/_group_chat/_base_group_chat.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,8 @@ def __init__(
7373
for agent in participants:
7474
for message_type in agent.produced_message_types:
7575
try:
76-
if issubclass(message_type, StructuredMessage):
76+
is_registered = self._message_factory.is_registered(message_type) # type: ignore[reportUnknownArgumentType]
77+
if issubclass(message_type, StructuredMessage) and not is_registered:
7778
self._message_factory.register(message_type) # type: ignore[reportUnknownArgumentType]
7879
except TypeError:
7980
# Not a class or not a valid subclassable type (skip)

0 commit comments

Comments
 (0)