Merge pull request #1 from qbit-/master

Fixing json decoding failure for Dify 0.6.6
haoyuhu · Jun 5, 2024 · 0ffc784 · 0ffc784
2 parents 246ec27 + e89e0c2
commit 0ffc784
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dify_client/_clientx.py b/dify_client/_clientx.py
@@ -130,7 +130,7 @@ def request_stream(self, endpoint: str, method: str,
                 errors.raise_for_status(event_source.response)
             for sse in event_source.iter_sse():
                 errors.raise_for_status(sse)
-                if sse.data in IGNORED_STREAM_EVENTS:
+                if sse.event in IGNORED_STREAM_EVENTS or sse.data in IGNORED_STREAM_EVENTS:
                     continue
                 yield sse
 
@@ -434,7 +434,7 @@ async def arequest_stream(self, endpoint: str, method: str,
                 errors.raise_for_status(event_source.response)
             async for sse in event_source.aiter_sse():
                 errors.raise_for_status(sse)
-                if sse.data in IGNORED_STREAM_EVENTS:
+                if sse.event in IGNORED_STREAM_EVENTS or sse.data in IGNORED_STREAM_EVENTS:
                     continue
                 yield sse