-
Notifications
You must be signed in to change notification settings - Fork 0
/
prompt.js
39 lines (32 loc) · 1.37 KB
/
prompt.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
function prmt() {}
prmt.conversationHistory = "";
prmt.promptFormatCML = function (systemPrompt, userPrompt, context, isFirstMessage = false) {
let formattedPrompt = "";
let embedding = `${context ? `\n Context: '${context}'` : ""}`;
formattedPrompt = `
<|im_start|>system
${systemPrompt}<|im_end|>
<|im_start|>user
${userPrompt} ${embedding} <|im_end|>
<|im_start|>assistant
`;
this.conversationHistory = formattedPrompt;
return formattedPrompt;
}
prmt.promptFormatLAMA3 = function (systemPrompt, userPrompt, context, isFirstMessage = false) {
let formattedPrompt = "";
let embedding = `${context ? `\n Context: '${context}'` : ""}`;
formattedPrompt = `<|begin_of_text|><|start_header_id|>system<|end_header_id|>\n\n${systemPrompt}<|eot_id|><|start_header_id|>user<|end_header_id|>\n\n${userPrompt} ${embedding}<|eot_id|><|start_header_id|>assistant<|end_header_id|>\n\n`;
this.conversationHistory += formattedPrompt;
return formattedPrompt;
}
prmt.promptFormatNONE= function (systemPrompt, userPrompt, context, isFirstMessage = false) {
let formattedPrompt = "";
let embedding = `${context ? `\n Context: '${context}'` : ""}`;
formattedPrompt = `${userPrompt} ${embedding}`;
this.conversationHistory = formattedPrompt;
return formattedPrompt;
}
try {
module.exports = exports = prmt;
} catch (e) {}