forked from phoenix/litellm-mirror
(test) gemini-pro-vision cost tracking
This commit is contained in:
parent
739d9e7a78
commit
06e4b301b4
1 changed files with 7 additions and 0 deletions
|
@ -294,6 +294,13 @@ def test_gemini_pro_vision():
|
||||||
],
|
],
|
||||||
)
|
)
|
||||||
print(resp)
|
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:
|
except Exception as e:
|
||||||
import traceback
|
import traceback
|
||||||
|
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue