blob42/langchain
1
0
Fork 0
forked from Archives/langchain
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix lint issues

Browse Source
This commit is contained in:
Shreya Rajpal 2022-12-28 19:54:48 +05:30
parent bb5427d2cd
commit a2c5680d9e
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
langchain/chains/fork_with_follow_ups.py
View File

@ -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: