Add flows and subflows for plan_oracle-code_debug_collab.

CF_CodeCriticWrongAttemptWithPlan.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_CodeCriticWrongAttemptWithPlan(OpenAIChatAtomicFlow):
+    def __init__(self, **kwargs):
+        super().__init__(**kwargs)

CF_CodeCriticWrongAttemptWithPlan.yaml

@@ -0,0 +1,95 @@
+name: "CodeCriticWrongAttemptWithPlan_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
+
+    Additionally, the user will provide you with a conceptual solution to the problem which should guide your reasoning.
+
+    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? Is the code implementation consistent with the conceptual solution? 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}}
+    
+    # Conceptual solution
+    {{plan}}
+    
+    # Solution attempt to be fixed
+    ```python
+    {{code}}
+    ```
+
+    {{testing_results_summary}}
+
+
+    Consider the problem statement, the conceptual solution, the code implementation 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"
+    - "plan"
+    - "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"
+  - "plan"
+  - "code"
+
+output_data_transformations:
+  - _target_: flows.data_transformations.KeyRename
+    old_key2new_key:
+      raw_response: "code_feedback"
+output_keys:
+  - "code_feedback"

CF_CodeDebugCollabWithPlan.py

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

CF_CodeDebugCollabWithPlan.yaml

@@ -0,0 +1,72 @@
+name: "CodeDebugCollabWithPlan_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"
+  - "plan"
+
+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_CodeWithPlan
+    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_CodeDebugCriticWithPlan

CF_CodeDebugCriticWithPlan.py

@@ -0,0 +1,13 @@
+from flows.base_flows import SequentialFlow
+
+
+class CF_CodeDebugCriticWithPlan(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_CodeDebugCriticWithPlan.yaml

@@ -0,0 +1,36 @@
+name: "CodeDebugCriticWithPlan_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"
+  - "plan"
+
+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_CodeCriticWrongAttemptWithPlan

__init__.py

@@ -32,3 +32,8 @@ from .CF_CodeDebug import CF_CodeDebug
 from .CF_CodeCriticWrongAttempt import CF_CodeCriticWrongAttempt
 from .CF_CodeDebugCritic import CF_CodeDebugCritic
 from .CF_CodeDebugCollab import CF_CodeDebugCollab
+
+# cf-plan_oracle-code_debug_collab
+from .CF_CodeCriticWrongAttemptWithPlan import CF_CodeCriticWrongAttemptWithPlan
+from .CF_CodeDebugCriticWithPlan import CF_CodeDebugCriticWithPlan
+from .CF_CodeDebugCollabWithPlan import CF_CodeDebugCollabWithPlan