CF_CodeDebug.yaml

name: "CodeDebug_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_keys:
  - "problem_description"
  - "input_description"
  - "output_description"
  - "io_examples_and_explanation"
  - "public_tests_individual_io"
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:
        template: |2-
          {{query}}


          Consider the problem statement, the last proposed solution, and its issue. Provide 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:
          - query
        partial_variables:
          code_placeholder: "{{python_code}}"
      input_data_transformations:
        - _target_: flows.data_transformations.KeyRename
          old_key2new_key:
            "test_results_summary": "query"
      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_CodeTesting
    overrides:
      name: "CodeTestingCritic"