name: "InteractiveCodeGenFlow"
description: "Generate code as goal requests, open up a temp file for the user to examine and provide feedback"

_target_: Tachi67.InteractiveCodeGenFlowModule.InteractiveCodeGenFlow.instantiate_from_default_config

memory_files: ???

input_interface:
  - "goal"

output_interface:
  - "code"
  - "feedback"
  - "temp_code_file_location"

subflows_config:
  MemoryReading:
    _target_: Tachi67.MemoryReadingFlowModule.MemoryReadingAtomicFlow.instantiate_from_default_config

  CodeGenerator:
    _target_: Tachi67.CodeGeneratorFlowModule.CodeGeneratorAtomicFlow.instantiate_from_default_config
    backend:
      api_infos: ???
      model_name:
        openai: gpt-4
        azure: azure/gpt-4

  CodeFileEditor:
    _target_: Tachi67.CodeFileEditFlowModule.CodeFileEditAtomicFlow.instantiate_from_default_config

  ParseFeedback:
    _target_: Tachi67.ParseFeedbackFlowModule.ParseFeedbackAtomicFlow.instantiate_from_default_config

early_exit_key: "EARLY_EXIT"

topology:
  - goal: "Read in necessary memory"
    input_interface:
      _target_: aiflows.interfaces.KeyInterface
      additional_transformations:
        - _target_: aiflows.data_transformations.KeyMatchInput
    flow: MemoryReading
    reset: false

  - goal: "Generate code to achieve the task."
    input_interface:
      _target_: aiflows.interfaces.KeyInterface
      additional_transformations:
        - _target_: aiflows.data_transformations.KeyMatchInput
    flow: CodeGenerator
    reset: false

  - goal: "Write the code generated to a temp file with instructions to the user"
    input_interface:
      _target_: aiflows.interfaces.KeyInterface
      additional_transformations:
        - _target_: aiflows.data_transformations.KeyMatchInput
    flow: CodeFileEditor
    reset: false

  - goal: "Parse user feedback from the temp file"
    input_interface:
      _target_: aiflows.interfaces.KeyInterface
      additional_transformations:
        - _target_: aiflows.data_transformations.KeyMatchInput
    flow: ParseFeedback
    reset: false