From f2ad13af6511181cf7318ac144eb9d418631f726 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Sat, 6 Jan 2024 17:55:32 +0530 Subject: [PATCH] fix(openai.py): fix image generation model dump --- litellm/llms/openai.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/llms/openai.py b/litellm/llms/openai.py index 91a79fa5795..3265b230f3f 100644 --- a/litellm/llms/openai.py +++ b/litellm/llms/openai.py @@ -711,7 +711,7 @@ class OpenAIChatCompletion(BaseLLM): original_response=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: exception_mapping_worked = True raise e