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

(test) cost calc on azure

Browse source
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
Download patch file Download diff file Expand all files Collapse all files

3
litellm/tests/test_completion.py
View file

@ -594,6 +594,9 @@ def test_completion_azure():
# )
# Add any assertions here to check the response
print(response)
cost = completion_cost(completion_response=response)
print("Cost for azure completion request", cost)
except Exception as e:
pytest.fail(f"Error occurred: {e}")

2
litellm/utils.py
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