summaryrefslogtreecommitdiff
path: root/continuedev/src
diff options
context:
space:
mode:
authorNate Sesti <sestinj@gmail.com>2023-09-06 20:56:51 -0700
committerNate Sesti <sestinj@gmail.com>2023-09-06 20:56:51 -0700
commit568771d9b94280f1cb47aae863e8faf168eb052b (patch)
treeb7c6aad451d2512529adfac979761faaff195d4e /continuedev/src
parentdb19f6bc98285d8ea45b4db16f619dffbec7c3db (diff)
downloadsncontinue-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.py16
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 ""