scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-10 22:41:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 129

fix(azure_ai): route only cohere parse deployment names to Cohere Parse

Browse source
This commit is contained in:
mateo-berri 2026-09-04 22:07:26 -07:00
parent 8426235290
commit d3a179f988
2 changed files with 5 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
litellm/llms/azure_ai/ocr/common_utils.py
View file

@ -25,7 +25,8 @@ def is_azure_document_intelligence_model(model: str) -> bool:
def is_azure_cohere_parse_model(model: str) -> bool:
return "parse" in model.lower()
lowered: Final = model.lower()
return "cohere" in lowered and "parse" in lowered
def get_azure_ai_ocr_config(model: str) -> Optional["BaseOCRConfig"]:

4
tests/test_litellm/llms/azure_ai/ocr/test_azure_ai_cohere_parse_transformation.py
View file

@ -40,7 +40,9 @@ def disable_aiohttp_transport(monkeypatch):
[
("Cohere-parse-v5", AzureAICohereParseConfig),
("cohere-parse-v5", AzureAICohereParseConfig),
("parse-v5", AzureAICohereParseConfig),
("cohere/parse-v5", AzureAICohereParseConfig),
("invoice-parser", AzureAIOCRConfig),
("parse-v5", AzureAIOCRConfig),
("mistral-ocr-4-0", AzureAIOCRConfig),
("mistral-document-ai-2512", AzureAIOCRConfig),
("doc-intelligence/prebuilt-read", AzureDocumentIntelligenceOCRConfig),