ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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