phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 11:14:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(utils.py): fix logging

Browse source
This commit is contained in:
Krrish Dholakia 2024-01-26 22:25:09 -08:00
parent e1f1e4fefb
commit 43e07503d6
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/utils.py
View file

@ -2936,7 +2936,7 @@ def cost_per_token(
f"Success: model={model_with_provider} in model_cost_map - {model_cost_ref[model_with_provider]}" f"Success: model={model_with_provider} in model_cost_map - {model_cost_ref[model_with_provider]}"
) )
print_verbose( print_verbose(
f"applying cost={model_cost_ref[model_with_provider]['input_cost_per_token']} for prompt_tokens={prompt_tokens}" f"applying cost={model_cost_ref[model_with_provider].get('input_cost_per_token', None)} for prompt_tokens={prompt_tokens}"
) )
prompt_tokens_cost_usd_dollar = ( prompt_tokens_cost_usd_dollar = (
model_cost_ref[model_with_provider]["input_cost_per_token"] * prompt_tokens model_cost_ref[model_with_provider]["input_cost_per_token"] * prompt_tokens
@ -2945,7 +2945,7 @@ def cost_per_token(
f"calculated prompt_tokens_cost_usd_dollar: {prompt_tokens_cost_usd_dollar}" f"calculated prompt_tokens_cost_usd_dollar: {prompt_tokens_cost_usd_dollar}"
) )
print_verbose( print_verbose(
f"applying cost={model_cost_ref[model_with_provider]['output_cost_per_token']} for completion_tokens={completion_tokens}" f"applying cost={model_cost_ref[model_with_provider].get('output_cost_per_token', None)} for completion_tokens={completion_tokens}"
) )
completion_tokens_cost_usd_dollar = ( completion_tokens_cost_usd_dollar = (
model_cost_ref[model_with_provider]["output_cost_per_token"] model_cost_ref[model_with_provider]["output_cost_per_token"]