o ưig @sdZddlmZmZmZddlmZddlmZddl m Z m Z m Z ddl mZde ded fd d Z dd ede deedeeeffddZ dd edeededeeeffddZ dd ededeedee def ddZd S)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_secondcCs|tjks |tjkr dSdS)Nrr)rZatranscriptionZ transcription)r r[/home/app/Keep/.python/lib/python3.10/site-packages/litellm/llms/openai/cost_calculation.py cost_routersrNmodelusage service_tiercCst||d|dS)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)rrrrrrrs rrdurationcCst||pdd}d}d}d|vr1|ddur1td|d|dd||d|}||fSd |vrT|d durTtd|d |d d||d |}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 rrrroutput_cost_per_secondN For model= - output_cost_per_second: ; duration: Zinput_cost_per_secondz - input_cost_per_second: )r rdebugget)rrr model_infoZ prompt_costZcompletion_costrrrrVs(     rduration_secondsr!cCs|dur t||p dd}|d}|dur'td|d|d|||S|d}|durBtd|d |d|||Std |d d S) 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 NrrZ output_cost_per_video_per_secondrz% - output_cost_per_video_per_second: rrrz*No cost information found for video model z<. Please add pricing to model_prices_and_context_window.jsonr)r r rrwarning)rr"rr!Zvideo_cost_per_secondrrrrvideo_generation_costs(   r$)N)r)NN)__doc__typingrrrZlitellm._loggingrZ.litellm.litellm_core_utils.llm_cost_calc.utilsrZlitellm.types.utilsrr r Z litellm.utilsr rstrfloatrrr$rrrrsP     B  0