CF_CodeDebugCollabWithPlan.py

from flows.base_flows import GeneratorCriticFlow


class CF_CodeDebugCollabWithPlan(GeneratorCriticFlow):
    def __init__(self, **kwargs):
        super().__init__(**kwargs)

    def _early_exit(self):
        if self.flow_state.get("all_tests_passed", False):
            return True

        return super()._early_exit()