CF_PlanCollab.yaml

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

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

output_keys:
  - "plan"
  - "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: ${oc.env:CC_FLOWS}
    class_name: CF_Plan
    overrides:
      name: "PlanGenerator"
      human_message_prompt_template:
        _target_: langchain.PromptTemplate
        template: |2-
          # Feedback on the last proposed conceptual solution
          {{query}}
    
    
          Consider the original problem statement, the last proposed solution and the provided feedback. Does the solution need to be updated? If so, provide the corrected version of the conceptual solution in the following format:
          # Conceptual solution
          {{plan_placeholder}}
          otherwise, reply:
          "Final answer."
        input_variables:
          - query
        partial_variables:
          plan_placeholder: "{{conceptual_solution}}"
        template_format: jinja2
      response_annotators:
        end_of_interaction_annotator:
          _target_: flows.message_annotators.EndOfInteraction
          end_of_interaction_message: "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_PlanCritic
    overrides:
      name: PlanCritic
      outputs_transformations:
        - _target_: flows.outputs_transformations.Rename
          old_key2new_key:
            raw_response: "query"