
    <                        S r SSKJr  SSKJr  SSKJr  SSKJr  SSKJr  SSK	J
r
  SSK	Jr  SS	K	Jr  SS
K	Jr  SS	KJr  SSKJr  \R$                  " \R&                  R(                  5       " S S\R*                  5      5       r\R$                  " \R&                  R.                  \R&                  R0                  5       " S S\5      5       rg)z:Command to check stream logs of a custom job in Vertex AI.    )absolute_import)division)unicode_literals)client)base)	constants)endpoint_util)flags)log_util)
validationc                   F    \ rS rSrSr\R                  r\S 5       r	S r
Srg)StreamLogsGA   Show stream logs from a running custom job.

## EXAMPLES

To stream logs of custom job ``123'' under project ``example'' in region
``us-central1'', run:

  $ {command} 123 --project=example --region=us-central1
c                     [         R                  " U S5        [        R                  " U 5        U R                  R                  [        R                  5        g )Nzto fetch stream log)custom_job_flagsAddCustomJobResourceArgcommon_flagsAddStreamLogsFlagsdisplay_info	AddFormatr   
LOG_FORMAT)parsers    )lib/surface/ai/custom_jobs/stream_logs.pyArgsStreamLogsGA.Args-   s;    ,,V5JK##F+
!!("5"56    c           	         UR                   R                  R                  5       nUR                  5       S   n[        R
                  " U5        [        R                  " U R                  US9   [        R                  " UR                  5       S   [        R                  " U R                  S9R                  UR                  5       5      UR                  UR                   UR"                  S9sS S S 5        $ ! , (       d  f       g = f)NlocationsId)versionregioncustomJobsId)r    )continue_functionpolling_interval	task_nameallow_multiline)CONCEPTS
custom_jobParseAsDictr   ValidateRegionr	   AiplatformEndpointOverrides_api_versionr   
StreamLogsr   CustomJobsClientCheckJobCompleteRelativeNamer$   r%   allow_multiline_logs)selfargscustom_job_refr!   s       r   RunStreamLogsGA.Run3   s    ]]--335N""$]3Ff%		2	2!!&
2  



!.
1"33''))9)9 --/*100NN335
2 
2 
2s   -B C77
D N)__name__
__module____qualname____firstlineno____doc__r   
GA_VERSIONr-   staticmethodr   r6   __static_attributes__r8   r   r   r   r      s+     %%,7 7
5r   r   c                   0    \ rS rSrSr\R                  rSrg)StreamLogsPreGAD   r   r8   N)	r9   r:   r;   r<   r=   r   BETA_VERSIONr-   r@   r8   r   r   rB   rB   D   s     '',r   rB   N)r=   
__future__r   r   r   %googlecloudsdk.api_lib.ai.custom_jobsr   googlecloudsdk.callioper   googlecloudsdk.command_lib.air   r	   r
   r   r   )googlecloudsdk.command_lib.ai.custom_jobsr   r   ReleaseTracksReleaseTrackGACommandr   BETAALPHArB   r8   r   r   <module>rP      s    A &  ' 8 ( 3 7 ? 2 O @ D%%(()!54<< !5 *!5H D%%**D,=,=,C,CD
(l 
( E
(r   