Update src/txagent/txagent.py

src/txagent/txagent.py

@@ -826,12 +826,11 @@ Generate **one summarized sentence** about "function calls' responses" with nece
                     history.extend(current_gradio_history)
                     if special_tool_call == 'Finish':
                         next_round = False
-                        conversation.extend(function_call_messages)
-                        if function_call_messages and len(function_call_messages) > 0:
+                        if function_call_messages:
+                            conversation.extend(function_call_messages)
                             return function_call_messages[0]['content']
                         else:
-                            print("[⚠️ Warning] function_call_messages is empty")
-                            return "An internal reasoning error occurred — no tool response was returned."
+                            return "No final response was generated."
                     elif special_tool_call == 'RequireClarification' or special_tool_call == 'DirectResponse':
                         history.append(
                             ChatMessage(role="assistant", content=history[-1].content))
@@ -874,9 +873,10 @@ Generate **one summarized sentence** about "function calls' responses" with nece
                     if each.metadata is not None:
                         each.metadata['status'] = 'done'
                 if '[FinalAnswer]' in last_thought:
-                    _, final_answer = last_thought.split('[FinalAnswer]', 1)
+                    parts = last_thought.split('[FinalAnswer]', 1)
+                    final_answer = parts[1].strip() if len(parts) > 1 else "No detailed final answer found."
                     history.append(
-                        ChatMessage(role="assistant", content=final_answer.strip())
+                        ChatMessage(role="assistant", content=final_answer)
                     )
                     yield history
                     return
@@ -894,10 +894,11 @@ Generate **one summarized sentence** about "function calls' responses" with nece
                     for each in history:
                         if each.metadata is not None:
                             each.metadata['status'] = 'done'
-                    if '[FinalAnswer]' in last_thought:
-                        _, final_answer = last_thought.split('[FinalAnswer]', 1)
+                    if '[FinalAnswer]' in last_outputs_str:
+                        parts = last_outputs_str.split('[FinalAnswer]', 1)
+                        final_answer = parts[1].strip() if len(parts) > 1 else "No final answer found."
                         history.append(
-                            ChatMessage(role="assistant", content=final_answer.strip())
+                            ChatMessage(role="assistant", content=final_answer)
                         )
                         yield history
                 else:
@@ -914,14 +915,15 @@ Generate **one summarized sentence** about "function calls' responses" with nece
                 for each in history:
                     if each.metadata is not None:
                         each.metadata['status'] = 'done'
-                if '[FinalAnswer]' in last_thought or '"name": "Finish",' in last_outputs_str:
-                    if '[FinalAnswer]' in last_thought:
-                        _, final_answer = last_thought.split('[FinalAnswer]', 1)
+                if '[FinalAnswer]' in last_outputs_str or '"name": "Finish",' in last_outputs_str:
+                    if '[FinalAnswer]' in last_outputs_str:
+                        parts = last_outputs_str.split('[FinalAnswer]', 1)
+                        final_answer = parts[1].strip() if len(parts) > 1 else "No final answer found."
                     else:
-                        final_answer = last_thought
+                        final_answer = last_outputs_str
                     history.append(
-                        ChatMessage(role="assistant", content=final_answer.strip())
+                        ChatMessage(role="assistant", content=final_answer)
                     )
                     yield history
             else:
-                return None
+                yield [{"role": "assistant", "content": f"⚠️ An error occurred: {e}"}]