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

version 0.0.49 (#436)

Browse Source
This commit is contained in:
Harrison Chase
2022-12-27 09:17:01 -05:00
committed by GitHub
parent f8b605293f
commit 0c5d3fd894
7 changed files with 1203 additions and 353 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/unit_tests/chains/test_conversation.py
View File

@@ -4,8 +4,8 @@ import pytest
from langchain.chains.base import Memory
from langchain.chains.conversation.base import ConversationChain
from langchain.chains.conversation.memory import (
ConversationalBufferWindowMemory,
ConversationBufferMemory,
ConversationBufferWindowMemory,
ConversationSummaryMemory,
)
from langchain.prompts.prompt import PromptTemplate
@@ -49,7 +49,7 @@ def test_conversation_chain_errors_bad_variable() -> None:
"memory",
[
ConversationBufferMemory(memory_key="baz"),
ConversationalBufferWindowMemory(memory_key="baz"),
ConversationBufferWindowMemory(memory_key="baz"),
ConversationSummaryMemory(llm=FakeLLM(), memory_key="baz"),
],
)
@@ -82,7 +82,7 @@ def test_conversation_memory(memory: Memory) -> None:
[
ConversationBufferMemory(memory_key="baz"),
ConversationSummaryMemory(llm=FakeLLM(), memory_key="baz"),
ConversationalBufferWindowMemory(memory_key="baz"),
ConversationBufferWindowMemory(memory_key="baz"),
],
)
def test_clearing_conversation_memory(memory: Memory) -> None: