scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-06 02:48:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

fix(ci): cap each sync PR body section so a large first sync stays under GitHub's body limit

Browse source
This commit is contained in:
mateo-berri 2026-09-04 19:17:58 -07:00
parent 5cc9690cb5
commit f41b19a120
2 changed files with 23 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
scripts/sync_cost_map.py
View file

@ -36,6 +36,7 @@ OPENROUTER_MODELS_URL: Final = "https://openrouter.ai/api/v1/models"
VERCEL_MODELS_URL: Final = "https://ai-gateway.vercel.sh/v1/models"
VERCEL_TYPE_TO_MODE: Final = MappingProxyType({"language": "chat", "embedding": "embedding"})
ADD_ONLY_FIELDS: Final = frozenset({"max_output_tokens", "max_tokens"})
PR_BODY_SECTION_LIMIT: Final = 30
Provider = Literal["openrouter", "vercel_ai_gateway"]
RegistryEntry = dict[str, object]
@ -385,8 +386,11 @@ def _ordered_result(cost_map: CostMap, result: CostMap, outcomes: Sequence[Provi
def _section_block(title: str, lines: Sequence[str], backtick: bool) -> str:
bullets: Final = "\n".join(f"- `{line}`" if backtick else f"- {line}" for line in lines) or "- none"
return f"### {title} ({len(lines)})\n{bullets}\n"
shown: Final = lines[:PR_BODY_SECTION_LIMIT]
bullets: Final = "\n".join(f"- `{line}`" if backtick else f"- {line}" for line in shown) or "- none"
overflow: Final = len(lines) - len(shown)
trailer: Final = f"\n- and {overflow} more, see the diff" if overflow else ""
return f"### {title} ({len(lines)})\n{bullets}{trailer}\n"
def _provider_body(outcome: ProviderOutcome) -> str:

17
tests/test_litellm/test_sync_cost_map.py
View file

@ -264,6 +264,23 @@ def test_pr_body_lists_changes_per_provider(sync: ModuleType) -> None:
assert "Catalog rows skipped: deprecated (1), no usable price (1), not token priced (1)" in body
def test_pr_body_caps_every_section_so_a_large_first_sync_fits_github_limit(sync: ModuleType) -> None:
lines: Final = tuple(f"provider/model-{index}: {'x' * 200}" for index in range(400))
outcome: Final = sync.SyncOutcome(
cost_map={},
providers=tuple(
sync.ProviderOutcome(provider=provider, added=lines, updated=lines, warnings=lines, skipped={})
for provider in ("openrouter", "vercel_ai_gateway")
),
)
body: Final = sync.render_pr_body(outcome)
assert len(body) < 65_536
assert body.count("### Added (400)") == 2 and body.count("- and 370 more, see the diff") == 6
assert body.count("- `provider/model-29: ") == 4 and "provider/model-30: " not in body
@pytest.mark.parametrize(
("loader", "raw"),
[