fix build

Author: ochafik

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": "../..",

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