github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-13 22:59:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

Lint

Browse Source
This commit is contained in:
Nuno Campos 2023-08-18 15:30:30 +01:00
parent 4452314aab
commit 354c42afd2
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
libs/langchain/langchain/schema/runnable/locals.py
View File

@ -63,7 +63,10 @@ class PutLocalVar(RunnablePassthrough):
return super().invoke(input, config=config) return super().invoke(input, config=config)
async def ainvoke( async def ainvoke(
self, input: Input, config: Optional[RunnableConfig] = None self,
input: Input,
config: Optional[RunnableConfig] = None,
**kwargs: Optional[Any],
) -> Input: ) -> Input:
self._concat_put(input, config=config, replace=True) self._concat_put(input, config=config, replace=True)
return await super().ainvoke(input, config=config) return await super().ainvoke(input, config=config)
@ -139,7 +142,10 @@ class GetLocalVar(
return self._call_with_config(self._get, input, config) return self._call_with_config(self._get, input, config)
async def ainvoke( async def ainvoke(
self, input: Input, config: Optional[RunnableConfig] = None self,
input: Input,
config: Optional[RunnableConfig] = None,
**kwargs: Optional[Any],
) -> Union[Output, Dict[str, Union[Input, Output]]]: ) -> Union[Output, Dict[str, Union[Input, Output]]]:
if config is None: if config is None:
raise ValueError( raise ValueError(