
    ~i2                        d dl mZ d dlmZ d dlmZmZ d dlZddlm	Z	m
Z
 ddlmZ ddlmZmZ dd	lmZmZ dd
lmZmZmZmZ ddlmZ ddlmZmZmZ ddlmZmZ ddl mZ  G d de          Z! G d de          Z"dS )    )annotations)Optional)LiteraloverloadN   )	NOT_GIVENNotGiven)make_request_options)apikey_requiredasync_apikey_required)SyncAPIResourceAsyncAPIResource)OptimizePromptOptions SequentialImageGenerationOptionsImagesResponseContentGenerationTool)ImageGenStreamEvent)BodyQueryHeaders)StreamAsyncStream)r   c                     e Zd Zeddddddddddddddddedd0d&            Zeddddddddddddddddd'd1d*            Zeddddddddddddddddd'd2d-            Zeddddddddddddddddedd3d/            ZdS )4ImagesNimageresponse_formatsizeseedguidance_scale	watermarkoptimize_promptoptimize_prompt_optionsextra_headersextra_query
extra_bodytimeoutsequential_image_generation#sequential_image_generation_optionstoolsoutput_formatstreammodelstrpromptr   str | list[str] | Noner   
str | Noner   r   
int | Noner    float | Noner!   bool | Noner"   r#   OptimizePromptOptions | Noner$   Headers | Noner%   Query | Noner&   Body | Noner'   float | httpx.Timeout | Noner(   r)   'SequentialImageGenerationOptions | Noner*   "list[ContentGenerationTool] | Noner+   r,   #Optional[Literal[False]] | NotGivenreturnr   c                   	 d S N selfr-   r/   r   r   r   r   r    r!   r"   r#   r$   r%   r&   r'   r(   r)   r*   r+   r,   s                       x/root/.openclaw/workspace/.venvs/ark-sdk/lib/python3.11/site-packages/volcenginesdkarkruntime/resources/images/images.pygeneratezImages.generate$   s    2 		 	    r   r   r   r   r    r!   r"   r#   r$   r%   r&   r'   r(   r)   r*   r+   Literal[True]Stream[ImageGenStreamEvent]c                   d S r?   r@   rA   s                       rC   rD   zImages.generateG   s	    0 '*crE   bool,ImagesResponse | Stream[ImageGenStreamEvent]c                   d S r?   r@   rA   s                       rC   rD   zImages.generatea   s	    0 8;srE   3Optional[Literal[False]] | Literal[True] | NotGivenc               (   |                      d|||||||||	|
|
                    d          nd |||                    d          nd |d |D             nd ||dt          ||||          t          |pdt          t
                             }|S )	N/images/generationsjsonmodec                :    g | ]}|                     d           S rP   rQ   
model_dump.0ts     rC   
<listcomp>z#Images.generate.<locals>.<listcomp>   &    >>>1Q\\v\..>>>rE   r-   r/   r   r   r   r   r    r!   r"   r#   r(   r)   r*   r+   r,   r$   r%   r&   r'   Fbodyoptionscast_tor,   
stream_cls)_postrV   r
   r   r   r   )rB   r-   r/   r   r   r   r   r    r!   r"   r#   r$   r%   r&   r'   r(   r)   r*   r+   r,   resps                        rC   rD   zImages.generate{   s    2 zz! #2"0&#2 /: ,66F6CCC/J ;F 8BBBOOO ( ?>>>>>!. 7 : )+'%	   #?U12O  (
 (
T rE   (r-   r.   r/   r.   r   r0   r   r1   r   r1   r   r2   r    r3   r!   r4   r"   r4   r#   r5   r$   r6   r%   r7   r&   r8   r'   r9   r(   r1   r)   r:   r*   r;   r+   r1   r,   r<   r=   r   )(r-   r.   r/   r.   r   r0   r   r1   r   r1   r   r2   r    r3   r!   r4   r"   r4   r#   r5   r$   r6   r%   r7   r&   r8   r'   r9   r(   r1   r)   r:   r*   r;   r+   r1   r,   rG   r=   rH   )(r-   r.   r/   r.   r   r0   r   r1   r   r1   r   r2   r    r3   r!   r4   r"   r4   r#   r5   r$   r6   r%   r7   r&   r8   r'   r9   r(   r1   r)   r:   r*   r;   r+   r1   r,   rJ   r=   rK   )(r-   r.   r/   r.   r   r0   r   r1   r   r1   r   r2   r    r3   r!   r4   r"   r4   r#   r5   r$   r6   r%   r7   r&   r8   r'   r9   r(   r1   r)   r:   r*   r;   r+   r1   r,   rM   r=   rK   )__name__
__module____qualname__r   r   rD   r   r@   rE   rC   r   r   #   s        )-&*'+!%'+@D(,$("&042648$(6?-          X D  )-&*'+!%'+@D(,$("&042648$(+* * * * * X*2  )-&*'+!%'+@D(,$("&042648$(+; ; ; ; ; X;2  )-&*'+!%'+@D(,$("&042648$(FO-B B B B B _B B BrE   r   c                     e Zd Zeddddddddddddddddedd0d&            Zeddddddddddddddddd'd1d*            Zeddddddddddddddddd'd2d-            Zeddddddddddddddddedd3d/            ZdS )4AsyncImagesNr   r-   r.   r/   r   r0   r   r1   r   r   r2   r    r3   r!   r4   r"   r#   r5   r$   r6   r%   r7   r&   r8   r'   r9   r(   r)   r:   r*   r;   r+   r,   r<   r=   r   c               
   K   d S r?   r@   rA   s                       rC   rD   zAsyncImages.generate   s      0 rE   rF   rG    AsyncStream[ImageGenStreamEvent]c               
   K   d S r?   r@   rA   s                       rC   rD   zAsyncImages.generate   s      0 ,/3rE   rJ   1ImagesResponse | AsyncStream[ImageGenStreamEvent]c               
   K   d S r?   r@   rA   s                       rC   rD   zAsyncImages.generate   s      0 =@CrE   rM   c               4  K   |                      d|||||||||	|
|
                    d          nd |||                    d          nd |d |D             nd ||dt          ||||          t          |pdt          t
                              d {V S )	NrO   rP   rQ   c                :    g | ]}|                     d           S rT   rU   rW   s     rC   rZ   z(AsyncImages.generate.<locals>.<listcomp>A  r[   rE   r\   r]   Fr^   )rc   rV   r
   r   r   r   rA   s                       rC   rD   zAsyncImages.generate  s     2 ZZ! #2"0&#2 /: ,66F6CCC/J ;F 8BBBOOO ( ?>>>>>!. 7 : )+'%	   #?U"#67O   (
 (
 (
 (
 (
 (
 (
 (
 (	
rE   re   )(r-   r.   r/   r.   r   r0   r   r1   r   r1   r   r2   r    r3   r!   r4   r"   r4   r#   r5   r$   r6   r%   r7   r&   r8   r'   r9   r(   r1   r)   r:   r*   r;   r+   r1   r,   rG   r=   rl   )(r-   r.   r/   r.   r   r0   r   r1   r   r1   r   r2   r    r3   r!   r4   r"   r4   r#   r5   r$   r6   r%   r7   r&   r8   r'   r9   r(   r1   r)   r:   r*   r;   r+   r1   r,   rJ   r=   rn   )(r-   r.   r/   r.   r   r0   r   r1   r   r1   r   r2   r    r3   r!   r4   r"   r4   r#   r5   r$   r6   r%   r7   r&   r8   r'   r9   r(   r1   r)   r:   r*   r;   r+   r1   r,   rM   r=   rn   )rf   rg   rh   r   r   rD   r   r@   rE   rC   rj   rj      s        )-&*'+!%'+@D(,$("&042648$(6?-     X2  )-&*'+!%'+@D(,$("&042648$(+/ / / / / X/2  )-&*'+!%'+@D(,$("&042648$(+@ @ @ @ @ X@2  )-&*'+!%'+@D(,$("&042648$(FO-@
 @
 @
 @
 @
 @
 @
 @
rE   rj   )#
__future__r   typingr   typing_extensionsr   r   httpx_typesr   r	   _base_clientr
   _utils._utilsr   r   	_resourcer   r   types.imagesr   r   r   r   #types.images.image_gen_stream_eventr   r   r   r   
_streamingr   r   types.images.imagesr   rj   r@   rE   rC   <module>r~      s   # " " " " "       / / / / / / / /  ) ) ) ) ) ) ) ) 0 0 0 0 0 0 C C C C C C C C : : : : : : : :            G F F F F F * * * * * * * * * * - - - - - - - - 8 8 8 8 8 8[ [ [ [ [_ [ [ [|P
 P
 P
 P
 P
" P
 P
 P
 P
 P
rE   