name: "CodeReflect_Flow"
verbose: True
description: "ToDO: add description"

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

output_keys:
  - "code"
  - "end_of_interaction"

reset_generator_every_round: False
reset_critic_every_round: True
max_rounds: 2 # ToDo: To increase to 4
early_exit_key: "end_of_interaction"

subflows_config:
  - _target_: flows.flow_verse.instantiate_flow
    repository_id: "martinjosifoski/CC_flows"
    # repository_id: "/Users/josifosk/Documents/PhD/CC_flows"
    class_name: CF_Code
    overrides:
      name: "CodeGenerator"
      model_name: "gpt-4"
      output_keys:
        - "code"
        - "end_of_interaction"
      response_annotators:
        end_of_interaction_annotator:
          _target_: flows.message_annotators.EndOfInteraction
          end_of_interaction_message: "Final answer"
          key: "end_of_interaction"
          verbose: True
  - _target_: flows.base_flows.FixedReplyAtomicFlow.instantiate_with_overrides
    overrides:
      name: "CodeFixedReplyCritic"
      description: "ToDo: Add description"
      input_keys: []
      output_keys:
        - "query"
      fixed_reply: |2-
        Consider the problem statement and the last proposed solution. Are you sure that the solution is provided in the requested format, and crucially, solves the problem?
        If that is not the case, provide the corrected version of the code in the following format:
        ```python
        {{python_code}}
        ```
        otherwise, reply:
        "Final answer."