_target_: Tachi67.ExtendLibraryFlowModule.CtrlExMem_ExtLib.instantiate_from_default_config
name: "CtrlExMem_ExtLibFlow"
description: "MemorizedControllerExecutor flow for ExtendLibrary flow"

subflows_config:
  Controller:
    _target_: Tachi67.ExtendLibraryFlowModule.ControllerFlow_ExtLib.instantiate_from_default_config
    backend:
      api_infos: ???
      model_name:
        openai: gpt-4
        azure: azure/gpt-4
  Executor:
    _target_: aiflows.base_flows.BranchingFlow.instantiate_from_default_config
    subflows_config:
      write_code:
        _target_: Tachi67.CodeWriterFlowModule.CodeWriterFlow.instantiate_from_default_config
        subflows_config:
          Controller:
            backend:
              api_infos: ???
              model_name:
                openai: gpt-4
                azure: azure/gpt-4
          Executor:
            subflows_config:
              write_code:
                subflows_config:
                  CodeGenerator:
                    backend:
                      api_infos: ???
                      model_name:
                        openai: gpt-4
                        azure: azure/gpt-4
      save_code:
        _target_: Tachi67.ExtendLibraryFlowModule.SaveCodeAtomicFlow.instantiate_from_default_config
      ask_user:
        _target_: Tachi67.ExtendLibraryFlowModule.ExtLibAskUserFlow.instantiate_from_default_config
      re_plan:
        _target_: Tachi67.ReplanningFlowModule.ReplanningFlow.instantiate_from_default_config
        subflows_config:
          Controller:
            backend:
              api_infos: ???
              model_name:
                openai: gpt-4
                azure: azure/gpt-4
          Executor:
            subflows_config:
              write_plan:
                subflows_config:
                  PlanGenerator:
                    backend:
                      api_infos: ???
                      model_name:
                        openai: gpt-4
                        azure: azure/gpt-4

      update_plan:
        _target_: Tachi67.ExtendLibraryFlowModule.UpdatePlanAtomicFlow.instantiate_from_default_config

topology:
  - goal: "Select the next action and prepare the input for the executor."
    input_interface:
      _target_: aiflows.interfaces.KeyInterface
      additional_transformations:
        - _target_: aiflows.data_transformations.KeyMatchInput
    flow: Controller
    output_interface:
      _target_: CtrlExMem_ExtLib.detect_finish_or_continue
    reset: false

  - goal: "Execute the action specified by the Controller."
    input_interface:
      _target_: aiflows.interfaces.KeyInterface
      keys_to_rename:
        command: branch
        command_args: branch_input_data
      keys_to_select: ["branch", "branch_input_data"]
    flow: Executor
    output_interface:
      _target_: aiflows.interfaces.KeyInterface
      keys_to_rename:
        branch_output_data: observation
      keys_to_unpack: ["observation"]
    reset: false

  - goal: "Write memory to memory files"
    input_interface:
      _target_: aiflows.interfaces.KeyInterface
      additional_transformations:
        - _target_: aiflows.data_transformations.KeyMatchInput
    flow: MemoryWriting
    reset: false

  - goal: "Read memory from memory files (flow_state)"
    input_interface:
      _target_: aiflows.interfaces.KeyInterface
      additional_transformations:
        - _target_: aiflows.data_transformations.KeyMatchInput
    flow: MemoryReading
    reset: false