From dc479a667d29cb8207fb88daaefc1ac699f85a9b Mon Sep 17 00:00:00 2001 From: Harrison Chase Date: Fri, 3 Mar 2023 16:19:29 -0800 Subject: [PATCH] cr --- docs/modules/agents/implementations/mrkl.ipynb | 2 +- docs/modules/agents/implementations/react.ipynb | 4 ++-- tests/unit_tests/chains/test_conversation.py | 4 ++-- 3 files changed, 5 insertions(+), 5 deletions(-) diff --git a/docs/modules/agents/implementations/mrkl.ipynb b/docs/modules/agents/implementations/mrkl.ipynb index f0ec3a37554..0afc3da0911 100644 --- a/docs/modules/agents/implementations/mrkl.ipynb +++ b/docs/modules/agents/implementations/mrkl.ipynb @@ -205,7 +205,7 @@ "name": "python", "nbconvert_exporter": "python", "pygments_lexer": "ipython3", - "version": "3.10.9" + "version": "3.9.1" } }, "nbformat": 4, diff --git a/docs/modules/agents/implementations/react.ipynb b/docs/modules/agents/implementations/react.ipynb index 3cfb8672257..8ced6b9f25d 100644 --- a/docs/modules/agents/implementations/react.ipynb +++ b/docs/modules/agents/implementations/react.ipynb @@ -81,7 +81,7 @@ ], "metadata": { "kernelspec": { - "display_name": "Python 3.9.0 64-bit ('llm-env')", + "display_name": "Python 3 (ipykernel)", "language": "python", "name": "python3" }, @@ -95,7 +95,7 @@ "name": "python", "nbconvert_exporter": "python", "pygments_lexer": "ipython3", - "version": "3.9.0" + "version": "3.9.1" }, "vscode": { "interpreter": { diff --git a/tests/unit_tests/chains/test_conversation.py b/tests/unit_tests/chains/test_conversation.py index c981289ae2b..66374aeb07c 100644 --- a/tests/unit_tests/chains/test_conversation.py +++ b/tests/unit_tests/chains/test_conversation.py @@ -16,14 +16,14 @@ def test_memory_ai_prefix() -> None: """Test that ai_prefix in the memory component works.""" memory = ConversationBufferMemory(memory_key="foo", ai_prefix="Assistant") memory.save_context({"input": "bar"}, {"output": "foo"}) - assert memory.buffer == "Human: bar\nAssistant: foo" + assert memory.buffer == "\nHuman: bar\nAssistant: foo" def test_memory_human_prefix() -> None: """Test that human_prefix in the memory component works.""" memory = ConversationBufferMemory(memory_key="foo", human_prefix="Friend") memory.save_context({"input": "bar"}, {"output": "foo"}) - assert memory.buffer == "Friend: bar\nAI: foo" + assert memory.buffer == "\nFriend: bar\nAI: foo" def test_conversation_chain_works() -> None: