uPj UdZddlmZmZerddlmZmZddedddedzd ee e ffd Z ddd d d e fd Z dZ ee d<dZee d<ddd edzfdZddd d d e fdZy)zi This file is used to calculate the cost of the Gemini API. Handles the context caching for Gemini API. ) TYPE_CHECKINGFinal) ModelInfoUsageNmodelusager service_tierreturnc&ddlm}|||d|S)z Calculates the cost per token for a given model, prompt tokens, and completion tokens. Follows the same logic as Anthropic's cost per token calculation. r)generic_cost_per_tokengemini)rrcustom_llm_providerr ).litellm.litellm_core_utils.llm_cost_calc.utilsr )rrr r s h/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/gemini/cost_calculator.pycost_per_tokenr s! V !$!   model_inforcddlm}ddlm}d}|j dxsi}|j d|}|d|j Xt |j |rBt|j dr,|j j|j jnd}||}|xs|xsd} |j d xsd } | dkDr| d k(rd n| } || zS) a% Calculates the cost of web search (grounding with Google Search). Billing mode is determined by ``web_search_billing_unit`` in model_info: - ``"per_query"``: charged per individual search query (Gemini 3.x). - ``"per_prompt"`` (default): charged per grounded prompt (Gemini 2.x), regardless of how many queries were executed internally. Reads the per-request cost from ``search_context_cost_per_query`` in ``model_info`` when available, falling back to $0.035 for models not yet updated in the pricing JSON. r)"get_web_search_requests_from_usagePromptTokensDetailsWrappergQ?search_context_cost_per_querysearch_context_size_mediumNweb_search_requestsweb_search_billing_unit per_prompt) rrlitellm.types.utilsrgetprompt_tokens_details isinstancehasattrr) rrrr _DEFAULT_COST search_costs_costrequests_from_prompt_detailsrequests_from_server_tool_usenumber_of_web_search_requests billing_modebillable_requestss rcost_per_web_search_requestr+s? M$..)HIORL##$@-PE  ++75668RS335JK++??K ##77!,Ne+T!+G+mKh+mlm!$..)BCS|L+a/LL4PWt $ $$rgy&1?,GOOGLE_MAPS_GROUNDING_DEFAULT_COST_PER_QUERYg?-GOOGLE_MAPS_GROUNDING_DEFAULT_COST_PER_PROMPTz Usage | Nonecxddlm}| |jnd}t||r t |dsy|j S)Nrrgoogle_maps_grounding_requests)rrr r!r"r/)rrdetailss rr/r/Ms;>494EU004G g9 :''SsBt  1 11rct|}|r|dkry|jdxsd}|dk(rtnt}|jd}||n|}|dk(r|nd}||zS)a Calculates the cost of Grounding with Google Maps. Billing follows ``web_search_billing_unit`` in model_info the same way Google Search grounding does: ``"per_query"`` (Gemini 3.x) multiplies the executed Maps queries, ``"per_prompt"`` (default, Gemini 2.x) charges one flat fee per grounded prompt. The rate comes from ``google_maps_grounding_cost_per_query`` in ``model_info``, falling back to Google's list price for that billing unit when the pricing JSON has no entry yet. rgrr per_query$google_maps_grounding_cost_per_queryr)r/rr,r-)rrrequestsr) default_costconfigured_costcostbilled_requestss r&cost_per_google_maps_grounding_requestr9Vs{5U;H x1}$..)BCS|L ; & 5 : (^^,RSO"1"9,D)5)DX!O / !!r)N)__doc__typingrrrrrstrtuplefloatrr+r,__annotations__r-intr/r9rrrBs (4 # g S4Z SXY^`eYeSf  )%w)%K)%E)%X7<,e;7<-u<2.2S4Z2"'"{"W\"r