This commit is contained in:
Nuno Campos 2023-09-01 15:46:53 +01:00
parent fc42726ea0
commit 81ebcc161e

View File

@ -1592,15 +1592,15 @@ class RunnableBinding(Serializable, Runnable[Input, Output]):
config: Optional[Union[RunnableConfig, List[RunnableConfig]]] = None, config: Optional[Union[RunnableConfig, List[RunnableConfig]]] = None,
**kwargs: Optional[Any], **kwargs: Optional[Any],
) -> List[Output]: ) -> List[Output]:
configs = cast( if isinstance(config, list):
List[RunnableConfig], configs = cast(
[self._merge_config(conf) for conf in config] List[RunnableConfig], [self._merge_config(conf) for conf in config]
if isinstance(config, list) )
else [ else:
configs = [
patch_config(self._merge_config(config), deep_copy_locals=True) patch_config(self._merge_config(config), deep_copy_locals=True)
for _ in range(len(inputs)) for _ in range(len(inputs))
], ]
)
return await self.bound.abatch(inputs, configs, **{**self.kwargs, **kwargs}) return await self.bound.abatch(inputs, configs, **{**self.kwargs, **kwargs})
def stream( def stream(