Add support for cf-code_debug_collab.

CF_CodeCriticWrongAttempt.py

@@ -0,0 +1,10 @@
+import os
+
+from flows.flow_verse import load_class
+repository_id = os.environ.get("OpenAIChatAtomicFlow")
+OpenAIChatAtomicFlow = load_class(repository_id, "OpenAIChatAtomicFlow")
+
+
+class CF_CodeCriticWrongAttempt(OpenAIChatAtomicFlow):
+    def __init__(self, **kwargs):
+        super().__init__(**kwargs)

CF_CodeCriticWrongAttempt.yaml

@@ -0,0 +1,88 @@
+name: "CodeCriticWrongAttempt_Flow"
+verbose: True
+description: "ToDO: add description"
+
+model_name: "gpt-4"
+generation_parameters:
+  n: 1
+  max_tokens: 3000
+  temperature: 0.3
+
+  model_kwargs:
+    top_p: 0.2
+    frequency_penalty: 0
+    presence_penalty: 0
+
+system_message_prompt_template:
+  _target_: langchain.PromptTemplate
+  template: |2-
+    Your goal is to identify the issues with an incorrect competitive programming solution attempt.
+  
+    The user will specify the problem by providing you with:
+      - the problem statement
+      - input description
+      - output description
+      - example test cases
+      - (optional) explanation of the test cases
+      - an incorrect Python solution attempt and a description of its issue
+  
+    Crucially, your goal is to consider all aspects of the problem and pinpoint the issues with the solution attempt, and not to provide the code implementation yourself.
+    Some aspects to consider: Is the input correctly parsed? Is the output correctly formatted? Are the corner cases correctly handled? Is there a logical mistake with the algorithm itself?
+    Use the code execution results provided in the issue description to guide your reasoning/debugging.
+  input_variables: []
+  template_format: jinja2
+
+human_message_prompt_template:
+  _target_: langchain.PromptTemplate
+  template: "{{query}}"
+  input_variables:
+    - "query"
+  template_format: jinja2
+
+query_message_prompt_template:
+  _target_: langchain.PromptTemplate
+  template: |2-
+    # Problem statement
+    {{problem_description}}
+
+    # Input description
+    {{input_description}}
+
+    # Output description
+    {{output_description}}
+
+    {{io_examples_and_explanation}}
+
+    # Solution attempt to be fixed
+    ```python
+    {{code}}
+    ```
+
+    {{testing_results_summary}}
+
+
+    Consider the problem statement, the solution attempt and the issue. Why is the solution attempt incorrect? How should it be fixed? Explain your reasoning very concisely, and do not provide code.
+  input_variables:
+    - "problem_description"
+    - "input_description"
+    - "output_description"
+    - "io_examples_and_explanation"
+    - "code"
+    - "testing_results_summary"
+  template_format: jinja2
+
+input_data_transformations: []
+input_keys:
+  - "problem_description"
+  - "input_description"
+  - "output_description"
+  - "io_examples_and_explanation"
+  - "testing_results_summary"
+  - "code"
+
+output_data_transformations:
+  - _target_: flows.data_transformations.KeyRename
+    old_key2new_key:
+      raw_response: "code_feedback"
+output_keys:
+  - "code_feedback"

CF_CodeDebugCollab.py

@@ -0,0 +1,12 @@
+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()

CF_CodeDebugCollab.yaml

@@ -0,0 +1,71 @@
+name: "CodeDebugCollab_Flow"
+verbose: True
+description: "ToDO: add description"
+
+reset_generator_every_round: False
+reset_critic_every_round: True
+max_rounds: 2 # ToDo: To increase to 4
+early_exit_key: "end_of_interaction"
+
+input_data_transformations: []
+input_keys:
+  - "problem_description"
+  - "input_description"
+  - "output_description"
+  - "io_examples_and_explanation"
+  - "public_tests_individual_io"
+
+output_data_transformations:
+  - _target_: flows.data_transformations.KeyRename
+    old_key2new_key:
+      raw_response.code: "code"
+output_keys:
+  - "code"
+
+subflows_config:
+  - _target_: flows.flow_verse.instantiate_flow
+    repository_id: ${oc.env:CC_FLOWS}
+    class_name: CF_Code
+    overrides:
+      name: "CodeGenerator"
+      model_name: "gpt-4"
+      human_message_prompt_template:
+        _target_: langchain.PromptTemplate
+        template: |2-
+          {{testing_results_summary}}
+        
+          {{code_feedback}}
+
+
+          Consider the problem statement, the last proposed solution, its issue and the provided feedback. Return a corrected version of the code that solves the original problem and resolves the issue, without any explanation, in the following format:
+          ```python
+          {{code_placeholder}}
+          ```
+        input_variables:
+          - code_feedback
+          - testing_results_summary
+        partial_variables:
+          code_placeholder: "{{python_code}}"
+        template_format: jinja2
+      default_human_input_keys:
+        - "code_feedback"
+        - "testing_results_summary"
+      output_data_transformations:
+        - _target_: flows.data_transformations.RegexFirstOccurrenceExtractor
+          regex: '(?<=```python)([\s\S]*?)(?=```)'
+          regex_fallback: '(?<=```)([\s\S]*?)(?=```)'
+          input_key: "raw_response"
+          output_key: "code"
+          strip: True
+          assert_unique: True
+          verbose: True
+        - _target_: flows.data_transformations.EndOfInteraction
+          end_of_interaction_string: "Final answer"
+          output_key: "end_of_interaction"
+          verbose: True
+      output_keys:
+        - "code"
+        - "end_of_interaction"
+  - _target_: flows.flow_verse.instantiate_flow
+    repository_id: ${oc.env:CC_FLOWS}
+    class_name: CF_CodeDebugCritic

CF_CodeDebugCritic.py

@@ -0,0 +1,13 @@
+from flows.base_flows import SequentialFlow
+
+
+class CF_CodeDebugCritic(SequentialFlow):
+    def __init__(self, **kwargs):
+        super().__init__(**kwargs)
+
+    def _early_exit(self):
+        if self.flow_state.get("all_tests_passed", False):
+            self.flow_state["code_feedback"] = None
+            return True
+
+        return super()._early_exit()

CF_CodeDebugCritic.yaml

@@ -0,0 +1,35 @@
+name: "CodeDebugCritic_Flow"
+verbose: True
+description: "ToDo: add description"
+
+early_exit_key: null
+
+input_data_transformations: []
+input_keys:
+  - "problem_description"
+  - "input_description"
+  - "output_description"
+  - "io_examples_and_explanation"
+  - "public_tests_individual_io"
+  - "code"
+
+output_data_transformations:
+  - _target_: flows.data_transformations.KeyRename
+    old_key2new_key:
+      raw_response.testing_results_summary: "testing_results_summary"
+      raw_response.all_tests_passed: "all_tests_passed"
+      raw_response.code_feedback: "code_feedback"
+output_keys:
+  - "testing_results_summary"
+  - "all_tests_passed"
+  - "code_feedback"
+
+subflows_config:
+  - _target_: flows.flow_verse.instantiate_flow
+    repository_id: ${oc.env:CC_FLOWS}
+    class_name: CF_CodeTesting
+    overrides:
+      name: "CodeTestingCritic"
+  - _target_: flows.flow_verse.instantiate_flow
+    repository_id: ${oc.env:CC_FLOWS}
+    class_name: CF_CodeCriticWrongAttempt

__init__.py

@@ -27,3 +27,8 @@ from .CF_PlanCollab_Code import CF_PlanCollab_Code
 # cf-code_debug
 from .CF_CodeTesting import CF_CodeTesting
 from .CF_CodeDebug import CF_CodeDebug
+
+# cf-code_debug_collab
+from .CF_CodeCriticWrongAttempt import CF_CodeCriticWrongAttempt
+from .CF_CodeDebugCritic import CF_CodeDebugCritic
+from .CF_CodeDebugCollab import CF_CodeDebugCollab