alihan/OpenPipe-llm
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Convert system to user prompt for airoboros

Browse Source
This commit is contained in:
David Corbitt
2023-08-17 03:10:55 -07:00
parent 687f3dd85f
commit edac8da4a8
1 changed files with 14 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
app/src/modelProviders/openpipe-chat/templatePrompt.ts
View File

@@ -156,7 +156,18 @@ export const templateAiroborosPrompt = (messages: OpenpipeChatInput["messages"])
} }
} }
let prompt = `${combinedSystemMessage}\n${conversationMessages.join(splitter)}`; let systemMessage = "";
if (combinedSystemMessage) {
// If there is no user message, add a user tag to the system message
if (conversationMessages.find((message) => message.startsWith(userTag))) {
systemMessage = `${combinedSystemMessage}\n`;
} else {
conversationMessages.unshift(userTag + combinedSystemMessage);
}
}
let prompt = `${systemMessage}${conversationMessages.join(splitter)}`;
// Ensure that the prompt ends with an assistant message // Ensure that the prompt ends with an assistant message
const lastUserIndex = prompt.lastIndexOf(userTag); const lastUserIndex = prompt.lastIndexOf(userTag);
@@ -166,5 +177,7 @@ export const templateAiroborosPrompt = (messages: OpenpipeChatInput["messages"])
prompt += splitter + assistantTag; prompt += splitter + assistantTag;
} }
console.log("prompt is", prompt);
return prompt; return prompt;
}; };