Update src/txagent/txagent.py

src/txagent/txagent.py

@@ -12,6 +12,10 @@ from tooluniverse import ToolUniverse
 from gradio import ChatMessage
 from .toolrag import ToolRAGModel
 import torch
+# near the top of txagent.py
+import logging
+logger = logging.getLogger(__name__)
+logging.basicConfig(level=logging.INFO)
 
 from .utils import NoRepeatSentenceProcessor, ReasoningTraceChecker, tool_result_format
 
@@ -922,8 +926,11 @@ 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 or '"name": "Finish",' in last_outputs_str:
-                    final_thought, final_answer = last_thought.split(
-                        '[FinalAnswer]')
+                    if '[FinalAnswer]' in last_thought:
+                        final_thought, final_answer = last_thought.split('[FinalAnswer]', 1)
+                    else:
+                        final_thought = ""
+                        final_answer = last_thought
                     history.append(
                         ChatMessage(role="assistant",
                                     content=final_thought.strip())
@@ -931,8 +938,8 @@ Generate **one summarized sentence** about "function calls' responses" with nece
                     yield history
                     history.append(
                         ChatMessage(
-                            role="assistant", content="**Answer**:\n"+final_answer.strip())
+                            role="assistant", content="**Answer**:\n" + final_answer.strip())
                     )
                     yield history
             else:
-                return None
+                return None