i/Aether
1
0
Fork 0
mirror of https://github.com/fawney19/Aether.git synced 2026-09-02 01:10:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #195 from AAEE86/keys

Browse Source 
feat(pool): 新增账号配额展示并清理 Codex 旧限额字段
This commit is contained in:
fawney19
2026-03-01 00:59:49 +08:00
committed by GitHub
parent 20ed9cd123 d7a8a89aa7
commit fea3d183bf
11 changed files with 349 additions and 68 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tests/services/test_aware_scheduler_quota_skipping.py
View File

@@ -68,7 +68,6 @@ def test_codex_weekly_quota_exhausted_skips(_mock_cb: MagicMock) -> None:
"codex": {
"primary_used_percent": 100.0,
"secondary_used_percent": 10.0,
"code_review_used_percent": 100.0,
}
}
)
@@ -114,14 +113,14 @@ def test_codex_5h_quota_exhausted_skips(_mock_cb: MagicMock) -> None:
"src.services.scheduling.candidate_builder.health_monitor.get_circuit_breaker_status",
return_value=(True, None),
)
def test_codex_ignores_code_review_quota(_mock_cb: MagicMock) -> None:
def test_codex_ignores_unrelated_metadata_fields(_mock_cb: MagicMock) -> None:
scheduler = CacheAwareScheduler()
key = _make_key(
upstream_metadata={
"codex": {
"primary_used_percent": 10.0,
"secondary_used_percent": 20.0,
"code_review_used_percent": 100.0,
"legacy_marker": "ignore-me",
}
}
)

6
tests/services/test_provider_keys_codex_realtime_quota.py
View File

@@ -114,7 +114,7 @@ def test_sync_codex_quota_from_headers_updates_and_preserves_existing_fields() -
upstream_metadata={
"codex": {
"primary_used_percent": 50.0,
"code_review_used_percent": 12.0,
"legacy_marker": "keep-me",
}
},
)
@@ -131,8 +131,8 @@ def test_sync_codex_quota_from_headers_updates_and_preserves_existing_fields() -
codex_meta = key.upstream_metadata["codex"]
assert codex_meta["primary_used_percent"] == 64.0
assert codex_meta["secondary_used_percent"] == 3.0
# 旧的 code_review 字段应被保留（wham/usage 补充信息）
assert codex_meta["code_review_used_percent"] == 12.0
# 旧字段应被保留（解析器只覆盖已知配额字段）
assert codex_meta["legacy_marker"] == "keep-me"
def test_sync_codex_quota_from_headers_skips_when_only_reset_seconds_changed() -> None: