Update parameter name in ChatVertex stream method
This commit is contained in:
parent
39a2801095
commit
a9bea1872a
1 changed files with 1 additions and 1 deletions
|
|
@ -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)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue