Remove unnecessary code for streaming to non-streamable component

This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-03-03 18:31:33 -03:00
commit 68d6c16e7c

View file

@ -609,11 +609,6 @@ 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."
)
def _reset(self, params_update: Optional[Dict[str, Any]] = None):
self._built = False