mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-03 02:22:24 +00:00
Merge a217303766 into 22b36cbcf6
This commit is contained in:
commit
dfa30bc7b5
1 changed files with 7 additions and 0 deletions
|
|
@ -308,10 +308,17 @@ class TestProcessResponse:
|
|||
)
|
||||
|
||||
|
||||
@pytest.mark.usefixtures("local_model_cost_map")
|
||||
class TestProcessEmbedContentResponseUsage:
|
||||
"""Gemini Embedding 2 embedContent usageMetadata must drive spend.
|
||||
|
||||
Regression for multimodal calls recording prompt_tokens=0 / spend=$0.
|
||||
|
||||
The spend assertions read the ``gemini-embedding-2`` rates, so they must be
|
||||
pinned to the bundled in-repo cost map: the default network-fetched ``main``
|
||||
copy is a different branch and already re-prices this model per modality
|
||||
token (no ``input_cost_per_image`` / ``input_cost_per_*_per_second``), which
|
||||
makes these cases fail for reasons unrelated to the transformation.
|
||||
"""
|
||||
|
||||
MODEL = "gemini-embedding-2"
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue