Update run_ExtendLibrary.py

run_ExtendLibrary.py

@@ -2,12 +2,13 @@ import os
 
 import hydra
 
-from flows.backends.api_info import ApiInfo
-from flows.messages import InputMessage
-from flows.utils.general_helpers import read_yaml_file
+from aiflows.backends.api_info import ApiInfo
+from aiflows.messages import InputMessage
+from aiflows.utils.general_helpers import read_yaml_file
 
-from flows import logging
-from flows.flow_cache import CACHING_PARAMETERS, clear_cache
+from aiflows import logging
+from aiflows.flow_cache import CACHING_PARAMETERS, clear_cache
+from aiflows.utils.general_helpers import update_api_infos
 
 CACHING_PARAMETERS.do_caching = False  # Set to True in order to disable caching
 # clear_cache() # Uncomment this line to clear the cache
@@ -26,7 +27,7 @@ dependencies = [
     {"url": "Tachi67/CodeWriterFlowModule", "revision": "main"},
 ]
 
-from flows import flow_verse
+from aiflows import flow_verse
 
 flow_verse.sync_dependencies(dependencies)
 
@@ -40,13 +41,7 @@ if __name__ == "__main__":
     cfg_path = os.path.join(current_dir, "ExtendLibraryFlow.yaml")
     cfg = read_yaml_file(cfg_path)
     # ~~~ 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
+    update_api_infos(cfg, api_information)
 
     # ~~~ setting memory files into config ~~~
     code_lib_file_loc = os.path.join(current_dir, "library.py")