feat(litellm_logging.py): cleanup payload + add response cost to logged payload

This commit is contained in:
Krrish Dholakia 2024-08-15 17:53:25 -07:00
parent 3ddeb3297d
commit f6dba82882
3 changed files with 51 additions and 15 deletions

View file

@ -1218,3 +1218,11 @@ def test_standard_logging_payload():
mock_client.call_args.kwargs["kwargs"]["standard_logging_object"]
)
json.loads(json_str_payload)
## response cost
assert (
mock_client.call_args.kwargs["kwargs"]["standard_logging_object"][
"response_cost"
]
> 0
)