o
    i                     @   s8  d dl mZ d dlmZmZ G dd deZG dd deZG dd deZG d	d
 d
eZG dd deZ	G dd deZ
G dd deZG dd deZG dd deZG dd deZG dd deZG dd deZG dd deZg dZg dZdd d!d"dd d!d"d#d$d%d"d&d'd(d"d)d(d*d"d+Zd,S )-    )Literal)	BaseModelFieldc                   @   s   e Zd ZU edZeed< edZeed< edZedB ed< edZ	edB ed< eddd	d
Z
edB ed< edddd
ZedB ed< edZedB ed< dS )Text2ImageTaskCreationRequest.modelprompturlNresponse_formatsizer   geleseedg      ?g      $@guidance_scaleF	watermark)__name__
__module____qualname__r   r   str__annotations__r   r	   r
   r   intr   floatr   bool r   r   </mnt/c/Users/fbmor/ComfyUI/comfy_api_nodes/apis/bytedance.pyr      s   
 r   c                   @      e Zd ZU edZeed< dS )Seedream4Options   
max_imagesN)r   r   r   r   r   r   r   r   r   r   r   r         
 r   c                   @   s   e Zd ZU edZeed< edZeed< edZeed< edddZ	e
e dB ed	< edZeed
< eddddZeed< edZeed< eeddZeed< edZeed< dZedB ed< dS )Seedream4TaskCreationRequest.r   r   r   r	   Nz
Image URLsdescriptionimager
   r   r   r   r   disabledsequential_image_generationr   )r   #sequential_image_generation_optionsFr   output_format)r   r   r   r   r   r   r   r   r	   r$   listr
   r   r   r&   r   r'   r   r   r(   r   r   r   r   r!      s   
 r!   c                   @   sZ   e Zd ZU edZeed< edddZeed< eg ddZ	e
ed< ei ddZeed	< d
S )ImageTaskCreationResponse.r   zIUnix timestamp (in seconds) indicating time when the request was created.r"   createdz2Contains information about the generated image(s).dataz6Contains `code` and `message` fields in case of error.errorN)r   r   r   r   r   r   r   r+   r   r,   r)   r-   dictr   r   r   r   r*   !   s
   
 r*   c                   @   s.   e Zd ZU edZeed< edZeed< dS )TaskTextContenttexttype.N)r   r   r   r   r1   r   r   r0   r   r   r   r   r/   (      
 r/   c                   @   r   )TaskImageContentUrl.r   N)r   r   r   r   r   r   r   r   r   r   r   r3   -   r    r3   c                   @   sF   e Zd ZU edZeed< edZeed< edZ	e
d dB ed< dS )TaskImageContent	image_urlr1   .N)first_frame
last_framereference_imagerole)r   r   r   r   r1   r   r   r5   r3   r9   r   r   r   r   r   r4   1   s   
 r4   c                   @   sJ   e Zd ZU edZeed< edddZee	 ed< edZ
edB ed< dS )Text2VideoTaskCreationRequest.r      
min_lengthcontentNgenerate_audio)r   r   r   r   r   r   r   r>   r)   r/   r?   r   r   r   r   r   r:   7   s   
 r:   c                   @   sN   e Zd ZU edZeed< edddZee	e
B  ed< edZedB ed< dS )Image2VideoTaskCreationRequest.r      r<   r>   Nr?   )r   r   r   r   r   r   r   r>   r)   r/   r4   r?   r   r   r   r   r   r@   =   s   
 r@   c                   @   r   )TaskCreationResponse.idN)r   r   r   r   rC   r   r   r   r   r   r   rB   C   r    rB   c                   @   s.   e Zd ZU edZeed< edZeed< dS )TaskStatusError.codemessageN)r   r   r   r   rE   r   r   rF   r   r   r   r   rD   G   r2   rD   c                   @   r   )TaskStatusResult.	video_urlN)r   r   r   r   rH   r   r   r   r   r   r   rG   L   r    rG   c                   @   sj   e Zd ZU edZeed< edZeed< edZe	d ed< edZ
edB ed< edZedB ed< dS )	TaskStatusResponse.rC   r   )queuedrunning	cancelled	succeededfailedstatusNr-   r>   )r   r   r   r   rC   r   r   r   rO   r   r-   rD   r>   rG   r   r   r   r   rI   P   s   
 rI   )
)z1024x1024 (1:1)   rP   )z864x1152 (3:4)`    )z1152x864 (4:3)rR   rQ   )z1280x720 (16:9)     )z720x1280 (9:16)rT   rS   )z832x1248 (2:3)@    )z1248x832 (3:2)rV   rU   )z1512x648 (21:9)i  i  z2048x2048 (1:1)   rX   CustomNN)rW   )z2304x1728 (4:3) 	    )z1728x2304 (3:4)r\   r[   )z2560x1440 (16:9) 
    )z1440x2560 (9:16)r^   r]   )z2496x1664 (3:2)	    )z1664x2496 (2:3)r`   r_   )z3024x1296 (21:9)i  i  )z3072x3072 (1:1)   ra   )z4096x4096 (1:1)   rb   rY   (   <   Z   )480p720p1080pF   U   s   2   A   d   P      )zseedance-1-0-lite-t2v-250428zseedance-1-0-lite-i2v-250428zseedance-1-0-pro-250528zseedance-1-0-pro-fast-251015zseedance-1-5-pro-251215N)typingr   pydanticr   r   r   r   r!   r*   r/   r3   r4   r:   r@   rB   rD   rG   rI   RECOMMENDED_PRESETSRECOMMENDED_PRESETS_SEEDREAM_4VIDEO_TASKS_EXECUTION_TIMEr   r   r   r   <module>   sL    

