From 487a57ffe6ba9b95196a0af8c041fc7746b57104 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=BC=A0=E5=9F=8E=E9=93=AD?= Date: Mon, 24 Apr 2023 12:04:59 +0800 Subject: [PATCH] Optimize code (#3412) Co-authored-by: assert --- langchain/chains/combine_documents/stuff.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/langchain/chains/combine_documents/stuff.py b/langchain/chains/combine_documents/stuff.py index 237ecc2d..9d0a141c 100644 --- a/langchain/chains/combine_documents/stuff.py +++ b/langchain/chains/combine_documents/stuff.py @@ -40,8 +40,8 @@ class StuffDocumentsChain(BaseCombineDocumentsChain): @root_validator(pre=True) def get_default_document_variable_name(cls, values: Dict) -> Dict: """Get default document variable name, if not provided.""" + llm_chain_variables = values["llm_chain"].prompt.input_variables if "document_variable_name" not in values: - llm_chain_variables = values["llm_chain"].prompt.input_variables if len(llm_chain_variables) == 1: values["document_variable_name"] = llm_chain_variables[0] else: @@ -50,7 +50,6 @@ class StuffDocumentsChain(BaseCombineDocumentsChain): "multiple llm_chain_variables" ) else: - llm_chain_variables = values["llm_chain"].prompt.input_variables if values["document_variable_name"] not in llm_chain_variables: raise ValueError( f"document_variable_name {values['document_variable_name']} was "