From 35b5d773c8c806a125480985d912caa1d7f59f71 Mon Sep 17 00:00:00 2001 From: Phodaie Date: Sun, 17 Sep 2023 15:40:36 +0000 Subject: [PATCH] code typo in falcon related prompt factory --- litellm/llms/prompt_templates/factory.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/litellm/llms/prompt_templates/factory.py b/litellm/llms/prompt_templates/factory.py index de5e6615cf..a7e041bf56 100644 --- a/litellm/llms/prompt_templates/factory.py +++ b/litellm/llms/prompt_templates/factory.py @@ -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):