Function test_task_tool_all_tools

tests/main/test_task_tool.py:182–269

Test that tools="all" enables all available tools for the sub-agent.

()

def test_task_tool_all_tools():
    """
    Test that tools="all" enables all available tools for the sub-agent.
    """
    # Create a main agent with multiple tools available
    main_config = ChatAgentConfig(
        llm=MockLMConfig(
            default_response=TaskTool(
                agent_name="Calculator",
                system_message=f"""
                    You are a multi-tool assistant. Use the appropriate tool
                    to complete the task, then use `{DoneTool.name()}` to return the 
                    result.
                    """,
                prompt="""
                    Multiply 4 and 6, call it x, then compute Nebrowski(x, 5)
                    """,
                model="gpt-4o-mini",
                tools=["ALL"],  # Enable all tools
                max_iterations=20,
            ).model_dump_json()
        ),
        name="MainAgent",
    )
    main_agent = ChatAgent(main_config)

    # Set up multiple tools for the main agent
    main_agent.enable_message(
        [TaskTool, MultiplierTool, NebrowskiTool], use=True, handle=True
    )

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

    # Run the task: input text is immaterial since the
    # MockLM is hard-coded to return the TaskTool request
    result = task.run(msg="Test all tools")

    # Verify that the sub-agent had access to all tools
    # Expected: Multiply 4 and 6 = 24, Nebrowski(3, 5) = 14
    assert result is not None, "Task should return a result"
    assert "77" in result.content, "Result should contain 77"

    # Verify that parent chain is maintained through TaskTool
    # When TaskTool creates a prompt ChatDocument with parent_id pointing to the
    # TaskTool message, and passes it to the subtask, the subtask's init() method
    # should preserve that parent_id even though it deep copies the message.
    # This ensures the parent chain is not broken.
    assert hasattr(result, "parent"), "Result should have a parent pointer"

    # Traverse up the parent chain to find the TaskTool message