(test) cost calc on azure

This commit is contained in:
ishaan-jaff 2023-11-23 13:49:51 -08:00
parent 5408a40cc8
commit da7c07574f
2 changed files with 4 additions and 1 deletions

View file

@ -1612,7 +1612,7 @@ def cost_per_token(model="gpt-3.5-turbo", prompt_tokens=0, completion_tokens=0):
)
return prompt_tokens_cost_usd_dollar, completion_tokens_cost_usd_dollar
else:
# calculate average input cost
# calculate average input cost, azure/gpt-deployments can potentially go here if users don't specify, gpt-4, gpt-3.5-turbo. LLMs litellm knows
input_cost_sum = 0
output_cost_sum = 0
model_cost_ref = litellm.model_cost