test_agent_iterator_reset() — langchain Function Reference

Architecture documentation for the test_agent_iterator_reset() function in test_agent_iterator.py from the langchain codebase.

Function python LangChainCore Runnables

Entity Profile

LangChainCore→ Runnables→ test_agent_iterator_reset() — langchain Function Reference

Dependency Diagram

graph TD
  55e2bbdb_4ee2_e468_9c6f_931c8b0bde82["test_agent_iterator_reset()"]
  2423e003_42f4_7c6a_3fba_30db54c07a0c["test_agent_iterator.py"]
  55e2bbdb_4ee2_e468_9c6f_931c8b0bde82 -->|defined in| 2423e003_42f4_7c6a_3fba_30db54c07a0c
  style 55e2bbdb_4ee2_e468_9c6f_931c8b0bde82 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/langchain/tests/unit_tests/agents/test_agent_iterator.py lines 274–295

def test_agent_iterator_reset() -> None:
    """Test reset functionality of AgentExecutorIterator."""
    agent = _get_agent()
    agent_iter = agent.iter(inputs="when was langchain made")
    assert isinstance(agent_iter, AgentExecutorIterator)

    # Perform one iteration
    iterator = iter(agent_iter)
    next(iterator)

    # Check if properties are updated
    assert agent_iter.iterations == 1
    assert agent_iter.time_elapsed > 0.0
    assert agent_iter.intermediate_steps

    # Reset the iterator
    agent_iter.reset()

    # Check if properties are reset
    assert agent_iter.iterations == 0
    assert agent_iter.time_elapsed == 0.0
    assert not agent_iter.intermediate_steps