Add an id attribute to Agent

Author: RussellLuo

coagent/agents/dynamic_triage.py

@@ -85,7 +85,7 @@ def client(self) -> ModelClient:
     async def _update_swarm_agent(self) -> None:
         agent_names = list(self._sub_agents.keys())
         logger.debug(
-            f"[{self.__class__.__name__}] Discovered sub-agents: {agent_names}"
+            f"[{self.__class__.__name__} {self.id}] Discovered sub-agents: {agent_names}"
         )
 
         tools = []

coagent/core/agent.py

@@ -131,6 +131,13 @@ def __init__(self, timeout: float = 60):
             **message_types,
         }
 
+    @property
+    def id(self) -> str:
+        if self.address.id:
+            return f"{self.address.name}.{self.address.id}"
+        else:
+            return self.address.name
+
     def init(self, channel: Channel, address: Address) -> None:
         self.channel = channel
         self.address = address
@@ -171,9 +178,8 @@ async def stopped(self) -> None:
         pass
 
     async def receive(self, raw: RawMessage) -> None:
-        logger.debug(
-            f"[{self.__class__.__name__}] Received a message: {raw.model_dump()}"
-        )
+        name: str = f"{self.__class__.__name__} {self.id}"
+        logger.debug(f"[{name}] Received a message: {raw.model_dump()}")
 
         async with self._lock:
             self._last_msg_received_at = time.time()

coagent/core/factory.py

@@ -94,7 +94,7 @@ async def _recycle(self) -> None:
 
             running_num = total_num - idle_num
             logger.debug(
-                f"[Factory {self.address.name}] Recycling agents: {running_num} running, {idle_num} idle"
+                f"[Factory {self.id}] Recycling agents: {running_num} running, {idle_num} idle"
             )
 
             deleted_agents: list[Agent] = []

coagent/core/types.py

@@ -136,6 +136,12 @@ class State(str, enum.Enum):
 
 
 class Agent(abc.ABC):
+    @property
+    @abc.abstractmethod
+    def id(self) -> str:
+        """Return the unique ID of the agent."""
+        pass
+
     @abc.abstractmethod
     def init(self, channel: Channel, address: Address) -> None:
         """Initialize the agent with the given channel and address."""