From e947f95c73f874f8f42175e20a2a0a1e06120b2a Mon Sep 17 00:00:00 2001
From: akshaym1shra <akshay.mishra@salescode.ai>
Date: Thu, 2 Jan 2025 22:00:10 +0530
Subject: [PATCH] Update stream.py to support parallel chatflow

---
 dify_client/models/stream.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/dify_client/models/stream.py b/dify_client/models/stream.py
index 137bd63..1b40932 100644
--- a/dify_client/models/stream.py
+++ b/dify_client/models/stream.py
@@ -26,6 +26,9 @@ class StreamEvent(StrEnum):
     MESSAGE_REPLACE = "message_replace"
     ERROR = "error"
     PING = "ping"
+    PARALLEL_BRANCH_STARTED = "parallel_branch_started"
+    PARALLEL_BRANCH_FINISHED = "parallel_branch_finished"
+    NODE_RETRY = "node_retry"
 
     @classmethod
     def new(cls, event: Union["StreamEvent", str]) -> "StreamEvent":