Delete the agent when it's cancelled

Author: RussellLuo

coagent/core/agent.py

@@ -245,6 +245,8 @@ async def _handle_control(self, msg: ControlMessage) -> None:
             case Cancel():
                 if self._handle_data_task:
                     self._handle_data_task.cancel()
+                # Delete the agent when cancelled.
+                await self.delete()
 
     async def _handle_data(self) -> None:
         """Handle DATA messages."""

coagent/core/messages.py

@@ -73,7 +73,7 @@ class ControlMessage(Message):
 
 
 class Cancel(ControlMessage):
-    """A control message to cancel the DATA message processing of an agent."""
+    """A control message to cancel the processing of an agent and delete it."""
 
     pass