diff --git a/server.py b/server.py index 50bdcad2b..a9ed4f314 100644 --- a/server.py +++ b/server.py @@ -286,7 +286,7 @@ class PromptServer(): async def send(self, event, data, sid=None): message = {"type": event, "data": data} - self.poll_messages.append(message) + self.poll_messages.append({"type": event, "data": data, "sid": sid}) if isinstance(message, str) == False: message = json.dumps(message)