tPjb UdZddlmZddlmZddlmZddlmZeZ ee d< dde d e e dzd e fd Zd ed e dzfd Zd ed e dzfdZd ed e dzfdZy)z4 Contains utils used by OpenAI compatible endpoints )Final)Request)SensitiveDataMasker_read_request_bodySENSITIVE_DATA_MASKERNdeployment_dict excluded_keysreturncf|djdd|djdd|djdd|djdd|djdd|djddd d h}|xs t|z}tj|d| |d<|S) at Removes sensitive information from a deployment dictionary. Args: deployment_dict (dict): The deployment dictionary to remove sensitive information from. excluded_keys (Optional[Set[str]]): Set of keys that should not be masked (exact match). Returns: dict: The modified deployment dictionary with sensitive information removed. litellm_paramsapi_keyN client_secretvertex_credentialsvertex_ai_credentialsaws_access_key_idaws_secret_access_keydefault_api_key_tpm_limitdefault_api_key_rpm_limit)r )popsetr mask_dict)r r _rate_limit_config_keys _excludeds u/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/proxy/common_utils/openai_endpoint_utils.py%remove_sensitive_info_from_deploymentrs$%)))T:$%))/4@$%))*>E$%))*A4H$%))*=tD$%))*A4H $#&&.2III(=(G(G())H)O$% requestcRKt|d{xsi}d|vr|dSy7w)z` Get the `custom_llm_provider` from the request body Safely reads the request body rNcustom_llm_providerr)r request_bodys r)get_custom_llm_provider_from_request_bodyr#4s7 '9&I I ORL ,122 !Js '%'c>d|jvr|jdSy)zx Get the `custom_llm_provider` from the request query parameters Safely reads the request query parameters r!N) query_paramsr s r*get_custom_llm_provider_from_request_queryr&@s'  4 44##$9:: rc>d|jvr|jdSy)zU Get the `custom_llm_provider` from the request header `custom-llm-provider` zcustom-llm-providerN)headersr s r,get_custom_llm_provider_from_request_headersr)Ks#/455 r)N)__doc__typingrfastapir0litellm.litellm_core_utils.sensitive_data_maskerr-litellm.proxy.common_utils.http_parsing_utilsrr__annotations__dictrstrrr#r&r)rrr3sPL24u4 &*""s8d?" "J W t C$J'cDjr