红尘
d73ff9b0fb
Deploy Site / deploy-vercel (push) Has been cancelled
Deploy Site / deploy-docs (push) Has been cancelled
Docker / shell lint / Lint Dockerfile (hadolint) (push) Has been cancelled
Docker / shell lint / Lint docker/ shell scripts (shellcheck) (push) Has been cancelled
Docker Build and Publish / build-amd64 (push) Has been cancelled
Docker Build and Publish / build-arm64 (push) Has been cancelled
Lint (ruff + ty) / ruff + ty diff (push) Has been cancelled
Lint (ruff + ty) / ruff enforcement (blocking) (push) Has been cancelled
Lint (ruff + ty) / Windows footguns (blocking) (push) Has been cancelled
Nix Lockfile Fix / auto-fix-main (push) Has been cancelled
Nix Lockfile Fix / fix (push) Has been cancelled
Nix / nix (macos-latest) (push) Has been cancelled
Nix / nix (ubuntu-latest) (push) Has been cancelled
OSV-Scanner / Scan lockfiles (push) Has been cancelled
Build Skills Index / build-index (push) Has been cancelled
Tests / test (1) (push) Has been cancelled
Tests / test (2) (push) Has been cancelled
Tests / test (3) (push) Has been cancelled
Tests / test (4) (push) Has been cancelled
Tests / test (5) (push) Has been cancelled
Tests / test (6) (push) Has been cancelled
Tests / e2e (push) Has been cancelled
uv.lock check / uv lock --check (push) Has been cancelled
Docker Build and Publish / merge (push) Has been cancelled
Build Skills Index / trigger-deploy (push) Has been cancelled
Tests / save-durations (push) Has been cancelled
79 lines
2.6 KiB
Python
79 lines
2.6 KiB
Python
"""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.
|
|
"""
|
|
import inspect
|
|
from unittest.mock import MagicMock
|
|
|
|
from agent import conversation_loop
|
|
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
|
|
|
|
|
|
def test_conversation_loop_resolves_pool_helper_through_run_agent_module():
|
|
"""Extracted conversation loop must honor tests/patches on run_agent.
|
|
|
|
conversation_loop intentionally lazy-loads run_agent via _ra(). If this
|
|
call site uses a bare imported helper, monkeypatching run_agent in tests (and
|
|
production wrappers that patch run_agent) will not propagate into the
|
|
extracted loop; older code also hit NameError in this branch.
|
|
"""
|
|
source = inspect.getsource(conversation_loop.run_conversation)
|
|
|
|
assert "_ra()._pool_may_recover_from_rate_limit(" in source
|
|
assert "pool_may_recover = _pool_may_recover_from_rate_limit(" not in source
|