Update src/txagent/txagent.py

src/txagent/txagent.py

@@ -317,7 +317,6 @@ class TxAgent:
                             )
                             sub_agent_task = "Sub TxAgent plan: " + \
                                 str(solution_plan)
-                            # When streaming, yield responses as they arrive.
                             call_result = yield from self.run_gradio_chat(
                                 full_message, history=[], temperature=temperature,
                                 max_new_tokens=1024, max_token=99999,
@@ -325,7 +324,7 @@ class TxAgent:
                                 conversation=None,
                                 sub_agent_task=sub_agent_task)
 
-                            if call_result is not None:
+                            if call_result is not None and isinstance(call_result, str):
                                 call_result = call_result.split('[FinalAnswer]')[-1]
                             else:
                                 call_result = "⚠️ No content returned from sub-agent."
@@ -345,7 +344,6 @@ class TxAgent:
                         if function_call_json[i]["name"] == 'Tool_RAG':
                             gradio_history.append(ChatMessage(role="assistant", content=str(call_result), metadata={
                                                   "title": "🧰 "+function_call_json[i]['name'], "log": str(function_call_json[i]['arguments'])}))
-
                         else:
                             gradio_history.append(ChatMessage(role="assistant", content=str(call_result), metadata={
                                                   "title": "⚒️ "+function_call_json[i]['name'], "log": str(function_call_json[i]['arguments'])}))
@@ -361,7 +359,6 @@ class TxAgent:
             "tool_calls": json.dumps(function_call_json)
         }] + call_results
 
-        # Yield the final result.
         if return_gradio_history:
             return revised_messages, existing_tools_prompt, special_tool_call, gradio_history
         else: