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,6 @@ Generate **one summarized sentence** about "function calls' responses" with nece
             yield "Please provide a valid message or upload files to analyze."
             return "Invalid input."
 
-        # Remove tool call traces or debug prefixes
         if message.startswith("[🧰 Tool_RAG") or message.startswith("⚒️"):
             return ""
 
@@ -827,16 +826,21 @@ Generate **one summarized sentence** about "function calls' responses" with nece
                         temperature=temperature)
 
                     history.extend(current_gradio_history)
-                    if special_tool_call == 'Finish':
+
+                    if special_tool_call == 'Finish' and function_call_messages:
                         yield history
                         next_round = False
                         conversation.extend(function_call_messages)
                         return function_call_messages[0]['content']
                     elif special_tool_call in ['RequireClarification', 'DirectResponse']:
-                        history.append(ChatMessage(role="assistant", content=history[-1].content))
-                        yield history
-                        next_round = False
-                        return history[-1].content
+                        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."
 
                     if (self.enable_summary or token_overflow) and not call_agent:
                         enable_summary = True
@@ -845,7 +849,7 @@ Generate **one summarized sentence** about "function calls' responses" with nece
                         conversation, status=last_status,
                         enable_summary=enable_summary)
 
-                    if function_call_messages is not None:
+                    if function_call_messages:
                         conversation.extend(function_call_messages)
                         yield history
                     else:
@@ -877,10 +881,12 @@ Generate **one summarized sentence** about "function calls' responses" with nece
                         msg.metadata['status'] = 'done'
 
                 if '[FinalAnswer]' in last_thought:
-                    final_thought, final_answer = last_thought.split('[FinalAnswer]', 1)
-                    history.append(ChatMessage(role="assistant", content=final_thought.strip()))
+                    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))
                     yield history
-                    history.append(ChatMessage(role="assistant", content="**🧠 Final Analysis:**\n" + final_answer.strip()))
+                    history.append(ChatMessage(role="assistant", content="**🧠 Final Analysis:**\n" + final_answer))
                     yield history
                 else:
                     history.append(ChatMessage(role="assistant", content=last_thought))
@@ -893,10 +899,12 @@ Generate **one summarized sentence** about "function calls' responses" with nece
                     last_outputs_str = self.get_answer_based_on_unfinished_reasoning(
                         conversation, temperature, max_new_tokens, max_token)
                     if '[FinalAnswer]' in last_outputs_str:
-                        final_thought, final_answer = last_outputs_str.split('[FinalAnswer]', 1)
-                        history.append(ChatMessage(role="assistant", content=final_thought.strip()))
+                        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))
                         yield history
-                        history.append(ChatMessage(role="assistant", content="**🧠 Final Analysis:**\n" + final_answer.strip()))
+                        history.append(ChatMessage(role="assistant", content="**🧠 Final Analysis:**\n" + final_answer))
                         yield history
                     else:
                         history.append(ChatMessage(role="assistant", content=last_outputs_str.strip()))
@@ -910,10 +918,12 @@ Generate **one summarized sentence** about "function calls' responses" with nece
                 last_outputs_str = self.get_answer_based_on_unfinished_reasoning(
                     conversation, temperature, max_new_tokens, max_token)
                 if '[FinalAnswer]' in last_outputs_str:
-                    final_thought, final_answer = last_outputs_str.split('[FinalAnswer]', 1)
-                    history.append(ChatMessage(role="assistant", content=final_thought.strip()))
+                    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))
                     yield history
-                    history.append(ChatMessage(role="assistant", content="**🧠 Final Analysis:**\n" + final_answer.strip()))
+                    history.append(ChatMessage(role="assistant", content="**🧠 Final Analysis:**\n" + final_answer))
                     yield history
             else:
                 yield f"An error occurred: {e}"