phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(utils.py): return additional kwargs from openai-like response body

Browse source 
Closes https://github.com/BerriAI/litellm/issues/4981
This commit is contained in:
Krrish Dholakia 2024-07-31 15:37:03 -07:00
parent bd68714f51
commit 09ee8c6e2d
2 changed files with 70 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/utils.py
View file

@ -5859,6 +5859,11 @@ def convert_to_model_response_object(
if _response_headers is not None:
model_response_object._response_headers = _response_headers
special_keys = litellm.ModelResponse.model_fields.keys()
for k, v in response_object.items():
if k not in special_keys:
setattr(model_response_object, k, v)
return model_response_object
elif response_type == "embedding" and (
model_response_object is None