From 199bfe612fcfe1a6fae703aff4fde7ae9a9b3d24 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 26 Jun 2024 12:57:09 -0700 Subject: [PATCH] fix add ollama codegemma --- litellm/model_prices_and_context_window_backup.json | 9 +++++++++ model_prices_and_context_window.json | 9 +++++++++ 2 files changed, 18 insertions(+) diff --git a/litellm/model_prices_and_context_window_backup.json b/litellm/model_prices_and_context_window_backup.json index d7a7a7dc80..acd03aeea8 100644 --- a/litellm/model_prices_and_context_window_backup.json +++ b/litellm/model_prices_and_context_window_backup.json @@ -3369,6 +3369,15 @@ "supports_function_calling": true, "supports_parallel_function_calling": true }, + "ollama/codegemma": { + "max_tokens": 8192, + "max_input_tokens": 8192, + "max_output_tokens": 8192, + "input_cost_per_token": 0.0, + "output_cost_per_token": 0.0, + "litellm_provider": "ollama", + "mode": "completion" + }, "ollama/llama2": { "max_tokens": 4096, "max_input_tokens": 4096, diff --git a/model_prices_and_context_window.json b/model_prices_and_context_window.json index d7a7a7dc80..acd03aeea8 100644 --- a/model_prices_and_context_window.json +++ b/model_prices_and_context_window.json @@ -3369,6 +3369,15 @@ "supports_function_calling": true, "supports_parallel_function_calling": true }, + "ollama/codegemma": { + "max_tokens": 8192, + "max_input_tokens": 8192, + "max_output_tokens": 8192, + "input_cost_per_token": 0.0, + "output_cost_per_token": 0.0, + "litellm_provider": "ollama", + "mode": "completion" + }, "ollama/llama2": { "max_tokens": 4096, "max_input_tokens": 4096,