test_memory_ai_prefix() — langchain Function Reference

Architecture documentation for the test_memory_ai_prefix() function in test_conversation.py from the langchain codebase.

Function python LangChainCore Runnables

Entity Profile

LangChainCore→ Runnables→ test_memory_ai_prefix() — langchain Function Reference

Dependency Diagram

graph TD
  9d61a560_46da_280d_7a44_d9e4bb67139c["test_memory_ai_prefix()"]
  bad7138b_753c_8c4e_915e_8bf58ada3cec["test_conversation.py"]
  9d61a560_46da_280d_7a44_d9e4bb67139c -->|defined in| bad7138b_753c_8c4e_915e_8bf58ada3cec
  style 9d61a560_46da_280d_7a44_d9e4bb67139c fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/langchain/tests/unit_tests/chains/test_conversation.py lines 42–46

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.load_memory_variables({}) == {"foo": "Human: bar\nAssistant: foo"}