Corrected missing variable from nested chat PR

Signed-off-by: Mark Sze <[email protected]>
ag2ai · Dec 6, 2024 · b9352da · b9352da
1 parent 482a60e
commit b9352da
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/autogen/agentchat/contrib/swarm_agent.py b/autogen/agentchat/contrib/swarm_agent.py
@@ -376,9 +376,13 @@ def __init__(
 
         self.after_work = None
 
-        # use in the tool execution agent to transfer to the next agent
+        # Used in the tool execution agent to transfer to the next agent
         self._next_agent = None
 
+        # Store nested chats hand offs as we'll establish these in the initiate_swarm_chat
+        # List of Dictionaries containing the nested_chats and condition
+        self._nested_chat_handoffs = []
+
         self.register_update_agent_before_reply(update_agent_before_reply)
 
     def register_update_agent_before_reply(self, functions: Optional[Union[List[Callable], Callable]]):