Update server.py

server.py

@@ -51,7 +51,11 @@ async def handle_sse(request: Request):
                             if placeholder_id in write_streams:
                                 write_streams[session_id] = write_streams.pop(placeholder_id)
                                 logger.info(f"Updated placeholder {placeholder_id} to session_id {session_id}")
-                    await sse_stream_writer.send({"event": "message", "data": message_data})
+                    # Send the event as a raw SSE event
+                    await sse_stream_writer.send({
+                        "event": message.get("event", "message"),
+                        "data": message.get("data", message_data)
+                    })
 
         sse_stream_writer, sse_stream_reader = anyio.create_memory_object_stream(0)
         async with transport.connect_sse(request.scope, request.receive, request._send) as (read_stream, write_stream):