
    ~i-                    *   d dl mZ d dlmZmZmZmZ d dlZd dlm	Z	 d dl
mZmZ d dlmZ ddlmZ dd	lmZmZ dd
lmZmZmZ ddlmZmZ ddlmZ ddlmZ ddlm Z   G d de          Z! G d de          Z" G d d          Z# G d d          Z$dS )    )annotations)IterableUnionListOptionalN)make_request_options)SyncAPIResourceAsyncAPIResource)CreateTaskContentParam   )cached_property)to_raw_response_wrapperasync_to_raw_response_wrapper)BodyQueryHeaders)apikey_requiredasync_apikey_required)ContentGenerationTask)ContentGenerationTaskID)"ListContentGenerationTasksResponsec                      e Zd Zed3d            Zeddddddddddddddddddd4d#            Zeddddd$d5d'            Ze	 	 	 	 	 	 	 	 	 	 d6d7d0            Ze	 	 	 	 d8d9d2            Z	dS ):TasksreturnTasksWithRawResponsec                     t          |           S N)r   selfs    /root/.openclaw/workspace/.venvs/ark-sdk/lib/python3.11/site-packages/volcenginesdkarkruntime/resources/content_generation/tasks.pywith_raw_responsezTasks.with_raw_response%   s    #D)))    Ncallback_urlreturn_last_frameservice_tierexecution_expires_aftergenerate_audiodraftcamera_fixed	watermarkseed
resolutionratiodurationframesextra_headersextra_query
extra_bodytimeoutmodelstrcontent Iterable[CreateTaskContentParam]r$   Optional[str]r%   Optional[bool]r&   r'   Optional[int]r(   r)   r*   r+   r,   r-   r.   r/   r0   r1   Headers | Noner2   Query | Noner3   Body | Noner4   float | httpx.Timeout | Noner   c                   |                      d|||||||||	|
|||||dt          ||||          t                    }|S N/contents/generations/tasks)r5   r7   r$   r%   r&   r'   r(   r)   r*   r+   r,   r-   r.   r/   r0   r1   r2   r3   r4   )bodyoptionscast_to_postr   r   r   r5   r7   r$   r%   r&   r'   r(   r)   r*   r+   r,   r-   r.   r/   r0   r1   r2   r3   r4   resps                        r    createzTasks.create)   s    0 zz)" ,%6 ,+B"0 ,&($  " )+'%	   ,3  
 
6 r"   rC   task_idr   c          	     h    |                      d| t          ||||          t                    }|S Nz/contents/generations/tasks/rC   )pathrE   rF   _getr   r   r   rL   r1   r2   r3   r4   rJ   s          r    getz	Tasks.get^   sQ     yy999(+'%	   *  	
 	
 r"   page_num
int | None	page_sizestatus
str | Nonetask_idsUnion[List[str], str] | Noner   c           
        g }|r|                     d|f           |r|                     d|f           |r|                     d|f           |r|                     d|f           |r|                     d|f           |r4t          |t                    r|g}|D ]}|                     d|f           |                     d|t	          ||	||
          t
          	          }|S 
NrT   rV   zfilter.statuszfilter.modelzfilter.service_tierzfilter.task_idsrB   rC   )rO   paramsrE   rF   append
isinstancer6   rQ   r   r   r   rT   rV   rW   rY   r5   r&   r1   r3   r2   r4   query_paramsrL   rJ   s                 r    listz
Tasks.listt   sH     	8X 6777 	:i 8999 	;& 9::: 	9 7888 	G!6 EFFF 	B(C(( &$:# B B##%6$@AAAAyy.(+'%	   7  

 

 r"   Nonec           	     h    |                      d| t          ||||          t                    }|S rN   _deleter   r   rR   s          r    deletezTasks.delete   sQ     ||999(+'%	   *  	
 	
 r"   )r   r   (r5   r6   r7   r8   r$   r9   r%   r:   r&   r9   r'   r;   r(   r:   r)   r:   r*   r:   r+   r:   r,   r;   r-   r9   r.   r9   r/   r;   r0   r;   r1   r<   r2   r=   r3   r>   r4   r?   r   r   rL   r6   r1   r<   r2   r=   r3   r>   r4   r?   r   r   
NNNNNNNNNNrT   rU   rV   rU   rW   rX   rY   rZ   r5   rX   r&   rX   r1   r<   r3   r>   r2   r=   r4   r?   r   r   NNNNrL   r6   r1   r<   r2   r=   r3   r>   r4   r?   r   rd   )
__name__
__module____qualname__r   r!   r   rK   rS   rc   rh    r"   r    r   r   $   sC       * * * _*  '+,0&*15)- $'+$("$(#"& $(,$("&04+2 2 2 2 2 _2h 
 )-$("&04     _*   $ $!15 #'(,"&$(04) ) ) ) _)V  )-$("&04    _  r"   r   c                      e Zd Zed3d            Zeddddddddddddddddddd4d#            Zeddddd$d5d'            Ze	 	 	 	 	 	 	 	 	 	 d6d7d0            Ze	 	 	 	 d8d9d2            Z	dS ):
AsyncTasksr   AsyncTasksWithRawResponsec                     t          |           S r   )ru   r   s    r    r!   zAsyncTasks.with_raw_response   s    (...r"   Nr#   r5   r6   r7   r8   r$   r9   r%   r:   r&   r'   r;   r(   r)   r*   r+   r,   r-   r.   r/   r0   r1   r<   r2   r=   r3   r>   r4   r?   r   c                  K   |                      d|||||||||	|
|||||dt          ||||          t                     d {V }|S rA   rG   rI   s                        r    rK   zAsyncTasks.create   s      0 ZZ)" ,%6 ,+B"0 ,&($  " )+'%	   ,3   
 
 
 
 
 
 
 
6 r"   rC   rL   r   c          	     x   K   |                      d| t          ||||          t                     d {V }|S rN   rP   rR   s          r    rS   zAsyncTasks.get   ss       YY999(+'%	   *  	
 	
 	
 	
 	
 	
 	
 	
 r"   rT   rU   rV   rW   rX   rY   rZ   r   c           
       K   g }|r|                     d|f           |r|                     d|f           |r|                     d|f           |r|                     d|f           |r|                     d|f           |r4t          |t                    r|g}|D ]}|                     d|f           |                     d|t	          ||	||
          t
          	           d {V }|S r\   r^   ra   s                 r    rc   zAsyncTasks.list  sj       	8X 6777 	:i 8999 	;& 9::: 	9 7888 	G!6 EFFF 	B(C(( &$:# B B##%6$@AAAAYY.(+'%	   7  

 

 

 

 

 

 

 

 r"   rd   c           	     x   K   |                      d| t          ||||          t                     d {V }|S rN   rf   rR   s          r    rh   zAsyncTasks.delete2  ss       \\999(+'%	   * " 	
 	
 	
 	
 	
 	
 	
 	
 r"   )r   ru   ri   rj   rk   rl   rm   rn   )
ro   rp   rq   r   r!   r   rK   rS   rc   rh   rr   r"   r    rt   rt      sG       / / / _/  '+,0&*15)- $'+$("$(#"& $(,$("&04+2 2 2 2 2 2h 
 )-$("&04     *   $ $!15 #'(,"&$(04) ) ) ) )V  )-$("&04      r"   rt   c                      e Zd ZddZdS )r   tasksr   r   rd   c                F    || _         t          |j                  | _        d S r   )_tasksr   rK   r   r|   s     r    __init__zTasksWithRawResponse.__init__I  s#    -L
 
r"   N)r|   r   r   rd   ro   rp   rq   r   rr   r"   r    r   r   H  (        
 
 
 
 
 
r"   r   c                      e Zd ZddZdS )ru   r|   rt   r   rd   c                F    || _         t          |j                  | _        d S r   )r~   r   rK   r   s     r    r   z"AsyncTasksWithRawResponse.__init__R  s#    3L
 
r"   N)r|   rt   r   rd   r   rr   r"   r    ru   ru   Q  r   r"   ru   )%
__future__r   typingr   r   r   r   httpx$volcenginesdkarkruntime._base_clientr   !volcenginesdkarkruntime._resourcer	   r
   Jvolcenginesdkarkruntime.types.content_generation.create_task_content_paramr   _compatr   	_responser   r   _typesr   r   r   _utils._utilsr   r   0types.content_generation.content_generation_taskr   3types.content_generation.content_generation_task_idr   ?types.content_generation.list_content_generation_tasks_responser   r   rt   r   ru   rr   r"   r    <module>r      s   # " " " " " 2 2 2 2 2 2 2 2 2 2 2 2  E E E E E E O O O O O O O O      ' & & & & & O O O O O O O O * * * * * * * * * * C C C C C C C C U U U U U U          
O O O O OO O O OdO O O O O! O O Od
 
 
 
 
 
 
 

 
 
 
 
 
 
 
 
 
r"   