scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-11 22:51:28 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 123

Merge pull request #3809 from BerriAI/litellm_add_pricing_forvertex_imagen

Browse source 
[Feat] Add Cost Tracking for `vertex_ai/imagegeneration@006`
This commit is contained in:
Ishaan Jaff 2024-05-23 16:40:25 -07:00 • committed by GitHub
commit 8febdadef4
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 17 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
litellm/model_prices_and_context_window_backup.json
View file

@ -1272,6 +1272,12 @@
"supports_function_calling": true,
"supports_vision": true
},
"vertex_ai/imagegeneration@006": {
"cost_per_image": 0.020,
"litellm_provider": "vertex_ai-image-models",
"mode": "image_generation",
"source": "https://cloud.google.com/vertex-ai/generative-ai/pricing"
},
"textembedding-gecko": {
"max_tokens": 3072,
"max_input_tokens": 3072,

5
litellm/utils.py
View file

@ -4659,6 +4659,11 @@ def completion_cost(
or call_type == CallTypes.aimage_generation.value
):
### IMAGE GENERATION COST CALCULATION ###
if custom_llm_provider == "vertex_ai":
# https://cloud.google.com/vertex-ai/generative-ai/pricing
# Vertex Charges Flat $0.20 per image
return 0.020
# fix size to match naming convention
if "x" in size and "-x-" not in size:
size = size.replace("x", "-x-")

6
model_prices_and_context_window.json
View file

@ -1272,6 +1272,12 @@
"supports_function_calling": true,
"supports_vision": true
},
"vertex_ai/imagegeneration@006": {
"cost_per_image": 0.020,
"litellm_provider": "vertex_ai-image-models",
"mode": "image_generation",
"source": "https://cloud.google.com/vertex-ai/generative-ai/pricing"
},
"textembedding-gecko": {
"max_tokens": 3072,
"max_input_tokens": 3072,