From 68113cc3839c26de3cafc655a391d4023d0dd9f4 Mon Sep 17 00:00:00 2001
From: Kwaroran <amasio1234@proton.me>
Date: Wed, 27 Nov 2024 08:03:17 +0900
Subject: [PATCH] Add system prompt handling to requestGoogleCloudVertex
 function

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

diff --git a/src/ts/process/request.ts b/src/ts/process/request.ts
index a4d8432f..374b3754 100644
--- a/src/ts/process/request.ts
+++ b/src/ts/process/request.ts
@@ -1365,6 +1365,12 @@ async function requestGoogleCloudVertex(arg:RequestDataArgumentExtended):Promise
 
     let reformatedChat:GeminiChat[] = []
     let pendingImage = ''
+    let systemPrompt = ''
+
+    if(formated[0].role === 'system'){
+        systemPrompt = formated[0].content
+        formated.shift()
+    }
 
     for(let i=0;i<formated.length;i++){
         const chat = formated[i]
@@ -1490,7 +1496,14 @@ async function requestGoogleCloudVertex(arg:RequestDataArgumentExtended):Promise
         }, ['temperature', 'top_p'], {
             'top_p': "topP"
         }, arg.mode),
-        safetySettings: uncensoredCatagory
+        safetySettings: uncensoredCatagory,
+        systemInstruction: {
+            parts: [
+                {
+                    "text": systemPrompt
+                }
+            ]
+        },
     }
 
     let headers:{[key:string]:string} = {}