@sjawhar/pi-legion-envoy 1.24.0 → 1.24.1

package/dist/legion.js

@@ -30665,6 +30665,30 @@ function createLegionDaemonClient(baseUrl, fetchFn = fetch, recovery) {
       throw new LegionDaemonApiError("POST", path, response.status, responseBody);
     return schema.parse(JSON.parse(responseBody));
   };
+  const recoveries = new Map;
+  const secretAfterRefusal = async (active, sessionId, refusedSecret) => {
+    let record = recoveries.get(sessionId);
+    if (record === undefined) {
+      record = {};
+      recoveries.set(sessionId, record);
+    }
+    if (record.latestSecret !== undefined && record.latestSecret !== refusedSecret) {
+      return record.latestSecret;
+    }
+    if (record.inFlight === undefined) {
+      const current = record;
+      const inFlight = postOnce(WORKER_SESSION_PATH, { sessionId, recoveryToken: active.recoveryToken(sessionId) }, LegionDaemonApi.WorkerSession.response).then((recovered) => {
+        active.onRecovered?.(sessionId, recovered);
+        current.latestSecret = recovered.secret;
+        return recovered;
+      }).finally(() => {
+        if (current.inFlight === inFlight)
+          current.inFlight = undefined;
+      });
+      record.inFlight = inFlight;
+    }
+    return (await record.inFlight).secret;
+  };
   const post = async (path, body, schema) => {
     try {
       return await postOnce(path, body, schema);
@@ -30673,12 +30697,8 @@ function createLegionDaemonClient(baseUrl, fetchFn = fetch, recovery) {
       if (path === WORKER_SESSION_PATH || recovery === undefined || capability === undefined || !isInvalidSessionSecret(error)) {
         throw error;
       }
-      const recovered = await postOnce(WORKER_SESSION_PATH, {
-        sessionId: capability.sessionId,
-        recoveryToken: recovery.recoveryToken(capability.sessionId)
-      }, LegionDaemonApi.WorkerSession.response);
-      recovery.onRecovered?.(capability.sessionId, recovered);
-      return await postOnce(path, { ...body, secret: recovered.secret }, schema);
+      const secret = await secretAfterRefusal(recovery, capability.sessionId, capability.secret);
+      return await postOnce(path, { ...body, secret }, schema);
     }
   };
   const get = async (path, schema) => {
@@ -34845,8 +34865,9 @@ function legionExtension(pi) {
     return subagentSession;
   };
   let phaseWorkerPane;
+  let daemonClient;
   const roleDaemon = () => {
-    return createLegionDaemonClient(requiredEnvironment(process.env, "LEGION_DAEMON_URL"), fetch, {
+    daemonClient ??= createLegionDaemonClient(requiredEnvironment(process.env, "LEGION_DAEMON_URL"), fetch, {
       recoveryToken: (sessionId) => {
         if (capability !== undefined && sessionId === capability.sessionID) {
           return requiredSecret(process.env, "LEGION_BOOT_TOKEN");
@@ -34863,6 +34884,7 @@ function legionExtension(pi) {
         capability = { ...capability, secret: recovered.secret };
       }
     });
+    return daemonClient;
   };
   const rerunReadyAfterRegain = async (endpoint, role, reason, call) => {
     try {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@sjawhar/pi-legion-envoy",
-  "version": "1.24.0",
+  "version": "1.24.1",
   "type": "module",
   "omp": {
     "extensions": [