scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-11 22:51:28 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 124

test: respect optional logging payload fields

Browse source
This commit is contained in:
Yuneng Jiang 2026-09-10 18:07:12 -07:00
parent 2f46425732
commit d1a1cda144
No known key found for this signature in database
2 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
tests/local_testing/test_custom_callback_input.py
View file

@ -1076,7 +1076,7 @@ def test_standard_logging_payload(model, turn_off_message_logging):
)
)
keys_list = list(StandardLoggingPayload.__annotations__.keys())
keys_list = list(StandardLoggingPayload.__required_keys__)
for k in keys_list:
assert (
@ -1190,7 +1190,7 @@ def test_standard_logging_payload_audio(turn_off_message_logging, stream):
)
)
keys_list = list(StandardLoggingPayload.__annotations__.keys())
keys_list = list(StandardLoggingPayload.__required_keys__)
for k in keys_list:
assert (

2
tests/logging_callback_tests/test_datadog.py
View file

@ -270,7 +270,7 @@ async def test_datadog_logging_http_request():
message = json.loads(body[0]["message"])
print("logged message", json.dumps(message, indent=4))
expected_message_fields = StandardLoggingPayload.__annotations__.keys()
expected_message_fields = StandardLoggingPayload.__required_keys__
for field in expected_message_fields:
assert field in message, f"Field '{field}' is missing from the message"