From 81fc749ead932533660f56ce2822cd9e7cae5893 Mon Sep 17 00:00:00 2001
From: yousonnet <75065489+yousonnet@users.noreply.github.com>
Date: Tue, 13 Feb 2024 13:24:46 +0800
Subject: [PATCH] add GPTAssistantAgent is_termination_msg valid (#1642)

* add GPTAssistantAgent is_termination_msg valid

* make is_termination_msg in GPTAssistantAgent valid and reformatted

* to pass the black format test
---
 autogen/agentchat/contrib/gpt_assistant_agent.py | 2 ++
 1 file changed, 2 insertions(+)

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,