github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2026-02-21 06:33:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

fmt

Browse Source
This commit is contained in:
Bagatur
2023-12-21 10:19:32 -05:00
parent 062ad47c65
commit 7f5faf1076
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
libs/langchain/langchain/chains/base.py
View File

@@ -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