From 070a2a044840e805928f3c4faf11d681d3f84178 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Wed, 28 Feb 2024 17:21:43 -0800 Subject: [PATCH] (feat) track if mode supports function calling --- litellm/model_prices_and_context_window_backup.json | 3 ++- model_prices_and_context_window.json | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/litellm/model_prices_and_context_window_backup.json b/litellm/model_prices_and_context_window_backup.json index ae25001f46e..7faa3b2753a 100644 --- a/litellm/model_prices_and_context_window_backup.json +++ b/litellm/model_prices_and_context_window_backup.json @@ -714,7 +714,8 @@ "input_cost_per_token": 0.00000025, "output_cost_per_token": 0.0000005, "litellm_provider": "vertex_ai-language-models", - "mode": "chat" + "mode": "chat", + "supports_function_calling": true }, "gemini-1.5-pro": { "max_tokens": 8192, diff --git a/model_prices_and_context_window.json b/model_prices_and_context_window.json index ae25001f46e..7faa3b2753a 100644 --- a/model_prices_and_context_window.json +++ b/model_prices_and_context_window.json @@ -714,7 +714,8 @@ "input_cost_per_token": 0.00000025, "output_cost_per_token": 0.0000005, "litellm_provider": "vertex_ai-language-models", - "mode": "chat" + "mode": "chat", + "supports_function_calling": true }, "gemini-1.5-pro": { "max_tokens": 8192,