scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-21 00:21:49 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 74

fix(param_utils): resolve type-discipline and basedpyright violations in sanitization utility and registry

Browse source
This commit is contained in:
mutnale_sushant 2026-07-10 11:55:41 +05:30
parent ae867e0eac
commit 59c8e36add
2 changed files with 18 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
litellm/litellm_core_utils/custom_logger_registry.py
View file

@ -60,7 +60,7 @@ class CustomLoggerRegistry:
Registry mapping the callback class string to the class type.
"""
CALLBACK_CLASS_STR_TO_CLASS_TYPE: dict[str, type[object]] = {
CALLBACK_CLASS_STR_TO_CLASS_TYPE: dict[str, type[object]] = { # mutable-ok: global callback registry mapping
"lago": LagoLogger,
"openmeter": OpenMeterLogger,
"braintrust": BraintrustLogger,

30
litellm/litellm_core_utils/param_utils.py
View file

@ -2,18 +2,22 @@ import logging
logger = logging.getLogger(__name__)
LITELLM_INTERNAL_PARAM_NAMES = {
"litellm_params",
"proxy_server_request",
"model_info",
"metadata",
"preset_cache_key",
"litellm_metadata",
"acompletion",
}
LITELLM_INTERNAL_PARAM_NAMES = frozenset(
(
"litellm_params",
"proxy_server_request",
"model_info",
"metadata",
"preset_cache_key",
"litellm_metadata",
"acompletion",
)
)
def strip_litellm_internal_params(data: dict[str, object]) -> dict[str, object]:
def strip_litellm_internal_params(
data: dict[str, object], # mutable-ok: interface expects mutable dict payload
) -> dict[str, object]: # mutable-ok: returns mutable dict payload
"""
Remove LiteLLM internal params (e.g. litellm_params, proxy_server_request, _litellm_ prefixed keys)
from request data before passing to client libraries (e.g. OpenAI).
@ -25,13 +29,13 @@ def strip_litellm_internal_params(data: dict[str, object]) -> dict[str, object]:
try:
# Create a shallow copy so we don't modify the input dictionary in-place
cleaned_data: dict[str, object] = {}
cleaned_data: dict[str, object] = {} # mutable-ok: building cleaned payload dict
for key, value in data.items():
if key in LITELLM_INTERNAL_PARAM_NAMES or key.startswith("_litellm_"):
continue
if key == "extra_body" and isinstance(value, dict):
cleaned_extra_body: dict[str, object] = {}
extra_body_dict: dict[object, object] = value # pyright: ignore[reportUnknownVariableType] # cast from dynamic dict
cleaned_extra_body: dict[str, object] = {} # mutable-ok: building cleaned extra_body dict
extra_body_dict: dict[object, object] = value # pyright: ignore[reportUnknownVariableType] # mutable-ok: reading from extra_body
for k, v in extra_body_dict.items():
if isinstance(k, str) and (k in LITELLM_INTERNAL_PARAM_NAMES or k.startswith("_litellm_")):
continue