
    +j#                        d dl mZ d dlmZmZ  G d de          Z G d de          Z G d de          Z G d	 d
e          Z G d de          Z	 G d de          Z
 G d de          Z G d de          Z G d de          Z G d de          Z G d de          Z G d de          Z G d de          Z G d de          Z G d d e          Z G d! d"e          Z G d# d$e          Z G d% d&e          Z G d' d(e          Z G d) d*e          Z G d+ d,e          Z G d- d.e          Z G d/ d0e          Z G d1 d2e          Z G d3 d4e          Zd5d6d7d8d9Zg d:Zg d;Z g d<Z!g d=Z"g d>Z#g d?Z$d@gZ%e"e#z   e$z   e%z   Z&e"e$z   e%z   Z'e!e"z   e$z   e%z   Z(dAdBdCdAdBdCdAdDdCdEdAdBdCdAdBdCdFdGZ)dHdIdJdEdHdIdJdEdKdLdMdEdNdOdPdEdQdPdRdEdSZ*dTS )U    )Literal)	BaseModelFieldc                      e Zd ZU  ed          Zeed<    ed          Zeed<    ed          Zedz  ed<    ed          Z	edz  ed<    eddd	
          Z
edz  ed<    eddd
          Zedz  ed<    ed          Zedz  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     >/home/wildlama/comfy/ComfyUI/comfy_api_nodes/apis/bytedance.pyr   r      s         sE3%**FC"'%,,OS4Z...uT{{D#*"""uQ1444D#*444#(5#>#>#>NEDL>>>"U5\\Itd{)))))r   r   c                   0    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       s*         eBiiJr   r    c                   t   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          Z	e
e         dz  ed	<    ed          Zeed
<    eddd          Zeed<    ed          Zeed<    e ed                    Zeed<    ed          Zeed<   dZedz  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.        sE3%**FC 5<<OS'''#eDlCCCE49tCCCc

D#caJ///D#///',uZ'8'8888<AEBRBR^`BaBaBa<b<b')9bbbeEllIt""" $M3:$$$$$r   r$   c                       e Zd ZU  ed          Zeed<    edd          Zeed<    eg d          Z	e
ed<    ei 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   r0   r,   r1   dictr   r   r   r.   r.   !   s         sE35*uvvvGSvvvr'[\\\D$\\\%(`aaaE4aaaaar   r.   c                   P    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   r6   r   r   r5   r   r   r   r4   r4   (   sB         fD#c

D#r   r4   c                   0    e Zd ZU  ed          Zeed<   dS )TaskImageContentUrl.r
   Nr   r   r   r   r
   r   r   r   r   r   r8   r8   -   *         uSzzCr   r8   c                       e Zd ZU  ed          Zeed<    ed          Zeed<    ed          Z	e
d         dz  ed<   dS )TaskImageContent	image_urlr6   .N)first_frame
last_framereference_imagerole)r   r   r   r   r6   r   r   r=   r8   rA   r   r   r   r   r<   r<   1   si         k""D#"""%*U3ZZI"///KP5QU;;D'@
AD
HVVVVVr   r<   c                   0    e Zd ZU  ed          Zeed<   dS )TaskVideoContentUrl.r
   Nr9   r   r   r   rC   rC   7   r:   r   rC   c                   p    e Zd ZU  ed          Zeed<    ed          Zeed<    ed          Z	eed<   dS )TaskVideoContent	video_urlr6   .reference_videorA   N)
r   r   r   r   r6   r   r   rF   rC   rA   r   r   r   rE   rE   ;   `         k""D#"""%*U3ZZI"///'((D#(((((r   rE   c                   0    e Zd ZU  ed          Zeed<   dS )TaskAudioContentUrl.r
   Nr9   r   r   r   rJ   rJ   A   r:   r   rJ   c                   p    e Zd ZU  ed          Zeed<    ed          Zeed<    ed          Z	eed<   dS )TaskAudioContent	audio_urlr6   .reference_audiorA   N)
r   r   r   r   r6   r   r   rM   rJ   rA   r   r   r   rL   rL   E   rH   r   rL   c                       e Zd ZU  ed          Zeed<    edd          Zee	         ed<    ed          Z
edz  ed<   dS )Text2VideoTaskCreationRequest.r      
min_lengthcontentNgenerate_audio)r   r   r   r   r   r   r   rT   r,   r4   rU   r   r   r   r   rP   rP   K   sj         sE3%*U31%=%=%=GT/"==="'%**ND4K,,,,,r   rP   c                       e Zd ZU  ed          Zeed<    edd          Zee	e
z           ed<    ed          Zedz  ed<   dS )Image2VideoTaskCreationRequest.r      rR   rT   NrU   )r   r   r   r   r   r   r   rT   r,   r4   r<   rU   r   r   r   r   rW   rW   Q   sp         sE38=ca8P8P8PGT/$445PPP"'%**ND4K,,,,,r   rW   c                   b   e Zd ZU  ed          Zeed<    edd          Zee	e
z  ez  ez           ed<    ed          Zedz  ed<    ed          Zedz  ed<    ed          Zedz  ed	<    edd
d          Zedz  ed<    eddd          Zedz  ed<    ed          Zedz  ed<   dS )Seedance2TaskCreationRequest.r   rQ   rR   rT   NrU   
resolutionratio   r!   r   durationr   r   r   r   )r   r   r   r   r   r   r   rT   r,   r4   r<   rE   rL   rU   r   r[   r\   r^   r   r   r   r   r   r   rZ   rZ   W   s        sE3^c^cdgtu^v^v^vGT/$447GGJZZ[vvv"'%++ND4K---"U4[[Jd
(((dE3:### 5!333HcDj333uTaJ777D#*777"U4[[Itd{(((((r   rZ   c                   0    e Zd ZU  ed          Zeed<   dS )TaskCreationResponse.idN)r   r   r   r   ra   r   r   r   r   r   r`   r`   b   s*         eCjjBr   r`   c                   P    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   rd   r   r   re   r   r   r   rc   rc   f   sB         c

D#5::GSr   rc   c                   0    e Zd ZU  ed          Zeed<   dS )TaskStatusResult.rF   N)r   r   r   r   rF   r   r   r   r   r   rg   rg   k   s*         U3ZZIsr   rg   c                   P    e Zd ZU  ed          Zeed<    ed          Zeed<   dS )TaskStatusUsager   completion_tokenstotal_tokensN)r   r   r   r   rj   r   r   rk   r   r   r   ri   ri   o   sC         "U1XXs%%%aL#     r   ri   c                       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z  ed<    ed          Zedz  ed<    ed          Zedz  ed	<   dS )
TaskStatusResponse.ra   r   )queuedrunning	cancelled	succeededfailedstatusNr1   rT   usage)r   r   r   r   ra   r   r   r   rs   r   r1   rc   rT   rg   rt   ri   r   r   r   rm   rm   t   s         eCjjBsE3OTuUXzzFGKLYYY$)E$KKE?T!///',uT{{G$222$)E$KKE?T!/////r   rm   c                      e Zd ZU  ed          Zeed<    ed          Zedz  ed<    ed          Zedz  ed<    ed          Z	eed<    ed          Z
eed<    ed          Zeed<    ed          Zedz  ed	<   dS )
GetAssetResponse.ra   Nnamer
   
asset_typegroup_idrs   r1   )r   r   r   r   ra   r   r   rw   r
   rx   ry   rs   r1   rc   r   r   r   rv   rv   }   s         eCjjBuT{{D#*"""eDkkCt!!!eCjjJ   E#JJHc%**FC$)E$KKE?T!/////r   rv   c                   P    e Zd ZU  ed          Zeed<    ed          Zeed<   dS )+SeedanceCreateVisualValidateSessionResponse.
session_idh5_linkN)r   r   r   r   r|   r   r   r}   r   r   r   r{   r{      sB         eCjjJ   5::GSr   r{   c                       e Zd ZU  ed          Zeed<    ed          Zeed<    ed          Zedz  ed<    ed          Z	edz  ed<    ed          Z
edz  ed<   dS )(SeedanceGetVisualValidateSessionResponse.r|   rs   Nry   
error_codeerror_message)r   r   r   r   r|   r   r   rs   ry   r   r   r   r   r   r   r      s         eCjjJ   %**FC 5;;HcDj&&&"U4[[Jd
((( %dM3:+++++r   r   c                       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          Z	edz  ed<    ed          Z
edz  ed	<   dS )
SeedanceCreateAssetRequest.ry   r
   rx   N@   )
max_lengthrw   project_name)r   r   r   r   ry   r   r   r
   rx   rw   r   r   r   r   r   r      s         E#JJHcuSzzCeCjjJ   uTb111D#*111$uT{{L#******r   r   c                   0    e Zd ZU  ed          Zeed<   dS )SeedanceCreateAssetResponse.asset_idN)r   r   r   r   r   r   r   r   r   r   r   r      s*         E#JJHcr   r   c                       e Zd ZU  edd          Zeed<    edd          Zeed<    edd          Zedz  ed	<   dS )
(SeedanceVirtualLibraryCreateAssetRequest.z/Publicly accessible URL of the asset to upload.r%   r
   zEDedup key. Re-submitting the same hash returns the existing asset id.hashNz@BytePlus asset type. Defaults to Image server-side when omitted.rx   )	r   r   r   r   r
   r   r   r   rx   r   r   r   r   r      sv         uS&WXXXCXXXc'noooD#ooo"U45wxxxJd
xxxxxr   r   gy&1|?g"uq?gǺv?gF%uk?))dreamina-seedance-2-0-260128F)r   T)!dreamina-seedance-2-0-fast-260128F)r   T)
)z1024x1024 (1:1)   r   )z864x1152 (3:4)`    )z1152x864 (4:3)r   r   )z1280x720 (16:9)     )z720x1280 (9:16)r   r   )z832x1248 (2:3)@    )z1248x832 (3:2)r   r   )z1512x648 (21:9)i  i  z2048x2048 (1:1)   r   CustomNN)r   )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)   r   )z4096x4096 (1:1)   r   r   ))z(1K) 1024x1024 (1:1)r   r   )z(1K) 864x1152 (3:4)r   r   )z(1K) 1152x864 (4:3)r   r   )z(1K) 1312x736 (16:9)     )z(1K) 736x1312 (9:16)r   r   )z(1K) 832x1248 (2:3)r   r   )z(1K) 1248x832 (3:2)r   r   )z(1K) 1568x672 (21:9)i   i  ))z(2K) 2048x2048 (1:1)r   r   )z(2K) 1728x2304 (3:4)r   r   )z(2K) 2304x1728 (4:3)r   r   )z(2K) 2848x1600 (16:9)   @  )z(2K) 1600x2848 (9:16)r   r   )z(2K) 1664x2496 (2:3)r   r   )z(2K) 2496x1664 (3:2)r   r   )z(2K) 3136x1344 (21:9)i@  i@  ))z(3K) 3072x3072 (1:1)r   r   )z(3K) 2592x3456 (3:4) 
    )z(3K) 3456x2592 (4:3)r   r   )z(3K) 4096x2304 (16:9)r   r   )z(3K) 2304x4096 (9:16)r   r   )z(3K) 2496x3744 (2:3)r     )z(3K) 3744x2496 (3:2)r   r   )z(3K) 4704x2016 (21:9)`  i  ))z(4K) 4096x4096 (1:1)r   r   )z(4K) 3520x4704 (3:4)  r   )z(4K) 4704x3520 (4:3)r   r   )z(4K) 5504x3040 (16:9)    )z(4K) 3040x5504 (9:16)r   r   )z(4K) 3328x4992 (2:3)     )z(4K) 4992x3328 (3:2)r   r   )z(4K) 6240x2656 (21:9)i`  i`
  r   i @ i& )minmaxi  )480p720p1080p)r   r   )r   r   (   <   Z   F   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.   r4   r8   r<   rC   rE   rJ   rL   rP   rW   rZ   r`   rc   rg   ri   rm   rv   r{   r   r   r   r   SEEDANCE2_PRICE_PER_1K_TOKENSRECOMMENDED_PRESETSRECOMMENDED_PRESETS_SEEDREAM_4_PRESETS_SEEDREAM_1K_PRESETS_SEEDREAM_2K_PRESETS_SEEDREAM_3K_PRESETS_SEEDREAM_4K_CUSTOM_PRESET#RECOMMENDED_PRESETS_SEEDREAM_5_LITE RECOMMENDED_PRESETS_SEEDREAM_4_5 RECOMMENDED_PRESETS_SEEDREAM_4_0 SEEDANCE2_REF_VIDEO_PIXEL_LIMITSVIDEO_TASKS_EXECUTION_TIMEr   r   r   <module>r      s         % % % % % % % %* * * * *I * * *         y      
% 
% 
% 
% 
%9 
% 
% 
%b b b b b	 b b b    i   
    )   W W W W Wy W W W    )   ) ) ) ) )y ) ) )    )   ) ) ) ) )y ) ) )- - - - -I - - -- - - - -Y - - -) ) ) ) )9 ) ) )    9       i   
         y      ! ! ! ! !i ! ! !
0 0 0 0 0 0 0 00 0 0 0 0y 0 0 0    )   
, , , , ,y , , ,+ + + + + + + +    )   y y y y yy y y y .3,22817	! !    " " " 	 	 	 	 	 	 	 	 	 	 	 	  )) //2FFW $ //.@ ! //2FFW !  0000 33% %  0000* *
$ 
$   % % % %     % %    +    r   