From bcffc704c166daf08ace1c25950c2c72927b1d94 Mon Sep 17 00:00:00 2001 From: Django <100112640+sixdjango@users.noreply.github.com> Date: Wed, 17 May 2023 08:37:56 +0800 Subject: [PATCH] fix: agenerate miss run_manager args in llm.py (#4566) # fix: agenerate miss run_manager args in llm.py Fixes # (issue) fix: agenerate miss run_manager args in llm.py --- langchain/chains/llm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/langchain/chains/llm.py b/langchain/chains/llm.py index ea008ee2..18d8f539 100644 --- a/langchain/chains/llm.py +++ b/langchain/chains/llm.py @@ -86,7 +86,7 @@ class LLMChain(Chain): run_manager: Optional[AsyncCallbackManagerForChainRun] = None, ) -> LLMResult: """Generate LLM result from inputs.""" - prompts, stop = await self.aprep_prompts(input_list) + prompts, stop = await self.aprep_prompts(input_list, run_manager=run_manager) return await self.llm.agenerate_prompt( prompts, stop, callbacks=run_manager.get_child() if run_manager else None )