uPj. UdZddlmZddlmZddlmZddlmZedddd d d d d Z ee d<de de fdZ de dede eeffdZy)zb Helper util for handling databricks-specific cost calculation - e.g.: handling 'dbrx-instruct-*' )MappingProxyType)Final)generic_cost_per_token)Usagezdatabricks-dbrx-instructz&databricks-meta-llama-3-1-70b-instructz'databricks-meta-llama-3-1-405b-instructz databricks-mixtral-8x7b-instructzdatabricks-bge-large-enzdatabricks-gte-large-enzdatabricks-llama-2-70b-chat)z dbrx-instructzmeta-llama-3.1-70b-instructzmeta-llama-3.1-405b-instructzmixtral-8x7b-instruct-v0.1z bge-large-enz gte-large-enzllama-2-70b-chat_LEGACY_ENDPOINT_NAMESmodelreturnct|jdtfdtjDS)Nz databricks/c3LK|]\}}j|s|yw)N) startswith).0prefixkeynames l/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/databricks/cost_calculator.py z _registry_key..s!Z$//RXBYZs$$) removeprefixnextritems)rrs @r _registry_keyrs5$$]3D Z 6 < < >Z  usagec0tt||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 databricks)rrcustom_llm_provider)rr)rrs rcost_per_tokenr!s "E"( rN)__doc__typesrtypingr.litellm.litellm_core_utils.llm_cost_calc.utilsrlitellm.types.utilsrr__annotations__strrtuplefloatrrrr'sz #Q% 03'O(Q&H119 ! #eeUl0Cr