Merge pull request #37419 from BerriAI/litellm_fix_ocr_tests_req_format

test(ocr): update Azure DI supported-params assertion for req_format
This commit is contained in:
Mateo Wang 2026-08-18 19:37:20 -07:00 committed by GitHub
commit 5f6a14dbb7
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -62,7 +62,7 @@ class TestAzureDocumentIntelligencePagesParam:
return AzureDocumentIntelligenceOCRConfig()
def test_get_supported_ocr_params_includes_pages_and_features(self, cfg):
assert cfg.get_supported_ocr_params("prebuilt-layout") == ["pages", "features"]
assert cfg.get_supported_ocr_params("prebuilt-layout") == ["pages", "features", "req_format"]
def test_map_ocr_params_mistral_zero_based_int_list(self, cfg):
mapped = cfg.map_ocr_params({"pages": [0, 1, 2]}, {}, "prebuilt-layout")