@tylerl0706/ahpx 0.2.8 → 0.2.9

package/dist/bin.js

@@ -11,7 +11,7 @@ import {
   createLogger,
   ensureFileUri,
   setVerbose
-} from "./chunk-B6RV43UP.js";
+} from "./chunk-TCJ6NC7B.js";
 
 // src/bin.ts
 import { randomUUID as randomUUID3 } from "crypto";
@@ -2680,26 +2680,32 @@ async function resolveOrCreateSession(client, serverInfo, opts, cfg, cwd, gitRoo
 function waitForReady(client, sessionUri) {
   return new Promise((resolve3, reject) => {
     const timeout = setTimeout(() => {
+      cleanup();
       reject(new TimeoutError("Timed out waiting for session to be ready"));
     }, 3e4);
-    client.on("action", (envelope) => {
+    const onAction = (envelope) => {
       const action = envelope.action;
       if (action.type === "session/ready" /* SessionReady */ && action.session === sessionUri) {
-        clearTimeout(timeout);
+        cleanup();
         resolve3();
       } else if (action.type === "session/creationFailed" /* SessionCreationFailed */ && action.session === sessionUri) {
-        clearTimeout(timeout);
+        cleanup();
         const sessionState2 = client.state.getSession(sessionUri);
         const errMsg = sessionState2?.creationError?.message ?? "Unknown error";
         reject(new AhpxError(`Session creation failed: ${errMsg}`, ExitCode.Error));
       }
-    });
+    };
+    const cleanup = () => {
+      clearTimeout(timeout);
+      client.removeListener("action", onAction);
+    };
+    client.on("action", onAction);
     const sessionState = client.state.getSession(sessionUri);
     if (sessionState?.lifecycle === "ready") {
-      clearTimeout(timeout);
+      cleanup();
       resolve3();
     } else if (sessionState?.lifecycle === "creationFailed") {
-      clearTimeout(timeout);
+      cleanup();
       const errMsg = sessionState?.creationError?.message ?? "Unknown error";
       reject(new AhpxError(`Session creation failed: ${errMsg}`, ExitCode.Error));
     }

package/dist/{chunk-B6RV43UP.js → chunk-TCJ6NC7B.js}

@@ -604,17 +604,41 @@ function sessionReducer(state, action, log7) {
       }
       return next;
     }
-    case "session/delta" /* SessionDelta */:
-      return updateResponsePart(state, action.turnId, action.partId, (part) => {
+    case "session/delta" /* SessionDelta */: {
+      const updated = updateResponsePart(state, action.turnId, action.partId, (part) => {
         if (part.kind === "markdown" /* Markdown */) {
           return { ...part, content: part.content + action.content };
         }
         return part;
       });
-    case "session/responsePart" /* SessionResponsePart */:
+      if (updated === state && state.activeTurn && state.activeTurn.id === action.turnId) {
+        return {
+          ...state,
+          activeTurn: {
+            ...state.activeTurn,
+            responseParts: [
+              ...state.activeTurn.responseParts,
+              { kind: "markdown" /* Markdown */, id: action.partId, content: action.content }
+            ]
+          }
+        };
+      }
+      return updated;
+    }
+    case "session/responsePart" /* SessionResponsePart */: {
       if (!state.activeTurn || state.activeTurn.id !== action.turnId) {
         return state;
       }
+      const newPartId = action.part.kind === "toolCall" /* ToolCall */ ? action.part.toolCall.toolCallId : "id" in action.part ? action.part.id : void 0;
+      if (newPartId !== void 0) {
+        const exists = state.activeTurn.responseParts.some((p) => {
+          const id = p.kind === "toolCall" /* ToolCall */ ? p.toolCall.toolCallId : "id" in p ? p.id : void 0;
+          return id === newPartId;
+        });
+        if (exists) {
+          return state;
+        }
+      }
       return {
         ...state,
         activeTurn: {
@@ -622,6 +646,7 @@ function sessionReducer(state, action, log7) {
           responseParts: [...state.activeTurn.responseParts, action.part]
         }
       };
+    }
     case "session/turnComplete" /* SessionTurnComplete */:
       return endTurn(state, action.turnId, "complete" /* Complete */, "idle" /* Idle */);
     case "session/turnCancelled" /* SessionTurnCancelled */:
@@ -904,6 +929,7 @@ var StateMirror = class {
   rootState = { agents: [] };
   sessions = /* @__PURE__ */ new Map();
   serverSeq = 0;
+  pendingActions = /* @__PURE__ */ new Map();
   /** Current root state (agents, active session count). */
   get root() {
     return this.rootState;
@@ -922,6 +948,7 @@ var StateMirror = class {
   }
   /**
    * Load a snapshot (from initialize, reconnect, or subscribe).
+   * After registering a session, replays any actions that arrived before the snapshot.
    */
   applySnapshot(snapshot) {
     if (snapshot.fromSeq > this.serverSeq) {
@@ -932,6 +959,21 @@ var StateMirror = class {
     } else if ("summary" in snapshot.state) {
       const sessionState = snapshot.state;
       this.sessions.set(snapshot.resource, sessionState);
+      const buffered = this.pendingActions.get(snapshot.resource);
+      if (buffered) {
+        this.pendingActions.delete(snapshot.resource);
+        for (const env of buffered) {
+          if (env.serverSeq > snapshot.fromSeq) {
+            const current = this.sessions.get(snapshot.resource);
+            if (current) {
+              this.sessions.set(
+                snapshot.resource,
+                sessionReducer(current, env.action)
+              );
+            }
+          }
+        }
+      }
     }
   }
   /**
@@ -949,6 +991,13 @@ var StateMirror = class {
         const current = this.sessions.get(sessionUri);
         if (current) {
           this.sessions.set(sessionUri, sessionReducer(current, sessionAction));
+        } else {
+          let buffer = this.pendingActions.get(sessionUri);
+          if (!buffer) {
+            buffer = [];
+            this.pendingActions.set(sessionUri, buffer);
+          }
+          buffer.push(envelope);
         }
       }
     }
@@ -958,6 +1007,7 @@ var StateMirror = class {
    */
   removeSession(uri) {
     this.sessions.delete(uri);
+    this.pendingActions.delete(uri);
   }
 };
 

package/dist/index.d.ts

@@ -2450,6 +2450,7 @@ declare class StateMirror {
     private rootState;
     private sessions;
     private serverSeq;
+    private pendingActions;
     /** Current root state (agents, active session count). */
     get root(): IRootState;
     /** Current server sequence number. */
@@ -2460,6 +2461,7 @@ declare class StateMirror {
     get sessionUris(): URI[];
     /**
      * Load a snapshot (from initialize, reconnect, or subscribe).
+     * After registering a session, replays any actions that arrived before the snapshot.
      */
     applySnapshot(snapshot: ISnapshot): void;
     /**

package/dist/index.js

@@ -25,7 +25,7 @@ import {
   ensureFileUri,
   fileUriToDisplayPath,
   truncatePreview
-} from "./chunk-B6RV43UP.js";
+} from "./chunk-TCJ6NC7B.js";
 
 // src/client/connection-pool.ts
 var ConnectionPool = class {

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@tylerl0706/ahpx",
-	"version": "0.2.8",
+	"version": "0.2.9",
 	"type": "module",
 	"description": "Agent Host Protocol client — use as a library or CLI to manage AHP server connections, sessions, and agent interactions",
 	"license": "MIT",