o $iD @sddlZddlZddlmZddlmZddlmZmZm Z m Z ddl Z ddl m Z ddlmZeeZdeddfd d Zd efd d ZdefddZdeedeefddZdeedeefddZdedefddZdedefddZde eefde eeffddZd e eeeefde efd!d"Zd#e eeeefde efd$d%Zd&eedeefd'd(Zd)eedeefd*d+Z d,eeefde eeeffd-d.Z!eee eeee!ed/Z"eed0Z#dS)1N) OrderedDict)Path)DictListOptionalUnion)is_path parse_pathpathreturncCs t|dS)z6Parse the path to ensure it is well-formed and exists.Nr )r r `/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/ray/_private/runtime_env/validation.py validate_paths ruricCsrzddlm}m}||\}}Wntyt|dw||vr3|ds5|ds7tddSdSdS)Nr)Protocol parse_uriz is not a valid URI. Passing directories or modules to be dynamically uploaded is only supported at the job level (i.e., passed to `ray.init`).z.zipz.whlz2Only .zip or .whl files supported for remote URIs.)"ray._private.runtime_env.packagingrr ValueErrorremote_protocolsendswith)rrrprotocolr r r r validate_uris   rrequirements_filecCs0t|}|st|d|dS)zIRead the given [requirements_file], and return all required dependencies.z is not a valid file )ris_filer read_textstripsplit)rrequirements_pathr r r#_handle_local_deps_requirement_file(sr py_modules_uriscCsRt|tstdt|d|D]}t|ts"tdt|dt|qdS)z^Parses and validates a 'py_modules' option. Expects py_modules to be a list of URIs. ,`py_modules` must be a list of strings, got ."`py_module` must be a string, got N) isinstancelist TypeErrortypestrr)r!moduler r rvalidate_py_modules_uris0s   r+ py_modulescCsdt|tstdt|d|D]}t|ts"tdt|dt|r+t|qt|q|S)zmParses and validates a 'py_modules' option. Expects py_modules to be a list of local paths or URIs. r"r#r$)r%r&r'r(r)rrr)r,r*r r rparse_and_validate_py_modulesBs    r-working_dir_uricCs*t|tstdt|dt|dS)z,Parses and validates a 'working_dir' option.$`working_dir` must be a string, got r#N)r%r)r'r(r)r.r r rvalidate_working_dir_uriYs  r0 working_dircCsJ|dusJt|tstdt|dt|rt||St||S)zSParses and validates a 'working_dir' option. This can be a URI or a path. Nr/r#)r%r)r'r(rrr)r1r r rparse_and_validate_working_dircs  r2condac Cs|dusJtjdkrtd|}t|trdt|}|jdvrN|s,t d|dz t | }W|St yM}z t d|d|dd}~ww|rb|s^t d |dt|}|St|trm|}|Std t|d) aParses and validates a user-provided 'conda' option. Conda can be one of three cases: 1) A dictionary describing the env. This is passed through directly. 2) A string referring to the name of a preinstalled conda env. 3) A string pointing to a local conda YAML file. This is detected by looking for a '.yaml' or '.yml' suffix. In this case, the file will be read as YAML and passed through as a dictionary. Nwin32runtime environment support is experimental on Windows. If you run into issues please file a report at https://github.com/ray-project/ray/issues.)z.yamlz.ymlzCan't find conda YAML file r#zFailed to read conda file z: zCan't find conda env directory z6runtime_env['conda'] must be of type str or dict, got )sysplatformloggerwarningr%r)rsuffixrryaml safe_loadr Exception is_absoluteis_dirdictr'r()r3result file_pather r rparse_and_validate_condaus:    rDuvcCs |dusJtjdkrtdd}t|tr"t|}t|dd}nt|tr7t dd|Dr7t|dd}nt|trt | hd rQt d t| d |vr\t d |d |vrrt|d t srtdt|d d|vrt|dtstdt|dd|vrt|dtstdt|dt|dD]\}}t|tstdt|d|dq|}|d d|d <|ddg|d<t|d tst dt|d n tdt|tt|d |d <t|d dkrd}td|d|d|S)uHParses and validates a user-provided 'uv' option. The value of the input 'uv' field can be one of two cases: 1) A List[str] describing the requirements. This is passed through. Example usage: ["tensorflow", "requests"] 2) a string containing the path to a local pip “requirements.txt” file. 3) A python dictionary that has one field: a) packages (required, List[str]): a list of uv packages, it same as 1). b) uv_check (optional, bool): whether to enable pip check at the end of uv install, default to False. c) uv_version (optional, str): user provides a specific uv to use; if unspecified, default version of uv will be used. d) uv_pip_install_options (optional, List[str]): user-provided options for `uv pip install` command, default to ["--no-cache"]. The returned parsed value will be a list of packages. If a Ray library (e.g. "ray[serve]") is specified, it will be deleted and replaced by its dependencies (e.g. "uvicorn", "requests"). Nr4r5F)packagesuv_checkcs|]}t|tVqdSNr%r).0depr r r z(parse_and_validate_uv..>rGrH uv_versionuv_pip_install_optionszrruntime_env['uv'] can only have these fields: packages, uv_check, uv_version and uv_pip_install_options, but got: rGz9runtime_env['uv'] must include field 'packages', but got rHz8runtime_env['uv']['uv_check'] must be of type bool, got rQz9runtime_env['uv']['uv_version'] must be of type str, got rRzJruntime_env['uv']['uv_pip_install_options'] must be of type list[str] got  for -th item.z --no-cachez9runtime_env['uv']['packages'] must be of type list, got: z:runtime_env['uv'] must be of type List[str], or dict, got rz$Rewrote runtime_env `uv` field from  to r#r6r7r8r9r%r)r r@r&allsetkeysrboolr'r( enumeratecopygetrfromkeyslendebug)rErAuv_listidxcur_optr r rparse_and_validate_uvs           rdpipcCs.|dusJd}tjdkrtdt|tr"t|}t|dd}nt|tr7t dd|Dr7t|dd}nt|trt | hdrQt d t| d |vrgt|d t sgtd t|d d |vr}t|d ts}td t|d d|vrt|dtstdt|dt|dD]\}}t|tstdt|d|dq|}|d d|d <d|vrt d|t|dtrt|d|d<nt|dtst dt|dn tdt|tt|d|d<t|ddkr d}td|d|d|S)aParses and validates a user-provided 'pip' option. The value of the input 'pip' field can be one of two cases: 1) A List[str] describing the requirements. This is passed through. 2) A string pointing to a local requirements file. In this case, the file contents will be read split into a list. 3) A python dictionary that has three fields: a) packages (required, List[str]): a list of pip packages, it same as 1). b) pip_check (optional, bool): whether to enable pip check at the end of pip install, default to False. c) pip_version (optional, str): the version of pip, ray will spell the package name 'pip' in front of the `pip_version` to form the final requirement string, the syntax of a requirement specifier is defined in full in PEP 508. d) pip_install_options (optional, List[str]): user-provided options for `pip install` command, defaults to ["--disable-pip-version-check", "--no-cache-dir"]. The returned parsed value will be a list of pip packages. If a Ray library (e.g. "ray[serve]") is specified, it will be deleted and replaced by its dependencies (e.g. "uvicorn", "requests"). Nr4r5F)rG pip_checkcsrIrJrKrLr r rrO(rPz)parse_and_validate_pip..>rGrf pip_versionpip_install_optionszrruntime_env['pip'] can only have these fields: packages, pip_check, pip_install_options and pip_version, but got: rfz:runtime_env['pip']['pip_check'] must be of type bool, got rgz;runtime_env['pip']['pip_version'] must be of type str, got rhzHruntime_env['pip']['pip_install_options'] must be of type list[str] got rSrTrGz:runtime_env['pip'] must include field 'packages', but got zAruntime_env['pip']['packages'] must be of type str of list, got: z9runtime_env['pip'] must be of type str or List[str], got rz%Rewrote runtime_env `pip` field from rUr#rV)rerApip_listrbrcr r rparse_and_validate_pips           rj containercCs|dusJ|S)zwParses and validates a user-provided 'container' option. This is passed through without validation (for now). Nr )rkr r rparse_and_validate_containerss rlexcludescCsX|dusJt|trt|dkrdSt|tr#tdd|Dr#|Stdt|)zParses and validates a user-provided 'excludes' option. This is validated to verify that it is of type List[str]. If an empty list is passed, we return `None` for consistency. NrcsrIrJrK)rMr r r rrOrPz.parse_and_validate_excludes..z7runtime_env['excludes'] must be of type List[str], got )r%r&r_rWr'r()rmr r rparse_and_validate_excludes|s rnenv_varscCs|dusJt|dkrdSt|tstdt||D]&\}}t|ts5td|dt|t|tsFtd|dt|q |S)a(Parses and validates a user-provided 'env_vars' option. This is validated to verify that all keys and vals are strings. If an empty dictionary is passed, we return `None` for consistency. Args: env_vars: A dictionary of environment variables to set in the runtime environment. Returns: The validated env_vars dictionary, or None if it was empty. Raises: TypeError: If the env_vars is not a dictionary of strings. The error message will include the type of the invalid value. NrzsD      "$+$bq &.