name: "CodeDebug_Flow"
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_: martinjosifoski.CC_flows.CF_Code.instantiate_from_default_config
    overrides:
      name: "CodeGenerator"
      model_name: "gpt-4"
      human_message_prompt_template:
        template: |2-
          {{testing_results_summary}}


          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:
          - testing_results_summary
        partial_variables:
          code_placeholder: "{{python_code}}"
      default_human_input_keys:
        - "testing_results_summary"
        - "all_tests_passed"
      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

        - _target_: flows.data_transformations.EndOfInteraction
          end_of_interaction_string: "Final answer"
          output_key: "end_of_interaction"

      output_keys:
        - "code"
        - "end_of_interaction"
  - _target_: martinjosifoski.CC_flows.CF_CodeTesting.instantiate_from_default_config
    overrides:
      name: "CodeTestingCritic"