From c511cefc67066f181b89c4307b14dca9004ebfeb Mon Sep 17 00:00:00 2001 From: AI Christianson Date: Thu, 13 Mar 2025 08:48:51 -0400 Subject: [PATCH] add check for fallback handler --- ra_aid/agent_backends/ciayn_agent.py | 3 ++- ra_aid/agent_utils.py | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/ra_aid/agent_backends/ciayn_agent.py b/ra_aid/agent_backends/ciayn_agent.py index 067e80d..9286653 100644 --- a/ra_aid/agent_backends/ciayn_agent.py +++ b/ra_aid/agent_backends/ciayn_agent.py @@ -825,7 +825,8 @@ class CiaynAgent: try: last_result = self._execute_tool(response) self.chat_history.append(response) - self.fallback_handler.reset_fallback_handler() + if hasattr(self.fallback_handler, 'reset_fallback_handler'): + self.fallback_handler.reset_fallback_handler() yield {} except ToolExecutionError as e: diff --git a/ra_aid/agent_utils.py b/ra_aid/agent_utils.py index fa1ff62..0abb164 100644 --- a/ra_aid/agent_utils.py +++ b/ra_aid/agent_utils.py @@ -464,7 +464,7 @@ def run_agent_with_retry( try: _run_agent_stream(agent, msg_list) - if fallback_handler: + if fallback_handler and hasattr(fallback_handler, 'reset_fallback_handler'): fallback_handler.reset_fallback_handler() should_break, prompt, auto_test, test_attempts = ( _execute_test_command_wrapper(