github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-27 14:26:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

Optimize code (#3412)

Browse Source 
Co-authored-by: assert <zhangchengming@kkguan.com>
This commit is contained in:
张城铭
2023-04-24 12:04:59 +08:00
committed by GitHub
parent 3d8243ec95
commit 487a57ffe6
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
langchain/chains/combine_documents/stuff.py
View File

@@ -40,8 +40,8 @@ class StuffDocumentsChain(BaseCombineDocumentsChain):
@root_validator(pre=True) @root_validator(pre=True)
def get_default_document_variable_name(cls, values: Dict) -> Dict: def get_default_document_variable_name(cls, values: Dict) -> Dict:
"""Get default document variable name, if not provided.""" """Get default document variable name, if not provided."""
llm_chain_variables = values["llm_chain"].prompt.input_variables
if "document_variable_name" not in values: if "document_variable_name" not in values:
llm_chain_variables = values["llm_chain"].prompt.input_variables
if len(llm_chain_variables) == 1: if len(llm_chain_variables) == 1:
values["document_variable_name"] = llm_chain_variables[0] values["document_variable_name"] = llm_chain_variables[0]
else: else:
@@ -50,7 +50,6 @@ class StuffDocumentsChain(BaseCombineDocumentsChain):
"multiple llm_chain_variables" "multiple llm_chain_variables"
) )
else: else:
llm_chain_variables = values["llm_chain"].prompt.input_variables
if values["document_variable_name"] not in llm_chain_variables: if values["document_variable_name"] not in llm_chain_variables:
raise ValueError( raise ValueError(
f"document_variable_name {values['document_variable_name']} was " f"document_variable_name {values['document_variable_name']} was "