@@ -53,22 +53,24 @@ class Gptimage(SimpleService): def make(self, input_message: Message, save: bool = True) -> list[Message]: info = input_message.info.copy() - version = info.get('version', 'image-generate') files = None for proxy in Proxy.objects.all(): json_data = { 'prompt': input_message.content, - 'background': info.get('background'), + 'background': info.get('background', 'auto'), 'model': 'gpt-image-1', 'output_format': 'png', - 'quality': info.get('quality'), - 'size': info.get('size'), + 'quality': info.get('quality', 'auto'), + 'size': info.get('size', 'auto'), } - if version == 'image-edit': + if input_message.file: + version = 'image-edit' file_bytes = input_message.file.read() files = { 'image': ('image.png', BytesIO(file_bytes), 'image/png'), } + else: + version = 'image-generate' with httpx.Client( base_url='https://api.openai.com/v1', proxy=f'{proxy.protocol}://{proxy.address}',