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

(tests) easily track cost for responses

Browse source
This commit is contained in:
ishaan-jaff 2023-10-28 15:08:34 -07:00
parent 0069b09caa
commit e80f07c76a
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/tests/test_completion.py
View file

@ -51,10 +51,12 @@ def test_completion_claude():
print(response.usage) print(response.usage)
print(response.usage.completion_tokens) print(response.usage.completion_tokens)
print(response["usage"]["completion_tokens"]) print(response["usage"]["completion_tokens"])
# print("new cost tracking")
print(response.cost())
except Exception as e: except Exception as e:
pytest.fail(f"Error occurred: {e}") pytest.fail(f"Error occurred: {e}")
# test_completion_claude() test_completion_claude()
# def test_completion_oobabooga(): # def test_completion_oobabooga():
# try: # try: