From ba15f8d9d92d99376ce3aad6bbb696757fad909e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?V=C3=ADctor=20Mayoral=20Vilches?= Date: Sun, 2 Nov 2025 14:43:12 +0100 Subject: [PATCH] Revert "Add retry logic for OpenAI timeout exceptions" (#326) Revert "Add retry logic for OpenAI timeout exceptions (#322)" This reverts commit 7a110a56fdebc2b38d627a9fd4cd96499de82c71. --- src/cai/cli.py | 31 ++++++++----------------------- 1 file changed, 8 insertions(+), 23 deletions(-) diff --git a/src/cai/cli.py b/src/cai/cli.py index d5cab9ff..86960a61 100644 --- a/src/cai/cli.py +++ b/src/cai/cli.py @@ -1625,29 +1625,14 @@ def run_cai_cli( raise else: # Use non-streamed response - from litellm.exceptions import Timeout - - max_retries = 3 - last_input = conversation_input - - for attempt in range(max_retries): - try: - response = asyncio.run(Runner.run(agent, last_input)) - break # Success, exit retry loop - except Timeout as e: - if attempt < max_retries - 1: - print(f"⏱️ Timeout on attempt {attempt + 1}/{max_retries}, retrying with 'continue'...") - last_input = "continue" - else: - print("❌ Max retries reached") - raise - except InputGuardrailTripwireTriggered as e: - # Display a user-friendly warning for input guardrails - reason = "Potential security threat detected in input" - if hasattr(e, 'guardrail_result') and e.guardrail_result: - if hasattr(e.guardrail_result, 'output') and e.guardrail_result.output: - reason = e.guardrail_result.output.output_info.get("reason", reason) - raise # Re-raise InputGuardrailTripwireTriggered + try: + response = asyncio.run(Runner.run(agent, conversation_input)) + except InputGuardrailTripwireTriggered as e: + # Display a user-friendly warning for input guardrails + reason = "Potential security threat detected in input" + if hasattr(e, 'guardrail_result') and e.guardrail_result: + if hasattr(e.guardrail_result, 'output') and e.guardrail_result.output: + reason = e.guardrail_result.output.output_info.get("reason", reason) # Use red color for the warning message print(f"\n\033[91m🛡️ INPUT SECURITY GUARDRAIL TRIGGERED\033[0m")