Add transformChats function to prepend system message in chatGPT temp…

…late
Protofy-xyz · Nov 7, 2024 · 081389d · 081389d
1 parent e24320b
commit 081389d
Showing 1 changed file with 13 additions and 2 deletions.
diff --git a/packages/protolib/src/bundles/chatbots/templates/chatgpt-chatbot.tpl b/packages/protolib/src/bundles/chatbots/templates/chatgpt-chatbot.tpl
@@ -11,11 +11,22 @@ const logger = getLogger()
 
 Protofy("type", "chatGPT")
 
+function transformChats(prevChats, prompt: string) {
+    const additionalSystemMessage = {
+        role: "system",
+        content: prompt
+    };
+    return [additionalSystemMessage, ...prevChats];
+}
+
 export default Protofy("code", async (app:Application, context: typeof APIContext) => {
     createChatbot(app, '{{codeNameLowerCase}}', async (req, res, chatbot) => {
-        const {session, token} = getAuth(req)
+        const { metadata, ...body } = req.body
+        const { session, token } = getAuth(req)
+
         context.chatGPT.chatGPTPrompt({
-            ...req.body,
+            ...body,
+            //messages: transformChats(body.messages, metadata.context),
             done: (response, message) => {
                 chatbot.end()
             },