Bläddra i källkod

fix bug that gemini has multiple candidates part

suruiqiang 10 månader sedan
förälder
incheckning
3fe55b4f7f
1 ändrade filer med 8 tillägg och 1 borttagningar
  1. 8 1
      app/client/platforms/google.ts

+ 8 - 1
app/client/platforms/google.ts

@@ -69,9 +69,16 @@ export class GeminiProApi implements LLMApi {
         .join("\n\n");
     };
 
+    let content = "";
+    if (Array.isArray(res)) {
+      res.map((item) => {
+        content += getTextFromParts(item?.candidates?.at(0)?.content?.parts);
+      });
+    }
+
     return (
       getTextFromParts(res?.candidates?.at(0)?.content?.parts) ||
-      getTextFromParts(res?.at(0)?.candidates?.at(0)?.content?.parts) ||
+      content || //getTextFromParts(res?.at(0)?.candidates?.at(0)?.content?.parts) ||
       res?.error?.message ||
       ""
     );