From b41c012d27d5495bec12f6aa6f9537ebb6873083 Mon Sep 17 00:00:00 2001
From: DDMeaqua <lzlloveyou@vip.qq.com>
Date: Wed, 13 Nov 2024 15:12:46 +0800
Subject: [PATCH] chore: shouldStream

---
 app/client/platforms/google.ts | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/app/client/platforms/google.ts b/app/client/platforms/google.ts
index 30f35359e89..a7bce4fc2d0 100644
--- a/app/client/platforms/google.ts
+++ b/app/client/platforms/google.ts
@@ -29,7 +29,7 @@ import { RequestPayload } from "./openai";
 import { fetch } from "@/app/utils/stream";
 
 export class GeminiProApi implements LLMApi {
-  path(path: string): string {
+  path(path: string, shouldStream = false): string {
     const accessStore = useAccessStore.getState();
 
     let baseUrl = "";
@@ -51,7 +51,7 @@ export class GeminiProApi implements LLMApi {
     console.log("[Proxy Endpoint] ", baseUrl, path);
 
     let chatPath = [baseUrl, path].join("/");
-    if (!chatPath.includes("gemini-pro")) {
+    if (shouldStream) {
       chatPath += chatPath.includes("?") ? "&alt=sse" : "?alt=sse";
     }
 
@@ -169,7 +169,10 @@ export class GeminiProApi implements LLMApi {
     options.onController?.(controller);
     try {
       // https://github.com/google-gemini/cookbook/blob/main/quickstarts/rest/Streaming_REST.ipynb
-      const chatPath = this.path(Google.ChatPath(modelConfig.model));
+      const chatPath = this.path(
+        Google.ChatPath(modelConfig.model),
+        shouldStream,
+      );
 
       const chatPayload = {
         method: "POST",