mirror of
https://github.com/NousResearch/hermes-agent.git
synced 2026-05-11 03:31:55 +00:00
Prefer fallback for Gemini CloudCode rate limits
This commit is contained in:
parent
12135b4c8a
commit
526742199b
2 changed files with 95 additions and 5 deletions
62
tests/agent/test_gemini_fast_fallback.py
Normal file
62
tests/agent/test_gemini_fast_fallback.py
Normal file
|
|
@ -0,0 +1,62 @@
|
|||
"""Regression tests for #13636 — CloudCode / Gemini CLI rate-limit fallback.
|
||||
|
||||
_pool_may_recover_from_rate_limit() is the hinge between credential-pool
|
||||
rotation and fallback-provider activation. For CloudCode (Gemini CLI /
|
||||
Gemini OAuth) the 429 is an account-wide throttle, so waiting for pool
|
||||
rotation is pointless — prefer fallback immediately.
|
||||
"""
|
||||
from unittest.mock import MagicMock
|
||||
|
||||
from run_agent import _pool_may_recover_from_rate_limit
|
||||
|
||||
|
||||
def _pool(entries: int = 2):
|
||||
p = MagicMock()
|
||||
p.has_available.return_value = True
|
||||
p.entries.return_value = list(range(entries))
|
||||
return p
|
||||
|
||||
|
||||
def test_cloudcode_provider_skips_pool_rotation():
|
||||
assert _pool_may_recover_from_rate_limit(
|
||||
_pool(entries=3),
|
||||
provider="google-gemini-cli",
|
||||
base_url="cloudcode-pa://google",
|
||||
) is False
|
||||
|
||||
|
||||
def test_cloudcode_base_url_skips_pool_rotation_even_on_alias_provider():
|
||||
# Even if the provider label is something else, a cloudcode-pa:// URL
|
||||
# signals the account-wide quota regime.
|
||||
assert _pool_may_recover_from_rate_limit(
|
||||
_pool(entries=3),
|
||||
provider="custom-provider",
|
||||
base_url="cloudcode-pa://google",
|
||||
) is False
|
||||
|
||||
|
||||
def test_non_cloudcode_multi_entry_pool_still_recovers():
|
||||
assert _pool_may_recover_from_rate_limit(
|
||||
_pool(entries=3),
|
||||
provider="openrouter",
|
||||
base_url="https://openrouter.ai/api/v1",
|
||||
) is True
|
||||
|
||||
|
||||
def test_single_entry_pool_skips_rotation_regardless_of_provider():
|
||||
# Pre-existing single-entry-pool exception (#11314) still holds.
|
||||
assert _pool_may_recover_from_rate_limit(
|
||||
_pool(entries=1),
|
||||
provider="openrouter",
|
||||
base_url="https://openrouter.ai/api/v1",
|
||||
) is False
|
||||
|
||||
|
||||
def test_exhausted_pool_skips_rotation():
|
||||
p = MagicMock()
|
||||
p.has_available.return_value = False
|
||||
assert _pool_may_recover_from_rate_limit(p) is False
|
||||
|
||||
|
||||
def test_no_pool_skips_rotation():
|
||||
assert _pool_may_recover_from_rate_limit(None) is False
|
||||
Loading…
Add table
Add a link
Reference in a new issue