From 7f5faf10764c466f6945c9fd892d3cd17003291d Mon Sep 17 00:00:00 2001 From: Bagatur Date: Thu, 21 Dec 2023 10:19:32 -0500 Subject: [PATCH] fmt --- libs/langchain/langchain/chains/base.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/libs/langchain/langchain/chains/base.py b/libs/langchain/langchain/chains/base.py index 931e2584779..3b8617b252d 100644 --- a/libs/langchain/langchain/chains/base.py +++ b/libs/langchain/langchain/chains/base.py @@ -701,11 +701,13 @@ class RunnableChain(Chain): def OutputType(self) -> Type[Dict[str, Any]]: return self.as_runnable().OutputType - def get_input_schema(self, config: RunnableConfig | None = None) -> Type[BaseModel]: + def get_input_schema( + self, config: Optional[RunnableConfig] = None + ) -> Type[BaseModel]: return self.as_runnable().get_input_schema(config) def get_output_schema( - self, config: RunnableConfig | None = None + self, config: Optional[RunnableConfig] = None ) -> Type[BaseModel]: return self.as_runnable().get_output_schema(config) @@ -789,7 +791,7 @@ class RunnableChain(Chain): input: Iterator[Dict[str, Any]], config: Optional[RunnableConfig] = None, return_only_outputs: bool = False, - **kwargs: Any | None, + **kwargs: Any, ) -> Iterator[Dict[str, Any]]: yield from self.as_runnable_wrapped(return_only_outputs).transform( input, config, **kwargs @@ -800,7 +802,7 @@ class RunnableChain(Chain): input: AsyncIterator[Dict[str, Any]], config: Optional[RunnableConfig] = None, return_only_outputs: bool = False, - **kwargs: Any | None, + **kwargs: Any, ) -> AsyncIterator[Dict[str, Any]]: async for chunk in self.as_runnable_wrapped(return_only_outputs).atransform( input, config, **kwargs