fix arg-type error reported by mypy

This commit is contained in:
italojohnny 2024-06-20 18:20:16 -03:00
commit 954c703cd8

View file

@ -136,7 +136,7 @@ class LCModelComponent(Component):
messages.append(input_value.to_lc_message())
else:
messages.append(HumanMessage(content=input_value))
inputs = messages or {}
inputs: Union[list, dict] = messages or {}
try:
if stream:
return runnable.stream(inputs)