From 4b8a1800c9ebb43f4c36f43799e12c348393d8af Mon Sep 17 00:00:00 2001 From: Ash Greh Date: Fri, 29 Dec 2023 20:45:25 -0400 Subject: [PATCH 1/3] Update model_prices_and_context_window.json Adds support for unicorn --- model_prices_and_context_window.json | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/model_prices_and_context_window.json b/model_prices_and_context_window.json index 44085a2ed6..5bbb2a3ea8 100644 --- a/model_prices_and_context_window.json +++ b/model_prices_and_context_window.json @@ -275,6 +275,13 @@ "litellm_provider": "vertex_ai-text-models", "mode": "completion" }, + "text-unicorn@001": { + "max_tokens": 8192, + "input_cost_per_token": 0.000001, + "output_cost_per_token": 0.000028, + "litellm_provider": "vertex_ai-chat-models", + "mode": "completion" + } "chat-bison": { "max_tokens": 4096, "input_cost_per_token": 0.000000125, From b0aea659bc1174d35dee87601d201adfc600b1e1 Mon Sep 17 00:00:00 2001 From: Ash Greh Date: Fri, 29 Dec 2023 20:47:12 -0400 Subject: [PATCH 2/3] Update model_prices_and_context_window.json Adds simplified name for Unicorn. --- model_prices_and_context_window.json | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/model_prices_and_context_window.json b/model_prices_and_context_window.json index 5bbb2a3ea8..6dc3113ec7 100644 --- a/model_prices_and_context_window.json +++ b/model_prices_and_context_window.json @@ -275,13 +275,20 @@ "litellm_provider": "vertex_ai-text-models", "mode": "completion" }, + "text-unicorn": { + "max_tokens": 8192, + "input_cost_per_token": 0.000001, + "output_cost_per_token": 0.000028, + "litellm_provider": "vertex_ai-chat-models", + "mode": "completion" + }, "text-unicorn@001": { "max_tokens": 8192, "input_cost_per_token": 0.000001, "output_cost_per_token": 0.000028, "litellm_provider": "vertex_ai-chat-models", "mode": "completion" - } + }, "chat-bison": { "max_tokens": 4096, "input_cost_per_token": 0.000000125, From f23f227087a6e34d9510567a6f9f8bddeea10b3c Mon Sep 17 00:00:00 2001 From: Ash Greh Date: Fri, 29 Dec 2023 20:53:39 -0400 Subject: [PATCH 3/3] Update model_prices_and_context_window.json Corrected the math. --- model_prices_and_context_window.json | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/model_prices_and_context_window.json b/model_prices_and_context_window.json index 6dc3113ec7..b946c09a52 100644 --- a/model_prices_and_context_window.json +++ b/model_prices_and_context_window.json @@ -277,14 +277,14 @@ }, "text-unicorn": { "max_tokens": 8192, - "input_cost_per_token": 0.000001, + "input_cost_per_token": 0.00001, "output_cost_per_token": 0.000028, "litellm_provider": "vertex_ai-chat-models", "mode": "completion" }, "text-unicorn@001": { "max_tokens": 8192, - "input_cost_per_token": 0.000001, + "input_cost_per_token": 0.00001, "output_cost_per_token": 0.000028, "litellm_provider": "vertex_ai-chat-models", "mode": "completion"