| name: "CodeCollab_Flow" | |
| description: "ToDO: add description" | |
| max_rounds: 2 # ToDO: To increase to 4 | |
| input_keys: | |
| - "problem_description" | |
| - "input_description" | |
| - "output_description" | |
| - "io_examples_and_explanation" | |
| output_keys: | |
| - "code" | |
| subflows_config: | |
| CodeGenerator: | |
| _target_: .CF_Code.instantiate_from_default_config | |
| overrides: | |
| human_message_prompt_template: | |
| _target_: langchain.PromptTemplate | |
| template: |2- | |
| # Feedback on the last proposed solution | |
| {{code_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 code in the following format: | |
| ```python | |
| {{code_placeholder}} | |
| ``` | |
| otherwise, reply: | |
| "Final answer." | |
| input_variables: | |
| - code_feedback | |
| partial_variables: | |
| code_placeholder: "{{python_code}}" | |
| template_format: jinja2 | |
| input_keys: | |
| - "code_feedback" | |
| CodeCritic: | |
| _target_: .CF_CodeCritic.instantiate_from_default_config | |
| topology: | |
| - flow: CodeGenerator | |
| reset_every_round: false | |
| output_transformations: | |
| - _target_: flows.data_transformations.EndOfInteraction | |
| end_of_interaction_string: "Final answer" | |
| input_key: "api_output" | |
| output_key: "end_of_interaction" | |
| - flow: CodeCritic | |
| reset_every_round: true | |
| output_transformations: | |
| - _target_: flows.data_transformations.KeyRename | |
| old_key2new_key: | |
| api_output: "code_feedback" | |
| early_exit_key: "end_of_interaction" | |