From 42574da37f4ffd9cacc9d40323026a70b4bc24e6 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Sun, 3 Mar 2024 18:23:55 -0300 Subject: [PATCH] Fix streaming issue for Chat Output vertex --- src/backend/langflow/graph/vertex/base.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/backend/langflow/graph/vertex/base.py b/src/backend/langflow/graph/vertex/base.py index 6fd66067e..f9f5849a1 100644 --- a/src/backend/langflow/graph/vertex/base.py +++ b/src/backend/langflow/graph/vertex/base.py @@ -609,6 +609,8 @@ class Vertex: raise ValueError( f"You are trying to stream to a {self.display_name}. Try using a Chat Output instead." ) + if self.display_name in ["Chat Output"]: + return raise ValueError( f"{self.display_name}: You are trying to stream to a non-streamable component." )