Gemini change thinking

kwaroran · Dec 19, 2024 · 01da904 · 01da904
1 parent c937a65
commit 01da904
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/src/ts/process/request.ts b/src/ts/process/request.ts
@@ -1459,7 +1459,9 @@ async function requestGoogleCloudVertex(arg:RequestDataArgumentExtended):Promise
                 parts: geminiParts,
             });
         } else if (prevChat?.role === qRole) {
-            reformatedChat[reformatedChat.length-1].parts[0].text += '\n' + chat.content
+            reformatedChat[reformatedChat.length-1].parts[
+                reformatedChat[reformatedChat.length-1].parts.length-1
+            ].text += '\n' + chat.content
             continue
         }
         else if(chat.role === 'system'){
@@ -1475,12 +1477,12 @@ async function requestGoogleCloudVertex(arg:RequestDataArgumentExtended):Promise
                 })
             }
         }
-        else if(chat.role === 'assistant' && arg.modelInfo.flags.includes(LLMFlags.geminiThinking) && chat.thoughts?.length > 0){
+        else if(chat.role === 'assistant' && arg.modelInfo.flags.includes(LLMFlags.geminiThinking)){
             reformatedChat.push({
                 role: 'MODEL',
-                parts: [{
+                parts: [chat.thoughts?.length > 0 ? {
                     text: chat.thoughts.join('\n\n')
-                }, {
+                } : null, {
                     text: chat.content
                 }]
             })