uPj\! RdZddlmZddlmZmZmZddlmZddl m Z ddl m Z m Z mZddlmZde d ed fd Z dd ededed zded zd eeeff dZdd eded zded eeeffdZded ed zfdZdeeefded zd ed zfdZ dd ededed zde d zded zd ef dZy )zR Helper util for handling openai-specific cost calculation - e.g.: prompt caching )Mapping)AnyFinalLiteral)verbose_loggergeneric_cost_per_token) CallTypes ModelInfoUsage)get_model_info call_typereturn)cost_per_tokencost_per_secondcR|tjk(s|tjk(ryy)Nrr)r atranscription transcription)rs i/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/openai/cost_calculation.py cost_routerrs#I,,, Y=T=T0T Nmodelusage service_tierdata_residencyc"t||d||S)a Calculates the cost per token for a given model, prompt tokens, and completion tokens. Input: - model: str, the model name without provider prefix - usage: LiteLLM Usage block, containing anthropic caching information - data_residency: optional OpenAI data-residency region (e.g. "eu", "us"), inferred from api_base. Applies the model's regional-processing uplift multiplier when set. Returns: Tuple[float, float] - prompt_cost_in_usd, completion_cost_in_usd openai)rrcustom_llm_providerrrr)rrrrs rrrs ( "$!%  rrdurationct||xsd}d}d}|jd}|&|dkDr!tjd|||||z}||fSd|vr6|d1tjd||jd||d|z}d}||fS) a Calculates the cost per second for a given model, prompt tokens, and completion tokens. Input: - model: str, the model name without provider prefix - custom_llm_provider: str, the custom llm provider - duration: float, the duration of the response in seconds Returns: Tuple[float, float] - prompt_cost_in_usd, completion_cost_in_usd rrroutput_cost_per_secondr7For model=%s - output_cost_per_second: %s; duration: %sinput_cost_per_secondz6For model=%s - input_cost_per_second: %s; duration: %s)r getrdebug)rrr model_info prompt_costcompletion_costr#s rrr^s'UH[Hg_ghJKO$.NN3K$L).Dq.H E  "   18;  '' !J .:>U3V3b D  NN2 3   !!89HD   ''r resolutionc|jj}|sydjd|D}|rt|dkDry|S)zSMap usage resolution to a safe suffix for ``output_cost_per_second_`` keys.Nc3LK|]}|js|dk(s|yw)_N)isalnum).0cs r z9_video_resolution_to_cost_field_suffix..sB!))+c!Bs$$)striplowerjoinlen)r+rsafes r&_video_resolution_to_cost_field_suffixr;sH!'')A ''BQBBD 3t9r> Krr(video_resolutionc|xsdjj}|r0t|}|#d|}|j|}| t |S|jd}| t |Sy)a Per-second video output rate from model_info. If ``video_resolution`` is set (e.g. ``1080p``, ``720p``, ``4k``), looks up ``output_cost_per_second_`` first (e.g. ``output_cost_per_second_1080p``), then falls back to ``output_cost_per_second``. r-Noutput_cost_per_second_r#)r5r6r;r&float)r(r<r9suffixtier_key tier_rateouts r_video_output_cost_per_secondrDs!&B--/557A>qA   7x@H)~~h7I$Y'' 89C Sz rduration_secondsc |t||xsd}|jd}|tjd|||||zSt ||}|tjd|||||zStj d|y)a Calculates the cost for video generation based on duration in seconds. Input: - model: str, the model name without provider prefix - duration_seconds: float, the duration of the generated video in seconds - custom_llm_provider: str, the custom llm provider - model_info: Optional[dict], deployment-level model info containing custom video pricing. When provided, skips the global get_model_info() lookup so that deployment-specific pricing is used. - video_resolution: Optional resolution label from usage (e.g. ``720p``, ``1080p``). Returns: float - total_cost_in_usd rr! output_cost_per_video_per_secondzAFor model=%s - output_cost_per_video_per_second: %s; duration: %sr$zhNo cost information found for video model %s. Please add pricing to model_prices_and_context_window.jsonr")r r&rr'rDwarning)rrErr(r<video_cost_per_secondr#s rvideo_generation_costrJs.#%EXEd\de $.>>2T#U( O  !   %'777$A*N^$_) EuNdfv &(888r  r)NN)r")NNN)__doc__collections.abcrtypingrrrlitellm._loggingr.litellm.litellm_core_utils.llm_cost_calc.utilsr litellm.types.utilsr r r litellm.utilsr rstrtupler?rrr;rDrJrrrUsY $&&+Q;;( 9 1T)U  $!%   *$J   5%< P'(3'(S4Z'(5'([`afhmam[n'(TssTzS!Dj T\8'+#'#' 1 11t1D 1 Dj 1  1r