From 3c3d8121dd03945438085567cedf6e7c2b48fd0c Mon Sep 17 00:00:00 2001 From: shima004 Date: Wed, 25 Sep 2024 17:06:44 +0900 Subject: [PATCH 1/2] feat: Update main.py to use AgentLauncher for launching agents --- adf_core_python/main.py | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) diff --git a/adf_core_python/main.py b/adf_core_python/main.py index d4f477a..c988075 100644 --- a/adf_core_python/main.py +++ b/adf_core_python/main.py @@ -1,5 +1,9 @@ import argparse +from adf_core_python.core.config.config import Config +from adf_core_python.core.launcher.agent_launcher import AgentLauncher +from adf_core_python.core.launcher.config_key import ConfigKey + class Main: def __init__(self) -> None: @@ -55,7 +59,23 @@ def __init__(self) -> None: ) args = parser.parse_args() print(args) + self.config = Config() + self.config.set_value(ConfigKey.KEY_KERNEL_HOST, args.host) + self.config.set_value(ConfigKey.KEY_KERNEL_PORT, args.port) + self.config.set_value(ConfigKey.KEY_AMBULANCE_CENTRE_COUNT, args.ambulance) + self.config.set_value(ConfigKey.KEY_FIRE_STATION_COUNT, args.firebrigade) + self.config.set_value(ConfigKey.KEY_POLICE_OFFICE_COUNT, args.policeforce) + self.config.set_value(ConfigKey.KEY_PRECOMPUTE, args.precompute) + self.config.set_value(ConfigKey.KEY_DEBUG_FLAG, args.verbose) + + def launch(self): + agent_launcher: AgentLauncher = AgentLauncher( + self.config, + ) + agent_launcher.init_connector() + agent_launcher.launch() if __name__ == "__main__": main = Main() + main.launch() From c6748ec2d7f7bf4822081b073185ef11800f708c Mon Sep 17 00:00:00 2001 From: shima004 Date: Wed, 25 Sep 2024 17:11:18 +0900 Subject: [PATCH 2/2] feat: Update launch method in main.py to use AgentLauncher --- adf_core_python/main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/adf_core_python/main.py b/adf_core_python/main.py index c988075..0e6d588 100644 --- a/adf_core_python/main.py +++ b/adf_core_python/main.py @@ -68,7 +68,7 @@ def __init__(self) -> None: self.config.set_value(ConfigKey.KEY_PRECOMPUTE, args.precompute) self.config.set_value(ConfigKey.KEY_DEBUG_FLAG, args.verbose) - def launch(self): + def launch(self) -> None: agent_launcher: AgentLauncher = AgentLauncher( self.config, )