add check for fallback handler
This commit is contained in:
parent
f08e9455b6
commit
c511cefc67
|
|
@ -825,6 +825,7 @@ class CiaynAgent:
|
||||||
try:
|
try:
|
||||||
last_result = self._execute_tool(response)
|
last_result = self._execute_tool(response)
|
||||||
self.chat_history.append(response)
|
self.chat_history.append(response)
|
||||||
|
if hasattr(self.fallback_handler, 'reset_fallback_handler'):
|
||||||
self.fallback_handler.reset_fallback_handler()
|
self.fallback_handler.reset_fallback_handler()
|
||||||
yield {}
|
yield {}
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -464,7 +464,7 @@ def run_agent_with_retry(
|
||||||
|
|
||||||
try:
|
try:
|
||||||
_run_agent_stream(agent, msg_list)
|
_run_agent_stream(agent, msg_list)
|
||||||
if fallback_handler:
|
if fallback_handler and hasattr(fallback_handler, 'reset_fallback_handler'):
|
||||||
fallback_handler.reset_fallback_handler()
|
fallback_handler.reset_fallback_handler()
|
||||||
should_break, prompt, auto_test, test_attempts = (
|
should_break, prompt, auto_test, test_attempts = (
|
||||||
_execute_test_command_wrapper(
|
_execute_test_command_wrapper(
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue