o ưi@sddlZddlZddlmZddlZddlmZddlmZddl m Z ddl m Z ddl mZddlmZmZGd d d e ZdS) N)Optional)verbose_logger) InMemoryCache)SECRET_MANAGER_REFRESH_INTERVAL) GCSBucketBase)_get_httpx_client)CommonProxyErrorsKeyManagementSystemcsNeZdZedfdeedeeddffdd Zdedeefd d Z Z S) GoogleSecretManagerFrefresh_intervalalways_read_secret_managerreturnNcsddlm}|durtdtjjttj dd|_ |j dur)tdt |_ |t_tjt_tj d|}|rAt|n|}t|d |_tj d }|r\|d kr\d|_dS|p_d |_dS) a! Args: refresh_interval (int, optional): The refresh interval in seconds. Defaults to 86400. (24 hours) always_read_secret_manager (bool, optional): Whether to always read from the secret manager. Defaults to False. Since we do want to cache values r) premium_userTz5Google Secret Manager requires an Enterprise License Z GOOGLE_SECRET_MANAGER_PROJECT_IDNzgGoogle Secret Manager requires a project ID, please set 'GOOGLE_SECRET_MANAGER_PROJECT_ID' in your .envZ&GOOGLE_SECRET_MANAGER_REFRESH_INTERVAL)Z default_ttlZ0GOOGLE_SECRET_MANAGER_ALWAYS_READ_SECRET_MANAGERtrueF)Zlitellm.proxy.proxy_serverr ValueErrorrZnot_premium_uservaluesuper__init__osenvironget PROJECT_IDrsync_httpx_clientlitellmZsecret_manager_clientr ZGOOGLE_SECRET_MANAGERZ_key_management_systemintrcachelowerr )selfr r rZ_refresh_intervalZ_always_read_secret_manager __class__d/home/app/Keep/.python/lib/python3.10/site-packages/litellm/secret_managers/google_secret_manager.pyrs<      zGoogleSecretManager.__init__ secret_namec Cs|jdur|j|}|dur|S||jjvr|Sd|jd|d}|}d|d}|jj||d}|jd krUt d t |j |j |dtd |d |j t d |j|}|did}|dur~t|d} |j || | S|j |dtd |d)z Retrieve a secret from Google Secret Manager or cache. Args: secret_name (str): The name of the secret. Returns: str: The secret value if successful, None otherwise. TNz projects/z /secrets/z/versions/latestz(https://secretmanager.googleapis.com/v1/z:access)urlheadersz)Google Secret Manager retrieval error: %szsecret z, not found in Google Secret Manager. Error: z8Google Secret Manager retrieval response status code: %spayloaddatazutf-8z# not found in Google Secret Manager)r rZ get_cacheZ cache_dictrZsync_construct_request_headersrr status_codererrorstrtextZ set_cacherdebugjsonbase64 b64decodedecode) rr"Z cached_secretZ _secret_namer$r#responseZ secret_dataZ_base64_encoded_valueZ_decoded_valuer r r!%get_secret_from_google_secret_manager>sF     z9GoogleSecretManager.get_secret_from_google_secret_manager) __name__ __module__ __qualname__rrrboolrr*r2 __classcell__r r rr!r s/r )r.rtypingrrZlitellm._loggingrZlitellm.caching.cachingrZlitellm.constantsrZ/litellm.integrations.gcs_bucket.gcs_bucket_baserZ&litellm.llms.custom_httpx.http_handlerrZlitellm.proxy._typesrr r r r r r!s