Update src/txagent/txagent.py

src/txagent/txagent.py

@@ -640,6 +640,7 @@ Summarize the function calls' responses in one sentence with all necessary infor
         try:
             while next_round and current_round < max_round:
                 current_round += 1
+                logger.debug("Starting round %d/%d", current_round, max_round)
                 if last_outputs:
                     function_call_messages, picked_tools_prompt, special_tool_call, current_gradio_history = yield from self.run_function_call_stream(
                         last_outputs, return_message=True,
@@ -651,6 +652,7 @@ Summarize the function calls' responses in one sentence with all necessary infor
                     history.extend(current_gradio_history)
 
                     if special_tool_call == 'Finish':
+                        logger.info("Finish tool called, ending chat")
                         yield history
                         next_round = False
                         conversation.extend(function_call_messages)
@@ -659,6 +661,7 @@ Summarize the function calls' responses in one sentence with all necessary infor
                     elif special_tool_call in ['RequireClarification', 'DirectResponse']:
                         last_msg = history[-1] if history else ChatMessage(role="assistant", content="Response needed.")
                         history.append(ChatMessage(role="assistant", content=last_msg.content))
+                        logger.info("Special tool %s called, ending chat", special_tool_call)
                         yield history
                         next_round = False
                         return last_msg.content
@@ -674,6 +677,7 @@ Summarize the function calls' responses in one sentence with all necessary infor
                     else:
                         next_round = False
                         conversation.append({"role": "assistant", "content": ''.join(last_outputs)})
+                        logger.info("No function call messages, ending chat")
                         return ''.join(last_outputs).replace("</s>", "")
 
                 last_outputs = []
@@ -711,6 +715,7 @@ Summarize the function calls' responses in one sentence with all necessary infor
                     history.append(ChatMessage(role="assistant", content=final_thought.strip()))
                     yield history
                     history.append(ChatMessage(role="assistant", content="**🧠 Final Analysis:**\n" + final_answer.strip()))
+                    logger.info("Final answer provided: %s", final_answer[:100])
                     yield history
                 else:
                     history.append(ChatMessage(role="assistant", content=last_thought))
@@ -727,6 +732,7 @@ Summarize the function calls' responses in one sentence with all necessary infor
                     history.append(ChatMessage(role="assistant", content=final_thought.strip()))
                     yield history
                     history.append(ChatMessage(role="assistant", content="**🧠 Final Analysis:**\n" + final_answer.strip()))
+                    logger.info("Forced final answer: %s", final_answer[:100])
                     yield history
                 else:
                     yield "Reasoning rounds exceeded limit."
@@ -744,5 +750,6 @@ Summarize the function calls' responses in one sentence with all necessary infor
                 history.append(ChatMessage(role="assistant", content=final_thought.strip()))
                 yield history
                 history.append(ChatMessage(role="assistant", content="**🧠 Final Analysis:**\n" + final_answer.strip()))
+                logger.info("Forced final answer after error: %s", final_answer[:100])
                 yield history
             return error_msg