Improve the CLI to handle the discovery agent specially

Author: RussellLuo

coagent/cli/main.py

@@ -50,6 +50,13 @@ async def run(
     else:
         agent_type, session_id = parts[0], uuid.uuid4().hex
 
+    probe = True
+    if agent_type == "discovery":
+        # The discovery agent is a system-provided agent, which is a distributed
+        # singleton and is always available.
+        session_id = ""
+        probe = False
+
     if server.startswith("nats://"):
         runtime = NATSRuntime.from_servers(server)
     elif server.startswith(("http://", "https://")):
@@ -62,7 +69,11 @@ async def run(
         try:
             if not stream:
                 response = await runtime.channel.publish(
-                    addr, msg, request=True, timeout=10
+                    addr,
+                    msg,
+                    request=True,
+                    timeout=10,
+                    probe=probe,
                 )
                 print_msg(response, oneline, filter)
             else:

examples/discovery/README.md

@@ -27,8 +27,14 @@ python examples/discovery/server.py team2.dev "Dev Engineer in Team 2."
 python examples/discovery/server.py team3.qa "QA Engineer in Team 3."
 ```
 
-Then start a client in another terminal.
+Then discover the available agent types using the `coagent` CLI:
 
 ```bash
-python examples/discovery/client.py team1
+coagent discovery -H type:DiscoveryQuery -F .content.agents -d '{"namespace":"team1"}'
+```
+
+Or start a more-friendly client in another terminal:
+
+```bash
+python examples/discovery/client.py --namespace team1
 ```