o ưiQ @stdZddlmZmZddlmZddlmZerddlmZde dedee e ffd d Z dd d d de fddZ dS)z Helper util for handling XAI-specific cost calculation - Uses the generic cost calculator which already handles tiered pricing correctly - Handles XAI-specific reasoning token billing (billed as part of completion tokens) ) TYPE_CHECKINGTuple)Usage)generic_cost_per_token) ModelInfomodelusagereturncCsxtt|ddpd}d}t|dr|jrtt|jddpd}||}t|j||j|jdd}t||dd\}}||fS) a Calculates the cost per token for a given XAI model, prompt tokens, and completion tokens. Uses the generic cost calculator for all pricing logic, with XAI-specific reasoning token handling. Input: - model: str, the model name without provider prefix - usage: LiteLLM Usage block, containing XAI-specific usage information Returns: Tuple[float, float] - prompt_cost_in_usd, completion_cost_in_usd completion_tokensrcompletion_tokens_detailsreasoning_tokensN) prompt_tokensr total_tokensprompt_tokens_detailsr Zxai)rrZcustom_llm_provider) intgetattrhasattrr rr rrr)rrr r Ztotal_completion_tokensZmodified_usageZ prompt_costZcompletion_costrW/home/app/Keep/.python/lib/python3.10/site-packages/litellm/llms/xai/cost_calculator.pycost_per_tokens$ rr model_inforcCsld}d}t|dr!|jdur!t|jdr!|jjdur!t|jj}nt|dr0|jdur0t|j}||}|S)aI Calculate the cost of web search requests for X.AI models. X.AI Live Search costs $25 per 1,000 sources used. Each source costs $0.025. The number of sources is stored in prompt_tokens_details.web_search_requests by the transformation layer to be compatible with the existing detection system. g?rrNweb_search_requestsnum_sources_used)rrrrr)rrZcost_per_sourcerZ total_costrrrcost_per_web_search_request6s     rN) __doc__typingrrZlitellm.types.utilsrZ.litellm.litellm_core_utils.llm_cost_calc.utilsrrstrfloatrrrrrrs   &