Function test_task_2_agent_2_tool

tests/main/test_task.py:546–707

QueryTool: Task A uses and handles (validates), Task B handles but doesn't use FeedbackTool: Task B uses and handles (validates), Task A handles but doesn't use

(
    test_settings: Settings,
    use_fn_api: bool,
    use_tools_api: bool,
    use_orch_tools: bool,
)

@pytest.mark.parametrize("use_tools_api", [True, False])
@pytest.mark.parametrize("use_orch_tools", [True, False])
def test_task_2_agent_2_tool(
    test_settings: Settings,
    use_fn_api: bool,
    use_tools_api: bool,
    use_orch_tools: bool,
):
    """
    QueryTool: Task A uses and handles (validates), Task B handles but doesn't use
    FeedbackTool: Task B uses and handles (validates), Task A handles but doesn't use
    """

    set_global(test_settings)

    class QueryTool(ToolMessage):
        request: str = "polinsky_query"
        purpose: str = """
            Ask whether the Polinsky transform of a <number> equals <value>.
            """
        number: int
        value: int

    class FeedbackTool(ToolMessage):
        request: str = "polinsky_feedback"
        purpose: str = """
            Given a <number>, respond with the Polinsky transform of the number.
            """
        feedback: str

    class Requestor(ChatAgent):
        def __init__(self, config: ChatAgentConfig):
            super().__init__(config)
            self.enable_message(QueryTool, use=True, handle=True)
            self.enable_message(FeedbackTool, use=False, handle=True)
            self.enable_message(DoneTool)

        def polinsky_query(self, msg: QueryTool) -> str | PassTool:
            # No validation err, so pass it on so other agent can respond
            return PassTool() if use_orch_tools else PASS

        def polinsky_feedback(self, msg: FeedbackTool) -> str:
            """Transmit feedback received from other agent, to this agent's LLM"""
            if msg.feedback.strip() == "":
                return f"""
                CORRECT, the value you gave IS the Polinsky transform of that number.
                Please proceed with requesting the Polinsky transform of 
                the NEXT number on your list, or if you&#x27;re finished, use the
                TOOL `{DoneTool.name()}` with `content` set to the summary of the
                transformations, in this format:
                '(number1, transform1), (number2, transform2)'
                """
            else:
                return f"""
                WRONG, please try again based on this feedback: 
                {msg.feedback}
                """

        def handle_message_fallback(self, msg: str | ChatDocument) -> Any:
            if isinstance(msg, ChatDocument) and msg.metadata.sender == Entity.LLM: