made output conform

1 year ago · bb5427d2cd
parent 3fcbae90ed
commit bb5427d2cd
1 changed files with 1 additions and 3 deletions
--- a/langchain/chains/fork_with_follow_ups.py
+++ b/langchain/chains/fork_with_follow_ups.py
@ -47,9 +47,7 @@ class ForkChain(Chain, BaseModel):

    def _call(self, inputs: Dict[str, str]) -> Dict[str, str]:
        decision_chain_output = self.decision_chain.run(inputs)
-        classification_output = decision_chain_output
-
        try:
-            return self.follow_up_chain[classification_output](inputs)
+            return self.follow_up_chains[decision_chain_output.strip()](inputs)
        except KeyError:
            return self.follow_up_chains['default'](inputs)