ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

(feat) track mistral model supports function calling

Browse Source
This commit is contained in:
ishaan-jaff 2024-02-28 17:15:50 -08:00
parent e5269fdb7c
commit 041e60a92c
3 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
litellm/model_prices_and_context_window_backup.json
View File

@ -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,

2
litellm/tests/test_function_calling.py
View File

@ -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

3
model_prices_and_context_window.json
View File

@ -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,