tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-07-15 14:21:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

(fix) linting error

Browse Source
This commit is contained in:
ishaan-jaff
2023-11-20 18:32:43 -08:00
parent 50f883a2fb
commit 4f46ac4ab5
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/utils.py
+3 -3
View File
@@ -3079,9 +3079,9 @@ def convert_to_model_response_object(response_object: Optional[dict]=None, model
model_response_object.choices = choice_list
if "usage" in response_object:
model_response_object.usage.completion_tokens = response_object["usage"]["completion_tokens"]
model_response_object.usage.prompt_tokens = response_object["usage"]["prompt_tokens"]
model_response_object.usage.total_tokens = response_object["usage"]["total_tokens"]
model_response_object.usage.completion_tokens = response_object["usage"].get("completion_tokens", 0)
model_response_object.usage.prompt_tokens = response_object["usage"].get("prompt_tokens", 0)
model_response_object.usage.total_tokens = response_object["usage"].get("total_tokens", 0)
if "id" in response_object:
model_response_object.id = response_object["id"]