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

revert(ocr): keep DeepSeek normalization separate

Browse source
This commit is contained in:
Yujong Lee 2026-09-01 18:19:42 -07:00
parent 0fe10f2bc6
commit 00cc169623
2 changed files with 1 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
litellm/llms/vertex_ai/ocr/deepseek_transformation.py
View file

@ -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]}],
}

1
tests/test_litellm/ocr/test_fixture_models.py
View file

@ -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]}]