From 750432457b8018eebe8bc771b21e3a7ff55d0c28 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Sat, 30 Dec 2023 12:44:54 +0530 Subject: [PATCH] fix(openai.py): fix async image gen call --- 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 bbdbd591d..efb7ee8f9 100644 --- a/litellm/llms/openai.py +++ b/litellm/llms/openai.py @@ -674,7 +674,7 @@ class OpenAIChatCompletion(BaseLLM): raise OpenAIError(status_code=422, message="max retries must be an int") if aimg_generation == True: - response = self.aimage_generation(data=data, input=input, logging_obj=logging_obj, model_response=model_response, api_base=api_base, api_key=api_key, timeout=timeout, client=client, max_retries=max_retries) # type: ignore + response = self.aimage_generation(data=data, prompt=prompt, logging_obj=logging_obj, model_response=model_response, api_base=api_base, api_key=api_key, timeout=timeout, client=client, max_retries=max_retries) # type: ignore return response if client is None: