Update src/txagent/txagent.py

src/txagent/txagent.py

@@ -428,7 +428,10 @@ class TxAgent:
                         if isinstance(function_call_messages[0]['content'], types.GeneratorType):
                             function_call_messages[0]['content'] = next(
                                 function_call_messages[0]['content'])
-                        return function_call_messages[0]['content'].split('[FinalAnswer]')[-1]
+                        content = function_call_messages[0]['content']
+                        if content is None:
+                            return "❌ No content returned after Finish tool call."
+                        return content.split('[FinalAnswer]')[-1]
 
                     if (self.enable_summary or token_overflow) and not call_agent:
                         if token_overflow:
@@ -462,9 +465,11 @@ class TxAgent:
                                                                   max_new_tokens=max_new_tokens, max_token=max_token,
                                                                   check_token_status=True)
                 if last_outputs_str is None:
-                    next_round = False
-                    print(
-                        "The number of tokens exceeds the maximum limit.")
+                    print("The number of tokens exceeds the maximum limit.")
+                    if self.force_finish:
+                        return self.get_answer_based_on_unfinished_reasoning(conversation, temperature, max_new_tokens, max_token)
+                    else:
+                        return "❌ Token limit exceeded — no further steps possible."
                 else:
                     last_outputs.append(last_outputs_str)
             if max_round == current_round: