i6 rdZddlmZmZddlmZddlmZerddlmZde dedee e ffd Z dd d d de fd Z y)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) ModelInfomodelusagereturncBtt|ddxsd}d}t|dr0|jr$tt|jddxsd}||z}t |j ||j |jd}t||d\}}||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 xai)rr custom_llm_provider) intgetattrhasattrr rrrrr)rr r rtotal_completion_tokensmodified_usage prompt_costcompletion_costs q/Users/manta/Documents/Projects/TheRoad-I1/.venv/lib/python3.12/site-packages/litellm/llms/xai/cost_calculator.pycost_per_tokenrsGE+>BGaHu12u7V7V E335G K Pq 02BB))1''#99"& N$:>u$ K  ''r model_inforc:d}d}t|drX|jLt|jdr6|jj t|jj}n-t|dr!|jt|j}||z}|S)aA 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?rrweb_search_requestsnum_sources_used)rrr rr!)r rcost_per_sourcer! total_costs rcost_per_web_search_requestr$6s$O ./  ' ' 3 E//1F G  ' ' ; ; Gu::NNO * +0F0F0Ru556 #33J rN) __doc__typingrrlitellm.types.utilsr.litellm.litellm_core_utils.llm_cost_calc.utilsrrstrfloatrr$rrr,sY (%Q-#(##(e#(eUl0C#(LwKEr