From 041e60a92c9aee986076e0eb0efae8dfb4f3ac05 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Wed, 28 Feb 2024 17:15:50 -0800 Subject: [PATCH] (feat) track mistral model supports function calling --- litellm/model_prices_and_context_window_backup.json | 3 ++- litellm/tests/test_function_calling.py | 2 +- model_prices_and_context_window.json | 3 ++- 3 files changed, 5 insertions(+), 3 deletions(-) diff --git a/litellm/model_prices_and_context_window_backup.json b/litellm/model_prices_and_context_window_backup.json index ad934b39fee..ae25001f46e 100644 --- a/litellm/model_prices_and_context_window_backup.json +++ b/litellm/model_prices_and_context_window_backup.json @@ -555,7 +555,8 @@ "input_cost_per_token": 0.000008, "output_cost_per_token": 0.000024, "litellm_provider": "mistral", - "mode": "chat" + "mode": "chat", + "supports_function_calling": true }, "mistral/mistral-embed": { "max_tokens": 8192, diff --git a/litellm/tests/test_function_calling.py b/litellm/tests/test_function_calling.py index ffef8f6594f..44cf9d8c675 100644 --- a/litellm/tests/test_function_calling.py +++ b/litellm/tests/test_function_calling.py @@ -71,7 +71,7 @@ def test_parallel_function_call(): } ] response = litellm.completion( - model="gpt-3.5-turbo-1106", + model="mistral/mistral-medium-latest", messages=messages, tools=tools, tool_choice="auto", # auto is default, but we'll be explicit diff --git a/model_prices_and_context_window.json b/model_prices_and_context_window.json index ad934b39fee..ae25001f46e 100644 --- a/model_prices_and_context_window.json +++ b/model_prices_and_context_window.json @@ -555,7 +555,8 @@ "input_cost_per_token": 0.000008, "output_cost_per_token": 0.000024, "litellm_provider": "mistral", - "mode": "chat" + "mode": "chat", + "supports_function_calling": true }, "mistral/mistral-embed": { "max_tokens": 8192,