Fix streaming issue for Chat Output vertex
This commit is contained in:
parent
af20ca0f6e
commit
42574da37f
1 changed files with 2 additions and 0 deletions
|
|
@ -609,6 +609,8 @@ class Vertex:
|
||||||
raise ValueError(
|
raise ValueError(
|
||||||
f"You are trying to stream to a {self.display_name}. Try using a Chat Output instead."
|
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(
|
raise ValueError(
|
||||||
f"{self.display_name}: You are trying to stream to a non-streamable component."
|
f"{self.display_name}: You are trying to stream to a non-streamable component."
|
||||||
)
|
)
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue