CC_flows / CF_PlanCollab.yaml
martinjosifoski's picture
Add early stopping for debugging on passed tests. Propagate minor changes to default human_message keys.
68afe75
name: "PlanCollab_Flow"
verbose: True
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.plan: "plan"
output_keys:
- "plan"
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
{{plan_feedback}}
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:
- plan_feedback
partial_variables:
plan_placeholder: "{{conceptual_solution}}"
template_format: jinja2
default_human_input_keys:
- "plan_feedback"
output_data_transformations:
- _target_: flows.data_transformations.RegexFirstOccurrenceExtractor
regex: '(?<=Conceptual solution)([\s\S]*?)(?=\n\n# [A-Z]|\Z)'
regex_fallback:
- '(?<=Conceptual solution:)([\s\S]*?)(?=\n\n# [A-Z]|\Z)'
output_key: "plan"
strip: True
assert_unique: True
verbose: True
- _target_: flows.data_transformations.EndOfInteraction
end_of_interaction_string: "Final answer"
output_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
output_data_transformations:
- _target_: flows.data_transformations.KeyRename
old_key2new_key:
raw_response: "plan_feedback"