From b05cc5685d529ee76744dbfada21b656f9a0f4e0 Mon Sep 17 00:00:00 2001
From: xv44586 <xv44586@gmail.com>
Date: Thu, 20 Feb 2025 15:01:35 +0800
Subject: [PATCH] fix: fix SSE formatting

---
 ktransformers/server/schemas/assistants/streaming.py | 2 +-
 ktransformers/server/schemas/endpoints/chat.py       | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/ktransformers/server/schemas/assistants/streaming.py b/ktransformers/server/schemas/assistants/streaming.py
index c5d8a045..0c3b1a7e 100644
--- a/ktransformers/server/schemas/assistants/streaming.py
+++ b/ktransformers/server/schemas/assistants/streaming.py
@@ -73,7 +73,7 @@ def to_stream_reply(self):
 
 class Done():
     def to_stream_reply(self):
-        return f"event: done\ndata: [DONE]\n\n"
+        return f"data: [DONE]\n\n"
 
 
 async def check_client_link(request: Request, async_events: AsyncIterable):
diff --git a/ktransformers/server/schemas/endpoints/chat.py b/ktransformers/server/schemas/endpoints/chat.py
index 5c4dc4ec..19505e23 100644
--- a/ktransformers/server/schemas/endpoints/chat.py
+++ b/ktransformers/server/schemas/endpoints/chat.py
@@ -75,4 +75,4 @@ def set_token(self,token:str):
         ]
 
     def to_stream_reply(self):
-        return f"data:{self.model_dump_json()}\n\n"
+        return f"data: {self.model_dump_json()}\n\n"