@apollo/client-ai-apps 0.5.3 → 0.6.0

package/src/testing/internal/mcp/mockMcpHost.ts

@@ -2,6 +2,7 @@ import {
   LATEST_PROTOCOL_VERSION,
   type McpUiHostCapabilities,
   type McpUiHostContext,
+  type McpUiHostContextChangedNotification,
   type McpUiToolResultNotification,
   type McpUiToolInputNotification,
 } from "@modelcontextprotocol/ext-apps";
@@ -15,6 +16,9 @@ import { invariant, promiseWithResolvers } from "../../../utilities/index.js";
 export interface MockMcpHost extends Disposable {
   sendToolResult(params: McpUiToolResultNotification["params"]): Promise<void>;
   sendToolInput(params: McpUiToolInputNotification["params"]): Promise<void>;
+  sendHostContextChanged(
+    params: McpUiHostContextChangedNotification["params"]
+  ): Promise<void>;
   mockToolCall(
     name: string,
     handler: (params: CallToolRequest["params"]) => CallToolResult
@@ -166,6 +170,14 @@ export async function mockMcpHost(
         params,
       });
     },
+    async sendHostContextChanged(params) {
+      await initialized;
+      window.postMessage({
+        jsonrpc: "2.0",
+        method: "ui/notifications/host-context-changed",
+        params,
+      });
+    },
     mockToolCall(name, handler) {
       if (toolCallHandlers.has(name)) {
         console.warn(

package/src/testing/internal/utilities/mockApplicationManifest.ts

@@ -29,6 +29,7 @@ export function mockApplicationManifest(
     ],
     resource: "index.html",
     csp: {
+      baseUriDomains: [],
       resourceDomains: [],
       connectDomains: [],
       frameDomains: [],

package/src/types/application-manifest.ts

@@ -48,6 +48,7 @@ export type ManifestExtraInput = {
 };
 
 export type ManifestCsp = {
+  baseUriDomains: string[];
   connectDomains: string[];
   frameDomains: string[];
   redirectDomains: string[];