
    +j                         S SK Jr  S SKJrJr   " S S\5      r " S S\5      r " S S\5      r " S	 S
\5      r " S S\5      r	 " S S\5      r
 " S S\5      r " S S\5      r " S S\5      r " S S\5      rg)    )Optional)	BaseModelFieldc                   v    \ rS rSr% \" S5      r\S-  \S'   \" S5      r\S-  \S'   \" S5      r	\S-  \S'   Sr
g)VeoRequestInstanceImage   NbytesBase64EncodedgcsUrimimeType )__name__
__module____qualname____firstlineno__r   r	   str__annotations__r
   r   __static_attributes__r       8/home/wildlama/comfy/ComfyUI/comfy_api_nodes/apis/veo.pyr   r      s9    %*4[d
0tFC$J$ ;HcDj&r   r   c                   n    \ rS rSr% \" S5      r\S-  \S'   \" S5      r\S-  \S'   \" SSS9r	\
\S'   S	rg)
VeoRequestInstance   Nimage	lastFrame.zText description of the videodescriptionpromptr   )r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r      s<    ,1$KE"T)705dI&-;)HIFCIr   r   c                      \ rS rSr% \" SS/S9r\\   \S'   Sr	\\
   \S'   Sr\\   \S'   \" SSS	9r\\   \S
'   Sr\\   \S'   \" SSS	9r\S-  \S'   Sr\\
   \S'   Sr\\
   \S'   \" SSS	9r\\   \S'   \" S5      r\S-  \S'   Srg)VeoRequestParameters   Nz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
resolutionr   )r   r   r   r   r   r"   r   r   r   r#   intr$   boolr%   r&   r'   r(   r)   r*   r+   r   r   r   r   r   r      s    !&tvh!?K#?%)OXc])$(M8D>($)S%M8D>  %)NHSM(#(;K#LcDjL!%K#%D(3- %J!J  #4[Jd
(r   r   c                   \    \ rS rSr% \" S5      r\\   S-  \S'   \" S5      r	\
S-  \S'   Srg)VeoGenVidRequest$   N	instances
parametersr   )r   r   r   r   r   r1   listr   r   r2   r   r   r   r   r   r/   r/   $   s/    16tIt&'$.<.3DkJ$t+9r   r/   c                   2    \ rS rSr% \" SSS/S9r\\S'   Srg)	VeoGenVidResponse)   .zOperation resource namez{projects/PROJECT_ID/locations/us-central1/publishers/google/models/MODEL_ID/operations/a1b07c8e-7b5a-4aba-bb34-3e1ccb8afcc8r   r!   namer   N)	r   r   r   r   r   r8   r   r   r   r   r   r   r5   r5   )   s$    - J
D# r   r5   c                   2    \ rS rSr% \" SSS/S9r\\S'   Srg)	VeoGenVidPollRequest3   .z+Full operation name (from predict response)zcprojects/PROJECT_ID/locations/us-central1/publishers/google/models/MODEL_ID/operations/OPERATION_IDr7   operationNamer   N)	r   r   r   r   r   r<   r   r   r   r   r   r   r:   r:   3   s!    Aq
M3 r   r:   c                   p    \ rS rSr% \" SSS9r\\   \S'   \" SSS9r	\\   \S'   \" SSS9r
\\   \S	'   S
rg)Video=   Nz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   r   r   r   r>   r>   =   sL    (-8)  "$4TUFHSMU#D6GHHhsmHr   r>   c                   R    \ rS rSr% \" SSS9r\\   \S'   \" SSS9r	\\
   \S'   Srg)	Error1E   Nz
Error coder   codezError messagemessager   )r   r   r   r   r   rC   r   r,   r   rD   r   r   r   r   r   rA   rA   E   s,    ,?D(3-?"4_EGXc]Er   rA   c                       \ rS rSr% \" SSS/S9r\\   \S'   \" SSS9r	\\
   \S	'   \" SS
S9r\\\      \S'   \" S5      r\\\      \S'   Srg)	Response1J   Nz@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videosr   )r   r   r   r   r   rI   r   r   r   rJ   r,   rK   r3   rL   r>   r   r   r   r   rF   rF   J   s     %T
!J  ,1N,8C=  49U4Xd3i0  %*$KFHT%[!/r   rF   c                   z    \ rS rSr% Sr\\   \S'   \" SSS9r	\\
   \S'   Sr\\   \S'   \" SSS9r\\   \S	'   S
rg)VeoGenVidPollResponse[   Ndonez!Error details if operation failedr   errorr8   z.The actual prediction response if done is trueresponser   )r   r   r   r   rP   r   r-   r   r   rQ   rA   r8   r   rR   rF   r   r   r   r   rN   rN   [   sX    D(4.#=E8F  D(3-$)J%Hhy! r   rN   N)typingr   pydanticr   r   r   r   r   r/   r5   r:   r>   rA   rF   rN   r   r   r   <module>rU      s     %'i 'J J)9 )$:y :
	 9 II IFY F
0	 0"I r   