name: "CodeDebug_Flow"
description: "ToDO: add description"

max_rounds: 2 # ToDo: To increase to 4

input_keys:
  - "problem_description"
  - "input_description"
  - "output_description"
  - "io_examples_and_explanation"
  - "public_tests_individual_io"

output_keys:
  - "code"

subflows_config:
  CodeGenerator:
    _target_: .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}}"
      input_keys:
        - "testing_results_summary"

  CodeTestingCritic:
    _target_: .CF_CodeTesting.instantiate_from_default_config

topology:
  - flow: CodeGenerator
    reset_every_round: false
    output_transformations:
      - _target_: flows.data_transformations.EndOfInteraction
        end_of_interaction_string: "Final answer"
        input_key: "api_output"
        output_key: "end_of_interaction"

  - flow: CodeTestingCritic
    reset_every_round: true

early_exit_key: "end_of_interaction"