Function test_done_sequence_specific_tool

tests/main/test_done_sequences.py:77–130

Test that task terminates only after specific tool

(test_settings: Settings)

def test_done_sequence_specific_tool(test_settings: Settings):
    """Test that task terminates only after specific tool"""
    set_global(test_settings)

    class AnotherTool(ToolMessage):
        request: str = "another_tool"
        purpose: str = "Another tool"
        data: str

        def handle(self) -> str:
            return f"Processed data: {self.data}"

    # Mock LLM that alternates between tools
    call_count = 0

    def mock_response(x):
        nonlocal call_count
        call_count += 1
        if call_count == 1:
            return '{"request": "another_tool", "data": "test"}'
        else:
            return '{"request": "simple_tool", "value": "test"}'

    agent = ChatAgent(
        ChatAgentConfig(
            name="TestAgent",
            llm=MockLMConfig(response_fn=mock_response),
        )
    )
    agent.enable_message(SimpleTool)
    agent.enable_message(AnotherTool)

    # Configure to be done only after simple_tool
    config = TaskConfig(
        done_sequences=[
            DoneSequence(
                name="specific_tool",
                events=[
                    AgentEvent(
                        event_type=EventType.SPECIFIC_TOOL, tool_name="simple_tool"
                    ),
                    AgentEvent(event_type=EventType.AGENT_RESPONSE),
                ],
            )
        ]
    )

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

    assert result is not None
    # Verify simple_tool was generated (it's in the last assistant message)
    last_assistant_msg = agent.message_history[-1]
    assert "simple_tool" in last_assistant_msg.content


def test_done_sequence_llm_agent_llm(test_settings: Settings):