Remove "predict_and_parse" from LLMRouterChain

langchain-ai · Sep 15, 2023 · 1ececb2 · 1ececb2
1 parent f91b9b6
commit 1ececb2
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/libs/langchain/langchain/chains/router/llm_router.py b/libs/langchain/langchain/chains/router/llm_router.py
@@ -55,8 +55,10 @@ def _call(
         callbacks = _run_manager.get_child()
         output = cast(
             Dict[str, Any],
-            self.llm_chain.predict_and_parse(callbacks=callbacks, **inputs),
+            self.llm_chain.predict(callbacks=callbacks, **inputs),
         )
+        if self.llm_chain.prompt.output_parser is not None:
+            output = self.llm_chain.prompt.output_parser.parse(output)
         return output
 
     async def _acall(
@@ -68,8 +70,10 @@ async def _acall(
         callbacks = _run_manager.get_child()
         output = cast(
             Dict[str, Any],
-            await self.llm_chain.apredict_and_parse(callbacks=callbacks, **inputs),
+            await self.llm_chain.apredict(callbacks=callbacks, **inputs),
         )
+        if self.llm_chain.prompt.output_parser is not None:
+            output = self.llm_chain.prompt.output_parser.parse(output)
         return output
 
     @classmethod