test_agent_tool_return_direct() — langchain Function Reference

Architecture documentation for the test_agent_tool_return_direct() function in test_agent_async.py from the langchain codebase.

Function python LangChainCore ApiManagement

Entity Profile

LangChainCore→ ApiManagement→ test_agent_tool_return_direct() — langchain Function Reference

Dependency Diagram

graph TD
  a71a3037_a762_54f8_86fd_7dcebf81d40a["test_agent_tool_return_direct()"]
  663245d2_2254_328f_5cf8_4bbd36fb32cc["test_agent_async.py"]
  a71a3037_a762_54f8_86fd_7dcebf81d40a -->|defined in| 663245d2_2254_328f_5cf8_4bbd36fb32cc
  style a71a3037_a762_54f8_86fd_7dcebf81d40a fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/langchain/tests/unit_tests/agents/test_agent_async.py lines 287–310

async def test_agent_tool_return_direct() -> None:
    """Test agent using tools that return directly."""
    tool = "Search"
    responses = [
        f"FooBarBaz\nAction: {tool}\nAction Input: misalignment",
        "Oh well\nFinal Answer: curses foiled again",
    ]
    fake_llm = FakeListLLM(responses=responses)
    tools = [
        Tool(
            name="Search",
            func=lambda x: x,
            description="Useful for searching",
            return_direct=True,
        ),
    ]
    agent = initialize_agent(
        tools,
        fake_llm,
        agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
    )

    output = await agent.arun("when was langchain made")
    assert output == "misalignment"