Function test_done_sequence_tool_then_agent

tests/main/test_done_sequences.py:39–74

Test that task terminates after tool followed by agent response

(test_settings: Settings)

def test_done_sequence_tool_then_agent(test_settings: Settings):
    """Test that task terminates after tool followed by agent response"""
    set_global(test_settings)

    # Mock LLM that always generates a tool
    agent = ChatAgent(
        ChatAgentConfig(
            name="TestAgent",
            llm=MockLMConfig(
                response_fn=lambda x: '{"request": "simple_tool", "value": "test"}'
            ),
        )
    )
    agent.enable_message(SimpleTool)

    # Configure task to be done after tool -> agent response
    config = TaskConfig(
        done_sequences=[
            DoneSequence(
                name="tool_then_agent",
                events=[
                    AgentEvent(event_type=EventType.TOOL),
                    AgentEvent(event_type=EventType.AGENT_RESPONSE),
                ],
            )
        ]
    )

    task = Task(agent, config=config, interactive=False)
    result = task.run("Generate a tool", turns=10)

    # Task should complete after tool generation and agent response
    assert result is not None
    # Should have: system, user, llm (with tool)
    # Note: agent response is not added to message history
    assert len(agent.message_history) == 3


def test_done_sequence_specific_tool(test_settings: Settings):