Refactor getting agent names

ag2ai · Jan 3, 2025 · 5bcf0ae · 5bcf0ae
1 parent 207114a
commit 5bcf0ae
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/autogen/messages/agent_messages.py b/autogen/messages/agent_messages.py
@@ -260,8 +260,8 @@ def __init__(self, *, uuid: Optional[UUID] = None, chat_info: dict[str, Any]):
         message = chat_info.get("message")
         verbose = chat_info.get("verbose", False)
 
-        sender_name = chat_info.get("sender").name  # type: ignore [union-attr]
-        recipient_name = chat_info.get("recipient").name  # type: ignore [union-attr]
+        sender_name = chat_info["sender"].name
+        recipient_name = chat_info["recipient"].name
         summary_args = chat_info.get("summary_args", None)
         max_turns = chat_info.get("max_turns", None)