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

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

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

output_keys:
  - "plan"

subflows_config:
  - _target_: flows.flow_verse.instantiate_flow
    repository_id: ${oc.env:CC_FLOWS}
    class_name: CF_Plan
    overrides:
      name: "PlanGenerator"
      model_name: "gpt-4"
      response_annotators:
        end_of_interaction_annotator:
          _target_: flows.message_annotators.EndOfInteraction
          end_of_interaction_string: "Final answer"
          key: "end_of_interaction"
          verbose: True
      output_keys:
        - "plan"
        - "end_of_interaction"
  - _target_: flows.flow_verse.instantiate_flow
    repository_id: ${oc.env:CC_FLOWS}
    class_name: CF_Reflect
    overrides:
      name: "PlanFixedReplyCritic"
      description: "ToDo: Add description"
      input_keys:
        - "plan"
      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 conceptual solution in the following format:
        # Conceptual solution
        {{conceptual_solution}}
        otherwise, reply:
        "Final answer."