From e1ac385395e0434065dba9115337107a62cc89db Mon Sep 17 00:00:00 2001 From: Anthony Date: Mon, 7 Apr 2025 12:43:56 +0200 Subject: [PATCH] following SSE standard with data: --- main.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/main.py b/main.py index f6bafd4..15d3ed9 100644 --- a/main.py +++ b/main.py @@ -685,8 +685,11 @@ async def generate_openai_chat_completion(form_data: OpenAIChatCompletionForm): def stream_content(): def __event_emitter__(event): - logging.error(f"stream_event:{event}") - return f"event: {json.dumps(event)}\n\n" + logging.info(f"stream_event:{event}") + eventData = { + "event" : event + } + return f"data: {json.dumps(eventData)}\n\n" res = pipe( user_message=user_message, @@ -716,7 +719,7 @@ async def generate_openai_chat_completion(form_data: OpenAIChatCompletionForm): logging.info(f"stream_content:Generator:{line}") - if line.startswith("data:") or line.startswith("event:"): + if line.startswith("data:"): yield f"{line}\n\n" else: line = stream_message_template(form_data.model, line)