Function _create_nebrowski_task

tests/main/test_task_tool.py:91–146

Helper function to create a Nebrowski task for both sync and async tests. Returns a configured Task ready to run.

()

def _create_nebrowski_task():
    """
    Helper function to create a Nebrowski task for both sync and async tests.
    Returns a configured Task ready to run.
    """
    # Configure the main agent with a real LLM
    main_config = ChatAgentConfig(
        llm=OpenAIGPTConfig(),  # Uses default model
        handle_llm_no_tool="you forgot to use one of your TOOLs!",
        system_message=f"""
        You are a Nebrowski operation specialist. The Nebrowski operation is an exotic 
        mathematical function that takes two numbers and produces a result.
        BUT you do NOT know how to compute it yourself!
        
        When the user asks you to compute nested Nebrowski operations like 
        Nebrowski(a, Nebrowski(b, c)), you MUST:
        
        1. Break it down into individual Nebrowski operations
        2. Use the TOOL `{TaskTool.name()}` to delegate each Nebrowski 
            operation to a sub-agent
        3. The sub-agent knows how to use the `{NebrowskiTool.name()}` tool
        
        For example, to compute Nebrowski(10, Nebrowski(3, 2)):
        - First compute inner: Nebrowski(3, 2) = result1 (using TaskTool)
        - Then compute outer: Nebrowski(10, result1) (using TaskTool)
        - Return the final result
        
        IMPORTANT: You must use TaskTool for EACH Nebrowski operation.
        Configure the TaskTool with:
        - system_message: Instructions for the sub-agent to compute Nebrowski
        - prompt: The specific Nebrowski task (e.g., "Compute Nebrowski(3, 2)")
        - tools: ["nebrowski_tool"]
        - model: "gpt-4o-mini"
        
        Remember: You cannot compute Nebrowski operations yourself - you must 
        delegate to sub-agents!
        
        You MUST use the TOOL `{DoneTool.name()}` to return the final result!
        """,
        name="NebrowskiAgent",
    )
    main_agent = ChatAgent(main_config)

    # Enable TaskTool and NebrowskiTool
    main_agent.enable_message(
        [DoneTool, TaskTool, NebrowskiTool], use=True, handle=True
    )

    # Create task with appropriate configuration
    task = Task(
        main_agent,
        name="NebrowskiTask",
        interactive=False,
    )

    return task