From 14740aaceaa2f84f1b15411098920213d009d00c Mon Sep 17 00:00:00 2001
From: Olivier Chafik <ochafik@anthropic.com>
Date: Mon, 24 Nov 2025 16:07:10 +0000
Subject: [PATCH] fix build

---
 examples/simple-host/package.json |  3 +--
 src/app-bridge.ts                 | 44 +++++++++++++++++++++++++++++++
 2 files changed, 45 insertions(+), 2 deletions(-)

diff --git a/examples/simple-host/package.json b/examples/simple-host/package.json
index 720d8e9..d207c34 100644
--- a/examples/simple-host/package.json
+++ b/examples/simple-host/package.json
@@ -6,10 +6,9 @@
   "scripts": {
     "start:server": "python3 serve.py",
     "start:mcp-server": "cd ../simple-server && npm install && npm run start",
-    "start:proxy": "python3 serve.py",
     "build": "concurrently 'INPUT=example-host.html vite build' 'INPUT=example-host-react.html vite build' 'INPUT=sandbox.html vite build'",
     "server": "bun server.ts",
-    "start": "NODE_ENV=development npm run build && concurrently 'npm run start:server' 'npm run start:mcp-server' 'npm run start:proxy'"
+    "start": "NODE_ENV=development npm run build && concurrently 'npm run start:server' 'npm run start:mcp-server'"
   },
   "dependencies": {
     "@modelcontextprotocol/ext-apps": "../..",
diff --git a/src/app-bridge.ts b/src/app-bridge.ts
index b36cb79..08df061 100644
--- a/src/app-bridge.ts
+++ b/src/app-bridge.ts
@@ -12,6 +12,8 @@ import {
   ListResourcesResultSchema,
   ListResourceTemplatesRequestSchema,
   ListResourceTemplatesResultSchema,
+  LoggingMessageNotification,
+  LoggingMessageNotificationSchema,
   Notification,
   PingRequest,
   PingRequestSchema,
@@ -40,6 +42,12 @@ import {
   McpUiInitializeRequest,
   McpUiInitializeRequestSchema,
   McpUiInitializeResult,
+  McpUiMessageRequest,
+  McpUiMessageRequestSchema,
+  McpUiMessageResult,
+  McpUiOpenLinkRequest,
+  McpUiOpenLinkRequestSchema,
+  McpUiOpenLinkResult,
   McpUiResourceTeardownRequest,
   McpUiResourceTeardownResultSchema,
   McpUiSandboxProxyReadyNotification,
@@ -101,6 +109,42 @@ export class AppBridge extends Protocol<Request, Notification, Result> {
       callback(n.params),
     );
   }
+  set onmessage(
+    callback: (
+      params: McpUiMessageRequest["params"],
+      extra: RequestExtra,
+    ) => Promise<McpUiMessageResult>,
+  ) {
+    this.setRequestHandler(
+      McpUiMessageRequestSchema,
+      async (request, extra) => {
+        return callback(request.params, extra);
+      },
+    );
+  }
+  set onopenlink(
+    callback: (
+      params: McpUiOpenLinkRequest["params"],
+      extra: RequestExtra,
+    ) => Promise<McpUiOpenLinkResult>,
+  ) {
+    this.setRequestHandler(
+      McpUiOpenLinkRequestSchema,
+      async (request, extra) => {
+        return callback(request.params, extra);
+      },
+    );
+  }
+  set onloggingmessage(
+    callback: (params: LoggingMessageNotification["params"]) => void,
+  ) {
+    this.setNotificationHandler(
+      LoggingMessageNotificationSchema,
+      async (notification) => {
+        callback(notification.params);
+      },
+    );
+  }
 
   assertCapabilityForMethod(method: Request["method"]): void {
     // TODO