Archives/langchain
2
0
Fork 1
mirror of https://github.com/hwchase17/langchain synced 2024-11-04 06:00:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use call directly for chain (#8655)

Browse Source 
for run_on_dataset since the `run()` method requires a single output
This commit is contained in:
William FH 2023-08-02 17:11:39 -07:00 committed by GitHub
parent 368aa4ede7
commit 7ea2b08d1f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
libs/langchain/langchain/smith/evaluation/runner_utils.py
View File

@ -604,11 +604,8 @@ async def _arun_chain(
inputs_, callbacks=callbacks, tags=tags
)
else:
if len(inputs) == 1:
inputs_ = next(iter(inputs.values()))
output = await chain.arun(inputs_, callbacks=callbacks, tags=tags)
else:
output = await chain.acall(inputs, callbacks=callbacks, tags=tags)
inputs_ = next(iter(inputs.values())) if len(inputs) == 1 else inputs
output = await chain.acall(inputs_, callbacks=callbacks, tags=tags)
return output
@ -926,11 +923,8 @@ def _run_chain(
inputs_ = input_mapper(inputs)
output: Union[dict, str] = chain(inputs_, callbacks=callbacks, tags=tags)
else:
if len(inputs) == 1:
inputs_ = next(iter(inputs.values()))
output = chain.run(inputs_, callbacks=callbacks, tags=tags)
else:
output = chain(inputs, callbacks=callbacks, tags=tags)
inputs_ = next(iter(inputs.values())) if len(inputs) == 1 else inputs
output = chain(inputs_, callbacks=callbacks, tags=tags)
return output