[fix] iterable error when use gemini-pro

This commit is contained in:
kmou424
2023-12-22 23:26:15 +08:00
parent 3e33501954
commit aa953e0634

View File

@@ -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