uPj@UdZddlmZddlmZmZddlmZddlm Z m Z erddlm Z dZ ee d<d e d eeefd zd d fd Zded e d eeeffdZddd efdZd dddd efdZy )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) )Mapping) TYPE_CHECKINGFinal)generic_cost_per_token)PromptTokensDetailsWrapperUsage) ModelInfog{Gzt?!_DEFAULT_WEB_SEARCH_COST_PER_CALLusagedetailsNreturnc|y||_ t|jdxsd}|dkry|j xs t }||_||_y#ttf$rYywxYw)z Attach server_side_tool_usage_details and mirror web_search_calls onto prompt_tokens_details.web_search_requests for built-in tool cost gating. Nweb_search_callsr)server_side_tool_usage_detailsintget TypeError ValueErrorprompt_tokens_detailsrweb_search_requests)r r rrs e/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/xai/cost_calculator.py-apply_server_side_tool_usage_details_to_usagers +2E("%gkk2D&E&J"K1#(#>#>#^B\B^0@-"7E z "sAA)(A)modelctt|ddxsd}tt|ddxsd}tt|ddxsd}t|dr0|jr$tt|jddxsdnd}|||zk(}|r|n||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 prompt_tokensrcompletion_tokens total_tokenscompletion_tokens_detailsreasoning_tokensN)rrrrrxai)rr custom_llm_provider) rgetattrhasattrrrrrrr) rr rrrralready_normalisedtotal_completion_tokensmodified_usage prompt_costcompletion_costs rcost_per_tokenr)'s wuoqAFQGM"752Eq#I#NQOge^Q?D1EL 55 65;Z;Z GE335G K PqQ  !- @Q0Q Q:L%6RcfvRv!))1''#99"& N$:^qv#w K  '' model_infor c|jd}t|tstSdD]+}|j|}| t |}|dkDs)|cStS#t t f$rYEwxYw)z Per-invocation web_search price from model_info when configured. Prefer ``search_context_cost_per_query`` (same shape as Gemini/Anthropic web search pricing in the model cost map). Fall back to current xAI list pricing. search_context_cost_per_query)search_context_size_mediumsearch_context_size_lowsearch_context_size_highr)r isinstancerr floatrr)r+ search_costskeyvaluecosts r)_web_search_cost_per_call_from_model_infor7Ps%..)HIL lG ,00    % =  rDs $'QA-,8!5788QTV\Q\I]`dId8im8&&(#&(e&(eUl0C&(R-+-%-6TwTKTETr*