Fix CoS examples

Author: RussellLuo

examples/cos/cos.js

@@ -122,22 +122,23 @@ class Agent {
 
     let result = this.handle(msg);
 
-    if (!msg.reply) {
+	const replyAddr = msg.reply? msg.reply.address : undefined;
+    if (!replyAddr) {
       return
     }
 
     if (isAsyncIterator(result)) {
       for await (let r of result) {
         console.log(`partial result: ${JSON.stringify(r)}`);
-        await this.channel.publish(msg.reply, r);
+        await this.channel.publish(replyAddr, r);
       }
       // End of the iteration, send an extra StopIteration message.
       const stop = {header: {type: 'StopIteration'}}
-      await this.channel.publish(msg.reply, stop);
+      await this.channel.publish(replyAddr, stop);
     } else {
       result = await result;
       console.log(`result: ${JSON.stringify(result)}`);
-      await this.channel.publish(msg.reply, result);
+      await this.channel.publish(replyAddr, result);
     }
   }
 }

examples/cos/cos.py

@@ -111,19 +111,20 @@ async def receive(self, msg: dict) -> None:
 
         result = self.handle(msg)
 
-        reply = msg.get("reply")
-        if not reply:
+        reply = msg.get("reply") or {}
+        reply_addr = reply.get("address")
+        if not reply_addr:
             return
 
         if is_async_iterator(result):
             async for x in result:
-                await self.channel.publish(reply, x)
+                await self.channel.publish(reply_addr, x)
             # End of the iteration, send an extra StopIteration message.
             stop = {"header": {"type": "StopIteration"}}
-            await self.channel.publish(reply, stop)
+            await self.channel.publish(reply_addr, stop)
         else:
             x = await result
-            await self.channel.publish(reply, x)
+            await self.channel.publish(reply_addr, x)
 
     @abc.abstractmethod
     async def handle(self, msg: dict) -> Any:

examples/cos/goagent/cos.go

@@ -228,11 +228,12 @@ func (a *Server) Receive(data string) {
 	slog.Info("[Server.Receive] handle result.", slog.Any("result", result))
 
 	slog.Info("[Server.Receive] reply.", slog.Any("msg.Reply", msg.Reply))
-	if len(msg.Reply) == 0 {
+	replyAddr, ok := msg.Reply["address"].(map[string]any)
+	if !ok || len(replyAddr) == 0 {
 		return
 	}
 
-	err = a.channel.Publish(msg.Reply, result)
+	err = a.channel.Publish(replyAddr, result)
 	if err != nil {
 		slog.Error("[Server.Receive] Error publishing reply.", slog.Any("err", err), slog.Any("msg.Reply", msg.Reply))
 	}