scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-11 22:51:28 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 124

fix(factory.py): fix mistral api prompt formatting

Browse source
This commit is contained in:
Krrish Dholakia 2024-03-13 11:34:45 -07:00
parent ac867daadc
commit dfac742324
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -137,6 +137,8 @@ def mistral_api_pt(messages):
return messages
elif c["type"] == "text" and isinstance(c["text"], str):
texts += c["text"]
elif isinstance(m["content"], str):
texts = m["content"]
new_m = {"role": m["role"], "content": texts}
new_messages.append(new_m)
return new_messages