Cleanup

autogen/agentchat/contrib/captainagent.py

@@ -126,7 +126,7 @@ class CaptainAgent(ConversableAgent):
     DEFAULT_DESCRIPTION = "A helpful AI assistant that can build a group of agents at a proper time to solve a task."
 
     # This is used to prompt the LLM to summarize the conversation history between CaptainAgent's tool execution history
-    DEFAULT_SUMMARY_PROMPT = "Read the following conversation history between an expert and a group of agent experts, summarize the conversation history. You should include the initial task, the experts' plan and the attempt, finally the results of the conversation."
+    DEFAULT_SUMMARY_PROMPT = "Read the following conversation history between an expert and a group of agent experts, summarize the conversation history. Your summarization should include the initial task, the experts' plan and the attempt, finally the results of the conversation. If the experts arrived at a conclusion, state it as it is without any modification."
 
     def __init__(
         self,
@@ -445,7 +445,9 @@ def _run_autobuild(self, group_name: str, execution_task: str, building_task: st
             **self._nested_config["group_chat_config"],
         )
         manager = autogen.GroupChatManager(
-            groupchat=nested_group_chat, llm_config=self._nested_config["group_chat_llm_config"]
+            groupchat=nested_group_chat,
+            llm_config=self._nested_config["group_chat_llm_config"],
+            is_termination_msg=lambda x: x.get("content", "") and "terminate" in x.get("content", "").lower(),
         )
         key = list(self.chat_messages.keys())[0]
         general_task = self.chat_messages[key][0]["content"]