From 3b105b4261f46887e4c0b2d0c9478ae8cc3dc534 Mon Sep 17 00:00:00 2001 From: "zaina.abushaban" Date: Thu, 30 Oct 2025 16:07:54 +0300 Subject: [PATCH] propagate error --- aixplain/modules/agent/__init__.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/aixplain/modules/agent/__init__.py b/aixplain/modules/agent/__init__.py index e08c47b3..c647a2f7 100644 --- a/aixplain/modules/agent/__init__.py +++ b/aixplain/modules/agent/__init__.py @@ -541,6 +541,7 @@ def sync_poll( error_message="No response from the service.", ) logging.error(f"Polling for Agent: polling for {name}: {e}") + return response_body break # Display completion message @@ -640,8 +641,8 @@ def run( result = self.sync_poll( poll_url, name=name, timeout=timeout, wait_time=wait_time, progress_verbosity=progress_verbosity ) - # if result.status == ResponseStatus.FAILED: - # raise Exception("Model failed to run with error: " + result.error_message) + if result.status == ResponseStatus.FAILED: + raise Exception("Model failed to run with error: " + result.error_message) result_data = result.get("data") or {} return AgentResponse( status=ResponseStatus.SUCCESS,