fix: deployment mode printing only to the file

agiresearch · dongyuanjushi · Apr 8, 2024 · Apr 2, 2024 · Apr 2, 2024 · Apr 2, 2024
commit 0f8557c3c2571552ef61269b0c3afa892fcfe16c
diff --git a/src/agents/agent_factory.py b/src/agents/agent_factory.py
@@ -10,6 +10,9 @@
 from concurrent.futures import ThreadPoolExecutor, as_completed
 
 from threading import Thread, Lock, Event
+
+from pympler import asizeof
+
 class AgentFactory:
     def __init__(self, llm, agent_process_queue, agent_log_mode):
         self.MAX_AID = MAX_AID
@@ -52,11 +55,15 @@ def activate_agent(self, agent_name, task_input):
         return agent
 
     def print_agent(self):
-        headers = ["Agent ID", "Agent Name", "Created Time", "Status"]
+        headers = ["Agent ID", "Agent Name", "Created Time", "Status", "Memory Usage"]
         data = []
         for id, agent in self.current_agents.items():
+            agent_name = agent.agent_name
+            created_time = agent.created_time
+            status = agent.status
+            memory_usage = f"{asizeof.asizeof(agent)} bytes"
             data.append(
-                [id, agent.agent_name, agent.created_time, agent.status]
+                [id, agent_name, created_time, status, memory_usage]
             )
         self.print(headers=headers, data=data)
 

diff --git a/src/agents/base.py b/src/agents/base.py
@@ -53,16 +53,24 @@ def run(self):
     def setup_logger(self):
         logger = logging.getLogger(f"{self.agent_name} Logger")
         # logger.setLevel(logging.INFO)  # Set the minimum logging level
-        logger.disabled = True
+        # logger.disabled = True
         date_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
         # Provide two log modes: console and file
 
+        # Ensure the logger doesn't propagate to the root logger
+        logger.propagate = False
+
+        # Remove all handlers associated with this logger
+        for handler in logger.handlers[:]:
+            logger.removeHandler(handler)
+
         if self.log_mode == "console":
-            logger.disabled = False
+            # logger.disabled = False
             handler = logging.StreamHandler()
             handler.setLevel(logging.INFO)  # Set logging level for console output
         else:
             assert self.log_mode == "file"
+            # logger.disabled = False
             log_dir = os.path.join(os.getcwd(), "logs", "agents",
                                     f"{self.agent_name}")
             if not os.path.exists(log_dir):

diff --git a/src/scheduler/base.py b/src/scheduler/base.py
@@ -29,9 +29,14 @@ def start(self):
     def setup_logger(self):
         logger = logging.getLogger(f"FIFO Scheduler Logger")
         # logger.setLevel(logging.INFO)  # Set the minimum logging level
-        logger.disabled = True
         date_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
         # Provide two log modes: console and file
+        # Ensure the logger doesn't propagate to the root logger
+        logger.propagate = False
+
+        # Remove all handlers associated with this logger
+        for handler in logger.handlers[:]:
+            logger.removeHandler(handler)
 
         if self.log_mode == "console":
             handler = logging.StreamHandler()