github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-04 04:07:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

remove chat-history (#15286)

Browse Source
This commit is contained in:
Harrison Chase 2023-12-28 14:22:16 -08:00 committed by GitHub
parent 90aa26a90e
commit bc5a0ef6ca
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 0 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
libs/langchain/langchain/chains/retrieval.py
View File

@ -63,7 +63,6 @@ def create_retrieval_chain(
retrieval_chain = (
RunnablePassthrough.assign(
context=retrieval_docs.with_config(run_name="retrieve_documents"),
chat_history=lambda x: x.get("chat_history", []),
).assign(answer=combine_docs_chain)
).with_config(run_name="retrieval_chain")

8
libs/langchain/tests/unit_tests/chains/test_retrieval.py
View File

@ -13,14 +13,6 @@ def test_create() -> None:
retriever = FakeParrotRetriever()
question_gen_prompt = PromptTemplate.from_template("hi! {input} {chat_history}")
chain = create_retrieval_chain(retriever, question_gen_prompt | llm)
expected_output = {
"answer": "I know the answer!",
"chat_history": [],
"context": [Document(page_content="What is the answer?")],
"input": "What is the answer?",
}
output = chain.invoke({"input": "What is the answer?"})
assert output == expected_output
expected_output = {
"answer": "I know the answer!",