From d37eb2ce181f6d6b67406fe88cf5db7fbccfa5cf Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 2 Sep 2026 20:20:38 -0700 Subject: [PATCH] test: retain azure ocr URL coverage --- .../core/src/providers/azure_ai/ocr/transformation.rs | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/litellm-rust/crates/core/src/providers/azure_ai/ocr/transformation.rs b/litellm-rust/crates/core/src/providers/azure_ai/ocr/transformation.rs index f9e1df4f709..886c1a9e9b9 100644 --- a/litellm-rust/crates/core/src/providers/azure_ai/ocr/transformation.rs +++ b/litellm-rust/crates/core/src/providers/azure_ai/ocr/transformation.rs @@ -583,6 +583,10 @@ mod tests { .data; assert_eq!(body["model"], "pixtral-12b-2409"); assert_eq!(body["include_image_base64"], true); + assert_eq!( + body["document"]["document_url"], + "data:application/pdf;base64,abc" + ); } #[test] @@ -821,8 +825,8 @@ mod tests { #[test] fn document_intelligence_url_combines_pages_and_features() { let url = complete_document_intelligence_url( - Some("https://example.cognitiveservices.azure.com"), - "prebuilt-layout", + Some("https://example.cognitiveservices.azure.com/"), + "azure_ai/doc-intelligence/prebuilt-layout", &mapped_params(json!({"pages": [0, 1, 2], "features": ["keyValuePairs", "languages"]})), &|_| None, )