j  dZddlmZmZmZddlmZddlmZddl m Z m Z ddl m Z de ded fd Z dd ed e deedeeeffdZ dd edeededeeeffdZ dd ededeedefdZy )zR Helper util for handling openai-specific cost calculation - e.g.: prompt caching )LiteralOptionalTuple)verbose_loggergeneric_cost_per_token) CallTypesUsage)get_model_info call_typereturn)cost_per_tokencost_per_secondcR|tjk(s|tjk(ryy)Nrr)r atranscription transcription)r s }/Users/manta/Documents/Projects/TheRoad-I1/backend/.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_secondsc0t||xsd}|jd}|#tjd|d|d|||zS|jd}|#tjd|d|d|||zStjd |d y ) aW 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 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_costr/s )<)HJ 'NN+MN(DEZD[[ghxgy z %'777(^^,DE):;Q:RR^_o^p q &(888 4UG;wx r)N)r)__doc__typingrrrlitellm._loggingr.litellm.litellm_core_utils.llm_cost_calc.utilsrlitellm.types.utilsr r litellm.utilsr rstrfloatrrr/rrr9s ,++Q0( 9 1T)U =A ,4SM 5%<DGJ*( *(%-c]*(>C*( 5%<*(\OS' '"''>Fsm' 'r