github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-19 13:23:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

Supports async in SequentialChain/SimpleSequentialChain (#3503)

Browse Source
This commit is contained in:
ccw630 2023-04-27 13:07:20 +08:00 committed by GitHub
parent 4a246e2fd6
commit cf384dcb7f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 24 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
langchain/chains/sequential.py
View File

@ -93,6 +93,13 @@ class SequentialChain(Chain):
known_values.update(outputs) known_values.update(outputs)
return {k: known_values[k] for k in self.output_variables} return {k: known_values[k] for k in self.output_variables}
async def _acall(self, inputs: Dict[str, str]) -> Dict[str, str]:
known_values = inputs.copy()
for i, chain in enumerate(self.chains):
outputs = await chain.acall(known_values, return_only_outputs=True)
known_values.update(outputs)
return {k: known_values[k] for k in self.output_variables}
class SimpleSequentialChain(Chain): class SimpleSequentialChain(Chain):
"""Simple chain where the outputs of one step feed directly into next.""" """Simple chain where the outputs of one step feed directly into next."""
@ -151,3 +158,20 @@ class SimpleSequentialChain(Chain):
_input, color=color_mapping[str(i)], end="\n", verbose=self.verbose _input, color=color_mapping[str(i)], end="\n", verbose=self.verbose
) )
return {self.output_key: _input} return {self.output_key: _input}
async def _acall(self, inputs: Dict[str, str]) -> Dict[str, str]:
_input = inputs[self.input_key]
color_mapping = get_color_mapping([str(i) for i in range(len(self.chains))])
for i, chain in enumerate(self.chains):
_input = await chain.arun(_input)
if self.strip_outputs:
_input = _input.strip()
if self.callback_manager.is_async:
await self.callback_manager.on_text(
_input, color=color_mapping[str(i)], end="\n", verbose=self.verbose
)
else:
self.callback_manager.on_text(
_input, color=color_mapping[str(i)], end="\n", verbose=self.verbose
)
return {self.output_key: _input}