fix(openai.py): fix async image gen call

This commit is contained in:
Krrish Dholakia 2023-12-30 12:44:54 +05:30
parent 2acd086596
commit 750432457b

View file

@ -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: