(tests) easily track cost for responses

This commit is contained in:
ishaan-jaff 2023-10-28 15:08:34 -07:00
parent 0069b09caa
commit e80f07c76a

View file

@ -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: