o iCi|@sddlmZddlmZmZmZmZddlmZm Z ddl Z ddl m Z ddl mZmZmZddlmZmZmZmZmZmZmZmZmZdd lmZmZmZmZm Z dd l!m"Z"dd l#m$Z$m%Z%dd l&m'Z'm(Z(dd l)m*Z*m+Z+ddl,m-Z-ddl.m/Z/ddl0m1Z1ddl2m3Z3ddl4m5Z5ddgZ6Gddde$Z7Gddde%Z8GdddZ9GdddZ:GdddZ;GdddZ str | Omitr?Headers | Noner@ Query | NonerA Body | NonerB'float | httpx.Timeout | None | NotGivenr%c Cslt||||||d} tttttf| dggd} ddi|p i}|jdt| tj | t ||| | dt dS) Creates a variation of a given image. This endpoint only supports `dall-e-2`. Args: image: The image to use as the basis for the variation(s). Must be a valid PNG file, less than 4MB, and square. model: The model to use for image generation. Only `dall-e-2` is supported at this time. n: The number of images to generate. Must be between 1 and 10. response_format: The format in which the generated images are returned. Must be one of `url` or `b64_json`. URLs are only valid for 60 minutes after the image has been generated. size: The size of the generated images. Must be one of `256x256`, `512x512`, or `1024x1024`. user: A unique identifier representing your end-user, which can help OpenAI to monitor and detect abuse. [Learn more](https://platform.openai.com/docs/guides/safety-best-practices#end-user-ids). extra_headers: Send extra headers extra_query: Add additional query parameters to the request extra_body: Add additional JSON properties to the request timeout: Override the client-level default timeout for this request, in seconds rCr:r;r<r=r>rCpaths Content-Typemultipart/form-data/images/variationsr?r@rArBbodyfilesoptionscast_to) rrrrstrobject_postrr ImageCreateVariationParamsr#r% r0rCr:r;r<r=r>r?r@rArBrVrWr1r1r2create_variation/s(0  zImages.create_variation backgroundinput_fidelitymaskr:r;output_compression output_formatpartial_imagesqualityr<r=streamr>r?r@rArB+Union[FileTypes, SequenceNotStr[FileTypes]]promptrZra9Optional[Literal['transparent', 'opaque', 'auto']] | Omitrb'Optional[Literal['high', 'low']] | OmitrcFileTypes | Omitrdre/Optional[Literal['png', 'jpeg', 'webp']] | OmitrfrgEOptional[Literal['standard', 'low', 'medium', 'high', 'auto']] | Omit]Optional[Literal['256x256', '512x512', '1024x1024', '1536x1024', '1024x1536', 'auto']] | OmitrhOptional[Literal[False]] | OmitcCdSaCreates an edited or extended image given one or more source images and a prompt. This endpoint only supports `gpt-image-1` and `dall-e-2`. Args: image: The image(s) to edit. Must be a supported image file or an array of images. For the GPT image models (`gpt-image-1`, `gpt-image-1-mini`, and `gpt-image-1.5`), each image should be a `png`, `webp`, or `jpg` file less than 50MB. You can provide up to 16 images. For `dall-e-2`, you can only provide one image, and it should be a square `png` file less than 4MB. prompt: A text description of the desired image(s). The maximum length is 1000 characters for `dall-e-2`, and 32000 characters for the GPT image models. background: Allows to set transparency for the background of the generated image(s). This parameter is only supported for the GPT image models. Must be one of `transparent`, `opaque` or `auto` (default value). When `auto` is used, the model will automatically determine the best background for the image. If `transparent`, the output format needs to support transparency, so it should be set to either `png` (default value) or `webp`. input_fidelity: Control how much effort the model will exert to match the style and features, especially facial features, of input images. This parameter is only supported for `gpt-image-1`. Unsupported for `gpt-image-1-mini`. Supports `high` and `low`. Defaults to `low`. mask: An additional image whose fully transparent areas (e.g. where alpha is zero) indicate where `image` should be edited. If there are multiple images provided, the mask will be applied on the first image. Must be a valid PNG file, less than 4MB, and have the same dimensions as `image`. model: The model to use for image generation. Only `dall-e-2` and the GPT image models are supported. Defaults to `dall-e-2` unless a parameter specific to the GPT image models is used. n: The number of images to generate. Must be between 1 and 10. output_compression: The compression level (0-100%) for the generated images. This parameter is only supported for the GPT image models with the `webp` or `jpeg` output formats, and defaults to 100. output_format: The format in which the generated images are returned. This parameter is only supported for the GPT image models. Must be one of `png`, `jpeg`, or `webp`. The default value is `png`. partial_images: The number of partial images to generate. This parameter is used for streaming responses that return partial images. Value must be between 0 and 3. When set to 0, the response will be a single image sent in one streaming event. Note that the final image may be sent before the full number of partial images are generated if the full image is generated more quickly. quality: The quality of the image that will be generated. `high`, `medium` and `low` are only supported for the GPT image models. `dall-e-2` only supports `standard` quality. Defaults to `auto`. response_format: The format in which the generated images are returned. Must be one of `url` or `b64_json`. URLs are only valid for 60 minutes after the image has been generated. This parameter is only supported for `dall-e-2`, as the GPT image models always return base64-encoded images. size: The size of the generated images. Must be one of `1024x1024`, `1536x1024` (landscape), `1024x1536` (portrait), or `auto` (default value) for the GPT image models, and one of `256x256`, `512x512`, or `1024x1024` for `dall-e-2`. stream: Edit the image in streaming mode. Defaults to `false`. See the [Image generation guide](https://platform.openai.com/docs/guides/image-generation) for more information. user: A unique identifier representing your end-user, which can help OpenAI to monitor and detect abuse. [Learn more](https://platform.openai.com/docs/guides/safety-best-practices#end-user-ids). extra_headers: Send extra headers extra_query: Add additional query parameters to the request extra_body: Add additional JSON properties to the request timeout: Override the client-level default timeout for this request, in seconds Nr1r0rCrjrarbrcr:r;rdrerfrgr<r=rhr>r?r@rArBr1r1r2editxpz Images.editrarbrcr:r;rdrerfrgr<r=r>r?r@rArB Literal[True]Stream[ImageEditStreamEvent]cCrraCreates an edited or extended image given one or more source images and a prompt. This endpoint only supports `gpt-image-1` and `dall-e-2`. Args: image: The image(s) to edit. Must be a supported image file or an array of images. For the GPT image models (`gpt-image-1`, `gpt-image-1-mini`, and `gpt-image-1.5`), each image should be a `png`, `webp`, or `jpg` file less than 50MB. You can provide up to 16 images. For `dall-e-2`, you can only provide one image, and it should be a square `png` file less than 4MB. prompt: A text description of the desired image(s). The maximum length is 1000 characters for `dall-e-2`, and 32000 characters for the GPT image models. stream: Edit the image in streaming mode. Defaults to `false`. See the [Image generation guide](https://platform.openai.com/docs/guides/image-generation) for more information. background: Allows to set transparency for the background of the generated image(s). This parameter is only supported for the GPT image models. Must be one of `transparent`, `opaque` or `auto` (default value). When `auto` is used, the model will automatically determine the best background for the image. If `transparent`, the output format needs to support transparency, so it should be set to either `png` (default value) or `webp`. input_fidelity: Control how much effort the model will exert to match the style and features, especially facial features, of input images. This parameter is only supported for `gpt-image-1`. Unsupported for `gpt-image-1-mini`. Supports `high` and `low`. Defaults to `low`. mask: An additional image whose fully transparent areas (e.g. where alpha is zero) indicate where `image` should be edited. If there are multiple images provided, the mask will be applied on the first image. Must be a valid PNG file, less than 4MB, and have the same dimensions as `image`. model: The model to use for image generation. Only `dall-e-2` and the GPT image models are supported. Defaults to `dall-e-2` unless a parameter specific to the GPT image models is used. n: The number of images to generate. Must be between 1 and 10. output_compression: The compression level (0-100%) for the generated images. This parameter is only supported for the GPT image models with the `webp` or `jpeg` output formats, and defaults to 100. output_format: The format in which the generated images are returned. This parameter is only supported for the GPT image models. Must be one of `png`, `jpeg`, or `webp`. The default value is `png`. partial_images: The number of partial images to generate. This parameter is used for streaming responses that return partial images. Value must be between 0 and 3. When set to 0, the response will be a single image sent in one streaming event. Note that the final image may be sent before the full number of partial images are generated if the full image is generated more quickly. quality: The quality of the image that will be generated. `high`, `medium` and `low` are only supported for the GPT image models. `dall-e-2` only supports `standard` quality. Defaults to `auto`. response_format: The format in which the generated images are returned. Must be one of `url` or `b64_json`. URLs are only valid for 60 minutes after the image has been generated. This parameter is only supported for `dall-e-2`, as the GPT image models always return base64-encoded images. size: The size of the generated images. Must be one of `1024x1024`, `1536x1024` (landscape), `1024x1536` (portrait), or `auto` (default value) for the GPT image models, and one of `256x256`, `512x512`, or `1024x1024` for `dall-e-2`. user: A unique identifier representing your end-user, which can help OpenAI to monitor and detect abuse. [Learn more](https://platform.openai.com/docs/guides/safety-best-practices#end-user-ids). extra_headers: Send extra headers extra_query: Add additional query parameters to the request extra_body: Add additional JSON properties to the request timeout: Override the client-level default timeout for this request, in seconds Nr1r0rCrjrhrarbrcr:r;rdrerfrgr<r=r>r?r@rArBr1r1r2rurvbool-ImagesResponse | Stream[ImageEditStreamEvent]cCrrrzr1r{r1r1r2ru\rvrCrjrh/Optional[Literal[False]] | Literal[True] | OmitcCst||||||||| | | | | ||d}tttttf|dgddgdggd}ddi|p.i}|jdt||r:tj ntj |t ||||d t |pId t td S N)rCrjrarbrcr:r;rdrerfrgr<r=rhr>rCzrcrOrQrRz /images/editsrTF)rVrWrXrYrh stream_cls)rrrrrZr[r\rr ImageEditParamsStreamingImageEditParamsNonStreamingr#r%r!r'r0rCrjrarbrcr:r;rdrerfrgr<r=rhr>r?r@rArBrVrWr1r1r2rusD(rar: moderationr;rdrerfrgr<r=rhstyler>r?r@rArBr'Optional[Literal['low', 'auto']] | OmitKOptional[Literal['standard', 'hd', 'low', 'medium', 'high', 'auto']] | OmitwOptional[Literal['auto', '1024x1024', '1536x1024', '1024x1536', '256x256', '512x512', '1792x1024', '1024x1792']] | Omitr,Optional[Literal['vivid', 'natural']] | OmitcCrra Creates an image given a prompt. [Learn more](https://platform.openai.com/docs/guides/images). Args: prompt: A text description of the desired image(s). The maximum length is 32000 characters for the GPT image models, 1000 characters for `dall-e-2` and 4000 characters for `dall-e-3`. background: Allows to set transparency for the background of the generated image(s). This parameter is only supported for the GPT image models. Must be one of `transparent`, `opaque` or `auto` (default value). When `auto` is used, the model will automatically determine the best background for the image. If `transparent`, the output format needs to support transparency, so it should be set to either `png` (default value) or `webp`. model: The model to use for image generation. One of `dall-e-2`, `dall-e-3`, or a GPT image model (`gpt-image-1`, `gpt-image-1-mini`, `gpt-image-1.5`). Defaults to `dall-e-2` unless a parameter specific to the GPT image models is used. moderation: Control the content-moderation level for images generated by the GPT image models. Must be either `low` for less restrictive filtering or `auto` (default value). n: The number of images to generate. Must be between 1 and 10. For `dall-e-3`, only `n=1` is supported. output_compression: The compression level (0-100%) for the generated images. This parameter is only supported for the GPT image models with the `webp` or `jpeg` output formats, and defaults to 100. output_format: The format in which the generated images are returned. This parameter is only supported for the GPT image models. Must be one of `png`, `jpeg`, or `webp`. partial_images: The number of partial images to generate. This parameter is used for streaming responses that return partial images. Value must be between 0 and 3. When set to 0, the response will be a single image sent in one streaming event. Note that the final image may be sent before the full number of partial images are generated if the full image is generated more quickly. quality: The quality of the image that will be generated. - `auto` (default value) will automatically select the best quality for the given model. - `high`, `medium` and `low` are supported for the GPT image models. - `hd` and `standard` are supported for `dall-e-3`. - `standard` is the only option for `dall-e-2`. response_format: The format in which generated images with `dall-e-2` and `dall-e-3` are returned. Must be one of `url` or `b64_json`. URLs are only valid for 60 minutes after the image has been generated. This parameter isn't supported for the GPT image models, which always return base64-encoded images. size: The size of the generated images. Must be one of `1024x1024`, `1536x1024` (landscape), `1024x1536` (portrait), or `auto` (default value) for the GPT image models, one of `256x256`, `512x512`, or `1024x1024` for `dall-e-2`, and one of `1024x1024`, `1792x1024`, or `1024x1792` for `dall-e-3`. stream: Generate the image in streaming mode. Defaults to `false`. See the [Image generation guide](https://platform.openai.com/docs/guides/image-generation) for more information. This parameter is only supported for the GPT image models. style: The style of the generated images. This parameter is only supported for `dall-e-3`. Must be one of `vivid` or `natural`. Vivid causes the model to lean towards generating hyper-real and dramatic images. Natural causes the model to produce more natural, less hyper-real looking images. user: A unique identifier representing your end-user, which can help OpenAI to monitor and detect abuse. [Learn more](https://platform.openai.com/docs/guides/safety-best-practices#end-user-ids). extra_headers: Send extra headers extra_query: Add additional query parameters to the request extra_body: Add additional JSON properties to the request timeout: Override the client-level default timeout for this request, in seconds Nr1r0rjrar:rr;rdrerfrgr<r=rhrr>r?r@rArBr1r1r2generatemzImages.generaterar:rr;rdrerfrgr<r=rr>r?r@rArBStream[ImageGenStreamEvent]cCrra Creates an image given a prompt. [Learn more](https://platform.openai.com/docs/guides/images). Args: prompt: A text description of the desired image(s). The maximum length is 32000 characters for the GPT image models, 1000 characters for `dall-e-2` and 4000 characters for `dall-e-3`. stream: Generate the image in streaming mode. Defaults to `false`. See the [Image generation guide](https://platform.openai.com/docs/guides/image-generation) for more information. This parameter is only supported for the GPT image models. background: Allows to set transparency for the background of the generated image(s). This parameter is only supported for the GPT image models. Must be one of `transparent`, `opaque` or `auto` (default value). When `auto` is used, the model will automatically determine the best background for the image. If `transparent`, the output format needs to support transparency, so it should be set to either `png` (default value) or `webp`. model: The model to use for image generation. One of `dall-e-2`, `dall-e-3`, or a GPT image model (`gpt-image-1`, `gpt-image-1-mini`, `gpt-image-1.5`). Defaults to `dall-e-2` unless a parameter specific to the GPT image models is used. moderation: Control the content-moderation level for images generated by the GPT image models. Must be either `low` for less restrictive filtering or `auto` (default value). n: The number of images to generate. Must be between 1 and 10. For `dall-e-3`, only `n=1` is supported. output_compression: The compression level (0-100%) for the generated images. This parameter is only supported for the GPT image models with the `webp` or `jpeg` output formats, and defaults to 100. output_format: The format in which the generated images are returned. This parameter is only supported for the GPT image models. Must be one of `png`, `jpeg`, or `webp`. partial_images: The number of partial images to generate. This parameter is used for streaming responses that return partial images. Value must be between 0 and 3. When set to 0, the response will be a single image sent in one streaming event. Note that the final image may be sent before the full number of partial images are generated if the full image is generated more quickly. quality: The quality of the image that will be generated. - `auto` (default value) will automatically select the best quality for the given model. - `high`, `medium` and `low` are supported for the GPT image models. - `hd` and `standard` are supported for `dall-e-3`. - `standard` is the only option for `dall-e-2`. response_format: The format in which generated images with `dall-e-2` and `dall-e-3` are returned. Must be one of `url` or `b64_json`. URLs are only valid for 60 minutes after the image has been generated. This parameter isn't supported for the GPT image models, which always return base64-encoded images. size: The size of the generated images. Must be one of `1024x1024`, `1536x1024` (landscape), `1024x1536` (portrait), or `auto` (default value) for the GPT image models, one of `256x256`, `512x512`, or `1024x1024` for `dall-e-2`, and one of `1024x1024`, `1792x1024`, or `1024x1792` for `dall-e-3`. style: The style of the generated images. This parameter is only supported for `dall-e-3`. Must be one of `vivid` or `natural`. Vivid causes the model to lean towards generating hyper-real and dramatic images. Natural causes the model to produce more natural, less hyper-real looking images. user: A unique identifier representing your end-user, which can help OpenAI to monitor and detect abuse. [Learn more](https://platform.openai.com/docs/guides/safety-best-practices#end-user-ids). extra_headers: Send extra headers extra_query: Add additional query parameters to the request extra_body: Add additional JSON properties to the request timeout: Override the client-level default timeout for this request, in seconds Nr1r0rjrhrar:rr;rdrerfrgr<r=rr>r?r@rArBr1r1r2r~r,ImagesResponse | Stream[ImageGenStreamEvent]cCrrrr1rr1r1r2rrcCsZ|jdt||||||||| | | | | |d| rtjntjt||||dt| p'dttdSNz/images/generations)rjrar:rr;rdrerfrgr<r=rhrr>rTF)rVrXrYrhr) r\rr ImageGenerateParamsStreamingImageGenerateParamsNonStreamingr#r%r!r&rr1r1r2r\s:)r+r,)r+r5rCrr:rDr;rEr<rFr=rGr>rHr?rIr@rJrArKrBrLr+r%(rCrirjrZrarkrbrlrcrmr:rDr;rErdrErernrfrErgror<rFr=rprhrqr>rHr?rIr@rJrArKrBrLr+r%)(rCrirjrZrhrxrarkrbrlrcrmr:rDr;rErdrErernrfrErgror<rFr=rpr>rHr?rIr@rJrArKrBrLr+ry)(rCrirjrZrhr|rarkrbrlrcrmr:rDr;rErdrErernrfrErgror<rFr=rpr>rHr?rIr@rJrArKrBrLr+r})(rCrirjrZrarkrbrlrcrmr:rDr;rErdrErernrfrErgror<rFr=rprhrr>rHr?rIr@rJrArKrBrLr+r}&rjrZrarkr:rDrrr;rErdrErernrfrErgrr<rFr=rrhrqrrr>rHr?rIr@rJrArKrBrLr+r%)&rjrZrhrxrarkr:rDrrr;rErdrErernrfrErgrr<rFr=rrrr>rHr?rIr@rJrArKrBrLr+r)&rjrZrhr|rarkr:rDrrr;rErdrErernrfrErgrr<rFr=rrrr>rHr?rIr@rJrArKrBrLr+r)&rjrZrarkr:rDrrr;rErdrErernrfrErgrr<rFr=rrhrrrr>rHr?rIr@rJrArKrBrLr+r __name__ __module__ __qualname__rr3r7rrr_rrurrr1r1r1r2r(F   Iqqq@nnnc@r*)Zr)r+AsyncImagesWithRawResponsecCr-r.)rr/r1r1r2r3r4zAsyncImages.with_raw_response AsyncImagesWithStreamingResponsecCr-r6)rr/r1r1r2r7r8z#AsyncImages.with_streaming_responseNr9rCrr:rDr;rEr<rFr=rGr>rHr?rIr@rJrArKrBrLr%c szt||||||d} tttttf| dggd} ddi|p!i}|jdt| tj IdH| t ||| | dt d IdHS) rMrNrCrOrQrRrSNrTrU) rrrrrZr[r\rr r]r#r%r^r1r1r2r_s*0  zAsyncImages.create_variationr`rirjrZrarkrbrlrcrmrdrernrfrgrorprhrqcdSrsr1rtr1r1r2rupzAsyncImages.editrwrx!AsyncStream[ImageEditStreamEvent]crrzr1r{r1r1r2rugrr|2ImagesResponse | AsyncStream[ImageEditStreamEvent]crrzr1r{r1r1r2rurr~rcst||||||||| | | | | ||d}tttttf|dgddgdggd}ddi|p/i}|jdt||r;tj ntj IdH|t ||||d t |pMd t td IdHSr)rrrrrZr[r\rr rrr#r%r"r'rr1r1r2ruKsF( rrrrrrrcrrr1rr1r1r2rmzAsyncImages.generater AsyncStream[ImageGenStreamEvent]crrr1rr1r1r2rr1ImagesResponse | AsyncStream[ImageGenStreamEvent]crrr1rr1r1r2rjrcsh|jdt||||||||| | | | | |d| rtjntjIdHt||||dt| p+dttdIdHSr) r\rr rrr#r%r"r&rr1r1r2rs< )r+r)r+rrr)(rCrirjrZrhrxrarkrbrlrcrmr:rDr;rErdrErernrfrErgror<rFr=rpr>rHr?rIr@rJrArKrBrLr+r)(rCrirjrZrhr|rarkrbrlrcrmr:rDr;rErdrErernrfrErgror<rFr=rpr>rHr?rIr@rJrArKrBrLr+r)(rCrirjrZrarkrbrlrcrmr:rDr;rErdrErernrfrErgror<rFr=rprhrr>rHr?rIr@rJrArKrBrLr+rr)&rjrZrhrxrarkr:rDrrr;rErdrErernrfrErgrr<rFr=rrrr>rHr?rIr@rJrArKrBrLr+r)&rjrZrhr|rarkr:rDrrr;rErdrErernrfrErgrr<rFr=rrrr>rHr?rIr@rJrArKrBrLr+r)&rjrZrarkr:rDrrr;rErdrErernrfrErgrr<rFr=rrhrrrr>rHr?rIr@rJrArKrBrLr+rrr1r1r1r2r)rc@eZdZdddZdS) r,imagesr(r+NonecC4||_t|j|_t|j|_t|j|_dSN)_imagesr to_raw_response_wrapperr_rurr0rr1r1r2__init__ zImagesWithRawResponse.__init__Nrr(r+rrrrrr1r1r1r2r,r,c@r) rrr)r+rcCrr)rr async_to_raw_response_wrapperr_rurrr1r1r2r%rz#AsyncImagesWithRawResponse.__init__Nrr)r+rrr1r1r1r2r$rrc@r) r5rr(r+rcC.||_t|j|_t|j|_t|j|_dSr)rrr_rurrr1r1r2r4 z$ImagesWithStreamingResponse.__init__Nrrr1r1r1r2r53rr5c@r) rrr)r+rcCrr)rr r_rurrr1r1r2rCrz)AsyncImagesWithStreamingResponse.__init__Nrrr1r1r1r2rBrr)= __future__rtypingrrrrtyping_extensionsrrhttpxr typesr r r _typesrrrrrrrrr_utilsrrrrr_compatr _resourcerr _responserr _streamingr!r" _base_clientr#types.image_modelr$types.images_responser%types.image_gen_stream_eventr&types.image_edit_stream_eventr'__all__r(r)r,rr5rr1r1r1r2sL  ,