test_fetch_last_ai_and_tool_messages_multiple_ai() — langchain Function Reference

Source Code

libs/langchain_v1/tests/unit_tests/agents/test_fetch_last_ai_and_tool_messages.py lines 30–45

def test_fetch_last_ai_and_tool_messages_multiple_ai() -> None:
    """Test that the last AIMessage is returned when multiple exist."""
    messages = [
        HumanMessage(content="First question"),
        AIMessage(content="First answer", id="ai1"),
        HumanMessage(content="Second question"),
        AIMessage(content="Second answer", id="ai2"),
    ]

    ai_msg, tool_msgs = _fetch_last_ai_and_tool_messages(messages)

    assert ai_msg is not None
    assert isinstance(ai_msg, AIMessage)
    assert ai_msg.content == "Second answer"
    assert ai_msg.id == "ai2"
    assert len(tool_msgs) == 0