From dc68a1d4d71f6c9ef65bc4fa2d67d90715a9b262 Mon Sep 17 00:00:00 2001
From: bruce <cnhxunoo@gmail.com>
Date: Mon, 29 Apr 2024 18:53:01 +0800
Subject: [PATCH] :bug: fix gemini end stream

---
 gpt_server/routers/gemini.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/gpt_server/routers/gemini.py b/gpt_server/routers/gemini.py
index bf11edf..e04d38c 100644
--- a/gpt_server/routers/gemini.py
+++ b/gpt_server/routers/gemini.py
@@ -119,8 +119,8 @@ async def get_content(data, url):
                         yield json.dumps(send_data)
                     except Exception:
                         yield '{"choices":[{"index":0,"delta":{"content":"error from gemini"}}]}'
-                        yield "[DONE]"
                         break
+                yield "[DONE]"
 
 
 @router.post("/chat", status_code=200)