Function test_task_tool_none_tools

tests/main/test_task_tool.py:272–315

Test that tools="none" disables all tools except DoneTool for the sub-agent.

()

def test_task_tool_none_tools():
    """
    Test that tools="none" disables all tools except DoneTool for the sub-agent.
    """
    # Create a main agent that delegates with no tools
    main_config = ChatAgentConfig(
        llm=MockLMConfig(
            default_response=TaskTool(
                agent_name="Calculator",
                system_message=f"""
                    You are an assistant with no tools. Just respond directly
                    to the prompt and use `{DoneTool.name()}` to return your answer.
                    """,
                prompt="What is 2 + 2? Just tell me the answer.",
                model="gpt-4o-mini",
                tools=["NONE"],  # Disable all tools except DoneTool
                max_iterations=20,
            ).model_dump_json()
        ),
        name="MainAgent",
    )
    main_agent = ChatAgent(main_config)

    # Enable TaskTool and other tools for the main agent
    # (sub-agent won't have access to these)
    main_agent.enable_message(
        [TaskTool, MultiplierTool, NebrowskiTool], use=True, handle=True
    )

    # Create task
    task = Task(
        main_agent,
        name="NoToolsTask",
        interactive=False,
        config=TaskConfig(
            done_sequences=["T,A"],  # LLM (Tool), Agent(Handled) -> done
        ),
    )

    # Run the task
    result = task.run(msg="Test no tools")

    # Verify that the task completed (sub-agent can still use DoneTool)
    assert result is not None, "Task should return a result"