scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-10 03:28:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 84

fix: keep excluded fields out of serialization fallbacks

Browse source
This commit is contained in:
r266-tech 2026-09-08 15:49:04 +08:00
parent c9dca39cf9
commit d52b2992ce
2 changed files with 7 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
litellm/litellm_core_utils/safe_json_dumps.py
View file

@ -20,7 +20,7 @@ def _dump_model(model: BaseModel) -> dict[str, object] | str:
try:
return model.model_dump(mode="json")
except (PydanticSerializationError, TypeError):
return str(model)
return "Unserializable Pydantic Model"
def safe_dumps(

13
tests/test_litellm/litellm_core_utils/test_safe_json_dumps.py
View file

@ -21,29 +21,28 @@ def test_repeated_pydantic_model_with_serialization_error(json_succeeds: bool) -
raise ValueError("serialization failed")
model: Final = FailingModel()
expected: Final = {"value": 42} if json_succeeds else str(model)
expected: Final = {"value": 42} if json_succeeds else "Unserializable Pydantic Model"
assert json.loads(safe_dumps([model, model])) == [expected, expected]
def test_pydantic_string_fallback_preserves_value_transform() -> None:
def test_pydantic_fallback_excludes_fields_and_preserves_value_transform() -> None:
from typing import Final
from pydantic import BaseModel, model_serializer
from pydantic import BaseModel, Field, model_serializer
class FailingModel(BaseModel):
password: str = Field(default="do-not-log\x00", exclude=True)
@model_serializer
def serialize_model(self) -> dict[str, object]:
raise ValueError("serialization failed")
def __str__(self) -> str:
return "secret\x00"
model: Final = FailingModel()
assert json.loads(
safe_dumps(
{"token": model, "other": model}, value_transform=lambda key, value: "redacted" if key == "token" else value
)
) == {"token": "redacted", "other": "secret"}
) == {"token": "redacted", "other": "Unserializable Pydantic Model"}
def test_primitive_types():