diff --git a/ktransformers/server/schemas/assistants/streaming.py b/ktransformers/server/schemas/assistants/streaming.py index c5d8a04..0c3b1a7 100644 --- a/ktransformers/server/schemas/assistants/streaming.py +++ b/ktransformers/server/schemas/assistants/streaming.py @@ -73,7 +73,7 @@ class RunStepDelta(Object): 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 5c4dc4e..19505e2 100644 --- a/ktransformers/server/schemas/endpoints/chat.py +++ b/ktransformers/server/schemas/endpoints/chat.py @@ -75,4 +75,4 @@ class ChatCompletionChunk(ChatCompletionBase): ] def to_stream_reply(self): - return f"data:{self.model_dump_json()}\n\n" + return f"data: {self.model_dump_json()}\n\n"