Update src/txagent/txagent.py
Browse files- 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
|
245 |
-
|
|
|
|
|
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:
|