Rename SetReplyAgent to SetReplyInfo

Author: RussellLuo

coagent/agents/parallel.py

@@ -9,7 +9,7 @@
     NO_REPLY,
     RawMessage,
     Reply,
-    SetReplyAgent,
+    SetReplyInfo,
 )
 
 
@@ -95,7 +95,7 @@ async def started(self) -> None:
             addr = Address(name=agent_type, id=self.address.id)
             await self.channel.publish(
                 addr,
-                SetReplyAgent(reply_info=aggregator_reply).encode(),
+                SetReplyInfo(reply_info=aggregator_reply).encode(),
             )
 
         # Set the current agent to no-reply mode.

coagent/agents/sequential.py

@@ -5,7 +5,7 @@
     GenericMessage,
     NO_REPLY,
     Reply,
-    SetReplyAgent,
+    SetReplyInfo,
     handler,
 )
 
@@ -25,7 +25,7 @@ async def started(self) -> None:
             reply = Reply(address=next_addr)
             await self.channel.publish(
                 addr,
-                SetReplyAgent(reply_info=reply).encode(),
+                SetReplyInfo(reply_info=reply).encode(),
             )
 
         # Set the current agent to no-reply mode.
@@ -42,7 +42,7 @@ async def handle(self, msg: GenericMessage, ctx: Context) -> None:
             last_addr = Address(name=self._agent_types[-1], id=self.address.id)
             await self.channel.publish(
                 last_addr,
-                SetReplyAgent(reply_info=reply).encode(),
+                SetReplyInfo(reply_info=reply).encode(),
             )
 
         # Send the message to the first agent in the list.

coagent/core/__init__.py

@@ -6,7 +6,7 @@
 )
 from .discovery import DiscoveryQuery, DiscoveryReply
 from .logger import logger, set_stderr_logger
-from .messages import Message, GenericMessage, SetReplyAgent, StopIteration
+from .messages import Message, GenericMessage, SetReplyInfo, StopIteration
 from .runtime import BaseRuntime, BaseChannel, QueueSubscriptionIterator
 from .types import (
     Address,

coagent/core/agent.py

@@ -15,7 +15,7 @@
     Message,
     Started,
     Stopped,
-    SetReplyAgent,
+    SetReplyInfo,
     StopIteration,
 )
 from .types import (
@@ -132,7 +132,7 @@ def __init__(self, timeout: float = 60):
         # this would result in a lot of messages.
         self._last_msg_received_at_lock: asyncio.Lock = asyncio.Lock()
 
-        # Normally `reply` is set by an orchestration agent by sending a `SetReplyAgent` message.
+        # Normally `reply` is set by an orchestration agent by sending a `SetReplyInfo` message.
         self.reply: Reply | None = None
         self._reply_lock: asyncio.Lock = asyncio.Lock()
 
@@ -144,7 +144,7 @@ def __init__(self, timeout: float = 60):
             "Cancel": Cancel,
             "Started": Started,
             "Stopped": Stopped,
-            "SetReplyAgent": SetReplyAgent,
+            "SetReplyInfo": SetReplyInfo,
             "ProbeAgent": ProbeAgent,
             "Empty": Empty,
             **message_types,
@@ -279,7 +279,7 @@ async def _handle_data(self) -> None:
                 case Stopped():
                     await self.stopped()
 
-                case SetReplyAgent():
+                case SetReplyInfo():
                     await self._set_reply_info(msg.reply_info)
 
                 case ProbeAgent() | Empty():

coagent/core/messages.py

@@ -122,7 +122,7 @@ class ProbeAgent(Message):
     pass
 
 
-class SetReplyAgent(Message):
+class SetReplyInfo(Message):
     """A message to set the reply information of an agent.
 
     This is mainly useful when orchestrating multiple agents to work together.