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)
|
await super()._run(*args, **kwargs)
|
||||||
|
|
||||||
async def stream(self):
|
async def stream(self):
|
||||||
iterator = self.params.get("message", None)
|
iterator = self.params.get("input_value", None)
|
||||||
if not isinstance(iterator, (AsyncIterator, Iterator)):
|
if not isinstance(iterator, (AsyncIterator, Iterator)):
|
||||||
raise ValueError("The message must be an iterator or an async iterator.")
|
raise ValueError("The message must be an iterator or an async iterator.")
|
||||||
is_async = isinstance(iterator, AsyncIterator)
|
is_async = isinstance(iterator, AsyncIterator)
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue