From 8e36f667e59965ed5bbd15f74f83f893ccdd7889 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Fri, 27 Oct 2023 09:39:55 -0700 Subject: [PATCH] (test) vertex ai use usage class --- litellm/tests/test_completion.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py index 86086489bb..239c490bb1 100644 --- a/litellm/tests/test_completion.py +++ b/litellm/tests/test_completion.py @@ -1031,6 +1031,9 @@ def test_completion_together_ai(): # print("making request", model) # response = completion(model="vertex_ai/codechat-bison-32k", messages=[{'role': 'user', 'content': 'hi'}]) # print(response) + +# print(response.usage.completion_tokens) +# print(response['usage']['completion_tokens']) # assert type(response.choices[0].message.content) == str # except Exception as e: # pytest.fail(f"Error occurred: {e}")