Remove BaseRuntime.delete()

Author: RussellLuo

coagent/core/runtime.py

@@ -6,7 +6,7 @@
 from .discovery import Discovery
 from .exceptions import BaseError
 from .messages import StopIteration, Error
-from .factory import Factory, DeleteAgent
+from .factory import Factory
 from .types import (
     AgentSpec,
     Channel,
@@ -69,11 +69,6 @@ async def deregister(self, *names: str) -> None:
     def channel(self) -> Channel:
         return self._channel
 
-    async def delete(self, addr: Address) -> None:
-        factory_addr = Address(name=addr.name)
-        msg = DeleteAgent(session_id=addr.id).encode()
-        await self._channel.publish(factory_addr, msg, probe=False)
-
 
 class BaseChannel(Channel):
     async def publish_multi(

coagent/cos/runtime.py

@@ -16,7 +16,6 @@
 )
 from coagent.core.messages import Cancel
 from coagent.core.exceptions import BaseError
-from coagent.core.factory import DeleteAgent
 from coagent.core.types import Runtime
 from coagent.core.util import clear_queue
 
@@ -122,23 +121,20 @@ async def event_stream() -> AsyncIterator[str]:
                 # Disconnected from the client.
 
                 # Delete the corresponding agent.
-                factory_addr = Address(name=addr.name)
-                delete_msg = DeleteAgent(session_id=addr.id).encode()
-                await self._runtime.channel.publish(
-                    factory_addr, delete_msg, probe=False
-                )
+                await agent.delete()
 
                 raise
 
         return EventSourceResponse(event_stream())
 
     async def publish(self, request: Request):
         data: dict = await request.json()
+        addr = Address.decode(data["addr"])
+        msg = RawMessage.decode(data["msg"])
+
         try:
-            msg = RawMessage.decode(data["msg"])
             await self._update_message_header_extensions(msg, request)
 
-            addr = Address.decode(data["addr"])
             resp: RawMessage | None = await self._runtime.channel.publish(
                 addr=addr,
                 msg=msg,