This commit is contained in:
qwertzman-png 2023-04-21 00:41:27 +02:00 committed by GitHub
parent 8346dae155
commit 3530a9532c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -287,6 +287,7 @@ class PromptServer():
async def send(self, event, data, sid=None): async def send(self, event, data, sid=None):
message = {"type": event, "data": data} message = {"type": event, "data": data}
self.poll_messages.append({"type": event, "data": data, "sid": sid, "poll_id": poll_id++})
if isinstance(message, str) == False: if isinstance(message, str) == False:
message = json.dumps(message) message = json.dumps(message)
@ -298,7 +299,6 @@ class PromptServer():
await self.sockets[sid].send_str(message) await self.sockets[sid].send_str(message)
def send_sync(self, event, data, sid=None): def send_sync(self, event, data, sid=None):
self.poll_messages.append({"type": event, "data": data, "sid": sid, "poll_id": poll_id++})
self.loop.call_soon_threadsafe( self.loop.call_soon_threadsafe(
self.messages.put_nowait, (event, data, sid)) self.messages.put_nowait, (event, data, sid))