
    +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dS )    )Optional)	BaseModelFieldc                       e Zd ZU  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 )VeoRequestInstanceImageNbytesBase64EncodedgcsUrimimeType)	__name__
__module____qualname__r   r   str__annotations__r	   r
        8/home/wildlama/comfy/ComfyUI/comfy_api_nodes/apis/veo.pyr   r      sg         %*U4[[d
000tFC$J$$$ 5;;HcDj&&&&&r   r   c                       e Zd ZU  ed          Zedz  ed<    ed          Zedz  ed<    edd          Ze	ed<   dS )VeoRequestInstanceNimage	lastFrame.zText description of the videodescriptionprompt)
r   r   r   r   r   r   r   r   r   r   r   r   r   r   r      sn         ,1E$KKE"T)77705dI&-;;;%)HIIIFCIIIIIr   r   c                   t   e Zd ZU  eddg          Zee         ed<   dZee	         ed<   dZ
ee         ed<    edd          Zee         ed	<   dZee         ed
<    edd          Zedz  ed<   dZee	         ed<   dZee	         ed<    edd          Zee         ed<    ed          Zedz  ed<   dS )VeoRequestParametersNz16:9)examplesaspectRatiodurationSecondsenhancePromptz=Generate audio for the video. Only supported by veo 3 models.r   generateAudionegativePromptzALLOW or BLOCKpersonGenerationsampleCountseedz.Optional Cloud Storage URI to upload the video
storageUri
resolution)r   r   r   r   r   r   r   r   r   intr   boolr    r!   r"   r#   r$   r%   r&   r   r   r   r   r      sB        !&tvh!?!?!?K#???%)OXc])))$(M8D>((($)ES% % %M8D>    %)NHSM(((#(5;K#L#L#LcDjLLL!%K#%%%D(3- %J! ! !J    #U4[[Jd
(((((r   r   c                   h    e Zd ZU  ed          Zee         dz  ed<    ed          Ze	dz  ed<   dS )VeoGenVidRequestN	instances
parameters)
r   r   r   r   r+   listr   r   r,   r   r   r   r   r*   r*   $   sS         16tIt&'$.<<<.3eDkkJ$t+99999r   r*   c                   8    e Zd ZU  edddg          Zeed<   dS )VeoGenVidResponse.zOperation resource namez{projects/PROJECT_ID/locations/us-central1/publishers/google/models/MODEL_ID/operations/a1b07c8e-7b5a-4aba-bb34-3e1ccb8afcc8r   r   nameN)r   r   r   r   r1   r   r   r   r   r   r/   r/   )   sN         - J
  D#     r   r/   c                   8    e Zd ZU  edddg          Zeed<   dS )VeoGenVidPollRequest.z+Full operation name (from predict response)zcprojects/PROJECT_ID/locations/us-central1/publishers/google/models/MODEL_ID/operations/OPERATION_IDr0   operationNameN)r   r   r   r   r4   r   r   r   r   r   r3   r3   3   sK         Aq
  M3     r   r3   c                       e Zd ZU  edd          Zee         ed<    edd          Zee         ed<    edd          Z	ee         ed<   dS )	VideoNzBase64-encoded video contentr   r   zCloud Storage URI of the videor	   zVideo MIME typer
   )
r   r   r   r   r   r   r   r   r	   r
   r   r   r   r6   r6   =   s         (-8) ) )    "E$4TUUUFHSMUUU#eD6GHHHHhsmHHHHHr   r6   c                   p    e Zd ZU  edd          Zee         ed<    edd          Zee	         ed<   dS )Error1Nz
Error coder   codezError messagemessage)
r   r   r   r   r9   r   r'   r   r:   r   r   r   r   r8   r8   E   sX         %,???D(3-???"U4_EEEGXc]EEEEEr   r8   c                       e Zd ZU  edddg          Zee         ed<    edd          Zee	         ed<    edd	          Z
eee                  ed
<    ed          Zeee                  ed<   dS )	Response1Nz@typezFtype.googleapis.com/cloud.ai.large_models.vision.GenerateVideoResponse)aliasr   
field_typez2Count of media filtered by responsible AI policiesr   raiMediaFilteredCountz9Reasons why media was filtered by responsible AI policiesraiMediaFilteredReasonsvideos)r   r   r   r   r>   r   r   r   r?   r'   r@   r-   rA   r6   r   r   r   r<   r<   J   s          %T
! ! !J    ,15N, , ,8C=    495U4 4 4Xd3i0    %*E$KKFHT%[!/////r   r<   c                       e Zd ZU dZee         ed<    edd          Zee	         ed<   dZ
ee         ed<    edd          Zee         ed<   dS )	VeoGenVidPollResponseNdonez!Error details if operation failedr   errorr1   z.The actual prediction response if done is trueresponse)r   r   r   rD   r   r(   r   r   rE   r8   r1   r   rF   r<   r   r   r   rC   rC   [   s         D(4.#e=  E8F    D(3-$)EJ% % %Hhy!     r   rC   N)typingr   pydanticr   r   r   r   r   r*   r/   r3   r6   r8   r<   rC   r   r   r   <module>rI      s         % % % % % % % %' ' ' ' 'i ' ' 'J J J J J J J J) ) ) ) )9 ) ) )$: : : : :y : : :
    	       9   I I I I II I I IF F F F FY F F F
0 0 0 0 0	 0 0 0"    I     r   