diff options
author | Nate Sesti <sestinj@gmail.com> | 2023-09-06 20:56:51 -0700 |
---|---|---|
committer | Nate Sesti <sestinj@gmail.com> | 2023-09-06 20:56:51 -0700 |
commit | 568771d9b94280f1cb47aae863e8faf168eb052b (patch) | |
tree | b7c6aad451d2512529adfac979761faaff195d4e /continuedev/src | |
parent | db19f6bc98285d8ea45b4db16f619dffbec7c3db (diff) | |
download | sncontinue-568771d9b94280f1cb47aae863e8faf168eb052b.tar.gz sncontinue-568771d9b94280f1cb47aae863e8faf168eb052b.tar.bz2 sncontinue-568771d9b94280f1cb47aae863e8faf168eb052b.zip |
feat: :globe_with_meridians: alpaca chat template
Diffstat (limited to 'continuedev/src')
-rw-r--r-- | continuedev/src/continuedev/libs/llm/prompts/chat.py | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/continuedev/src/continuedev/libs/llm/prompts/chat.py b/continuedev/src/continuedev/libs/llm/prompts/chat.py index 1329a2ff..03230499 100644 --- a/continuedev/src/continuedev/libs/llm/prompts/chat.py +++ b/continuedev/src/continuedev/libs/llm/prompts/chat.py @@ -21,6 +21,22 @@ def anthropic_template_messages(messages: List[Dict[str, str]]) -> str: return prompt +def template_alpaca_messages(msgs: List[Dict[str, str]]) -> str: + prompt = "" + + if msgs[0]["role"] == "system": + prompt += f"{msgs[0]['content']}\n" + msgs.pop(0) + + prompt += "### Instruction:\n" + for msg in msgs: + prompt += f"{msg['content']}\n" + + prompt += "### Response:\n" + + return prompt + + def llama2_template_messages(msgs: List[Dict[str, str]]) -> str: if len(msgs) == 0: return "" |