from flows.base_flows import GeneratorCriticFlow class CF_CodeDebugCollab(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()