github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-19 13:23:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

update keys for chain (#5164)

Browse Source 
Co-authored-by: Bagatur <baskaryan@gmail.com>
This commit is contained in:
Harrison Chase 2023-08-11 16:25:13 -07:00 committed by GitHub
parent aca8cb5fba
commit 1b58460fe3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 14 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
libs/langchain/langchain/chains/combine_documents/stuff.py
View File

@ -100,6 +100,13 @@ class StuffDocumentsChain(BaseCombineDocumentsChain):
) )
return values return values
@property
def input_keys(self) -> List[str]:
extra_keys = [
k for k in self.llm_chain.input_keys if k != self.document_variable_name
]
return super().input_keys + extra_keys
def _get_inputs(self, docs: List[Document], **kwargs: Any) -> dict: def _get_inputs(self, docs: List[Document], **kwargs: Any) -> dict:
"""Construct inputs from kwargs and docs. """Construct inputs from kwargs and docs.

7
libs/langchain/tests/unit_tests/chains/test_combine_documents.py
View File

@ -9,8 +9,10 @@ from langchain.chains.combine_documents.reduce import (
_collapse_docs, _collapse_docs,
_split_list_of_docs, _split_list_of_docs,
) )
from langchain.chains.qa_with_sources import load_qa_with_sources_chain
from langchain.docstore.document import Document from langchain.docstore.document import Document
from langchain.schema import format_document from langchain.schema import format_document
from tests.unit_tests.llms.fake_llm import FakeLLM
def _fake_docs_len_func(docs: List[Document]) -> int: def _fake_docs_len_func(docs: List[Document]) -> int:
@ -21,6 +23,11 @@ def _fake_combine_docs_func(docs: List[Document], **kwargs: Any) -> str:
return "".join([d.page_content for d in docs]) return "".join([d.page_content for d in docs])
def test_multiple_input_keys() -> None:
chain = load_qa_with_sources_chain(FakeLLM(), chain_type="stuff")
assert chain.input_keys == ["input_documents", "question"]
def test__split_list_long_single_doc() -> None: def test__split_list_long_single_doc() -> None:
"""Test splitting of a long single doc.""" """Test splitting of a long single doc."""
docs = [Document(page_content="foo" * 100)] docs = [Document(page_content="foo" * 100)]