&`i dZddlZddlmZddlmZmZmZddlm Z ddl m Z ddl m Z mZmZddlmZejeZdd Zd efd Z dd eeeefded dfdZdS)zAuthentication token setup for Ray. This module provides functions to generate and save authentication tokens for Ray's token-based authentication system. Token loading and caching is handled by the C++ AuthenticationTokenLoader. N)Path)AnyDictOptional)3TOKEN_AUTH_ENABLED_BUT_NO_TOKEN_FOUND_ERROR_MESSAGE)!generate_new_authentication_token)AuthenticationModeAuthenticationTokenLoaderget_authentication_mode)AuthenticationErrorreturncJt}t} |jddt |d5}||dddn #1swxYwYt d|dS#t$rwxYw)zGenerate a new random token and save it in the default token path. Returns: The newly generated authentication token. T)parentsexist_okwNz0Generated new authentication token and saved to ) r_get_default_token_pathparentmkdiropenwriteloggerinfo Exception)token token_pathfs /home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/_private/authentication/authentication_token_setup.pygenerate_and_save_tokenrs . / /E(**J t<<<*c " " a GGENNN                 SzSSTTTTT  s/,B A, B,A00B3A04 B B"c4tjdz dz S)zyGet the default token file path (~/.ray/auth_token). Returns: Path object pointing to ~/.ray/auth_token z.ray auth_token)rhomerrr3s 9;; , ..r#T system_configcreate_token_if_missingcJttjkr#|rd|vr|ddkrtddSt j}|ds:|r$t|dSttdS)aCheck authentication settings and set up token resources if authentication is enabled. Ray calls this early during ray.init() to do the following for token-based authentication: 1. Check whether you enabled token-based authentication. 2. Make sure a token is available if authentication is enabled. 3. Generate and save a default token for new local clusters if one doesn't already exist. Args: system_config: Ray raises an error if you set AUTH_MODE in system_config instead of the environment. create_token_if_missing: Generate a new token if one doesn't already exist. Raises: RuntimeError: Ray raises this error if authentication is enabled but no token is found when connecting to an existing cluster. AUTH_MODEdisabledzsSet authentication mode can only be set with the `RAY_AUTH_MODE` environment variable, not using the system_config.NT)ignore_auth_mode) r r TOKEN RuntimeErrorr instance has_tokenr reset_cacher r)r$r% token_loaders rensure_token_if_auth_enabledr0<s(  $6$<<<  },,k*j88F  ,577L  ! !4 ! 8 8  "  # % % %  $ $ & & & & &%C   r#)r N)NT)__doc__loggingpathlibrtypingrrr4ray._private.authentication.authentication_constantsr:ray._private.authentication.authentication_token_generatorr ray._rayletr r r ray.exceptionsr getLogger__name__rrrstrboolr0r"r#rr=s\&&&&&&&&&& /.....  8 $ $./////UY++DcN++MQ+ ++++++r#