phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix(openai.py): fix image generation model dump

Browse source
This commit is contained in:
Krrish Dholakia 2024-01-06 17:55:32 +05:30
parent 2d8d7e3569
commit f2ad13af65
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/llms/openai.py
View file

@ -711,7 +711,7 @@ class OpenAIChatCompletion(BaseLLM):
original_response=response, original_response=response,
) )
# return response # return response
return convert_to_model_response_object(response_object=model_dump(), model_response_object=model_response, response_type="image_generation") # type: ignore return convert_to_model_response_object(response_object=response.model_dump(), model_response_object=model_response, response_type="image_generation") # type: ignore
except OpenAIError as e: except OpenAIError as e:
exception_mapping_worked = True exception_mapping_worked = True
raise e raise e