fix: Remove precompute method from agent classes and update logging in connection handling

Author: shima004

.gitignore

@@ -172,3 +172,4 @@ cython_debug/
 
 # ADF
 agent.log*
+precompute

adf_core_python/core/agent/agent.py

@@ -154,7 +154,6 @@ def post_connect(self) -> None:
             self._agent_info,
         )
 
-        self.logger.info("Agent running in %s mode", self._mode)
         self.logger.debug(f"agent_config: {self.config}")
 
     def update_step_info(
@@ -213,10 +212,6 @@ def update_step_info(
     def think(self) -> None:
         pass
 
-    @abstractmethod
-    def precompute(self) -> None:
-        pass
-
     @abstractmethod
     def get_requested_entities(self) -> list[EntityURN]:
         pass
@@ -269,12 +264,12 @@ def handle_connect_ok(self, msg: Any) -> None:
         self.send_acknowledge(msg.request_id)
         self.post_connect()
         self.logger.info(
-            f"Connected to kernel: {self.__class__.__qualname__} (request_id: {msg.request_id})",
+            f"Connected to kernel: {self.__class__.__qualname__} (request_id: {msg.request_id}, agent_id: {self.agent_id}, mode: {self.mode})",
             request_id=msg.request_id,
         )
-        if self.precompute_flag:
-            print("self.precompute_flag: ", self.precompute_flag)
-            self.precompute()
+        if self.is_precompute:
+            self.logger.info("Precompute finished")
+            exit(0)
 
         self.finish_post_connect_event.set()
 

adf_core_python/core/agent/office/office_fire.py

@@ -31,8 +31,5 @@ def __init__(
             finish_post_connect_event,
         )
 
-    def precompute(self) -> None:
-        pass
-
     def get_requested_entities(self) -> list[EntityURN]:
         return [EntityURN.FIRE_STATION]

adf_core_python/core/agent/office/office_police.py

@@ -31,8 +31,5 @@ def __init__(
             finish_post_connect_event,
         )
 
-    def precompute(self) -> None:
-        pass
-
     def get_requested_entities(self) -> list[EntityURN]:
         return [EntityURN.POLICE_OFFICE]

adf_core_python/core/agent/platoon/platoon_ambulance.py

@@ -31,8 +31,5 @@ def __init__(
             finish_post_connect_event,
         )
 
-    def precompute(self) -> None:
-        pass
-
     def get_requested_entities(self) -> list[EntityURN]:
         return [EntityURN.AMBULANCE_TEAM]

adf_core_python/core/agent/platoon/platoon_fire.py

@@ -31,8 +31,5 @@ def __init__(
             finish_post_connect_event,
         )
 
-    def precompute(self) -> None:
-        pass
-
     def get_requested_entities(self) -> list[EntityURN]:
         return [EntityURN.FIRE_BRIGADE]

adf_core_python/core/agent/platoon/platoon_police.py

@@ -31,8 +31,5 @@ def __init__(
             finish_post_connect_event,
         )
 
-    def precompute(self) -> None:
-        pass
-
     def get_requested_entities(self) -> list[EntityURN]:
         return [EntityURN.POLICE_FORCE]