Ali2206 commited on
Commit
11fdfd1
·
verified ·
1 Parent(s): fe33e9e

Update src/txagent/txagent.py

Browse files
Files changed (1) hide show
  1. src/txagent/txagent.py +4 -2
src/txagent/txagent.py CHANGED
@@ -241,8 +241,10 @@ class TxAgent:
241
  full_message, temperature=temperature,
242
  max_new_tokens=1024, max_token=99999,
243
  call_agent=False, call_agent_level=call_agent_level)
244
- call_result = call_result.split(
245
- '[FinalAnswer]')[-1].strip()
 
 
246
  else:
247
  call_result = "Error: The CallAgent has been disabled. Please proceed with your reasoning process to solve this question."
248
  else:
 
241
  full_message, temperature=temperature,
242
  max_new_tokens=1024, max_token=99999,
243
  call_agent=False, call_agent_level=call_agent_level)
244
+ if call_result is None:
245
+ call_result = "⚠️ No content returned from sub-agent."
246
+ else:
247
+ call_result = call_result.split('[FinalAnswer]')[-1].strip()
248
  else:
249
  call_result = "Error: The CallAgent has been disabled. Please proceed with your reasoning process to solve this question."
250
  else: