@feynmanzhang/open-party 0.1.8 → 0.1.9

package/dist/party-server.js

@@ -3994,9 +3994,15 @@ agentRoutes.post("/heartbeat", async (c) => {
 agentRoutes.get("/list", async (c) => {
   const { registry: registry2, discovery: discovery2 } = await Promise.resolve().then(() => (init_state(), state_exports));
   const localAgents = registry2.listAll();
-  const remoteAgents = discovery2.getReachableRemoteAgents();
-  const allAgents = localAgents.concat(remoteAgents);
-  return c.json({ agents: sanitizeAgentList(allAgents), count: allAgents.length });
+  const remoteEntries = discovery2.getRemoteAgentEntries();
+  const remoteAgents = remoteEntries.filter((e) => e.reachable).map((e) => ({ ...sanitizeAgentInfo(e.agentInfo), source_peer_ip: e.sourcePeerIp }));
+  const allAgents = sanitizeAgentList(localAgents).concat(remoteAgents);
+  return c.json({
+    agents: allAgents,
+    count: allAgents.length,
+    local_count: localAgents.length,
+    remote_count: remoteAgents.length
+  });
 });
 agentRoutes.post("/send", async (c) => {
   const { registry: registry2, messageQueue: messageQueue2, discovery: discovery2 } = await Promise.resolve().then(() => (init_state(), state_exports));
@@ -4082,6 +4088,11 @@ proxyRoutes.get("/list_agents", async (c) => {
   const agents = registry.listAll();
   return c.json({ agents, count: agents.length });
 });
+proxyRoutes.get("/peers", async (c) => {
+  const { discovery: discovery2 } = await Promise.resolve().then(() => (init_state(), state_exports));
+  const peerStates = discovery2.getPeerStates();
+  return c.json({ peers: peerStates, total: peerStates.length });
+});
 proxyRoutes.post("/receive", async (c) => {
   const envelope = await c.req.json();
   const stamped = { ...envelope, timestamp: envelope.timestamp ?? Date.now() / 1e3 };