< dZddlmZddlmZddlmZddlmZddlmZddl m Z ddl m Z dd l m Z dd l mZdd lm Zdd lmZej$ej&j(Gd d ej*Zej$ej&j.ej&j0GddeZy)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>eZdZdZej ZedZdZ y) StreamLogsGAShow 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 ctj|dtj||jj t jy)Nzto fetch stream log)custom_job_flagsAddCustomJobResourceArg common_flagsAddStreamLogsFlags display_info AddFormatr LOG_FORMAT)parsers )lib/surface/ai/custom_jobs/stream_logs.pyArgszStreamLogsGA.Args-s;,,V5JK##F+ !!("5"56c|jjj}|jd}t j |t j|j|5tj|jdtj|jj|j|j|j |j"cdddS#1swYyxYw)N locationsId)versionregion customJobsId)r)continue_functionpolling_interval task_nameallow_multiline)CONCEPTS custom_jobParseAsDictr ValidateRegionr AiplatformEndpointOverrides _api_versionr StreamLogsrCustomJobsClientCheckJobComplete RelativeNamer"r#allow_multiline_logs)selfargscustom_job_refrs rRunzStreamLogsGA.Run3s]]--335N  " " $] 3Ff%  2 2!!& 2     !. 1"33''))9)9 --/*100NN335 2 2 2s .BC<<DN) __name__ __module__ __qualname____doc__r GA_VERSIONr+ staticmethodrr4rrrrs+%%,77 5rrc(eZdZdZej Zy)StreamLogsPreGArN)r5r6r7r8r BETA_VERSIONr+r;rrr=r=Ds'',rr=N)r8 __future__rrr%googlecloudsdk.api_lib.ai.custom_jobsrgooglecloudsdk.calliopergooglecloudsdk.command_lib.airr r rr )googlecloudsdk.command_lib.ai.custom_jobsrr ReleaseTracks ReleaseTrackGACommandrBETAALPHAr=r;rrrJsA&'8(37?2O@D%%(()!54<<!5*!5HD%%**D,=,=,C,CD (l (E (r