uPjrFdZddlmZmZmZerddlmZneZGddZy)z Cost calculator for A2A (Agent-to-Agent) calls. Supports dynamic cost parameters that allow platform owners to define custom costs per agent query or per token. ) TYPE_CHECKINGAnyFinal)Loggingc ZeZdZededzdefdZedededzdedzdefdZy) A2ACostCalculatorlitellm_logging_objNreturncF|y|j}|jdd}| t|S|jdixsi}|jdt|dS|jd}|jd}||tj |||Sy) a Calculate the cost of an A2A send_message call. Supports multiple cost parameters for platform owners: - cost_per_query: Fixed cost per query - input_cost_per_token + output_cost_per_token: Token-based pricing Priority order: 1. response_cost - if set directly (backward compatibility) 2. cost_per_query - fixed cost per query 3. input_cost_per_token + output_cost_per_token - token-based cost 4. Default to 0.0 Args: litellm_logging_obj: The LiteLLM logging object containing call details Returns: float: The cost of the A2A call N response_costlitellm_paramscost_per_queryinput_cost_per_tokenoutput_cost_per_token)model_call_detailsrr)rgetfloatr_calculate_token_based_cost)r rr rrrs i/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/a2a_protocol/cost_calculator.pycalculate_a2a_costz$A2ACostCalculator.calculate_a2a_costs.  &$7$J$J 255otL  $' '!3 6 67G L RPR   . / ;(89: :'5&8&89O&P'5'9'9:Q'R  +/D/P$@@#5%9&;A rrrc|jd}|yt|ddxsd}t|ddxsd}||r t|ndz}||r t|ndz}||zS)a Calculate cost based on token usage and per-token pricing. Args: model_call_details: The model call details containing usage input_cost_per_token: Cost per input token (can be None, defaults to 0) output_cost_per_token: Cost per output token (can be None, defaults to 0) Returns: float: The calculated cost usager prompt_tokensrcompletion_tokens)rgetattrr)rrrrrr input_cost output_costs rrz-A2ACostCalculator._calculate_token_based_costIs$*--g6 = 'uoqAFQ #*52Eq#I#NQ*L`U3G-Hfij .Rg%8M2Nmpq K''r) __name__ __module__ __qualname__ staticmethodLitellmLoggingObjectrrdictrrrrrsj31D83 33j( (#dl( %t|(  ((rrN) __doc__typingrrr*litellm.litellm_core_utils.litellm_loggingrr$rr&rrr*s.-,U(U(r