From 0e4dda49f8eab0f5a40b22c4ccf25b3eb026333d Mon Sep 17 00:00:00 2001
From: kwaroran <kwaworan@gmail.com>
Date: Thu, 14 Dec 2023 00:13:52 +0900
Subject: [PATCH] [fix] mistral

---
 src/ts/process/request.ts | 46 ++++++++++++++++++++++++++++++++++++++-
 1 file changed, 45 insertions(+), 1 deletion(-)

diff --git a/src/ts/process/request.ts b/src/ts/process/request.ts
index 943c1cb8..f7ec831a 100644
--- a/src/ts/process/request.ts
+++ b/src/ts/process/request.ts
@@ -286,11 +286,55 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
 
             if(aiModel.startsWith('mistral')){
                 requestModel = aiModel
+
+                let reformatedChat:OpenAIChatExtra[] = []
+
+                for(let i=0;i<formatedChat.length;i++){
+                    const chat = formatedChat[i]
+                    if(i === 0){
+                        if(chat.role === 'user' || chat.role === 'system'){
+                            reformatedChat.push({
+                                role: chat.role,
+                                content: chat.content
+                            })
+                        }
+                        else{
+                            reformatedChat.push({
+                                role: 'system',
+                                content:  chat.role + ':' + chat.content
+                            })
+                        }
+                    }
+                    else{
+                        const prevChat = reformatedChat[i-1]
+                        if(prevChat.role === chat.role){
+                            reformatedChat[i-1].content += '\n' + chat.content
+                            continue
+                        }
+                        else if(chat.role === 'system'){
+                            if(prevChat.role === 'user'){
+                                reformatedChat[i-1].content += '\nSystem:' + chat.content
+                            }
+                            else{
+                                reformatedChat.push({
+                                    role: 'user',
+                                    content: 'System:' + chat.content
+                                })
+                            }
+                        }
+                        else{
+                            reformatedChat.push({
+                                role: chat.role,
+                                content: chat.content
+                            })
+                        }
+                    }
+                }
             
                 const res = await globalFetch("https://api.mistral.ai/v1/chat/completions", {
                     body: {
                         model: requestModel,
-                        messages: formatedChat,
+                        messages: reformatedChat,
                         temperature: temperature,
                         max_tokens: maxTokens,
                         top_p: db.top_p,