test_agent_name_set_on_ai_message() — langchain Function Reference

Architecture documentation for the test_agent_name_set_on_ai_message() function in test_agent_name.py from the langchain codebase.

Function python LangChainCore MessageInterface

Entity Profile

LangChainCore→ MessageInterface→ test_agent_name_set_on_ai_message() — langchain Function Reference

Dependency Diagram

graph TD
  7bd254e9_3cf1_4b8b_d051_2a06094243fc["test_agent_name_set_on_ai_message()"]
  855858ac_82a7_eb82_4d23_1dd8507dbcbb["test_agent_name.py"]
  7bd254e9_3cf1_4b8b_d051_2a06094243fc -->|defined in| 855858ac_82a7_eb82_4d23_1dd8507dbcbb
  style 7bd254e9_3cf1_4b8b_d051_2a06094243fc fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/langchain_v1/tests/unit_tests/agents/test_agent_name.py lines 25–37

def test_agent_name_set_on_ai_message() -> None:
    """Test that agent name is set on AIMessage when name is provided."""
    tool_calls: list[list[ToolCall]] = [[]]
    agent = create_agent(
        model=FakeToolCallingModel(tool_calls=tool_calls),
        name="test_agent",
    )

    result = agent.invoke({"messages": [HumanMessage("Hello")]})

    ai_messages = [m for m in result["messages"] if isinstance(m, AIMessage)]
    assert len(ai_messages) == 1
    assert ai_messages[0].name == "test_agent"