Add early stopping for debugging on passed tests. Propagate minor changes to default human_message keys.

CF_CodeCollab.yaml

@@ -46,7 +46,8 @@ subflows_config:
         partial_variables:
           code_placeholder: "{{python_code}}"
         template_format: jinja2
-      default_human_input_key: "code_feedback"
+      default_human_input_keys:
+        - "code_feedback"
       output_data_transformations:
         - _target_: flows.data_transformations.RegexFirstOccurrenceExtractor
           regex: '(?<=```python)([\s\S]*?)(?=```)'

CF_CodeDebug.py

@@ -4,3 +4,9 @@ from flows.base_flows import GeneratorCriticFlow
 class CF_CodeDebug(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()

CF_CodeDebug.yaml

@@ -42,7 +42,9 @@ subflows_config:
           - testing_results_summary
         partial_variables:
           code_placeholder: "{{python_code}}"
-      default_human_input_key: "testing_results_summary"
+      default_human_input_keys:
+        - "testing_results_summary"
+        - "all_tests_passed"
       output_data_transformations:
         - _target_: flows.data_transformations.RegexFirstOccurrenceExtractor
           regex: '(?<=```python)([\s\S]*?)(?=```)'

CF_PlanCollab.yaml

@@ -44,7 +44,8 @@ subflows_config:
         partial_variables:
           plan_placeholder: "{{conceptual_solution}}"
         template_format: jinja2
-      default_human_input_key: "plan_feedback"
+      default_human_input_keys:
+        - "plan_feedback"
       output_data_transformations:
         - _target_: flows.data_transformations.RegexFirstOccurrenceExtractor
           regex: '(?<=Conceptual solution)([\s\S]*?)(?=\n\n# [A-Z]|\Z)'