18 lines
640 B
Python
18 lines
640 B
Python
"""Automated cost budget verification."""
|
|
|
|
from src.runtime.core.config import RuntimePricing
|
|
|
|
|
|
def test_per_article_cost_within_budget():
|
|
# 2000 input prompt tokens, 500 completion tokens on llama-3.1-8b-instant ($0.05 / $0.08 per 1M)
|
|
pricing = RuntimePricing(
|
|
primary_input_1k=0.00005,
|
|
primary_output_1k=0.00008,
|
|
fallback_input_1k=0.00014,
|
|
fallback_output_1k=0.00028,
|
|
)
|
|
cost = (2000 / 1000.0) * pricing.primary_input_1k + (500 / 1000.0) * pricing.primary_output_1k
|
|
|
|
# Assert cost per article is well within $0.0006 limit
|
|
assert cost < 0.0006, f"Cost ${cost} exceeds budget $0.0006"
|