RomanNum3ral
/
cai
mirror of https://github.com/aliasrobotics/cai.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'main' of https://github.com/aliasrobotics/cai into github

This commit is contained in:
Víctor Mayoral Vilches 2025-11-02 14:24:34 +00:00
parent fde4c4aef4 ba15f8d9d9
commit 6ce8125f12
1 changed files with 8 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
src/cai/cli.py
View File

@ -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")