From c46a21c1e47b730a995ee23aeeff68ace7adf7d0 Mon Sep 17 00:00:00 2001
From: bruce <cnhxunoo@gmail.com>
Date: Mon, 29 Apr 2024 18:58:03 +0800
Subject: [PATCH] ...

---
 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 e04d38c..bf11edf 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)