Update parameter name in ChatVertex stream method

This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-27 18:08:50 -03:00
commit a9bea1872a

View file

@ -405,7 +405,7 @@ class ChatVertex(StatelessVertex):
await super()._run(*args, **kwargs)
async def stream(self):
iterator = self.params.get("message", None)
iterator = self.params.get("input_value", None)
if not isinstance(iterator, (AsyncIterator, Iterator)):
raise ValueError("The message must be an iterator or an async iterator.")
is_async = isinstance(iterator, AsyncIterator)