From 8a8d3f686c87eb5c892bf8dd19469dfec40fe45e Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Sat, 28 Oct 2023 15:08:34 -0700 Subject: [PATCH] (tests) easily track cost for responses --- litellm/tests/test_completion.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py index 239c490bb16..a6d44efc3ee 100644 --- a/litellm/tests/test_completion.py +++ b/litellm/tests/test_completion.py @@ -51,10 +51,12 @@ def test_completion_claude(): print(response.usage) print(response.usage.completion_tokens) print(response["usage"]["completion_tokens"]) + # print("new cost tracking") + print(response.cost()) except Exception as e: pytest.fail(f"Error occurred: {e}") -# test_completion_claude() +test_completion_claude() # def test_completion_oobabooga(): # try: