o ưi @spdZddlmZmZmZerddlmZmZ ddedddeed ee e ffd d Z ddd d d e fddZ dS)zi This file is used to calculate the cost of the Gemini API. Handles the context caching for Gemini API. ) TYPE_CHECKINGOptionalTuple) ModelInfoUsageNmodelusager service_tierreturncCsddlm}|||d|dS)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)rrZcustom_llm_providerr )Z.litellm.litellm_core_utils.llm_cost_calc.utilsr )rrr r r Z/home/app/Keep/.python/lib/python3.10/site-packages/litellm/llms/gemini/cost_calculator.pycost_per_token s r model_inforcCsdddlm}d}d}|dur*|jdur*t|j|r*t|jdr*|jjdur*|jj}nd}||}|S)zm Calculates the cost per web search request for a given model, prompt tokens, and completion tokens. r)PromptTokensDetailsWrappergQ?Nweb_search_requests)litellm.types.utilsrZprompt_tokens_details isinstancehasattrr)rrrcost_per_web_search_requestZnumber_of_web_search_requestsZ total_costr r rrs      r)N) __doc__typingrrrrrrstrfloatrrr r r rs