@multi-agent-protocol/sdk 0.1.6 → 0.1.8

package/dist/testing.cjs

@@ -4817,6 +4817,7 @@ var AgentConnection = class _AgentConnection {
   #options;
   #messageHandlers = /* @__PURE__ */ new Set();
   #reconnectionHandlers = /* @__PURE__ */ new Set();
+  #notificationHandlers = /* @__PURE__ */ new Map();
   #scopeMemberships = /* @__PURE__ */ new Set();
   #agentId = null;
   #sessionId = null;
@@ -5237,6 +5238,47 @@ var AgentConnection = class _AgentConnection {
     this.#messageHandlers.delete(handler);
     return this;
   }
+  /**
+   * Register a handler for a specific notification method.
+   *
+   * Handles server-to-agent notifications that aren't standard MAP methods
+   * (e.g., `trajectory/content.request`). When the server sends a JSON-RPC
+   * notification with the specified method, the handler fires with the params.
+   *
+   * @param method - The JSON-RPC notification method name to handle
+   * @param handler - Async function called with the notification params
+   */
+  onNotification(method, handler) {
+    if (!this.#notificationHandlers.has(method)) {
+      this.#notificationHandlers.set(method, /* @__PURE__ */ new Set());
+    }
+    this.#notificationHandlers.get(method).add(handler);
+    return this;
+  }
+  /**
+   * Remove a notification handler for a specific method.
+   */
+  offNotification(method, handler) {
+    const handlers = this.#notificationHandlers.get(method);
+    if (handlers) {
+      handlers.delete(handler);
+      if (handlers.size === 0) this.#notificationHandlers.delete(method);
+    }
+    return this;
+  }
+  /**
+   * Send a raw JSON-RPC notification to the server.
+   *
+   * Used for custom protocol notifications (e.g., `trajectory/content.response`)
+   * that aren't standard MAP methods. The notification is sent directly on the
+   * underlying connection without any MAP-level wrapping.
+   *
+   * @param method - The JSON-RPC notification method name
+   * @param params - The notification parameters
+   */
+  async sendNotification(method, params) {
+    return this.#connection.sendNotification(method, params);
+  }
   // ===========================================================================
   // State Management
   // ===========================================================================
@@ -5701,8 +5743,19 @@ var AgentConnection = class _AgentConnection {
         }
         break;
       }
-      default:
-        console.warn("MAP: Unknown notification:", method);
+      default: {
+        const handlers = this.#notificationHandlers.get(method);
+        if (handlers?.size) {
+          for (const handler of handlers) {
+            try {
+              await handler(params);
+            } catch (error) {
+              console.error("MAP: Notification handler error:", error);
+            }
+          }
+        }
+        break;
+      }
     }
   }
   /**