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

Avoid recursive Bedrock logging sanitizer

Browse source 
Co-authored-by: ishaan-berri <ishaan-berri@users.noreply.github.com>
This commit is contained in:
oss-agent-shin 2026-05-11 21:36:24 +00:00
parent 4f587087ff
commit 14ea4f57f4
No known key found for this signature in database
1 changed files with 9 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
litellm/llms/bedrock/batches/handler.py
View file

@ -1,3 +1,4 @@
import json
from datetime import datetime
from typing import Any, Optional, cast
@ -76,16 +77,14 @@ def _to_epoch(value: Any) -> Optional[int]:
def _sanitize_response_for_logging(value: Any) -> Any:
if isinstance(value, datetime):
return value.isoformat()
if isinstance(value, dict):
return {
key: _sanitize_response_for_logging(response_value)
for key, response_value in value.items()
}
if isinstance(value, (list, tuple)):
return [_sanitize_response_for_logging(item) for item in value]
return value
def _default(obj: Any) -> Any:
if isinstance(obj, datetime):
return obj.isoformat()
raise TypeError(
f"Object of type {obj.__class__.__name__} is not JSON serializable"
)
return json.loads(json.dumps(value, default=_default))
class BedrockBatchesHandler: