diff --git a/openai/api_resources/image.py b/openai/api_resources/image.py index abd32c3f8b..39a5b6f616 100644 --- a/openai/api_resources/image.py +++ b/openai/api_resources/image.py @@ -156,7 +156,7 @@ async def acreate_variation( def _prepare_create_edit( cls, image, - mask, + mask=None, api_key=None, api_base=None, api_type=None, @@ -179,14 +179,15 @@ def _prepare_create_edit( for key, value in params.items(): files.append((key, (None, value))) files.append(("image", ("image", image, "application/octet-stream"))) - files.append(("mask", ("mask", mask, "application/octet-stream"))) + if mask is not None: + files.append(("mask", ("mask", mask, "application/octet-stream"))) return requestor, url, files @classmethod def create_edit( cls, image, - mask, + mask=None, api_key=None, api_base=None, api_type=None, @@ -215,7 +216,7 @@ def create_edit( async def acreate_edit( cls, image, - mask, + mask=None, api_key=None, api_base=None, api_type=None, diff --git a/openai/cli.py b/openai/cli.py index eaa49c713a..0ebb0537ce 100644 --- a/openai/cli.py +++ b/openai/cli.py @@ -240,8 +240,10 @@ def create_variation(cls, args): def create_edit(cls, args): with open(args.image, "rb") as file_reader: image_reader = BufferReader(file_reader.read(), desc="Upload progress") - with open(args.mask, "rb") as file_reader: - mask_reader = BufferReader(file_reader.read(), desc="Upload progress") + mask_reader = None + if args.mask is not None: + with open(args.mask, "rb") as file_reader: + mask_reader = BufferReader(file_reader.read(), desc="Upload progress") resp = openai.Image.create_edit( image=image_reader, mask=mask_reader, @@ -893,7 +895,7 @@ def help(args): "-M", "--mask", type=str, - required=True, + required=False, help="Path to a mask image. It should be the same size as the image you're editing and a RGBA PNG image. The Alpha channel acts as the mask.", ) sub.set_defaults(func=Image.create_edit)