
    +jg                        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d$S )%    )	BaseModelFieldc                   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 )MultiPromptEntry.indexpromptdurationN)
__name__
__module____qualname__r   r   int__annotations__r   strr	        :/home/wildlama/comfy/ComfyUI/comfy_api_nodes/apis/kling.pyr   r      sZ         sE3%**FC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ed<    ed          Z	eed	<    ed
          Z
eed<    ed          Zedz  ed<    ed          Zee         dz  ed<    ed          Zedz  ed<    edd          Zeed<   dS )OmniProText2VideoRequest.kling-video-o1description
model_name'16:9', '9:16' or '1:1'aspect_ratio'5' or '10'r	   r   promodeN
multi_shotmulti_prompt	shot_type'on' or 'off'sound)r
   r   r   r   r   r   r   r   r	   r   r   r   boolr   listr   r    r"   r   r   r   r   r   
   s        eC-=>>>J>>>c/HIIIL#IIIE#=999Hc999%**FCeD##eDkkJt)))27%++L$'(4/===!E$KKIsTz'''s888E388888r   r   c                   Z    e Zd ZU  ed          Zeed<    edd          Zedz  ed<   dS )OmniParamImage.	image_urlNz#Can be 'first_frame' or 'end_frame'r   type)r
   r   r   r   r'   r   r   r(   r   r   r   r&   r&      sN         U3ZZIsuT/TUUUD#*UUUUUr   r&   c                   ~    e Zd ZU  ed          Zeed<    edd          Zedz  ed<    edd          Zeed<   dS )	OmniParamVideo.	video_urlzCan be 'base' or 'feature'r   N
refer_typez'yes' or 'no'keep_original_sound)	r
   r   r   r   r+   r   r   r,   r-   r   r   r   r*   r*      sn         U3ZZIs"U34PQQQJd
QQQ$uSoFFFFFFFFr   r*   c                   r   e Zd ZU  edd          Zeed<    eddd          Zee	         ed<    ed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<    ed          Zee         dz  ed<    ed          Zedz  ed<   dS )OmniProFirstLastFrameRequest.r   r   r         )
min_length
max_length
image_listr   r	   r   r   r   Nr!   r"   r   r   r    )r
   r   r   r   r   r   r   r4   r$   r&   r	   r   r   r"   r   r#   r   r   r    r   r   r   r/   r/   !   s%        eC-=>>>J>>>',uSQ1'M'M'MJ^$MMME#=999Hc999%**FCeD#d@@@E3:@@@#eDkkJt)))27%++L$'(4/===!E$KKIsTz'''''r   r/   c                      e Zd ZU  edd          Zeed<    edd          Zedz  ed<    eddd	
          Ze	e
         dz  ed<    edd          Ze	e         dz  ed<    edd          Zedz  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<    ed          Ze	e         dz  ed<    ed          Zedz  ed<   dS )OmniProReferences2VideoRequest.r   r   r   r   Nr   r1   z#Max length 4 when video is present.)r3   r   r4   r0   r3   
video_listzFrom 3 to 10.r	   r   r   r   r!   r"   r   r   r    )r
   r   r   r   r   r   r   r   r4   r$   r&   r8   r*   r	   r   r   r"   r   r#   r   r   r    r   r   r   r6   r6   -   s        eC-=>>>J>>>$uS6OPPPL#*PPP.3e(M/ / /J^$t+    /4eDQ.G.G.GJ^$t+GGG 5/BBBHcDjBBB%**FCeD#d@@@E3:@@@#eDkkJt)))27%++L$'(4/===!E$KKIsTz'''''r   r6   c                       e Zd ZU  edd          Zedz  ed<    edd          Zedz  ed<    edd          Zedz  ed<   dS )	TaskStatusVideoResultNzTotal video durationr   r	   zGenerated video IDidzURL for generated videourl)	r
   r   r   r   r	   r   r   r;   r<   r   r   r   r:   r:   =   s~          53IJJJHcDjJJJU4-ABBBBd
BBBeD.GHHHCtHHHHHr   r:   c                   X    e Zd ZU  edd          Zeed<    edd          Zeed<   dS )TaskStatusImageResult.u   Image Number，0-9r   r   zURL for generated imager<   N)	r
   r   r   r   r   r   r   r<   r   r   r   r   r>   r>   C   sR         s(<===E3===uS&?@@@C@@@@@r   r>   c                       e Zd ZU  ed          Zee         dz  ed<    ed          Zee	         dz  ed<    ed          Z
ee	         dz  ed<   dS )TaskStatusResultsNvideosimagesseries_images)r
   r   r   r   rA   r$   r:   r   rB   r>   rC   r   r   r   r@   r@   H   s{         16tFD&'$.<<<16tFD&'$.<<<8=dM4-.5CCCCCr   r@   c                       e Zd ZU  edd          Zedz  ed<    edd          Zedz  ed<   dZe	dz  ed<    edd          Z
e	dz  ed	<    edd
          Ze	dz  ed<    ed          Zedz  ed<   dS )TaskStatusResponseDataNzTask creation timer   
created_atzTask update time
updated_attask_statusz5Additional failure reason. Only for polling endpoint.task_status_msgzTask IDtask_idtask_result)r
   r   r   r   rF   r   r   rG   rH   r   rI   rJ   rK   r@   r   r   r   rE   rE   N   s         "U45IJJJJd
JJJ"U45GHHHJd
HHH"Kt""""'%:q"r"r"rOS4Zrrr%)<<<GS4Z<<<,1E$KKK"T)77777r   rE   c                       e Zd ZU  edd          Zedz  ed<    edd          Zedz  ed<    edd          Z	edz  ed<    ed          Z
edz  ed	<   dS )
TaskStatusResponseNz
Error coder   codezError messagemessagez
Request ID
request_iddata)r
   r   r   r   rN   r   r   rO   r   rP   rQ   rE   r   r   r   rM   rM   W   s         uT|<<<D#*<<<%/BBBGS4ZBBB"U4\BBBJd
BBB*/%++D
 4
'55555r   rM   c                   0    e Zd ZU  ed          Zeed<   dS )OmniImageParamImage.imageN)r
   r   r   r   rT   r   r   r   r   r   rS   rS   ^   s*         sE3r   rS   c                   n   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ed<    ed          Z
eed<    ed	d
          Zedz  ed<    edd          Zee         dz  ed<    edd          Zedz  ed<    eddd
d          Zedz  ed<   dS )OmniProImageRequest.r   
resolutionNr   r   r   r   r0   	   )len
   r7   r4   z%Set to 'series' for series generationr   result_type   zNumber of images in a series)gerY   r   series_amount)r
   r   r   r   r   r   r   rW   r   r   r   rZ   r   r4   r$   rS   r\   r_   r   r   r   rV   rV   b   s$        eCjjJ   eCjjJ   $uSzzL#*)))%**FCeD#E!NNNAsTz"""3853L3L3LJ()D0LLL#eD6]^^^Kt^^^ %dqQDb c c cM3:cccccr   rV   c                      e Zd ZU  ed          Zeed<    ed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ed<    edd          Zeed<    ed          Zedz  ed<    ed          Zee         dz  ed<    ed          Zedz  ed<   dS )TextToVideoWithAudioRequest.r   r   r   r   r	   Nr   negative_promptr   r   r!   r"   r   r   r    )r
   r   r   r   r   r   r   r   r	   r   rb   r   r"   r   r#   r   r$   r   r    r   r   r   ra   ra   n   s,        eCjjJ   c/HIIIL#IIIE#JJHcsFC$J###"'%++OS4Z---eD#s888E3888#eDkkJt)))27%++L$'(4/===!E$KKIsTz'''''r   ra   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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d          Zeed<    ed          Zedz  ed<    ed          Zee         dz  ed<    ed          Zedz  ed<   dS )ImageToVideoWithAudioRequest.r   rT   N
image_tailr	   r   rb   r   r   r!   r   r"   r   r   r    )r
   r   r   r   r   r   r   rT   re   r	   r   rb   r   r"   r   r#   r   r$   r   r    r   r   r   rd   rd   {   s@        eCjjJ   sE3"U4[[Jd
(((E#JJHcsFC$J###"'%++OS4Z---eD#s888E3888#eDkkJt)))27%++L$'(4/===!E$KKIsTz'''''r   rd   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ed<   dS )KlingAvatarRequest.rT   
sound_fileNr   r   )
r
   r   r   r   rT   r   r   rh   r   r   r   r   r   rg   rg      sv         sE3eCjjJ   tFC$J$$$c

D#r   rg   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          Z	eed<    ed          Z
eed<    edd          Zeed	<    ed          Zeed
<   dS )MotionControlRequest.r   r'   r+   r-   character_orientationz'pro' or 'std'r   r   r   N)r
   r   r   r   r   r   r   r'   r+   r-   rk   r   r   r   r   r   rj   rj      s         %**FCU3ZZIsU3ZZIs$uSzz)))!&s3+++c'7888D#888eCjjJ     r   rj   N)pydanticr   r   r   r   r&   r*   r/   r6   r:   r>   r@   rE   rM   rS   rV   ra   rd   rg   rj   r   r   r   <module>rm      s!   % % % % % % % %    y   	9 	9 	9 	9 	9y 	9 	9 	9V V V V VY V V V
G G G G GY G G G	( 	( 	( 	( 	(9 	( 	( 	(( ( ( ( (Y ( ( ( I I I I II I I IA A A A AI A A A
D D D D D	 D D D8 8 8 8 8Y 8 8 86 6 6 6 6 6 6 6    )   	d 	d 	d 	d 	d) 	d 	d 	d
( 
( 
( 
( 
() 
( 
( 
(( ( ( ( (9 ( ( (       ! ! ! ! !9 ! ! ! ! !r   