From 04fd08d5ba60bcbb7c7470387393e38f7e4cc84e Mon Sep 17 00:00:00 2001 From: bennykok Date: Thu, 6 Jun 2024 19:26:11 -0700 Subject: [PATCH] fix: streaming event format --- custom_routes.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/custom_routes.py b/custom_routes.py index 3301830..3dc1abd 100644 --- a/custom_routes.py +++ b/custom_routes.py @@ -362,7 +362,8 @@ async def stream_response(request): try: result = await stream_prompt(data=data) - await response.write(json.dumps(result).encode('utf-8')) + await response.write(f"event: event_update\ndata: {json.dumps(result)}\n\n".encode('utf-8')) + # await response.write(.encode('utf-8')) await response.drain() # Ensure the buffer is flushed while pending: @@ -372,7 +373,7 @@ async def stream_response(request): logfire.info(data["event"], data=json.dumps(data)) # logger.info("listener", data) - await response.write(json.dumps(data).encode('utf-8')) + await response.write(f"event: event_update\ndata: {json.dumps(data)}\n\n".encode('utf-8')) await response.drain() # Ensure the buffer is flushed if data["event"] == "status":