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)