🐛 fix gemini end stream

bruceunx · Apr 29, 2024 · dc68a1d · dc68a1d
1 parent d83d593
commit dc68a1d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)