phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 03:04:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

add cost tracking for pass through imagen

Browse source
This commit is contained in:
Ishaan Jaff 2024-09-02 18:10:46 -07:00
parent 9fcab392a4
commit 4a0fdc40f1
4 changed files with 12 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/tests/test_get_llm_provider.py
View file

@ -72,6 +72,6 @@ def test_get_llm_provider_deepseek_custom_api_base():
def test_get_llm_provider_vertex_ai_image_models():
model, custom_llm_provider, dynamic_api_key, api_base = litellm.get_llm_provider(
model="imagegeneration@006",
model="imagegeneration@006", custom_llm_provider=None
)
assert custom_llm_provider == "vertex_ai"