LC_CodeDebugCollab.py

import importlib
import os

from flows.base_flows import GeneratorCriticFlow

from martinjosifoski.OpenAIChatAtomicFlow import OpenAIChatAtomicFlow

class LC_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()