From a2c5680d9e4832e22a5320023032efc06ab28609 Mon Sep 17 00:00:00 2001 From: Shreya Rajpal Date: Wed, 28 Dec 2022 19:54:48 +0530 Subject: [PATCH] Fix lint issues --- langchain/chains/fork_with_follow_ups.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/langchain/chains/fork_with_follow_ups.py b/langchain/chains/fork_with_follow_ups.py index 8af8d1a5..a7caf955 100644 --- a/langchain/chains/fork_with_follow_ups.py +++ b/langchain/chains/fork_with_follow_ups.py @@ -20,7 +20,7 @@ class ForkChain(Chain, BaseModel): arbitrary_types_allowed = True @validator("follow_up_chains") - def default_in_follow_up_chains(cls, v): + def default_in_follow_up_chains(cls, v: Dict[str, Chain]) -> Dict[str, Chain]: if "default" not in v: raise ValueError( "`follow_up_chains` must contain a 'default' option. " @@ -46,7 +46,7 @@ class ForkChain(Chain, BaseModel): return [] def _call(self, inputs: Dict[str, str]) -> Dict[str, str]: - decision_chain_output = self.decision_chain.run(inputs) + decision_chain_output = self.decision_chain.run(**inputs) try: return self.follow_up_chains[decision_chain_output.strip()](inputs) except KeyError: