o ưi @sZdZddlmZmZddlmZ   d dedeeded eed ee e ff d d Z dS)z( Cost calculation for search providers. )OptionalTuple)get_model_infoNmodelcustom_llm_providernumber_of_queriesoptional_paramsreturnc Cst||d}|d}|r@t|tr@|pidd}d}|D]}|d\} } | |kr0| kr8nq|d}nq|dd}n t|dpGd}||} | dfS) a` Calculate cost for search-only providers. Returns (input_cost, output_cost) where input_cost = queries * cost_per_query Supports tiered pricing based on max_results parameter. Args: model: Model name (e.g., "exa_ai/search", "tavily/search") custom_llm_provider: Provider name (e.g., "exa_ai", "tavily") number_of_queries: Number of search queries performed (default: 1) optional_params: Optional parameters including max_results for tiered pricing Returns: Tuple of (input_cost, output_cost) where output_cost is always 0.0 )rrtiered_pricing max_results gZmax_results_rangeZinput_cost_per_query)rget isinstancelistfloat) rrrr Z model_infor r Zcost_per_queryZtierZ range_minZ range_maxZ total_costrU/home/app/Keep/.python/lib/python3.10/site-packages/litellm/search/cost_calculator.pysearch_provider_cost_per_query s    r)NrN) __doc__typingrrZ litellm.utilsrstrintdictrrrrrrs"