add GPTAssistantAgent is_termination_msg valid (microsoft#1642)

* add GPTAssistantAgent is_termination_msg valid * make is_termination_msg in GPTAssistantAgent valid and reformatted * to pass the black format test
gunnarku · Feb 13, 2024 · 81fc749 · 81fc749
1 parent fb22f78
commit 81fc749
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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,