scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-08 03:08:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 115

refactor: use method-level import for MockOpenAITransport

Browse source
This commit is contained in:
Ryan Crabbe 2026-02-23 09:58:46 -08:00
parent 95d9514054
commit fb123ec52b
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
litellm/llms/openai/common_utils.py
View file

@ -22,7 +22,6 @@ from litellm.llms.custom_httpx.http_handler import (
AsyncHTTPHandler,
get_ssl_configuration,
)
from litellm.llms.custom_httpx.mock_transport import MockOpenAITransport
def _get_client_init_params(cls: type) -> Tuple[str, ...]:
@ -207,6 +206,8 @@ class BaseOpenAILLM:
return litellm.aclient_session
if getattr(litellm, "network_mock", False):
from litellm.llms.custom_httpx.mock_transport import MockOpenAITransport
return httpx.AsyncClient(transport=MockOpenAITransport())
# Get unified SSL configuration
@ -230,6 +231,8 @@ class BaseOpenAILLM:
return litellm.client_session
if getattr(litellm, "network_mock", False):
from litellm.llms.custom_httpx.mock_transport import MockOpenAITransport
return httpx.Client(transport=MockOpenAITransport())
# Get unified SSL configuration