o ưi#1@s*UdZddlZddlZddlmZddlmZmZmZm Z ddl Z ddl m Z da eeed<GdddZd ed efd d Zd efd dZd efddZded efddZdeeded eefddZdeded efddZdeded eefddZdeded efddZded eefd d!ZdS)"aR Centralized manager for Anthropic beta headers across different providers. This module provides utilities to: 1. Load beta header configuration from JSON (mapping of supported headers per provider) 2. Filter and map beta headers based on provider support 3. Handle provider-specific header name mappings (e.g., advanced-tool-use -> tool-search-tool) 4. Support remote fetching and caching similar to model cost map Design: - JSON config contains mapping of beta headers for each provider - Keys are input header names, values are provider-specific header names (or null if unsupported) - Only headers present in mapping keys with non-null values can be forwarded - This enforces stricter validation than the previous unsupported list approach Configuration can be loaded from: - Remote URL (default): Fetches from GitHub repository - Local file: Set LITELLM_LOCAL_ANTHROPIC_BETA_HEADERS=True to use bundled config only Environment Variables: - LITELLM_LOCAL_ANTHROPIC_BETA_HEADERS: Set to "True" to disable remote fetching - LITELLM_ANTHROPIC_BETA_HEADERS_URL: Custom URL for remote config (optional) N)files)DictListOptionalSet)verbose_logger_BETA_HEADERS_CONFIGc@sjeZdZdZedefddZededefddZ e dedefdd Z edd e d e defd dZdS)GetAnthropicBetaHeadersConfigz Handles fetching, validating, and loading the Anthropic beta headers configuration. Similar to GetModelCostMap, this class manages the lifecycle of the beta headers configuration with support for remote fetching and local fallback. returnc Csjzttddjdd}|WSty4}ztd|iiiiiidWYd}~Sd}~ww)zCLoad the local backup beta headers config bundled with the package.litellmz"anthropic_beta_headers_config.jsonzutf-8)encodingz*Failed to load local beta headers config: ) anthropicazure_aibedrockbedrock_converse vertex_aiprovider_aliasesN)jsonloadsrjoinpath read_text Exceptionrerror)contenter]/home/app/Keep/.python/lib/python3.10/site-packages/litellm/anthropic_beta_headers_manager.pyload_local_beta_headers_config.s&zXszEGetAnthropicBetaHeadersConfig._check_is_valid_dict..zYLiteLLM: Fetched beta headers config missing provider keys. Falling back to local backup.T) isinstancedictrwarningtype__name__lenany)rZ provider_keysZ has_providerrr"r_check_is_valid_dictDs&  z2GetAnthropicBetaHeadersConfig._check_is_valid_dictcCs ||S)z Validate the integrity of a fetched beta headers config. Returns True if all checks pass, False otherwise. )r+)clsrrrrvalidate_beta_headers_configcs z:GetAnthropicBetaHeadersConfig.validate_beta_headers_configurltimeoutcCstj||d}||S)z Fetch the beta headers config from a remote URL. Returns the parsed JSON dict. Raises on network/parse errors (caller is expected to handle). )r0)httpxgetraise_for_statusr)r/r0responserrr fetch_remote_beta_headers_configlsz>GetAnthropicBetaHeadersConfig.fetch_remote_beta_headers_configN)r.)r( __module__ __qualname____doc__ staticmethodrrr%boolr+ classmethodr-strintr5rrrrr &sr r/r c CstdddkrtSzt|}Wnty4}ztd|t |tWYd}~Sd}~wwtj |dsEtd|tS|S)u Public entry point — returns the beta headers config dict. 1. If ``LITELLM_LOCAL_ANTHROPIC_BETA_HEADERS`` is set, uses the local backup only. 2. Otherwise fetches from ``url``, validates integrity, and falls back to the local backup on any failure. Args: url: URL to fetch the remote beta headers configuration from Returns: Dict containing the beta headers configuration Z$LITELLM_LOCAL_ANTHROPIC_BETA_HEADERStruez^LiteLLM: Failed to fetch remote beta headers config from %s: %s. Falling back to local backup.Nr"z_LiteLLM: Fetched beta headers config failed integrity check. Using local backup instead. url=%s) osgetenvlowerr rr5rrr&r<r-)r/rrrrrget_beta_headers_configys( rCcCs0tdurtSddlm}t|datdtS)a  Load the beta headers configuration. Uses caching to avoid repeated fetches/file reads. This function is called by all public API functions and manages the global cache. Returns: Dict containing the beta headers configuration Nranthropic_beta_headers_url)r/z%Loaded and cached beta headers config)rr rErCrdebugrDrrr_load_beta_headers_configs     rGcCsdatdtS)z Force reload the beta headers configuration from source (remote or local). Clears the cache and fetches fresh configuration. Returns: Dict containing the newly loaded beta headers configuration Nz-Reloading beta headers config (cache cleared))rrinforGrrrrreload_beta_headers_configs rIprovidercCst}|di}|||S)z Resolve provider aliases to canonical provider names. Args: provider: Provider name (may be an alias) Returns: Canonical provider name r)rGr2)rJconfigaliasesrrrget_provider_names  rM beta_headerscCs|sgSt}t|}||i}t}|D]1}|}||vr-td|d|dq||}|durBtd|d|dq||qtt |S)aC Filter and transform beta headers based on provider's mapping configuration. This function: 1. Only allows headers that are present in the provider's mapping keys 2. Filters out headers with null values (unsupported) 3. Maps headers to provider-specific names (e.g., advanced-tool-use -> tool-search-tool) Args: beta_headers: List of Anthropic beta header values provider: Provider name (e.g., "anthropic", "bedrock", "vertex_ai") Returns: List of filtered and transformed beta headers for the provider zDropping unknown beta header 'z' for provider 'z' (not in mapping)Nz"Dropping unsupported beta header '') rGrMr2setstriprrFaddsortedlist)rNrJrKprovider_mappingZfiltered_headersheaderZ mapped_headerrrr!filter_and_transform_beta_headerss*   rW beta_headercCs.t}t|}||i}||vo||duS)a Check if a specific beta header is supported by a provider. Args: beta_header: The Anthropic beta header value provider: Provider name Returns: True if the header is in the mapping with a non-null value, False otherwise NrGrMr2)rXrJrKrUrrris_beta_header_supporteds rZanthropic_beta_headercCs.t}t|}||i}||vrdS||S)a Get the provider-specific beta header name for a given Anthropic beta header. This function handles header transformations/mappings (e.g., advanced-tool-use -> tool-search-tool). Args: anthropic_beta_header: The Anthropic beta header value provider: Provider name Returns: The provider-specific header name if supported, or None if unsupported/unknown NrY)r[rJrKrUrrrget_provider_beta_header&s  r\headerscCsX|d}|s |Sdd|dD}t||d}|r$d||d<|S|dd|S)a4 Update headers dict by filtering and transforming anthropic-beta header values. Modifies the headers dict in place and returns it. Args: headers: Request headers dict (will be modified in place) provider: Provider name Returns: Updated headers dict zanthropic-betacSsg|] }|r|qSr)rQ)r brrr Xz5update_headers_with_filtered_beta..,)rNrJN)r2splitrWjoinpop)r]rJZ existing_betaZ beta_valuesZfiltered_beta_valuesrrr!update_headers_with_filtered_betaDs  recCs,t}t|}||i}dd|DS)z Get all beta headers that are unsupported by a provider (have null values in mapping). Args: provider: Provider name Returns: List of unsupported Anthropic beta header names cSsg|] \}}|dur|qSrr)r rVvaluerrrr_yr`z+get_unsupported_headers..)rGrMr2items)rJrKrUrrrget_unsupported_headersjs  rh)r8rr@importlib.resourcesrtypingrrrrr1Z*litellm.litellm_core_utils.litellm_loggingrr__annotations__r r<r%rCrGrIrMrWr:rZr\rerhrrrrsT  S* 8   &