From 06e4b301b402eb695d5f8cd6e64def253f040115 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Fri, 29 Dec 2023 16:31:28 +0530 Subject: [PATCH] (test) gemini-pro-vision cost tracking --- litellm/tests/test_amazing_vertex_completion.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/litellm/tests/test_amazing_vertex_completion.py b/litellm/tests/test_amazing_vertex_completion.py index 755788be4..a56e0343c 100644 --- a/litellm/tests/test_amazing_vertex_completion.py +++ b/litellm/tests/test_amazing_vertex_completion.py @@ -294,6 +294,13 @@ def test_gemini_pro_vision(): ], ) print(resp) + + prompt_tokens = resp.usage.prompt_tokens + + # DO Not DELETE this ASSERT + # Google counts the prompt tokens for us, we should ensure we use the tokens from the orignal response + assert prompt_tokens == 263 # the gemini api returns 263 to us + except Exception as e: import traceback