scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-09 03:18:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 108

fix(factory.py): handle special keys for mistral chat template

Browse source
This commit is contained in:
Krrish Dholakia 2024-07-29 17:47:33 -07:00
parent c3ac6b098e
commit 528741a809
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
litellm/llms/prompt_templates/factory.py
View file

@ -235,7 +235,7 @@ def mistral_api_pt(messages):
"""
new_messages = []
for m in messages:
special_keys = ["role", "content", "tool_calls"]
special_keys = ["role", "content", "tool_calls", "function_call"]
extra_args = {}
if isinstance(m, dict):
for k, v in m.items():