@tylerl0706/ahpx 0.2.8 → 0.2.10

package/dist/bin.js

@@ -11,7 +11,7 @@ import {
   createLogger,
   ensureFileUri,
   setVerbose
-} from "./chunk-B6RV43UP.js";
+} from "./chunk-646D2XNX.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-646D2XNX.js}

@@ -39,6 +39,19 @@ var ActionType = /* @__PURE__ */ ((ActionType2) => {
   return ActionType2;
 })(ActionType || {});
 
+// src/protocol/notifications.ts
+var AuthRequiredReason = /* @__PURE__ */ ((AuthRequiredReason2) => {
+  AuthRequiredReason2["Required"] = "required";
+  AuthRequiredReason2["Expired"] = "expired";
+  return AuthRequiredReason2;
+})(AuthRequiredReason || {});
+var NotificationType = /* @__PURE__ */ ((NotificationType2) => {
+  NotificationType2["SessionAdded"] = "notify/sessionAdded";
+  NotificationType2["SessionRemoved"] = "notify/sessionRemoved";
+  NotificationType2["AuthRequired"] = "notify/authRequired";
+  return NotificationType2;
+})(NotificationType || {});
+
 // src/client/protocol.ts
 import { EventEmitter } from "events";
 var RpcError = class extends Error {
@@ -139,6 +152,12 @@ var ProtocolLayer = class extends EventEmitter {
 };
 
 // src/protocol/state.ts
+var PolicyState = /* @__PURE__ */ ((PolicyState2) => {
+  PolicyState2["Enabled"] = "enabled";
+  PolicyState2["Disabled"] = "disabled";
+  PolicyState2["Unconfigured"] = "unconfigured";
+  return PolicyState2;
+})(PolicyState || {});
 var PendingMessageKind = /* @__PURE__ */ ((PendingMessageKind2) => {
   PendingMessageKind2["Steering"] = "steering";
   PendingMessageKind2["Queued"] = "queued";
@@ -156,6 +175,18 @@ var SessionStatus = /* @__PURE__ */ ((SessionStatus2) => {
   SessionStatus2["Error"] = "error";
   return SessionStatus2;
 })(SessionStatus || {});
+var TurnState = /* @__PURE__ */ ((TurnState2) => {
+  TurnState2["Complete"] = "complete";
+  TurnState2["Cancelled"] = "cancelled";
+  TurnState2["Error"] = "error";
+  return TurnState2;
+})(TurnState || {});
+var AttachmentType = /* @__PURE__ */ ((AttachmentType2) => {
+  AttachmentType2["File"] = "file";
+  AttachmentType2["Directory"] = "directory";
+  AttachmentType2["Selection"] = "selection";
+  return AttachmentType2;
+})(AttachmentType || {});
 var ResponsePartKind = /* @__PURE__ */ ((ResponsePartKind2) => {
   ResponsePartKind2["Markdown"] = "markdown";
   ResponsePartKind2["ContentRef"] = "contentRef";
@@ -172,6 +203,32 @@ var ToolCallStatus = /* @__PURE__ */ ((ToolCallStatus2) => {
   ToolCallStatus2["Cancelled"] = "cancelled";
   return ToolCallStatus2;
 })(ToolCallStatus || {});
+var ToolCallConfirmationReason = /* @__PURE__ */ ((ToolCallConfirmationReason2) => {
+  ToolCallConfirmationReason2["NotNeeded"] = "not-needed";
+  ToolCallConfirmationReason2["UserAction"] = "user-action";
+  ToolCallConfirmationReason2["Setting"] = "setting";
+  return ToolCallConfirmationReason2;
+})(ToolCallConfirmationReason || {});
+var ToolCallCancellationReason = /* @__PURE__ */ ((ToolCallCancellationReason2) => {
+  ToolCallCancellationReason2["Denied"] = "denied";
+  ToolCallCancellationReason2["Skipped"] = "skipped";
+  ToolCallCancellationReason2["ResultDenied"] = "result-denied";
+  return ToolCallCancellationReason2;
+})(ToolCallCancellationReason || {});
+var ToolResultContentType = /* @__PURE__ */ ((ToolResultContentType2) => {
+  ToolResultContentType2["Text"] = "text";
+  ToolResultContentType2["EmbeddedResource"] = "embeddedResource";
+  ToolResultContentType2["Resource"] = "resource";
+  ToolResultContentType2["FileEdit"] = "fileEdit";
+  return ToolResultContentType2;
+})(ToolResultContentType || {});
+var CustomizationStatus = /* @__PURE__ */ ((CustomizationStatus2) => {
+  CustomizationStatus2["Loading"] = "loading";
+  CustomizationStatus2["Loaded"] = "loaded";
+  CustomizationStatus2["Degraded"] = "degraded";
+  CustomizationStatus2["Error"] = "error";
+  return CustomizationStatus2;
+})(CustomizationStatus || {});
 
 // src/client/session-handle.ts
 import { randomUUID } from "crypto";
@@ -604,17 +661,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 +703,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 +986,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 +1005,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 +1016,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 +1048,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 +1064,7 @@ var StateMirror = class {
    */
   removeSession(uri) {
     this.sessions.delete(uri);
+    this.pendingActions.delete(uri);
   }
 };
 
@@ -2354,14 +2461,23 @@ function encodeFilePathComponents(filePath) {
 
 export {
   ActionType,
+  AuthRequiredReason,
+  NotificationType,
   RpcError,
   RpcTimeoutError,
   ProtocolLayer,
+  PolicyState,
   PendingMessageKind,
   SessionLifecycle,
   SessionStatus,
+  TurnState,
+  AttachmentType,
   ResponsePartKind,
   ToolCallStatus,
+  ToolCallConfirmationReason,
+  ToolCallCancellationReason,
+  ToolResultContentType,
+  CustomizationStatus,
   SessionHandle,
   StateMirror,
   Transport,

package/dist/index.d.ts

@@ -111,7 +111,7 @@ interface IProtectedResourceMetadata {
  *
  * @category Root State
  */
-declare const enum PolicyState {
+declare enum PolicyState {
     Enabled = "enabled",
     Disabled = "disabled",
     Unconfigured = "unconfigured"
@@ -181,7 +181,7 @@ interface ISessionModelInfo {
  *
  * @category Pending Message Types
  */
-declare const enum PendingMessageKind {
+declare enum PendingMessageKind {
     /** Injected into the current turn at a convenient point */
     Steering = "steering",
     /** Sent automatically as a new turn after the current turn finishes */
@@ -207,7 +207,7 @@ interface IPendingMessage {
  *
  * @category Session State
  */
-declare const enum SessionLifecycle {
+declare enum SessionLifecycle {
     Creating = "creating",
     Ready = "ready",
     CreationFailed = "creationFailed"
@@ -217,7 +217,7 @@ declare const enum SessionLifecycle {
  *
  * @category Session State
  */
-declare const enum SessionStatus {
+declare enum SessionStatus {
     Idle = "idle",
     InProgress = "in-progress",
     Error = "error"
@@ -300,7 +300,7 @@ interface ISessionSummary {
  *
  * @category Turn Types
  */
-declare const enum TurnState {
+declare enum TurnState {
     Complete = "complete",
     Cancelled = "cancelled",
     Error = "error"
@@ -310,7 +310,7 @@ declare const enum TurnState {
  *
  * @category Turn Types
  */
-declare const enum AttachmentType {
+declare enum AttachmentType {
     File = "file",
     Directory = "directory",
     Selection = "selection"
@@ -383,7 +383,7 @@ interface IMessageAttachment {
  *
  * @category Response Parts
  */
-declare const enum ResponsePartKind {
+declare enum ResponsePartKind {
     Markdown = "markdown",
     ContentRef = "contentRef",
     ToolCall = "toolCall",
@@ -457,7 +457,7 @@ type IResponsePart = IMarkdownResponsePart | IResourceReponsePart | IToolCallRes
  *
  * @category Tool Call Types
  */
-declare const enum ToolCallStatus {
+declare enum ToolCallStatus {
     Streaming = "streaming",
     PendingConfirmation = "pending-confirmation",
     Running = "running",
@@ -474,7 +474,7 @@ declare const enum ToolCallStatus {
  *
  * @category Tool Call Types
  */
-declare const enum ToolCallConfirmationReason {
+declare enum ToolCallConfirmationReason {
     NotNeeded = "not-needed",
     UserAction = "user-action",
     Setting = "setting"
@@ -484,7 +484,7 @@ declare const enum ToolCallConfirmationReason {
  *
  * @category Tool Call Types
  */
-declare const enum ToolCallCancellationReason {
+declare enum ToolCallCancellationReason {
     Denied = "denied",
     Skipped = "skipped",
     ResultDenied = "result-denied"
@@ -709,7 +709,7 @@ interface IToolAnnotations {
  *
  * @category Tool Result Content
  */
-declare const enum ToolResultContentType {
+declare enum ToolResultContentType {
     Text = "text",
     EmbeddedResource = "embeddedResource",
     Resource = "resource",
@@ -824,7 +824,7 @@ interface ICustomizationRef {
  *
  * @category Customization Types
  */
-declare const enum CustomizationStatus {
+declare enum CustomizationStatus {
     /** Plugin is being loaded */
     Loading = "loading",
     /** Plugin is fully operational */
@@ -906,7 +906,7 @@ interface ISnapshot {
  *
  * @category Actions
  */
-declare const enum ActionType {
+declare enum ActionType {
     RootAgentsChanged = "root/agentsChanged",
     RootActiveSessionsChanged = "root/activeSessionsChanged",
     SessionReady = "session/ready",
@@ -1551,7 +1551,7 @@ interface IInitializeResult {
  *
  * @category Commands
  */
-declare const enum ReconnectResultType {
+declare enum ReconnectResultType {
     Replay = "replay",
     Snapshot = "snapshot"
 }
@@ -1718,7 +1718,7 @@ interface IListSessionsResult {
  *
  * @category Commands
  */
-declare const enum ContentEncoding {
+declare enum ContentEncoding {
     Base64 = "base64",
     Utf8 = "utf-8"
 }
@@ -2058,7 +2058,7 @@ interface IAuthenticateResult {
  *
  * @category Protocol Notifications
  */
-declare const enum AuthRequiredReason {
+declare enum AuthRequiredReason {
     /** The client has not yet authenticated for the resource */
     Required = "required",
     /** A previously valid token has expired or been revoked */
@@ -2069,7 +2069,7 @@ declare const enum AuthRequiredReason {
  *
  * @category Protocol Notifications
  */
-declare const enum NotificationType {
+declare enum NotificationType {
     SessionAdded = "notify/sessionAdded",
     SessionRemoved = "notify/sessionRemoved",
     AuthRequired = "notify/authRequired"
@@ -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;
     /**
@@ -3394,4 +3396,4 @@ declare function ensureFileUri(pathOrUri: string): string;
  */
 declare function fileUriToDisplayPath(uri: string): string;
 
-export { ActionType, ActiveClientManager, AhpClient, type AhpClientEvents, type AhpClientOptions, type AhpxEvent, AuthHandler, type AuthHandlerOptions, ConnectionPool, type ConnectionPoolOptions, type ConnectionProfile, ContentEncoding, type EventForwarder, FleetManager, type FleetManagerOptions, ForwardingFormatter, type ForwardingFormatterOptions, HealthChecker, type IActionEnvelope, type IActiveTurn, type IAgentInfo, type ICreateSessionParams, type ICustomizationRef, type IErrorInfo, type IFetchTurnsResult, type IInitializeResult, type IListSessionsResult, type IMarkdownResponsePart, type IPendingMessage, type IProtocolNotification, type IReasoningResponsePart, type IResourceCopyParams, type IResourceCopyResult, type IResourceDeleteParams, type IResourceDeleteResult, type IResourceListResult, type IResourceMoveParams, type IResourceMoveResult, type IResourceReadResult, type IResourceWriteParams, type IResourceWriteResult, type IResponsePart, type IRootState, type ISessionCustomization, type ISessionForkSource, type ISessionState, type ISessionSummary, type ISnapshot, type IStateAction, type ISubscribeResult, type IToolAnnotations, type IToolCallResponsePart, type IToolCallState, type IToolDefinition, type ITurn, type IUsageInfo, type IUserMessage, type Icon, type OpenSessionOptions, PendingMessageKind, type PromptOptions, ProtocolLayer, type ProtocolLayerOptions, ReconnectManager, type ReconnectOptions, type ReconnectOutcome, ResponsePartKind, type ResumeOutcome, type RoutingStrategy, RpcError, RpcTimeoutError, type ServerHealth, type ServerRequirements, type SessionFilter, SessionHandle, type SessionHandleEvents, SessionLifecycle, SessionPersistence, type SessionRecord, SessionStatus, SessionStore, type TurnResult as SessionTurnResult, StateMirror, type SyncResult, ToolCallStatus, Transport, type TransportOptions, type TurnSummary, type URI, WebSocketForwarder, type WebSocketForwarderOptions, WebhookForwarder, type WebhookForwarderOptions, buildTurnSummary, ensureFileUri, fileUriToDisplayPath, truncatePreview };
+export { ActionType, ActiveClientManager, AhpClient, type AhpClientEvents, type AhpClientOptions, type AhpxEvent, AttachmentType, AuthHandler, type AuthHandlerOptions, AuthRequiredReason, ConnectionPool, type ConnectionPoolOptions, type ConnectionProfile, ContentEncoding, CustomizationStatus, type EventForwarder, FleetManager, type FleetManagerOptions, ForwardingFormatter, type ForwardingFormatterOptions, HealthChecker, type IActionEnvelope, type IActiveTurn, type IAgentInfo, type ICreateSessionParams, type ICustomizationRef, type IErrorInfo, type IFetchTurnsResult, type IInitializeResult, type IListSessionsResult, type IMarkdownResponsePart, type IPendingMessage, type IProtocolNotification, type IReasoningResponsePart, type IResourceCopyParams, type IResourceCopyResult, type IResourceDeleteParams, type IResourceDeleteResult, type IResourceListResult, type IResourceMoveParams, type IResourceMoveResult, type IResourceReadResult, type IResourceWriteParams, type IResourceWriteResult, type IResponsePart, type IRootState, type ISessionCustomization, type ISessionForkSource, type ISessionState, type ISessionSummary, type ISnapshot, type IStateAction, type ISubscribeResult, type IToolAnnotations, type IToolCallResponsePart, type IToolCallState, type IToolDefinition, type ITurn, type IUsageInfo, type IUserMessage, type Icon, NotificationType, type OpenSessionOptions, PendingMessageKind, PolicyState, type PromptOptions, ProtocolLayer, type ProtocolLayerOptions, ReconnectManager, type ReconnectOptions, type ReconnectOutcome, ReconnectResultType, ResponsePartKind, type ResumeOutcome, type RoutingStrategy, RpcError, RpcTimeoutError, type ServerHealth, type ServerRequirements, type SessionFilter, SessionHandle, type SessionHandleEvents, SessionLifecycle, SessionPersistence, type SessionRecord, SessionStatus, SessionStore, type TurnResult as SessionTurnResult, StateMirror, type SyncResult, ToolCallCancellationReason, ToolCallConfirmationReason, ToolCallStatus, ToolResultContentType, Transport, type TransportOptions, TurnState, type TurnSummary, type URI, WebSocketForwarder, type WebSocketForwarderOptions, WebhookForwarder, type WebhookForwarderOptions, buildTurnSummary, ensureFileUri, fileUriToDisplayPath, truncatePreview };

package/dist/index.js

@@ -2,10 +2,15 @@ import {
   ActionType,
   ActiveClientManager,
   AhpClient,
+  AttachmentType,
   AuthHandler,
+  AuthRequiredReason,
+  CustomizationStatus,
   ForwardingFormatter,
   HealthChecker,
+  NotificationType,
   PendingMessageKind,
+  PolicyState,
   ProtocolLayer,
   ReconnectManager,
   ResponsePartKind,
@@ -17,15 +22,19 @@ import {
   SessionStatus,
   SessionStore,
   StateMirror,
+  ToolCallCancellationReason,
+  ToolCallConfirmationReason,
   ToolCallStatus,
+  ToolResultContentType,
   Transport,
+  TurnState,
   WebSocketForwarder,
   WebhookForwarder,
   buildTurnSummary,
   ensureFileUri,
   fileUriToDisplayPath,
   truncatePreview
-} from "./chunk-B6RV43UP.js";
+} from "./chunk-646D2XNX.js";
 
 // src/client/connection-pool.ts
 var ConnectionPool = class {
@@ -191,6 +200,11 @@ var FleetManager = class {
 };
 
 // src/protocol/commands.ts
+var ReconnectResultType = /* @__PURE__ */ ((ReconnectResultType2) => {
+  ReconnectResultType2["Replay"] = "replay";
+  ReconnectResultType2["Snapshot"] = "snapshot";
+  return ReconnectResultType2;
+})(ReconnectResultType || {});
 var ContentEncoding = /* @__PURE__ */ ((ContentEncoding2) => {
   ContentEncoding2["Base64"] = "base64";
   ContentEncoding2["Utf8"] = "utf-8";
@@ -200,15 +214,21 @@ export {
   ActionType,
   ActiveClientManager,
   AhpClient,
+  AttachmentType,
   AuthHandler,
+  AuthRequiredReason,
   ConnectionPool,
   ContentEncoding,
+  CustomizationStatus,
   FleetManager,
   ForwardingFormatter,
   HealthChecker,
+  NotificationType,
   PendingMessageKind,
+  PolicyState,
   ProtocolLayer,
   ReconnectManager,
+  ReconnectResultType,
   ResponsePartKind,
   RpcError,
   RpcTimeoutError,
@@ -218,8 +238,12 @@ export {
   SessionStatus,
   SessionStore,
   StateMirror,
+  ToolCallCancellationReason,
+  ToolCallConfirmationReason,
   ToolCallStatus,
+  ToolResultContentType,
   Transport,
+  TurnState,
   WebSocketForwarder,
   WebhookForwarder,
   buildTurnSummary,

package/package.json

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