From 983cf2278c6bee4790f61186d42649ae9a886edb Mon Sep 17 00:00:00 2001 From: Sameer Kankute Date: Mon, 4 May 2026 17:41:05 +0530 Subject: [PATCH] fix: restore missing test function header; remove stale openai_tool_name_mappings lazy-load Co-authored-by: Cursor --- litellm/__init__.py | 14 -------------- tests/test_litellm/test_utils.py | 4 ++++ 2 files changed, 4 insertions(+), 14 deletions(-) diff --git a/litellm/__init__.py b/litellm/__init__.py index 2874ff88b5d..28112f5c12a 100644 --- a/litellm/__init__.py +++ b/litellm/__init__.py @@ -1961,8 +1961,6 @@ if TYPE_CHECKING: bedrock_tool_name_mappings: InMemoryCache - openai_tool_name_mappings: InMemoryCache - # Azure exception class (lazy-loaded) from litellm.llms.azure.common_utils import AzureOpenAIError @@ -2055,18 +2053,6 @@ def __getattr__(name: str) -> Any: _globals["bedrock_tool_name_mappings"] = _bedrock_tool_name_mappings return _globals["bedrock_tool_name_mappings"] - if name == "openai_tool_name_mappings": - from ._lazy_imports import _get_litellm_globals - - _globals = _get_litellm_globals() - if "openai_tool_name_mappings" not in _globals: - from .litellm_core_utils.openai_tool_name_mapping import ( - openai_tool_name_mappings as _openai_tool_name_mappings, - ) - - _globals["openai_tool_name_mappings"] = _openai_tool_name_mappings - return _globals["openai_tool_name_mappings"] - # Lazy load AzureOpenAIError exception class if name == "AzureOpenAIError": from ._lazy_imports import _get_litellm_globals diff --git a/tests/test_litellm/test_utils.py b/tests/test_litellm/test_utils.py index 0d0ee51e333..c917bbde1b7 100644 --- a/tests/test_litellm/test_utils.py +++ b/tests/test_litellm/test_utils.py @@ -4024,7 +4024,11 @@ def test_validate_and_fix_openai_tools_sanitizes_when_enabled(): assert out[0]["function"]["name"] == "invalid_name" assert original_name == "invalid.name" + +def test_openai_tool_name_mapping_per_request_context(): + from litellm.litellm_core_utils.openai_tool_name_mapping import ( begin_openai_tool_name_mapping_scope, + restore_openai_tool_name_for_user, ) from litellm.utils import validate_and_fix_openai_tools