ó
    Ù^íhr	  ã                   óL   • S r SSKJr  SSKJr  SSKJr  S\S\S\\\4   4S jr	g	)
zb
Helper util for handling databricks-specific cost calculation
- e.g.: handling 'dbrx-instruct-*'
é    )ÚTuple)ÚUsage)Úget_model_infoÚmodelÚusageÚreturnc                 ó>  • U nU R                  S5      (       d  U R                  S5      (       a  SnGOIU R                  S5      (       d  U R                  S5      (       a  SnGOU R                  S5      (       d  U R                  S5      (       a  S	nOêU R                  S
5      (       d  U R                  S5      (       a  SnO»U R                  S
5      (       d  U R                  S5      (       a  SnOŒU R                  S5      (       d  U R                  S5      (       a  SnO]U R                  S5      (       d  U R                  S5      (       a  SnO.U R                  S5      (       d  U R                  S5      (       a  Sn[        USS9nUS   US   -  nUS   US   -  nXE4$ )a3  
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
zdatabricks/dbrx-instructzdbrx-instructzdatabricks-dbrx-instructz&databricks/meta-llama-3.1-70b-instructzmeta-llama-3.1-70b-instructz&databricks-meta-llama-3-1-70b-instructz'databricks/meta-llama-3.1-405b-instructzmeta-llama-3.1-405b-instructz'databricks-meta-llama-3-1-405b-instructz%databricks/mixtral-8x7b-instruct-v0.1zmixtral-8x7b-instruct-v0.1z databricks-mixtral-8x7b-instructzdatabricks/bge-large-enzbge-large-enzdatabricks-bge-large-enzdatabricks/gte-large-enzgte-large-enzdatabricks-gte-large-enzdatabricks/llama-2-70b-chatzllama-2-70b-chatzdatabricks-llama-2-70b-chatÚ
databricks)r   Úcustom_llm_providerÚprompt_tokensÚinput_cost_per_tokenÚcompletion_tokensÚoutput_cost_per_token)Ú
startswithr   )r   r   Ú
base_modelÚ
model_infoÚprompt_costÚcompletion_costs         Úa/home/james-whalen/.local/lib/python3.13/site-packages/litellm/llms/databricks/cost_calculator.pyÚcost_per_tokenr      sÂ  € ð €JØ×ÑÐ2×3Ñ3°u×7GÑ7GØ÷8ñ 8ð 0Š
Ø	×	Ñ	ÐB×	CÑ	CÀu×GWÑGWØ%÷Hñ Hð >Š
Ø	×	Ñ	Ø1÷
ñ 
à	×	Ñ	Ð8×	9Ñ	9Ø>‰
Ø	×	Ñ	ÐA×	BÑ	BÀe×FVÑFVØ$÷Gñ Gð 8‰
Ø	×	Ñ	ÐA×	BÑ	BÀe×FVÑFVØ$÷Gñ Gð 8‰
Ø	×	Ñ	Ð3×	4Ñ	4¸×8HÑ8HØ÷9ñ 9ð /‰
Ø	×	Ñ	Ð3×	4Ñ	4¸×8HÑ8HØ÷9ñ 9ð /‰
Ø	×	Ñ	Ð7×	8Ñ	8¸E×<LÑ<LØ÷=ñ =ð 3ˆ
ä jÀlÑS€Jð ˜Ñ/°*Ð=SÑ2TÑT€Kð Ð/Ñ0°:Ð>UÑ3VÑV€OàÐ'Ð'ó    N)
Ú__doc__Útypingr   Úlitellm.types.utilsr   Úlitellm.utilsr   ÚstrÚfloatr   © r   r   Ú<module>r      s8   ðñõ
 å %Ý (ð6(˜#ð 6( eð 6(°°e¸U°lÑ0Cõ 6(r   