Skip to content

Commit 29f24f6

Browse files
author
junjie.miao
committed
fix sse get non-exist key get lots error logs on the plugin daemon
1 parent b43cf03 commit 29f24f6

File tree

2 files changed

+6
-11
lines changed

2 files changed

+6
-11
lines changed

endpoints/sse.py

Lines changed: 5 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -23,16 +23,11 @@ def generate():
2323
yield create_sse_message("endpoint", endpoint)
2424

2525
while True:
26-
message = None
27-
try:
26+
if self.session.storage.exist(session_id):
2827
message = self.session.storage.get(session_id)
29-
except:
30-
pass
31-
if message is None:
32-
time.sleep(0.5)
33-
continue
34-
message = message.decode()
35-
self.session.storage.delete(session_id)
36-
yield create_sse_message("message", message)
28+
message = message.decode()
29+
self.session.storage.delete(session_id)
30+
yield create_sse_message("message", message)
31+
time.sleep(0.5)
3732

3833
return Response(generate(), status=200, content_type="text/event-stream")

requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
dify_plugin>=0.1.0,<0.2.0
1+
dify_plugin>=0.2.1,<0.3.0

0 commit comments

Comments
 (0)