mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-09 03:18:44 +00:00
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:
parent
4676d54171
commit
983cf2278c
2 changed files with 4 additions and 14 deletions
|
|
@ -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
|
||||
|
|
|
|||
|
|
@ -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
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue