ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

[pre-commit.ci] auto fixes from pre-commit.com hooks

Browse source 
for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2026-05-24 14:08:56 +00:00
commit da82a68174
2 changed files with 6 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

4
studio/backend/core/inference/pricing.py
View file

@ -156,9 +156,7 @@ def _lookup(provider: str, model: str) -> Optional[dict[str, float]]:
# date / variant suffixes, never on a longer numeric or alpha
# continuation of the same component.
for key in sorted(table, key = len, reverse = True):
if model.startswith(key) and (
len(model) == len(key) or model[len(key)] == "-"
):
if model.startswith(key) and (len(model) == len(key) or model[len(key)] == "-"):
return table[key]
return None

9
studio/backend/tests/test_pricing_edge.py
View file

@ -85,9 +85,9 @@ def test_prefix_match_requires_dash_boundary_pro_lookalike():
# lands on the canonical ``gpt-5.5`` row instead.
prices = _lookup("openai", "gpt-5.5-prod")
assert prices is not None
assert prices["input_per_mtok"] == OPENAI_PRICING["gpt-5.5"]["input_per_mtok"], (
"expected fallback to gpt-5.5 base ($5), not gpt-5.5-pro ($30)"
)
assert (
prices["input_per_mtok"] == OPENAI_PRICING["gpt-5.5"]["input_per_mtok"]
), "expected fallback to gpt-5.5 base ($5), not gpt-5.5-pro ($30)"
out = calculate_cost(
"openai",
"gpt-5.5-prod",
@ -324,7 +324,8 @@ def test_cache_creation_as_int_does_not_crash():
)
# Falls back to the 5m default for the whole cache_creation bucket.
assert _isclose(
out["cache_write_usd"], 1_000_000 / 1_000_000.0 * base * ANTHROPIC_CACHE_5M_WRITE_MULT
out["cache_write_usd"],
1_000_000 / 1_000_000.0 * base * ANTHROPIC_CACHE_5M_WRITE_MULT,
)