zach
/
RA.Aid
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Better handle 429 errors on openrouter.

This commit is contained in:
AI Christianson 2025-01-09 11:42:13 -05:00
parent d115b8d5fe
commit 80d2d2fa60
2 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
CHANGELOG.md
View File

@ -7,6 +7,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
## [Unreleased] ## [Unreleased]
- Improve prompts for better open model support. - Improve prompts for better open model support.
- Better handle 429 errors on openrouter.
## [0.12.1] - 2025-01-08 ## [0.12.1] - 2025-01-08
- Fix bug where directories are added as related files. - Fix bug where directories are added as related files.

4
ra_aid/agent_utils.py
View File

@ -552,7 +552,9 @@ def run_agent_with_retry(agent, prompt: str, config: dict) -> Optional[str]:
return "Agent run completed successfully" return "Agent run completed successfully"
except (KeyboardInterrupt, AgentInterrupt): except (KeyboardInterrupt, AgentInterrupt):
raise raise
except (InternalServerError, APITimeoutError, RateLimitError, APIError) as e: except (InternalServerError, APITimeoutError, RateLimitError, APIError, ValueError) as e:
if isinstance(e, ValueError) and 'code 429' not in str(e):
raise # Re-raise ValueError if it's not a Lambda 429
if attempt == max_retries - 1: if attempt == max_retries - 1:
logger.error("Max retries reached, failing: %s", str(e)) logger.error("Max retries reached, failing: %s", str(e))
raise RuntimeError(f"Max retries ({max_retries}) exceeded. Last error: {e}") raise RuntimeError(f"Max retries ({max_retries}) exceeded. Last error: {e}")