ig dZddlmZmZmZddlmZddlmZddl m Z m Z m Z ddl mZde ded fd Z dd ed e deedeeeffdZ dd edeededeeeffdZ dd ededeedee def dZy )zR Helper util for handling openai-specific cost calculation - e.g.: prompt caching )LiteralOptionalTuple)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)r s u/Users/manta/Documents/Projects/TheRoad-I1/.venv/lib/python3.12/site-packages/litellm/llms/openai/cost_calculation.py cost_routerrs#I,,, Y=T=T0T Nmodelusage service_tierc t||d|S)aO 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 Returns: Tuple[float, float] - prompt_cost_in_usd, completion_cost_in_usd openai)rrcustom_llm_providerrr)rrrs rrrs "$!  rrdurationc8t||xsd}d}d}d|vr>|d9tjd|d|jdd||d|z}||fSd|vr<|d7tjd|d |jdd||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_second For model= - output_cost_per_second: ; duration: input_cost_per_secondz - input_cost_per_second: )r rdebugget)rrr model_info prompt_costcompletion_costs rrrVs  )<)HJKO !J. / 0 <::>>Jb;c:ddpqypz { %%=>I  '' :- . / ;9*..I`:a9bbnownx y !!89HD   ''rduration_secondsr(c4|t||xsd}|jd}|#tjd|d|d|||zS|jd}|#tjd|d|d|||zStjd |d y ) a7 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. Returns: float - total_cost_in_usd rr output_cost_per_video_per_secondr"z% - output_cost_per_video_per_second: r$r!r#z*No cost information found for video model z<. Please add pricing to model_prices_and_context_window.jsonr )r r'rr&warning)rr+rr(video_cost_per_secondr!s rvideo_generation_costr0s*#-@-LH 'NN+MN(DEZD[[ghxgy z %'777(^^,DE):;Q:RR^_o^p q &(888 4UG;wx r)N)r )NN)__doc__typingrrrlitellm._loggingr.litellm.litellm_core_utils.llm_cost_calc.utilsrlitellm.types.utilsr r r litellm.utilsr rstrfloatrrr0rrr:s ,++Q;;( 9 1T)U =A ,4SM 5%<DGJ*( *(%-c]*(>C*( 5%<*(`*.&* . .."#.# .  .r