diff --git a/autogen/agentchat/contrib/gpt_assistant_agent.py b/autogen/agentchat/contrib/gpt_assistant_agent.py index b588b2b59f5..dc2967e103e 100644 --- a/autogen/agentchat/contrib/gpt_assistant_agent.py +++ b/autogen/agentchat/contrib/gpt_assistant_agent.py @@ -145,6 +145,8 @@ def __init__( self._openai_threads = {} self._unread_index = defaultdict(int) self.register_reply(Agent, GPTAssistantAgent._invoke_assistant) + self.register_reply(Agent, GPTAssistantAgent.check_termination_and_human_reply) + self.register_reply(Agent, GPTAssistantAgent.a_check_termination_and_human_reply) def _invoke_assistant( self,