o ưi8@s@dZddlmZmZmZerddlmZneZGdddZdS)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_CHECKINGAnyOptional)Loggingc @sLeZdZedeedefddZededeedeedefdd Z d S) A2ACostCalculatorlitellm_logging_objreturncCs|durdS|j}|dd}|durt|S|dipi}|ddur,t|dS|d}|d}|dus>|durFtj|||dSdS) 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_paramsZcost_per_queryinput_cost_per_tokenoutput_cost_per_token)model_call_detailsr r )rgetfloatr_calculate_token_based_cost)rrr r r r r[/home/app/Keep/.python/lib/python3.10/site-packages/litellm/a2a_protocol/cost_calculator.pycalculate_a2a_costs$    z$A2ACostCalculator.calculate_a2a_costrr r cCsf|d}|dur dSt|ddpd}t|ddpd}||r"t|nd}||r,t|nd}||S)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 usageNr prompt_tokensrcompletion_tokens)rgetattrr)rr r rrrZ input_costZ output_costrrrrIs z-A2ACostCalculator._calculate_token_based_costN) __name__ __module__ __qualname__ staticmethodrLitellmLoggingObjectrrdictrrrrrrs" 5rN) __doc__typingrrrZ*litellm.litellm_core_utils.litellm_loggingrrrrrrrs