test_conversation_chain_works() — langchain Function Reference

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

Function python LangChainCore Runnables

Entity Profile

LangChainCore→ Runnables→ test_conversation_chain_works() — langchain Function Reference

Dependency Diagram

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

Relationship Graph

Source Code

libs/langchain/tests/unit_tests/chains/test_conversation.py lines 64–75

async def test_conversation_chain_works() -> None:
    """Test that conversation chain works in basic setting."""
    llm = DummyLLM()
    prompt = PromptTemplate(input_variables=["foo", "bar"], template="{foo} {bar}")
    memory = ConversationBufferMemory(memory_key="foo")
    chain = ConversationChain(llm=llm, prompt=prompt, memory=memory, input_key="bar")
    chain.run("aaa")
    assert llm.last_prompt == " aaa"
    chain.run("bbb")
    assert llm.last_prompt == "Human: aaa\nAI: dummy bbb"
    await chain.arun("ccc")
    assert llm.last_prompt == "Human: aaa\nAI: dummy\nHuman: bbb\nAI: dummy ccc"