fix: restore missing test function header; remove stale openai_tool_name_mappings lazy-load

Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
Sameer Kankute 2026-05-04 17:41:05 +05:30
parent 4676d54171
commit 983cf2278c
No known key found for this signature in database
2 changed files with 4 additions and 14 deletions

View file

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

View file

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