tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-07-20 00:18:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #388 from Phodaie/main

Browse Source 
Fix typo in prompt factory templates
This commit is contained in:
Krish Dholakia
2023-09-17 08:49:54 -07:00
committed by GitHub
parent abe3286ec6 35b5d773c8
commit 0bdfd5b366
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/llms/prompt_templates/factory.py
+5 -2
View File
@@ -26,21 +26,24 @@ def falcon_instruct_pt(messages):
prompt = ""
for message in messages:
if message["role"] == "system":
prompt += messages["content"]
prompt += message["content"]
else:
prompt += message['role']+":"+ message["content"].replace("\r\n", "\n").replace("\n\n", "\n")
prompt += "\n\n"
return prompt
def falcon_chat_pt(messages):
prompt = ""
for message in messages:
if message["role"] == "system":
prompt += "System: " + messages["content"]
prompt += "System: " + message["content"]
elif message["role"] == "assistant":
prompt += "Falcon: " + message["content"]
elif message["role"] == "user":
prompt += "User: " + message["content"]
return prompt
# MPT prompt template - from https://github.com/lm-sys/FastChat/blob/main/fastchat/conversation.py#L110
def mpt_chat_pt(messages):