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()) messages.append(input_value.to_lc_message())
else: else:
messages.append(HumanMessage(content=input_value)) messages.append(HumanMessage(content=input_value))
inputs = messages or {} inputs: Union[list, dict] = messages or {}
try: try:
if stream: if stream:
return runnable.stream(inputs) return runnable.stream(inputs)