chore: regenerate cost-map schema and UI API types, drop test banner comments

This commit is contained in:
mubashir1osmani 2026-09-10 13:35:35 -04:00
parent c246f75e3e
commit 91e7df3d8f
5 changed files with 16 additions and 16 deletions

View file

@ -53,6 +53,10 @@
"type": "number",
"minimum": 0
},
"annotation_cost_per_page_batches": {
"type": "number",
"minimum": 0
},
"audio_transcription_config": {
"type": "string"
},
@ -432,6 +436,10 @@
"type": "number",
"minimum": 0
},
"ocr_cost_per_page_batches": {
"type": "number",
"minimum": 0
},
"output_cost_per_audio_token": {
"type": "number",
"minimum": 0

View file

@ -1809,11 +1809,6 @@ class TestBatchCostIsFinal:
assert bu.batch_cost_is_final(_retrieved_batch(status)) is True
# =========================================================================== #
# OCR batch output lines (Mistral /v1/ocr batches) are billed per page, not per token
# =========================================================================== #
def _ocr_row(pages_processed, annotation_pages=None, model="mistral-ocr-latest"):
usage_info = {"pages_processed": pages_processed, "doc_size_bytes": 4096}
if annotation_pages is not None:

View file

@ -76,11 +76,6 @@ def test_custom_llm_provider(config):
assert config.custom_llm_provider == LlmProviders.MISTRAL
# --------------------------------------------------------------------------- #
# create
# --------------------------------------------------------------------------- #
def test_create_request_maps_openai_fields_onto_mistral_job(config):
data = CreateBatchRequest(
completion_window="24h",
@ -175,11 +170,6 @@ def test_create_response_maps_job_onto_openai_batch(config):
assert batch.metadata == {"job_type": "testing"}
# --------------------------------------------------------------------------- #
# retrieve
# --------------------------------------------------------------------------- #
def test_retrieve_request_is_presigned_get_with_auth(config, api_key):
req = config.transform_retrieve_batch_request(
batch_id="job/with slash", optional_params={}, litellm_params={"api_base": "https://api.mistral.ai"}

View file

@ -63,7 +63,6 @@ def test_ocr4_cost_scales_with_pages(model: str, pages_processed: int) -> None:
assert cost == pytest.approx(OCR4_COST_PER_PAGE * pages_processed)
@pytest.mark.parametrize("cost_map_path", [MAIN_COST_MAP, BACKUP_COST_MAP])
def test_ocr3_pricing_entry(cost_map_path: Path) -> None:
with open(cost_map_path) as f:

View file

@ -29493,6 +29493,8 @@ export interface components {
allow_client_keepalive_override: boolean | null;
/** Annotation Cost Per Page */
annotation_cost_per_page?: number | null;
/** Annotation Cost Per Page Batches */
annotation_cost_per_page_batches?: number | null;
/** Api Base */
api_base?: string | null;
/** Api Key */
@ -29704,6 +29706,8 @@ export interface components {
ocr_cost_per_credit?: number | null;
/** Ocr Cost Per Page */
ocr_cost_per_page?: number | null;
/** Ocr Cost Per Page Batches */
ocr_cost_per_page_batches?: number | null;
/** Organization */
organization?: string | null;
/** Otpm */
@ -39684,6 +39688,8 @@ export interface components {
allow_client_keepalive_override: boolean | null;
/** Annotation Cost Per Page */
annotation_cost_per_page?: number | null;
/** Annotation Cost Per Page Batches */
annotation_cost_per_page_batches?: number | null;
/** Api Base */
api_base?: string | null;
/** Api Key */
@ -39895,6 +39901,8 @@ export interface components {
ocr_cost_per_credit?: number | null;
/** Ocr Cost Per Page */
ocr_cost_per_page?: number | null;
/** Ocr Cost Per Page Batches */
ocr_cost_per_page_batches?: number | null;
/** Organization */
organization?: string | null;
/** Otpm */