phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 03:34:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

(fix) track EmbeddingResponse cost

Browse source
This commit is contained in:
ishaan-jaff 2024-01-18 12:29:21 -08:00
parent 88cdfedf84
commit 4eb9221121
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/utils.py
View file

@ -8478,6 +8478,8 @@ def get_logging_payload(kwargs, response_obj, start_time, end_time):
if param in payload and type(payload[param]) != Json: if param in payload and type(payload[param]) != Json:
if type(payload[param]) == ModelResponse: if type(payload[param]) == ModelResponse:
payload[param] = payload[param].model_dump_json() payload[param] = payload[param].model_dump_json()
if type(payload[param]) == EmbeddingResponse:
payload[param] = payload[param].model_dump_json()
elif type(payload[param]) == Usage: elif type(payload[param]) == Usage:
payload[param] = payload[param].model_dump_json() payload[param] = payload[param].model_dump_json()
else: else: