Update src/txagent/txagent.py

src/txagent/txagent.py

@@ -755,17 +755,17 @@ Generate **one summarized sentence** about "function calls' responses" with nece
         return updated_attributes
 
     def run_gradio_chat(self, message: str,
-                            history: list,
-                            temperature: float,
-                            max_new_tokens: int,
-                            max_token: int,
-                            call_agent: bool,
-                            conversation: gr.State,
-                            max_round: int = 20,
-                            seed: int = None,
-                            call_agent_level: int = 0,
-                            sub_agent_task: str = None,
-                            uploaded_files: list = None) -> str:
+                        history: list,
+                        temperature: float,
+                        max_new_tokens: int,
+                        max_token: int,
+                        call_agent: bool,
+                        conversation: gr.State,
+                        max_round: int = 20,
+                        seed: int = None,
+                        call_agent_level: int = 0,
+                        sub_agent_task: str = None,
+                        uploaded_files: list = None) -> str:
         """
         Generate a streaming response using the loaded model.
         Args:
@@ -783,7 +783,7 @@ Generate **one summarized sentence** about "function calls' responses" with nece
             yield "Please provide a valid message or upload files to analyze."
             return "Invalid input."
 
-        if message.startswith("[🧰 Tool_RAG") or message.startswith("⚒️"):
+        if message.startswith("[\U0001f9f0 Tool_RAG") or message.startswith("⚒️"):
             return ""
 
         outputs = []
@@ -832,15 +832,13 @@ Generate **one summarized sentence** about "function calls' responses" with nece
                         next_round = False
                         conversation.extend(function_call_messages)
                         return function_call_messages[0]['content']
+
                     elif special_tool_call in ['RequireClarification', 'DirectResponse']:
-                        if history:
-                            history.append(ChatMessage(role="assistant", content=history[-1].content))
-                            yield history
-                            next_round = False
-                            return history[-1].content
-                        else:
-                            yield "I need more information to proceed."
-                            return "Missing data."
+                        last_msg = history[-1] if history else ChatMessage(role="assistant", content="Response needed.")
+                        history.append(ChatMessage(role="assistant", content=last_msg.content))
+                        yield history
+                        next_round = False
+                        return last_msg.content
 
                     if (self.enable_summary or token_overflow) and not call_agent:
                         enable_summary = True
@@ -882,11 +880,13 @@ Generate **one summarized sentence** about "function calls' responses" with nece
 
                 if '[FinalAnswer]' in last_thought:
                     parts = last_thought.split('[FinalAnswer]', 1)
-                    final_thought = parts[0].strip()
-                    final_answer = parts[1].strip() if len(parts) > 1 else ""
-                    history.append(ChatMessage(role="assistant", content=final_thought))
+                    if len(parts) == 2:
+                        final_thought, final_answer = parts
+                    else:
+                        final_thought, final_answer = last_thought, ""
+                    history.append(ChatMessage(role="assistant", content=final_thought.strip()))
                     yield history
-                    history.append(ChatMessage(role="assistant", content="**🧠 Final Analysis:**\n" + final_answer))
+                    history.append(ChatMessage(role="assistant", content="**🧠 Final Analysis:**\n" + final_answer.strip()))
                     yield history
                 else:
                     history.append(ChatMessage(role="assistant", content=last_thought))
@@ -900,11 +900,13 @@ Generate **one summarized sentence** about "function calls' responses" with nece
                         conversation, temperature, max_new_tokens, max_token)
                     if '[FinalAnswer]' in last_outputs_str:
                         parts = last_outputs_str.split('[FinalAnswer]', 1)
-                        final_thought = parts[0].strip()
-                        final_answer = parts[1].strip() if len(parts) > 1 else ""
-                        history.append(ChatMessage(role="assistant", content=final_thought))
+                        if len(parts) == 2:
+                            final_thought, final_answer = parts
+                        else:
+                            final_thought, final_answer = last_outputs_str, ""
+                        history.append(ChatMessage(role="assistant", content=final_thought.strip()))
                         yield history
-                        history.append(ChatMessage(role="assistant", content="**🧠 Final Analysis:**\n" + final_answer))
+                        history.append(ChatMessage(role="assistant", content="**🧠 Final Analysis:**\n" + final_answer.strip()))
                         yield history
                     else:
                         history.append(ChatMessage(role="assistant", content=last_outputs_str.strip()))
@@ -919,11 +921,13 @@ Generate **one summarized sentence** about "function calls' responses" with nece
                     conversation, temperature, max_new_tokens, max_token)
                 if '[FinalAnswer]' in last_outputs_str:
                     parts = last_outputs_str.split('[FinalAnswer]', 1)
-                    final_thought = parts[0].strip()
-                    final_answer = parts[1].strip() if len(parts) > 1 else ""
-                    history.append(ChatMessage(role="assistant", content=final_thought))
+                    if len(parts) == 2:
+                        final_thought, final_answer = parts
+                    else:
+                        final_thought, final_answer = last_outputs_str, ""
+                    history.append(ChatMessage(role="assistant", content=final_thought.strip()))
                     yield history
-                    history.append(ChatMessage(role="assistant", content="**🧠 Final Analysis:**\n" + final_answer))
+                    history.append(ChatMessage(role="assistant", content="**🧠 Final Analysis:**\n" + final_answer.strip()))
                     yield history
             else:
-                yield f"An error occurred: {e}"
+                yield f"An error occurred: {e}"