&`i$ @ddlZddlZddlZddlmZddlmZmZmZm Z m Z ddl m Z ddl mZmZmZmZmZmZddlmZddlmZddlmZejeZeGd d eZGd d ZGd dZefdddedededej f dZ!dS)N)ABC)AnyDictListOptionalType) import_attr) RAY_RUNTIME_ENV_CLASS_FIELD_NAME'RAY_RUNTIME_ENV_PLUGIN_DEFAULT_PRIORITY#RAY_RUNTIME_ENV_PLUGIN_MAX_PRIORITY#RAY_RUNTIME_ENV_PLUGIN_MIN_PRIORITYRAY_RUNTIME_ENV_PLUGINS_ENV_VAR#RAY_RUNTIME_ENV_PRIORITY_FIELD_NAME)RuntimeEnvContext)URICache) DeveloperAPIc eZdZUdZdZeed<eZe ed<e de ddfdZ dd de efd Zd eed ed ejdefdZde edd d ed ejddf dZd ed ejdefdZdS)RuntimeEnvPluginz4Abstract base class for runtime environment plugins.Nnamepriorityruntime_env_dictreturncdS)a Validate user entry for this plugin. The method is invoked upon installation of runtime env. Args: runtime_env_dict: The user-supplied runtime environment dict. Raises: ValueError: If the validation fails. N)rs s/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/_private/runtime_env/plugin.pyvalidatezRuntimeEnvPlugin.validates   runtime_env RuntimeEnvcgSNr)selfrs rget_uriszRuntimeEnvPlugin.get_uris,s ruricontextloggerc KdS)a{Create and install the runtime environment. Gets called in the runtime env agent at install time. The URI can be used as a caching mechanism. Args: uri: A URI uniquely describing this resource. runtime_env: The RuntimeEnv object. context: Auxiliary information supplied by Ray. logger: A logger to log messages during the context modification. Returns: float: The disk space taken up by this plugin installation for this environment. e.g. for working_dir, this downloads the files to the local node. rr)r"r$rr%r&s rcreatezRuntimeEnvPlugin.create/s .qruriscdS)aModify context to change worker startup behavior. For example, you can use this to prepend "cd " command to worker startup, or add new environment variables. Args: uris: The URIs used by this resource. runtime_env: The RuntimeEnv object. context: Auxiliary information supplied by Ray. logger: A logger to log messages during the context modification. Nr)r"r)rr%r&s rmodify_contextzRuntimeEnvPlugin.modify_contextHs $ rcdS)aDelete the runtime environment given uri. Args: uri: A URI uniquely describing this resource. logger: The logger used to log messages during the deletion. Returns: float: The amount of space reclaimed by the deletion. rr)r"r$r&s r delete_urizRuntimeEnvPlugin.delete_uri\s qr)__name__ __module__ __qualname____doc__rstr__annotations__r rint staticmethoddictrrr#rrloggingLoggerfloatr(r+r-rrrrrsC>>D#;Hc;;;  4  D    \  LT#Y c]#     23i"#     ( c 7> e      rrc&eZdZdedededefdZdS)PluginSetupContextrclass_instancer uri_cachec>||_||_||_||_dSr!)rr<rr=)r"rr<rr=s r__init__zPluginSetupContext.__init__js% ,  "rN)r.r/r0r2rr4rr?rrrr;r;isN # #) # #  # # # # # #rr;ceZdZdZdZdeeddfdZdeddfdZ d e e ddfd Z d eddfd Z d edefd Zde efdZdS)RuntimeEnvPluginManagerz:This manager is used to load plugins in runtime env agent.ci|_tjt}|r+t j|}||dSdSr!)pluginsosenvirongetrjsonloads load_plugins)r"plugin_config_strplugin_configss rr?z RuntimeEnvPluginManager.__init__zsZ68 JNN+JKK  .!Z(9::N   n - - - - - . .r plugin_classrNct|tstd|d|jstd|d|j|jvr&td|d|j|jddS)Nz!Invalid runtime env plugin class zQ. The plugin class must inherit ray._private.runtime_env.plugin.RuntimeEnvPlugin.z$No valid name in runtime env plugin .zThe name of runtime env plugin z conflicts with ) issubclassr RuntimeErrorrrC)r"rLs rvalidate_plugin_classz-RuntimeEnvPluginManager.validate_plugin_classs,(899 DLDDD    WUlUUUVV V   , ,;,;; \%67;;;  - ,rrc t|tr|tks |tkr#t d|dtdtddS)NzInvalid runtime env priority z", it should be an integer between z and rN) isinstancer4r r rP)r"rs rvalidate_priorityz)RuntimeEnvPluginManager.validate_prioritysx8S)) ======>>>6>>;>>> >=rrKc |D]}t|tr t|vrtd|dtdt |t}||t |vr|t }n|j}|||}t|j ||| ||j |j <dS)z8Load runtime env plugins and create URI caches for them.z"Invalid runtime env plugin config z+, it should be a object which contains the z field.N) rSr6r rPr rQrrrTr;rcreate_uri_cache_for_pluginrC)r"rK plugin_configrLrr<s rrIz$RuntimeEnvPluginManager.load_pluginss+  M}d33 3=HH"AAA7AAA '}5U'VWWL  & &| 4 4 4 3mCC()LM'0  " "8 , , ,)\^^N.@!00@@ //DL* + +/  rpluginct|}||||jt |j||j|||j|j<dS)zAdd a plugin to the manager and create a URI cache for it. Args: plugin: The class instance of the plugin. N)typerQrTrr;rrVrC)r"rXrLs r add_pluginz"RuntimeEnvPluginManager.add_pluginsy F||  ""<000 |4555*<     !  , ,V 4 4 + +  \&'''rc d|jd}tdttj|dz}t|j|S)zCreate a URI cache for a plugin. Args: plugin_name: The name of the plugin. Returns: The created URI cache for the plugin. RAY_RUNTIME_ENV__CACHE_SIZE_GBi@ ) rupperr4r9rDrErFrr-)r"rXcache_size_env_varcache_size_bytess rrVz3RuntimeEnvPluginManager.create_uri_cache_for_pluginskL KKKQQSS bjnn-?DDEE E  )+;<<zFRuntimeEnvPluginManager.sorted_plugin_setup_contexts..s1:r)key)sortedrCvalues)r"s rsorted_plugin_setup_contextsz4RuntimeEnvPluginManager.sorted_plugin_setup_contextss) dl))++1E1EFFFFr)r.r/r0r1r?rrrQrrTrrrIr[rrVr;rjrrrrArAwsDD... $7G2H T     # $    4:$@ !1 d    =2B=x==== Gd3E.FGGGGGGrrArrrXr=r%r&c`K|j|vs ||jdS||||}|s=|d|jd|d|||d{V|D]}||vrQ|d|d|||||d{V}||||W|d|jd|d ||||||||dS) zISet up the environment using the plugin if not already set up and cached.NzNo URIs for runtime env plugin z+; create always without checking the cache.)r&zCache miss for URI rNz Runtime env  zq is already installed and will be reused. Search all runtime_env_setup-*.log to find the corresponding setup log.) rrr#debugr(addinfo mark_usedr+)rrXr=r%r&r)r$ size_bytess rcreate_for_plugin_if_neededrrs{+%%V[)A)I OOK   ??; ' 'D G  8fk 8 8 8   mmD+wvmFFFFFFFFF 4 4 i   LL5s555 6 6 6%}}S+wv}VVVVVVVVJ MM#z&M 9 9 9 9 KKSv{SSSSSS      F  3 3 3 3 $ Wf=====r)"rGr7rDabcrtypingrrrrrray._common.utilsr "ray._private.runtime_env.constantsr r r r rr ray._private.runtime_env.contextr"ray._private.runtime_env.uri_cacherray.util.annotationsr getLoggerr.default_loggerrr;rAr8rrrrrr|s  22222222222222))))))?>>>>>777777------""8,,NNNNNsNNNb # # # # # # # #lGlGlGlGlGlGlGlGh, #>#>#> #>#> #> N #>#>#>#>#>#>r