From acf1ecc08173da4e7a6caafb11ab0ac757f2ec40 Mon Sep 17 00:00:00 2001 From: Roshan Santhosh Date: Mon, 11 Mar 2024 22:30:07 -0700 Subject: [PATCH] langchain[patch]: update llm_router.py (#18865) Issue : _call method of LLMRouterChain uses predict_and_parse, which is slated for deprecation. Description : Instead of using predict_and_parse, this replaces it with individual predict and parse functions. --- libs/langchain/langchain/chains/router/llm_router.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/libs/langchain/langchain/chains/router/llm_router.py b/libs/langchain/langchain/chains/router/llm_router.py index 9443321827a..dd10c155742 100644 --- a/libs/langchain/langchain/chains/router/llm_router.py +++ b/libs/langchain/langchain/chains/router/llm_router.py @@ -56,9 +56,11 @@ class LLMRouterChain(RouterChain): ) -> Dict[str, Any]: _run_manager = run_manager or CallbackManagerForChainRun.get_noop_manager() callbacks = _run_manager.get_child() + + prediction = self.llm_chain.predict(callbacks=callbacks, **inputs) output = cast( Dict[str, Any], - self.llm_chain.predict_and_parse(callbacks=callbacks, **inputs), + self.llm_chain.prompt.output_parser.parse(prediction), ) return output