From 00cc169623f3d04b00ae97a3e3426cb9603c84ab Mon Sep 17 00:00:00 2001 From: Yujong Lee Date: Tue, 1 Sep 2026 18:19:42 -0700 Subject: [PATCH] revert(ocr): keep DeepSeek normalization separate --- litellm/llms/vertex_ai/ocr/deepseek_transformation.py | 3 +-- tests/test_litellm/ocr/test_fixture_models.py | 1 - 2 files changed, 1 insertion(+), 3 deletions(-) diff --git a/litellm/llms/vertex_ai/ocr/deepseek_transformation.py b/litellm/llms/vertex_ai/ocr/deepseek_transformation.py index b57a87c3325..2603552152d 100644 --- a/litellm/llms/vertex_ai/ocr/deepseek_transformation.py +++ b/litellm/llms/vertex_ai/ocr/deepseek_transformation.py @@ -177,9 +177,8 @@ class VertexAIDeepSeekOCRConfig(BaseOCRConfig): content_item = {"type": "image_url", "image_url": document_url} # Build DeepSeek OCR request - provider_model: Final = model if model.startswith("deepseek-ai/") else f"deepseek-ai/{model}" data: Final = { - "model": provider_model, + "model": "deepseek-ai/" + model, "messages": [{"role": "user", "content": [content_item]}], } diff --git a/tests/test_litellm/ocr/test_fixture_models.py b/tests/test_litellm/ocr/test_fixture_models.py index f97cf27db66..12e88de5d1e 100644 --- a/tests/test_litellm/ocr/test_fixture_models.py +++ b/tests/test_litellm/ocr/test_fixture_models.py @@ -492,7 +492,6 @@ def test_vertex_deepseek_request_maps_both_document_types_to_image_content( data: Final = cast(dict[str, object], request.data) messages: Final = cast(list[dict[str, object]], data["messages"]) content: Final = cast(list[dict[str, object]], messages[0]["content"]) - assert data["model"] == "deepseek-ai/deepseek-ocr-maas" assert content == [{"type": "image_url", "image_url": document[source_key]}]