def structured_predict(
        self, output_cls: BaseModel, prompt: PromptTemplate, **prompt_args: Any
    ) -> BaseModel:
        """This is fixed so that it goes through 2 Reflections and 1 Correction."""
        if output_cls == Reflection:
            return mock_reflections_queue.popleft()
        elif output_cls == Correction:
            return mock_correction
        else:
            raise ValueError("Unexpected output_cls type for this test.")
This is fixed so that it goes through 2 Reflections and 1 Correction.
llama-index-integrations/agent/llama-index-agent-introspective/tests/test_self_reflection.py