&`i89 ddlZddlZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z m Z m Z mZddlZddlmZddlZddlmZmZddlmZmZmZmZmZddlmZddlmZm Z dd l!m"Z"dd l#m$Z$dd l%m&Z&m'Z'm(Z(m)Z)ej*e+Z,ej-d kZ.d e/fdZ0dZ1dZ2e,fdeej3fdZ4dZ5e,fdeej3d ee/fdZ6 d de e e fde/dee e/d e e e ffdZ7de d e/fdZ8de d ee/fdZ9e,fdddeej3d e e/e ffdZ:Gdde"Z;dS)!N)Path)AnyDictListOptional)FileLock)get_or_create_event_looptry_to_create_directory)create_conda_env_if_neededdelete_conda_envget_conda_activate_commandsget_conda_envsget_conda_info_json)RuntimeEnvContext)Protocol parse_uri)RuntimeEnvPlugin)parse_and_validate_conda)get_directory_size_bytesget_master_wheel_urlget_release_wheel_urlget_wheel_filenamentreturnctjtjtjddS)Nr)ospathsplitray__file__r/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/_private/runtime_env/conda.py_resolve_current_ray_pathr$+s2 7==s|44Q7 8 8 ;;r"ct}tj|d}t j|dS)zFind the Ray setup_spec from the currently running Ray. This function works even when Ray is built from source with pip install -e. zsetup.py setup_spec)r$rrjoinrunpyrun_path)ray_source_python_path setup_py_paths r#_get_ray_setup_specr,5s; 788GLL!7DDM >- ( ( 66r"ctjtjdz}tt |S)z> K     )**H NN 2X 2 2=O 2 2 2 GLL!3U;;M w}}]##%M8MMNNN m$$$ bgll-/?@@# F F! sE**E.1E.cvdtttjddS)N.)r'mapstrsys version_infor!r"r#_current_py_versionrRms* 88CS-bqb122 3 33r"ctjdratjt t jj ddtSt j dkr:tjddkr| ddSd t j vrtStS) aThe pip requirement specifier for the running version of Ray. Returns: A string which can be passed to `pip install` to install the currently running Ray version, or None if running on a version built from source locally (likely if you are developing Ray). Examples: Returns "https://s3-us-west-2.amazonaws.com/ray-wheels/[..].whl" if running a stable release, a nightly or a specific commit RAY_CI_POST_WHEEL_TESTSz.whlz{{RAY_COMMIT_SHA}}RAY_RUNTIME_ENV_LOCAL_DEV_MODE1zCurrent Ray version could not be detected, most likely because you have manually built Ray from source. To use runtime_env in this case, set the environment variable RAY_RUNTIME_ENV_LOCAL_DEV_MODE=1.Ndev)renvirongetrr'rrr resolveparentsr __commit__r> __version__rrr5s r#current_ray_pip_specifierr`qs z~~/00'w||    & & ( ( 0 3V=O=Q=Q    / / / :>>: ; ;s B B NN4    t #/ ! !#%%%$&&&r" conda_dict py_versionpip_dependenciesc|g}|dg|d<|d}|d|d|vr|dd}|D]Y}t|trB|dr-t|dtr||dz|d<d}nZ|s|d|i|S)aAdd Ray, Python and (optionally) extra pip dependencies to a conda dict. Args: conda_dict: A dict representing the JSON-serialized conda environment YAML file. This dict will be modified and returned. py_version: A string representing a Python version to inject into the conda dependencies, e.g. "3.7.7" pip_dependencies (List[str]): A list of pip dependencies that will be prepended to the list of pip dependencies in the conda dict. If the conda dict does not already have a "pip" field, one will be created. Returns: The modified dict. (Note: the input argument conda_dict is modified and returned.) N dependencieszpython=pipFT)rZappend isinstancedictr1)rarbrcr3found_pip_dictdeps r#inject_dependenciesrls(~~n%%-%' >" n %D  KK&*&&''' D EN c4  SWWU^^  3u:t8T8T )CJ6CJ!N E / U,-... r"ctj|d}tj|d}|S)NT) sort_keyszutf-8)jsondumpshashlibsha1encode hexdigest)raserialized_conda_spechashs r#_get_conda_env_hashrwsF JzTBBB <-44W== > > H H J JD Kr" runtime_envc|d}|ft|trd}nPt|trdt |}n't dt |jdd}|S)zDReturn `"conda://"`, or None if no GC required.condaNzconda://)razAconda field received by RuntimeEnvAgent must be str or dict, not rL)rZrhrOrirw TypeErrortype__name__)rxrzuris r#get_urirs OOG $ $E  eS ! ! CC t $ $ D0EBBBDDCC<$(KK$8<<<   Jr" RuntimeEnvctj|}|Jt|}|r|dg}n&|drt }ng}t |t|}|S)zEReturns the conda spec with the Ray and `python` dependency inserted.Nr_z ray[default]_inject_current_ray)roloads conda_configr` get_extensionr4rlrR)rxr5raray_pipextra_pip_dependenciess r#!_get_conda_dict_with_ray_insertedrs K446677J  ! ! !'v666G$")>!:  " "#8 9 9$!N!P!P!#$'))+AJ r"c eZdZdZdefdZdedefdZdddeefd Ze fd ed e e j de fd Ze fd e eddd ed e j de f dZe fdeeddd ed e e j fdZdS) CondaPluginrz resources_dirctj|d|_t |jtj|jd|_t |_dS)Nrzz%ray-conda-installs-and-deletions.lock)rrr'_resources_dirr !_installs_and_deletions_file_lockr2_validated_named_conda_env)selfrs r#__init__zCondaPlugin.__init__sb gll='BB 3444 24  !H2 2 .+.%%'''r"rvrcLtj|j|S)aGenerate a path from the hash of a conda or pip spec. The output path also functions as the name of the conda environment when using the `--prefix` option to `conda create` and `conda remove`. Example output: /tmp/ray/session_2021-11-03_16-33-59_356303_41018/runtime_resources /conda/ray-9a7972c3a75f55e976e620484f58410c920db091 )rrr'r)rrvs r#_get_path_from_hashzCondaPlugin._get_path_from_hashsw||D/666r"rxrc8|}|r|gSgS)zFReturn the conda URI from the RuntimeEnv if it exists, else return [].) conda_uri)rrxrs r#get_uriszCondaPlugin.get_uriss())++  ;  r"r~r5c|d|t|\}}|tjkrt d|d|||}t |}t|j5t||}dddn #1swxYwY|s| d|ddS|S) z2Delete URI and return the number of bytes deleted.zGot request to delete URI zICondaPlugin can only delete URIs with protocol conda. Received protocol z, URI prefixr5NzError when deleting conda env z. r) inforrCONDA ValueErrorrrrrr r>)rr~r5protocolrvconda_env_pathlocal_dir_size successfuls r# delete_urizCondaPlugin.delete_uri"sM  666777"3$ x~ % %C-5CC=@CC  11$771.AA d< = = P P)OOOJ P P P P P P P P P P P P P P P  NNNNNNN O O O1sB&&B*-B*contextcKsdSfd}t}|d|d{VS)Nrc t dttrx jvrdSt }t |}tfd|Dstdd dd j dS d t \}} |}t } d t! j5 t$j jd }t-|d 5}t/j||dddn #1swxYwYt3|| t%j|n#t%j|wxYw drt9| dddn #1swxYwY d|t;|S)Nrzrc3HK|]}|dkp |dkVdS)rNr!).0envresults r# z6CondaPlugin.create.._create..Ss:PPC6SV+?vQ/?PPPPPPr"zThe given conda environment 'z' from the runtime env z doesn't exist from the output of `conda info --json`. You can only specify an env that already exists. Please make sure to create an env  z"Setting up conda for runtime_env: r_z"Setting up conda environment with zenvironment.ymlr8rr)rDr5z'Finished creating conda environment at )rrZrhrOrrranyradddebug serializerrrrrrrrr'rrByamldumpr removerrJr) conda_infoenvsrrvconda_env_nameraconda_yaml_filefilerr5rxrr~s @r#_createz#CondaPlugin.create.._createCs-koog.F.FGGF&#&&  T<<<1022 %j11PPPP4PPPPP$GGG0;GG>D GGG/33F;;;q LLQ 8M8M8O8OQQ   's^^NHd!55d;;N:;vVVVJ KKJ[JJ K K K$@AA X X /&(gll+->''Oos334t *d333444444444444444.'vIo....BIo....,,-BCCX-PVWWWW X X X X X X X X X X X X X X X KKR.RR S S S+N;; ;sNH5G 7F G F G  F !G 6H G!!)HHH) has_condar run_in_executor)rr~rxrr5rloops``` ` r#createzCondaPlugin.create9s$$&& 14 <4 <4 <4 <4 <4 <4 <4 |sdS|r|}n9t|\}}||}d|_|xjt|z c_dS)Nr7)rrrrr py_executablecommand_prefixr )rrrxrr5rrrvs r#modify_contextzCondaPlugin.modify_context|s$$&&  F  % % ' ' <(7799NN&{'<'<'>'>??NHd!55d;;N ("=n"M"MMr"N)r} __module__ __qualname__namerOrrrrdefault_loggerrloggingLoggerintrrrrr!r"r#rrss D0c0000$ 7 7 7 7 7 7LT#Yrs   ,,,,,,,,,,,,  ?>>>>>BBBBBBBB<<<<<<HHHHHH#"8,, D<3<<<<7774B"" 0""""J444 (6#'#' W^ $#' c]#'#'#'#'R-1//S#X//tCy)/ #s(^ ////dDS(3-,(6 W^ $ #s(^*SNSNSNSNSN"SNSNSNSNSNr"