Skip to content

Commit 487a57f

Browse files
authored
Optimize code (langchain-ai#3412)
Co-authored-by: assert <zhangchengming@kkguan.com>
1 parent 3d8243e commit 487a57f

File tree

1 file changed

+1
-2
lines changed
  • langchain/chains/combine_documents

1 file changed

+1
-2
lines changed

langchain/chains/combine_documents/stuff.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,8 +40,8 @@ class Config:
4040
@root_validator(pre=True)
4141
def get_default_document_variable_name(cls, values: Dict) -> Dict:
4242
"""Get default document variable name, if not provided."""
43+
llm_chain_variables = values["llm_chain"].prompt.input_variables
4344
if "document_variable_name" not in values:
44-
llm_chain_variables = values["llm_chain"].prompt.input_variables
4545
if len(llm_chain_variables) == 1:
4646
values["document_variable_name"] = llm_chain_variables[0]
4747
else:
@@ -50,7 +50,6 @@ def get_default_document_variable_name(cls, values: Dict) -> Dict:
5050
"multiple llm_chain_variables"
5151
)
5252
else:
53-
llm_chain_variables = values["llm_chain"].prompt.input_variables
5453
if values["document_variable_name"] not in llm_chain_variables:
5554
raise ValueError(
5655
f"document_variable_name {values['document_variable_name']} was "

0 commit comments

Comments
 (0)