_target_: Tachi67.JarvisFlowModule.CtrlExMem_JarvisFlow.instantiate_from_default_config
name: "CtrlExMem_JarvisFlow"
description: "MemorizedControllerExecutor flow for Jarvis flow"

input_interface:
  - "plan"
  - "logs"
  - "memory_files"
  - "goal"

subflows_config:
  Controller:
    _target_: Tachi67.JarvisFlowModule.Controller_JarvisFlow.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:
      Coder:
        _target_: Tachi67.CoderFlowModule.CoderFlow.instantiate_from_default_config
        memory_files: ???
        subflows_config:
          Planner:
            #TODO remove this line
            _target_: Tachi67.CoderFlowModule.Planner_CoderFlow.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

          CtrlExMem:
            subflows_config:
              Controller:
                backend:
                  api_infos: ???
                  model_name:
                    openai: gpt-4
                    azure: azure/gpt-4
              Executor:
                subflows_config:
                  extend_library:
                    memory_files: ???
                    subflows_config:
                      Planner:
                        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
                      CtrlExMem:
                        subflows_config:
                          Controller:
                            backend:
                              api_infos: ???
                              model_name:
                                openai: gpt-4
                                azure: azure/gpt-4
                          Executor:
                            subflows_config:
                              write_code:
                                subflows_config:
                                  Controller:
                                    backend:
                                      api_infos: ???
                                      model_name:
                                        openai: gpt-4
                                        azure: azure/gpt-4
                                  Executor:
                                    subflows_config:
                                      write_code:
                                        memory_files: ???
                                        subflows_config:
                                          CodeGenerator:
                                            backend:
                                              api_infos: ???
                                              model_name:
                                                openai: gpt-4
                                                azure: azure/gpt-4
                                      test:
                                        memory_files: ???
                              re_plan:
                                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
                  re_plan:
                    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
      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.JarvisFlowModule.UpdatePlanAtomicFlow.instantiate_from_default_config
      intermediate_answer:
        _target_: Tachi67.JarvisFlowModule.IntermediateAns_Jarvis.instantiate_from_default_config
      final_answer:
        _target_: Tachi67.JarvisFlowModule.FinalAns_Jarvis.instantiate_from_default_config


  MemoryReading:
    _target_: Tachi67.MemoryReadingFlowModule.MemoryReadingAtomicFlow.instantiate_from_default_config
    output_interface:
      - "plan"
      - "logs"

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_JarvisFlow.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.plan: plan
        branch_output_data.status: status
        branch_output_data.summary: summary
        branch_output_data.result: result
      keys_to_delete: ["branch_output_data"]
    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