scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-08 03:08:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 115

feat(registry): add azure_ai/mistral-ocr-4-0 page and annotation prices from Azure Retail Prices

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
mateo 2026-09-03 13:48:25 +00:00
parent f26407aa8c
commit 840173e778
3 changed files with 34 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
litellm/model_prices_and_context_window_backup.json
View file

@ -10037,6 +10037,16 @@
],
"source": "https://ai.azure.com/catalog/models/mistral-document-ai-2512"
},
"azure_ai/mistral-ocr-4-0": {
"litellm_provider": "azure_ai",
"ocr_cost_per_page": 0.004,
"annotation_cost_per_page": 0.005,
"mode": "ocr",
"supported_endpoints": [
"/v1/ocr"
],
"source": "https://azure.microsoft.com/en-us/pricing/details/ai-foundry-models/mistral/"
},
"azure_ai/doc-intelligence/prebuilt-read": {
"litellm_provider": "azure_ai",
"ocr_cost_per_page": 0.0015,

10
model_prices_and_context_window.json
View file

@ -10037,6 +10037,16 @@
],
"source": "https://ai.azure.com/catalog/models/mistral-document-ai-2512"
},
"azure_ai/mistral-ocr-4-0": {
"litellm_provider": "azure_ai",
"ocr_cost_per_page": 0.004,
"annotation_cost_per_page": 0.005,
"mode": "ocr",
"supported_endpoints": [
"/v1/ocr"
],
"source": "https://azure.microsoft.com/en-us/pricing/details/ai-foundry-models/mistral/"
},
"azure_ai/doc-intelligence/prebuilt-read": {
"litellm_provider": "azure_ai",
"ocr_cost_per_page": 0.0015,

14
tests/test_litellm/llms/mistral/ocr/test_mistral_ocr_cost.py
View file

@ -15,6 +15,7 @@ from litellm.cost_calculator import completion_cost
from litellm.llms.base_llm.ocr.transformation import OCRPage, OCRResponse, OCRUsageInfo
OCR4_COST_PER_PAGE = 0.004
OCR4_ANNOTATION_COST_PER_PAGE = 0.005
REPO_ROOT = Path(__file__).parents[5]
MAIN_COST_MAP = REPO_ROOT / "model_prices_and_context_window.json"
@ -133,3 +134,16 @@ def test_azure_doc_ai_annotation_pages_fall_back_to_ocr_rate(local_model_cost_ma
call_type="ocr",
)
assert cost == pytest.approx(AZURE_DOC_AI_COST_PER_PAGE)
def test_azure_ocr4_bills_ocr_and_annotation_pages_at_their_own_rates(local_model_cost_map) -> None:
info = litellm.get_model_info(model="azure_ai/mistral-ocr-4-0", custom_llm_provider="azure_ai")
assert info["ocr_cost_per_page"] == OCR4_COST_PER_PAGE
assert info["annotation_cost_per_page"] == OCR4_ANNOTATION_COST_PER_PAGE
cost = completion_cost(
completion_response=_annotated_ocr_response("mistral-ocr-4-0", 2, 3),
model="azure_ai/mistral-ocr-4-0",
custom_llm_provider="azure_ai",
call_type="ocr",
)
assert cost == pytest.approx(2 * OCR4_COST_PER_PAGE + 3 * OCR4_ANNOTATION_COST_PER_PAGE)