From aa953e0634696dec24800c2b2c79ca1dd73da278 Mon Sep 17 00:00:00 2001
From: kmou424 <me@kmou424.moe>
Date: Fri, 22 Dec 2023 23:26:15 +0800
Subject: [PATCH] [fix] iterable error when use gemini-pro

---
 src/ts/process/request.ts | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/src/ts/process/request.ts b/src/ts/process/request.ts
index d2644b11..ab62dd9a 100644
--- a/src/ts/process/request.ts
+++ b/src/ts/process/request.ts
@@ -1194,8 +1194,7 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
 
             let fullRes = ''
 
-            for(const data of res.data){
-
+            const processDataItem = (data:any) => {
                 if(data?.candidates?.[0]?.content?.parts?.[0]?.text){
                     fullRes += data.candidates[0].content.parts[0].text
                 }
@@ -1213,6 +1212,15 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
                 }
             }
 
+            // traverse responded data if it contains multipart contents
+            if (typeof (res.data)[Symbol.iterator] === 'function') {
+                for(const data of res.data){
+                    processDataItem(data)
+                }
+            } else {
+                processDataItem(res.data)
+            }
+
             return {
                 type: 'success',
                 result: fullRes