o 1 iW>0@sHddlZddlZddlZddlZddlZddlmZddlmZm Z m Z m Z m Z ddl Z ddlmmZddlmZmZddlmZmZddlmZmZmZddlmZddlmZdd lm Z dd l!m"Z"m#Z#dd l$m%Z%  dade e&de'de e&de e'e&fde#f ddZ(de e&de e e&effddZ)ddZ*ddZ+de#de&de"fddZ,de#de&de"fdd Z-e .d!d"d#Z/e/0e j1d$e&dd d%d&e j1d'e&dd d(d&e j1d)e&dd d*d&e j1d+e&dd d,d&e j1d-e&dd d.d&e j1d/e&dd d0d&e j1d1e&dd d2d&e j1d3d e2d4d5e j1d6d e2d7d5e j1d8d e3d9d5e j1d:d e&d;d5e j1deee j4d?d@d e j5dAe%de e&de e&dBe e&dCe e&dDe e&dEe e&dFe e&d?e e&dGe e e3e2fdHe e e3e2fdIe e3dJe e&dKe'de e'e&fde e&fdLdMZ6e/0e j1d$e&dd d%d&e j4dNe&dOeee%dPdQde e&de&de e&de e'e&ffdRdSZ7e/0e j1d$e&dd d%d&e j1de j4dNe&dOeee%dPdQde e&dKe'de&de e&de e'e&ff dUdVZ8e/0e j1d$e&dd d%d&e j4dNe&dOeee%dPdQde e&de&de e&de e'e&ffdWdXZ9e/0e j1d$e&dd d%d&e j4dNe&dOe j1dYdZd e'd d[d>eee%dPdQde e&de&d\e'de e&de e'e&ff d]d^Z:e/0e j1d$e&dd d%d&eee%dPdQde e&de e&de e'e&ffd_d`Z;dS)bN) list2cmdline)AnyDictOptionalTupleUnion)get_or_create_event_loop load_class)parse_metadata_jsonparse_resources_json)add_click_logging_optionscf cli_logger)parse_runtime_env_args)add_common_job_options)redact_url_password) JobStatusJobSubmissionClient) PublicAPIFTaddresscreate_cluster_if_neededheadersverifyreturncCs0t||t||d}|}tdt||S)NrrzJob submission server address)r_handle_headersZ get_addressrZ labeled_valuer)rrrrclientclient_addressri/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/ray/dashboard/modules/job/cli.py_get_sdk_clientsr c Cs^|durdtjvrtjd}|dur-zt|WSty,}ztd||d}~wwdS)NZRAY_JOB_HEADERSzbFailed to parse headers into JSON. Expected format: {{"KEY": "VALUE"}}, got {}, {})osenvironjsonloads Exception ValueErrorformat)rexcrrrr/s  rcCBttdt|t|tdt|tdSN-)rnewlinesuccesslenZ success_msgrrr_log_big_success_msg?   r0cCr)r*)rr,errorr.r/rrr_log_big_error_msgGr1r3rjob_idcCs||}|jtjkrtd|d|jS|jtjkr(td|d|jS|jtjkrIt d|d|j durFtj d|j dd|jSt d|d |j|j duretj d|j dd|jS) NJob 'z ' succeededz ' was stoppedz' failedzStatus message: T no_formatzStatus for job 'z': ) Z get_job_infostatusr SUCCEEDEDr0STOPPEDrwarningFAILEDr3messageprint)rr4inforrr_log_job_statusOs      r@cs2||2z 3dHW}t|ddq6t||S)N)end)Z tail_job_logsr>r@)rr4linesrrr _tail_logsas  rDZjobcCsdS)z'Submit, stop, delete, or list Ray jobs.Nrrrrr job_cli_grouphsrEz --addresszAddress of the Ray cluster to connect to. Can also be specified using the RAY_API_SERVER_ADDRESS environment variable (falls back to RAY_ADDRESS).)typedefaultrequiredhelpz--job-idz*DEPRECATED: Use `--submission-id` instead.z--submission-idzMSubmission ID to specify for the job. If not provided, one will be generated.z --runtime-envz>Path to a local YAML file containing a runtime_env definition.z--runtime-env-jsonz'JSON-serialized runtime_env dictionary.z --working-dirzDirectory containing files that your job will run in. Can be a local directory or a remote URI to a .zip file (S3, GS, HTTP). If specified, this overrides the option in `--runtime-env`.z--metadata-jsonzZindentedboldr,flush get_versionintrrun_until_completerDrr<sysexit)rr4rMrNrOrPrQrJrRrSrTrUrVrrrZfinal_runtime_env sdk_versionZ job_statusrrrsubmitns~          rczjob-id)rFZstable)Z stabilitycCst|||d}t||dS)z`Queries for the current status of a job. Example: `ray job status ` rN)r r@rr4rrrrrrr8Hsr8z*If set, will not wait for the job to exit.cCst|||d}td|d|||rdStd|d ||}|tjtjtjhvr9t ||dStd|t d q#) zMAttempts to stop a job. Example: `ray job stop ` rzAttempting to stop job ''NzWaiting for job 'z#' to exit (disable with --no-wait):Tz Job has not exited yet. Status: rX) r rr>Zstop_jobZget_job_statusrr:r9r<r@timesleep)rrVr4rrrr8rrrstopfs "     rhcCs.t|||d}||td|ddS)aDeletes a stopped job and its associated data from memory. Only supported for jobs that are already in a terminal state. Fails with exit code 1 if the job is not already stopped. Does not delete job logs from disk. Submitting a job with the same submission ID as a previously deleted job is not supported and may lead to unexpected behavior. Example: ray job delete rr5z' deleted successfullyN)r Z delete_jobrr>rdrrrdeletes riz-fz--followz)If set, follow the logs (like `tail -f`).followcCslt|||d}|}|r)t|dkrtt||dStd|ddStj| |ddddS) zMGets the logs of a job. Example: `ray job logs ` rrzZ get_job_logs)rr4rjrrrrbrrrlogss# rkcCs*t|||d}tjt|dddS)zWLists all running jobs and their information. Example: `ray job list` rTr6N)r rr>pprintpformatZ list_jobs)rrrrrrrlistsrn)FNT)s       "          ,u     !     4