name: "CodeReflect_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"

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"
      input_data_transformations:
        - _target_: flows.data_transformations.KeyRename
          old_key2new_key:
            code_reflect_message: "query"
      output_data_transformations:
        - _target_: flows.data_transformations.RegexFirstOccurrenceExtractor
          regex: '(?<=```python)([\s\S]*?)(?=```)'
          regex_fallback: '(?<=```)([\s\S]*?)(?=```)'
          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.FixedReply_CodeReflect.instantiate_from_default_config