scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-17 23:51:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 67

refactor(router): validate classifier snapshots without recursion

Browse source
This commit is contained in:
moe-berri 2026-09-10 12:35:59 -07:00
parent 41a147f2a0
commit 33954d9cfe
1 changed files with 6 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
litellm/litellm_core_utils/classifier_logging.py
View file

@ -13,13 +13,13 @@ _JSON_OBJECT: Final = TypeAdapter(dict[str, JsonValue])
def classifier_input_snapshot(value: object, *, openai_sdk: bool = False) -> Mapping[str, JsonValue] | None:
if openai_sdk and isinstance(value, Mapping):
body: Final = {
key: item for key, item in value.items() if key not in ("extra_headers", "extra_query", "extra_body")
}
extra_body: Final = value.get("extra_body")
return classifier_input_snapshot({**body, **extra_body} if isinstance(extra_body, Mapping) else body)
try:
if openai_sdk and isinstance(value, Mapping):
body: Final = {
key: item for key, item in value.items() if key not in ("extra_headers", "extra_query", "extra_body")
}
extra_body: Final = value.get("extra_body")
return _JSON_OBJECT.validate_python({**body, **extra_body} if isinstance(extra_body, Mapping) else body)
return (
_JSON_OBJECT.validate_json(value)
if isinstance(value, (str, bytes))