U ܟjd@sDddlmZddlmZmZmZmZddlmZddl Z ddl m Z ddl m Z mZmZddlmZmZmZmZmZmZdd lmZmZmZmZdd lmZdd lmZm Z dd l!m"Z"m#Z#dd l$m%Z%ddl&m'Z'ddl(m)Z)ddgZ*GdddeZ+Gddde Z,GdddZ-GdddZ.GdddZ/GdddZ0dS)) annotations)UnionMappingOptionalcast)LiteralN)_legacy_response)image_edit_paramsimage_generate_paramsimage_create_variation_params) NOT_GIVENBodyQueryHeadersNotGiven FileTypes) extract_filesmaybe_transformdeepcopy_minimalasync_maybe_transform)cached_property)SyncAPIResourceAsyncAPIResource)to_streamed_response_wrapper"async_to_streamed_response_wrapper)make_request_options) ImageModel)ImagesResponseImages AsyncImagesc@seZdZeddddZeddddZeeeeeddded d d d d dddddddd ddZeeeeeeddded d ddd d d dddddddd ddZeeeeeeeddded dd d dd d d!ddddddd" d#d$Z dS)%rImagesWithRawResponsereturncCst|Sa This property can be used as a prefix for any HTTP method call to return the raw response object instead of the parsed content. For more information, see https://www.github.com/openai/openai-python#accessing-raw-response-data-eg-headers )r!selfr';/tmp/pip-unpacked-wheel-2qj3ofc_/openai/resources/images.pywith_raw_responseszImages.with_raw_responseImagesWithStreamingResponsecCst|Sz An alternative to `.with_raw_response` that doesn't eagerly read the response body. For more information, see https://www.github.com/openai/openai-python#with_streaming_response )r*r%r'r'r(with_streaming_response(szImages.with_streaming_responseN modelnresponse_formatsizeuser extra_headers extra_query extra_bodytimeoutr'Union[str, ImageModel, None] | NotGivenOptional[int] | NotGiven1Optional[Literal[('url', 'b64_json')]] | NotGivenAOptional[Literal[('256x256', '512x512', '1024x1024')]] | NotGivenstr | NotGivenHeaders | None Query | None Body | None'float | httpx.Timeout | None | NotGivenr imager.r/r0r1r2r3r4r5r6r#c Cslt||||||d} tttttf| dggd} ddi|p@i}|jdt| tj | t ||| | dt dS)  Creates a variation of a given image. 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. For `dall-e-3`, only `n=1` is supported. 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 rAr.r/r0r1r2rApaths Content-Typemultipart/form-data/images/variationsr3r4r5r6bodyfilesoptionscast_to) rrrrstrobject_postrr ImageCreateVariationParamsrr r&rAr.r/r0r1r2r3r4r5r6rKrLr'r'r(create_variation1s.0  zImages.create_variation maskr.r/r0r1r2r3r4r5r6rOFileTypes | NotGiven rApromptrVr.r/r0r1r2r3r4r5r6r#c  Cstt||||||||d} tttttf| dgdggd}ddi| pHi} |jdt| tj |t | | | | dt d S)  Creates an edited or extended image given an original image and a prompt. Args: image: The image to edit. Must be a valid PNG file, less than 4MB, and square. If mask is not provided, image must have transparency, which will be used as the mask. prompt: A text description of the desired image(s). The maximum length is 1000 characters. mask: An additional image whose fully transparent areas (e.g. where alpha is zero) indicate where `image` should be edited. 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` 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 rArYrVr.r/r0r1r2rArVrDrFrG /images/editsrIrJ) rrrrrOrPrQrr ImageEditParamsrrr&rArYrVr.r/r0r1r2r3r4r5r6rKrLr'r'r(editzs28 " z Images.edit r.r/qualityr0r1styler2r3r4r5r6&Literal[('standard', 'hd')] | NotGiven[Optional[Literal[('256x256', '512x512', '1024x1024', '1792x1024', '1024x1792')]] | NotGiven2Optional[Literal[('vivid', 'natural')]] | NotGiven rYr.r/rar0r1rbr2r3r4r5r6r#c Cs8|jdt||||||||dtjt| | | | dtdS) Creates an image given a prompt. Args: prompt: A text description of the desired image(s). The maximum length is 1000 characters for `dall-e-2` and 4000 characters for `dall-e-3`. model: The model to use for image generation. n: The number of images to generate. Must be between 1 and 10. For `dall-e-3`, only `n=1` is supported. quality: The quality of the image that will be generated. `hd` creates images with finer details and greater consistency across the image. This param is only supported for `dall-e-3`. 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` for `dall-e-2`. Must be one of `1024x1024`, `1792x1024`, or `1024x1792` for `dall-e-3` models. style: The style of the generated images. 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. This param is only supported for `dall-e-3`. 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 /images/generationsrYr.r/rar0r1rbr2rIrKrMrN)rQrr ImageGenerateParamsrr r&rYr.r/rar0r1rbr2r3r4r5r6r'r'r(generates,;  zImages.generate __name__ __module__ __qualname__rr)r,r rTr_rmr'r'r'r(rsJ  &N*Wc@seZdZeddddZeddddZeeeeeddded d d d d dddddddd ddZeeeeeeddded d ddd d d dddddddd ddZeeeeeeeddded dd d dd d d!ddddddd" d#d$Z dS)%r AsyncImagesWithRawResponser"cCst|Sr$)rrr%r'r'r(r)szAsyncImages.with_raw_response AsyncImagesWithStreamingResponsecCst|Sr+)rsr%r'r'r(r,)sz#AsyncImages.with_streaming_responseNr-rr7r8r9r:r;r<r=r>r?rr@c sxt||||||d} tttttf| dggd} ddi|p@i}|jdt| tj IdH| t ||| | dt d IdHS) rBrCrArDrFrGrHNrIrJ) rrrrrOrPrQrr rRrrrSr'r'r(rT2s.0 zAsyncImages.create_variationrUrOrWrXc  st||||||||d} tttttf| dgdggd}ddi| pHi} |jdt| tj IdH|t | | | | d t d IdHS) rZr[rArVrDrFrGr\NrIrJ) rrrrrOrPrQrr r]rrr^r'r'r(r_{s28 "zAsyncImages.editr`rcrdrerfc sD|jdt||||||||dtjIdHt| | | | dtdIdHS)rgrhriNrIrj)rQrr rkrrrlr'r'r(rms,;  zAsyncImages.generaternr'r'r'r(r sJ  &N*Wc@seZdZdddddZdS)r!rNoneimagesr#cCs4||_t|j|_t|j|_t|j|_dSN)_imagesr Zto_raw_response_wrapperrTr_rmr&rvr'r'r(__init__ szImagesWithRawResponse.__init__Nrorprqrzr'r'r'r(r!sr!c@seZdZdddddZdS)rrr rtrucCs4||_t|j|_t|j|_t|j|_dSrw)rxr Zasync_to_raw_response_wrapperrTr_rmryr'r'r(rz/sz#AsyncImagesWithRawResponse.__init__Nr{r'r'r'r(rr.srrc@seZdZdddddZdS)r*rrtrucCs.||_t|j|_t|j|_t|j|_dSrw)rxrrTr_rmryr'r'r(rz>sz$ImagesWithStreamingResponse.__init__Nr{r'r'r'r(r*=sr*c@seZdZdddddZdS)rsr rtrucCs.||_t|j|_t|j|_t|j|_dSrw)rxrrTr_rmryr'r'r(rzMsz)AsyncImagesWithStreamingResponse.__init__Nr{r'r'r'r(rsLsrs)1 __future__rtypingrrrrtyping_extensionsrZhttpxr typesr r r _typesr rrrrr_utilsrrrr_compatrZ _resourcerr _responserrZ _base_clientrZtypes.image_modelrZtypes.images_responser__all__rr r!rrr*rsr'r'r'r(s0