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

revert(ocr): stop forwarding client= on the Python path

Browse source 
Python becomes a thin SDK interface over Rust, so a live Python HTTP client has no effect on either route. This puts the Python OCR path back to what main does
This commit is contained in:
Yujong Lee 2026-09-18 19:39:07 -07:00
parent 0119f50015
commit fb41bc3ed6
2 changed files with 0 additions and 37 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
litellm/ocr/main.py
View file

@ -25,7 +25,6 @@ from litellm.llms.base_llm.ocr.transformation import (
OCRResponse,
parse_ocr_request_format,
)
from litellm.llms.custom_httpx.http_handler import AsyncHTTPHandler, HTTPHandler
from litellm.llms.custom_httpx.llm_http_handler import BaseLLMHTTPHandler
from litellm.types.router import GenericLiteLLMParams
from litellm.types.utils import CustomPricingLiteLLMParams
@ -53,11 +52,6 @@ class _PreparedOCRRequest:
litellm_logging_obj: LiteLLMLoggingObj
def _supplied_client(kwargs: Mapping[str, object]) -> HTTPHandler | AsyncHTTPHandler | None:
candidate: Final = kwargs.get("client")
return candidate if isinstance(candidate, (HTTPHandler, AsyncHTTPHandler)) else None
def _prepare_ocr_request(
model: str,
document: Mapping[str, object],
@ -244,7 +238,6 @@ async def aocr(
api_key=prepared.api_key,
api_base=prepared.api_base,
custom_llm_provider=prepared.custom_llm_provider,
client=_supplied_client(kwargs),
aocr=True,
headers=prepared.extra_headers,
provider_config=prepared.provider_config,
@ -411,7 +404,6 @@ def ocr(
api_key=prepared.api_key,
api_base=prepared.api_base,
custom_llm_provider=prepared.custom_llm_provider,
client=_supplied_client(kwargs),
aocr=_is_async,
headers=prepared.extra_headers,
provider_config=prepared.provider_config,

29
tests/test_litellm/ocr/test_main.py
View file

@ -113,35 +113,6 @@ async def test_python_request_response_and_callbacks(
assert logger.log_pre_api_call.call_count == 1
@pytest.mark.asyncio
@pytest.mark.parametrize("asynchronous", [False, True])
async def test_python_uses_the_supplied_client(provider: Mock, asynchronous: bool) -> None:
supplied: Final = Mock(return_value=provider.return_value)
transport: Final = httpx.MockTransport(supplied)
arguments: Final = {
"model": "mistral/mistral-ocr-latest",
"document": dict(PRICING_DOCUMENT),
"api_key": "test-key",
"api_base": "https://ocr.test/v1",
}
async def call() -> OCRResponse:
if not asynchronous:
with httpx.Client(transport=transport) as sync_client:
return litellm.ocr(**arguments, client=HTTPHandler(client=sync_client))
async with httpx.AsyncClient(transport=transport) as async_client:
handler: Final = AsyncHTTPHandler()
await handler.client.aclose()
handler.client = async_client
return await litellm.aocr(**arguments, client=handler)
response: Final = await call()
assert response.pages[0].markdown == "parsed document"
assert supplied.call_count == 1
assert str(supplied.call_args.args[0].url) == "https://ocr.test/v1/ocr"
assert provider.call_count == 0
@pytest.mark.asyncio
@pytest.mark.parametrize("asynchronous", [False, True])
async def test_python_provider_errors_keep_public_exception(provider: Mock, asynchronous: bool) -> None: