h `dZddlZddlmZmZmZmZmZmZm Z m Z ddl m Z de de fdZde fdZy) z1 Handles calculating cost for together ai models N)TOGETHER_AI_4_BTOGETHER_AI_8_BTOGETHER_AI_21_BTOGETHER_AI_41_BTOGETHER_AI_80_BTOGETHER_AI_110_BTOGETHER_AI_EMBEDDING_150_MTOGETHER_AI_EMBEDDING_350_M) CallTypes call_typereturnc|tjk(s|tjk(r t|S|j }t j d|}d}|t|jd}|jdd}| t|}n|S|tkrd}n;|tkrd}n/|tkrd }n#|tkrd }n|tkrd }n |t krd }||S|S) z Helper function for calculating together ai pricing. Returns - str - model pricing category if mapped else received model name ) model_namez(\d+b)Nbztogether-ai-up-to-4bztogether-ai-4.1b-8bztogether-ai-8.1b-21bztogether-ai-21.1b-41bztogether-ai-41.1b-80bztogether-ai-81.1b-110b)r embedding aembedding(get_model_params_and_category_embeddingslowerresearchstrgroupreplacefloatrrrrrr)rr re_params_matchcategory params_matchparams_billions d/var/www/Befach/backend/env/lib/python3.12/site-packages/litellm/llms/together_ai/cost_calculator.pyget_model_params_and_categoryr"sI'''9 8L8L+L7:NN!!#Jii:OH"?0034 #++C4  #"<0N  _ ,-H  .,H / /-H / /.H / /.H 0 0/H  O c|j}tjd|}d}|Wt|j d}|j dd}| t |}n|S|tkrd}n |tkrd}||S|S)z Helper function for calculating together ai embedding pricing. Returns - str - model pricing category if mapped else received model name z(\d+m)Nrmrz together-ai-embedding-up-to-150mz"together-ai-embedding-151m-to-350m) rrrrrrrr r )rrrrparams_millions r!rr>s!!#Jii:OH"?0034 #++C4  #"<0N  8 89H : :;H  O r#)__doc__rlitellm.constantsrrrrrrr r litellm.types.utilsr rr"rr#r!r+sC    * %%s%PCr#