ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

(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
Show Stats 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 type(payload[param]) == ModelResponse:
payload[param] = payload[param].model_dump_json()
if type(payload[param]) == EmbeddingResponse:
payload[param] = payload[param].model_dump_json()
elif type(payload[param]) == Usage:
payload[param] = payload[param].model_dump_json()
else: