Upload run_ExtendLibrary.py

run_ExtendLibrary.py

@@ -16,6 +16,8 @@ logging.set_verbosity_debug()
 logging.auto_set_dir()
 
 dependencies = [
+    {"url": "aiflows/HumanStandardInputFlowModule", "revision": "5683a922372c5fa90be9f6447d6662d8d80341fc"},
+    {"url": "aiflows/ChatFlowModule", "revision": "a749ad10ed39776ba6721c37d0dc22af49ca0f17"},
     {"url": "Tachi67/ExtendLibraryFlowModule", "revision": "main"},
     {"url": "Tachi67/AbstractBossFlowModule", "revision": "main"},
     {"url": "Tachi67/MemoryReadingFlowModule", "revision": "main"},
@@ -34,30 +36,51 @@ if __name__ == "__main__":
     api_information = [ApiInfo(backend_used="openai", api_key=os.getenv("OPENAI_API_KEY"))]
     path_to_output_file = None
 
-    # ~~~ setting api information into config ~~~
     current_dir = os.getcwd()
-    cfg_path = os.path.join(current_dir, "CodeWriterFlow.yaml")
+    cfg_path = os.path.join(current_dir, "ExtendLibraryFlow.yaml")
     cfg = read_yaml_file(cfg_path)
-    cfg["subflows_config"]["Controller"]["backend"]["api_infos"] = api_information
-    cfg["subflows_config"]["Executor"]["subflows_config"]["write_code"]["subflows_config"]["CodeGenerator"]["backend"]["api_infos"] = api_information
+    # ~~~ setting api information into config ~~~
+    cfg["subflows_config"]["Planner"]["subflows_config"]["Controller"]["backend"]["api_infos"] = api_information
+    cfg["subflows_config"]["Planner"]["subflows_config"]["Executor"]["subflows_config"]["write_plan"]["subflows_config"]["PlanGenerator"]["backend"]["api_infos"] = api_information
+    cfg["subflows_config"]["CtrlExMem"]["subflows_config"]["Controller"]["backend"]["api_infos"] = api_information
+    cfg["subflows_config"]["CtrlExMem"]["subflows_config"]["Executor"]["subflows_config"]["write_code"]["subflows_config"]["Controller"]["backend"]["api_infos"] = api_information
+    cfg["subflows_config"]["CtrlExMem"]["subflows_config"]["Executor"]["subflows_config"]["write_code"]["subflows_config"]["Executor"]["subflows_config"]["write_code"]["subflows_config"]["CodeGenerator"]["backend"]["api_infos"] = api_information
+    cfg["subflows_config"]["CtrlExMem"]["subflows_config"]["Executor"]["subflows_config"]["re_plan"]["subflows_config"]["Controller"]["backend"]["api_infos"] = api_information
+    cfg["subflows_config"]["CtrlExMem"]["subflows_config"]["Executor"]["subflows_config"]["re_plan"]["subflows_config"]["Executor"]["subflows_config"]["write_plan"]["subflows_config"]["PlanGenerator"]["backend"]["api_infos"] = api_information
 
-    # ~~~ setting code library into config ~~~
+    # ~~~ setting memory files into config ~~~
     code_lib_file_loc = os.path.join(current_dir, "library.py")
+    plan_file_loc = os.path.join(current_dir, "plan_extlib.txt")
+    logs_file_loc = os.path.join(current_dir, "logs_extlib.txt")
     with open(code_lib_file_loc, 'w') as file:
         pass
-    cfg["subflows_config"]["Executor"]["subflows_config"]["write_code"]["memory_files"] = {"code_library": code_lib_file_loc}
-    cfg["subflows_config"]["Executor"]["subflows_config"]["test"]["memory_files"] = {"code_library": code_lib_file_loc}
+    with open(plan_file_loc, 'w') as file:
+        pass
+    with open(logs_file_loc, 'w') as file:
+        pass
+    memfiles_extlib = {}
+    memfiles_writecode_interactivecoder = {}
+    memfiles_writecode_test = {}
+    memfiles_extlib["plan"] = plan_file_loc
+    memfiles_extlib["logs"] = logs_file_loc
+    memfiles_extlib["code_library"] = code_lib_file_loc
+    memfiles_writecode_interactivecoder["code_library"] = code_lib_file_loc
+    memfiles_writecode_test["code_library"] = code_lib_file_loc
+    cfg["memory_files"] = memfiles_extlib
+    cfg["subflows_config"]["CtrlExMem"]["subflows_config"]["Executor"]["subflows_config"]["write_code"]["subflows_config"]["Executor"]["subflows_config"]["write_code"]["memory_files"] = memfiles_writecode_interactivecoder
+    cfg["subflows_config"]["CtrlExMem"]["subflows_config"]["Executor"]["subflows_config"]["write_code"]["subflows_config"]["Executor"]["subflows_config"]["test"]["memory_files"] = memfiles_writecode_test
+
 
     # ~~~ instantiating the flow and input data ~~~
-    CodeWriterFlow = hydra.utils.instantiate(cfg, _recursive_=False, _convert_="partial")
+    ExtendLibraryFlow = hydra.utils.instantiate(cfg, _recursive_=False, _convert_="partial")
     input_data = {
-        "goal": "create a function that adds two numbers and returns the result",
+        "goal": "Write a function that adds up two numbers",
     }
     input_message = InputMessage.build(
         data_dict=input_data,
         src_flow="Launcher",
-        dst_flow=CodeWriterFlow.name
+        dst_flow=ExtendLibraryFlow.name
     )
 
     # ~~~ calling the flow ~~~
-    output_message = CodeWriterFlow(input_message)
+    output_message = ExtendLibraryFlow(input_message)