@apollo/client-ai-apps 0.6.4 → 0.7.0

package/src/testing/internal/tests/eachHostEnv.ts

@@ -0,0 +1,22 @@
+import { describe } from "vitest";
+import { createHostEnv } from "../utilities/createHostEnv";
+import { AbstractApolloClient } from "../../../core/AbstractApolloClient";
+import { ApolloClient as McpApolloClient } from "../../../mcp/core/ApolloClient.js";
+import { ApolloClient as OpenAiApolloClient } from "../../../openai/core/ApolloClient.js";
+
+export function eachHostEnv(
+  suite: (
+    setupHost: ReturnType<typeof createHostEnv>,
+    ApolloClient: new (
+      options: AbstractApolloClient.Options
+    ) => AbstractApolloClient,
+    opts: { hostEnv: "mcp" | "openai" }
+  ) => void | PromiseLike<void>
+) {
+  describe.each([
+    ["mcp", McpApolloClient],
+    ["openai", OpenAiApolloClient],
+  ] as const)("Host: %s", (hostEnv, ApolloClient) => {
+    return suite(createHostEnv(hostEnv), ApolloClient, { hostEnv });
+  });
+}

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

@@ -0,0 +1,117 @@
+import type { McpUiHostContext } from "@modelcontextprotocol/ext-apps";
+import type { AbstractApolloClient } from "../../../core/AbstractApolloClient";
+import { minimalHostContextWithToolName } from "../mcp/minimalHostContextWithToolName";
+import { mockMcpHost } from "../mcp/mockMcpHost";
+import { stubOpenAiGlobals } from "../openai/stubOpenAiGlobals";
+import type { ApolloMcpServerApps } from "../../../core/types";
+import type { OpenAiGlobals } from "../../../openai/types";
+
+export declare namespace createHostEnv {
+  export namespace setupHost {
+    export interface MockToolResult extends Pick<
+      ApolloMcpServerApps.CallToolResult,
+      "structuredContent" | "isError"
+    > {
+      _meta?: Omit<ApolloMcpServerApps.Meta, "toolName"> & {
+        toolName?: string;
+      };
+    }
+
+    export interface Options {
+      /**
+       * The ApolloClient instance under test. Passed here so that
+       * `client.stop()` is called automatically when the env is disposed,
+       * preventing MCP event listener leaks between tests.
+       */
+      client: AbstractApolloClient;
+      hostContext?: McpUiHostContext;
+      toolCall?: {
+        name?: string;
+        input?: Record<string, unknown>;
+        result?: setupHost.MockToolResult;
+      };
+      openai?: {
+        /**
+         * When true, sets window.openai.toolOutput to toolCall.result.structuredContent.
+         * Simulates the page-refresh scenario where ChatGPT does not re-send the
+         * tool-result notification. Call host.sendToolInput(params.toolInput) but
+         * omit host.sendToolResult() in the test.
+         */
+        toolOutput?: true;
+      };
+    }
+  }
+}
+
+export function createHostEnv(hostEnv: "openai" | "mcp") {
+  return async function setupHost(options: createHostEnv.setupHost.Options) {
+    const { client, toolCall } = options;
+
+    const toolName = toolCall?.name;
+    const toolInput = toolCall?.input;
+
+    const mockOptions: mockMcpHost.Options = {};
+
+    const hostContext =
+      toolName ?
+        { ...minimalHostContextWithToolName(toolName), ...options.hostContext }
+      : options.hostContext;
+
+    if (hostContext) {
+      mockOptions.hostContext = hostContext;
+    }
+
+    const toolResult =
+      toolName ?
+        {
+          ...options.toolCall?.result,
+          structuredContent: {
+            ...options.toolCall?.result?.structuredContent,
+            toolName,
+          },
+          _meta: { ...options.toolCall?.result?._meta, toolName },
+        }
+      : options.toolCall?.result;
+
+    const host = await mockMcpHost(mockOptions);
+
+    const params = {
+      toolInput: toolInput ? { arguments: toolInput } : {},
+      toolResult: { ...toolResult },
+    };
+
+    if (hostEnv === "openai") {
+      // OpenAI doesn't set _meta in the notification
+      delete params.toolResult._meta;
+
+      stubOpenAiGlobals((defaults) => {
+        const globals: Partial<OpenAiGlobals> = { ...defaults };
+
+        if (toolInput) {
+          globals.toolInput = toolInput;
+        }
+
+        const _meta = toolResult?._meta;
+
+        if (_meta) {
+          globals.toolResponseMetadata = _meta;
+        }
+
+        if (options.openai?.toolOutput && toolResult?.structuredContent) {
+          globals.toolOutput = toolResult.structuredContent;
+        }
+
+        return globals;
+      });
+    }
+
+    return {
+      host,
+      params,
+      [Symbol.dispose]() {
+        host[Symbol.dispose]();
+        client.stop();
+      },
+    };
+  };
+}

package/src/types/application-manifest.ts

@@ -32,6 +32,7 @@ export type ManifestTool = {
   name: string;
   description: string;
   extraInputs?: ManifestExtraInput[];
+  extraOutputs?: Record<string, unknown>;
   labels?: ManifestLabels;
 };
 

package/src/utilities/connectToHost.ts

@@ -0,0 +1,13 @@
+import { PostMessageTransport, type App } from "@modelcontextprotocol/ext-apps";
+
+export async function connectToHost(app: App) {
+  try {
+    return await app.connect(
+      new PostMessageTransport(window.parent, window.parent)
+    );
+  } catch (e) {
+    const error = e instanceof Error ? e : new Error("Failed to connect");
+
+    throw error;
+  }
+}

package/src/utilities/index.ts

@@ -1,4 +1,5 @@
 export { aiClientSymbol } from "./constants.js";
+export { connectToHost } from "./connectToHost.js";
 export { getVariablesForOperationFromToolInput } from "./getVariablesForOperationFromToolInput.js";
 export { getToolNamesFromDocument } from "./getToolNamesFromDocument.js";
 export { getVariableNamesFromDocument } from "./getVariableNamesFromDocument.js";

package/src/vite/__tests__/apolloClientAiApps.test.ts

@@ -143,6 +143,35 @@ describe("operations", () => {
     `);
   });
 
+  test("writes extraOutputs to the manifest", async () => {
+    vol.fromJSON({
+      "package.json": mockPackageJson(),
+      "src/my-component.tsx": declareOperation(gql`
+        query HelloWorldQuery
+        @tool(
+          name: "hello-world"
+          description: "This is an awesome tool!"
+          extraOutputs: { foo: "bar", nested: { label: "count" } }
+        ) {
+          helloWorld
+        }
+      `),
+    });
+
+    await using server = await setupServer({
+      plugins: [
+        apolloClientAiApps({ targets: ["mcp"], appsOutDir: "dist/apps" }),
+      ],
+    });
+    await server.listen();
+
+    const manifest = readManifestFile();
+    expect(manifest.operations[0].tools[0].extraOutputs).toEqual({
+      foo: "bar",
+      nested: { label: "count" },
+    });
+  });
+
   test("handles operations with fragments in the same file", async () => {
     vol.fromJSON({
       "package.json": mockPackageJson(),
@@ -890,6 +919,33 @@ describe("@tool validation", () => {
     );
   });
 
+  test("errors when extraOutputs is not an object", async () => {
+    vol.fromJSON({
+      "package.json": mockPackageJson(),
+      "src/my-component.tsx": declareOperation(gql`
+        query HelloWorldQuery
+        @tool(
+          name: "hello-world"
+          description: "hello"
+          extraOutputs: [1, 2, 3]
+        ) {
+          helloWorld
+        }
+      `),
+    });
+
+    await expect(async () => {
+      await using server = await setupServer({
+        plugins: [
+          apolloClientAiApps({ targets: ["mcp"], appsOutDir: "dist/apps" }),
+        ],
+      });
+      await server.listen();
+    }).rejects.toThrowErrorMatchingInlineSnapshot(
+      `[Error: Expected argument 'extraOutputs' to be of type 'ObjectValue' but found 'ListValue' instead.]`
+    );
+  });
+
   test("errors when extraInputs is not an array", async () => {
     vol.fromJSON({
       "package.json": mockPackageJson(),

package/src/vite/apolloClientAiApps.ts

@@ -874,6 +874,10 @@ const processQueryLink = new ApolloLink((operation) => {
         getDirectiveArgument("extraInputs", directive),
         Kind.LIST
       ),
+      extraOutputs: maybeGetArgumentValue(
+        getDirectiveArgument("extraOutputs", directive),
+        Kind.OBJECT
+      ),
       labels: maybeGetArgumentValue(
         getDirectiveArgument("labels", directive),
         Kind.OBJECT
@@ -1044,5 +1048,6 @@ const ToolDirectiveSchema = z.strictObject({
       })
     )
   ),
+  extraOutputs: z.optional(z.record(z.string(), z.unknown())),
   labels: ApolloClientAiAppsConfigSchema.shape.labels.optional(),
 });

package/tsconfig.vite.json

@@ -4,7 +4,7 @@
   "include": [
     "./src/vite",
     "./src/config",
-    "./src/utilities/",
+    "./src/utilities/invariant.ts",
     "./src/types/application-manifest.ts"
   ],
   "exclude": [

package/vitest.config.ts

@@ -8,5 +8,18 @@ export default defineConfig({
     setupFiles: ["./vitest-setup.ts"],
     mockReset: true,
     unstubGlobals: true,
+    tags: [
+      {
+        name: "flaky",
+        retry: 3,
+        description:
+          "Tests that might intermittently fail due to e.g. timeout issues.",
+      },
+      {
+        name: "fs",
+        timeout: 10_000,
+        description: "Tests that write to the filesystem.",
+      },
+    ],
   },
 });

package/dist/mcp/core/McpAppManager.d.ts

@@ -1,30 +0,0 @@
-import { App, type McpUiHostContextChangedNotification } from "@modelcontextprotocol/ext-apps";
-import type { ApplicationManifest } from "../../types/application-manifest";
-import type { FormattedExecutionResult } from "graphql";
-import type { DocumentNode, OperationVariables } from "@apollo/client";
-import type { ApolloMcpServerApps } from "../../core/types";
-/** @internal */
-export declare class McpAppManager {
-    #private;
-    readonly app: App;
-    constructor(manifest: ApplicationManifest);
-    get toolName(): string | undefined;
-    get toolMetadata(): ApolloMcpServerApps.Meta | undefined;
-    get toolInput(): Record<string, unknown> | undefined;
-    onHostContextChanged(cb: (params: McpUiHostContextChangedNotification["params"]) => void): () => void;
-    connect: (() => Promise<{
-        toolName: string | undefined;
-        args: Record<string, unknown> | undefined;
-        result: FormattedExecutionResult;
-        prefetch?: Record<string, FormattedExecutionResult>;
-    }>) & {
-        reset: () => void;
-    };
-    close(): Promise<void>;
-    executeQuery({ query, variables, }: {
-        query: DocumentNode;
-        variables: OperationVariables | undefined;
-    }): Promise<FormattedExecutionResult<import("graphql/jsutils/ObjMap").ObjMap<unknown>, import("graphql/jsutils/ObjMap").ObjMap<unknown>>>;
-    private connectToHost;
-}
-//# sourceMappingURL=McpAppManager.d.ts.map

package/dist/mcp/core/McpAppManager.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"McpAppManager.d.ts","sourceRoot":"","sources":["../../../src/mcp/core/McpAppManager.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,GAAG,EAEH,KAAK,mCAAmC,EACzC,MAAM,gCAAgC,CAAC;AAExC,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,kCAAkC,CAAC;AAC5E,OAAO,KAAK,EAAE,wBAAwB,EAAE,MAAM,SAAS,CAAC;AACxD,OAAO,KAAK,EAAE,YAAY,EAAE,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AAGvE,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,kBAAkB,CAAC;AAM5D,gBAAgB;AAChB,qBAAa,aAAa;;IACxB,QAAQ,CAAC,GAAG,EAAE,GAAG,CAAC;gBAUN,QAAQ,EAAE,mBAAmB;IAIzC,IAAI,QAAQ,uBAEX;IAED,IAAI,YAAY,yCAEf;IAED,IAAI,SAAS,wCAEZ;IAED,oBAAoB,CAClB,EAAE,EAAE,CAAC,MAAM,EAAE,mCAAmC,CAAC,QAAQ,CAAC,KAAK,IAAI;IASrE,OAAO;;;;;;;MAyCJ;IAEH,KAAK;IAIC,YAAY,CAAC,EACjB,KAAK,EACL,SAAS,GACV,EAAE;QACD,KAAK,EAAE,YAAY,CAAC;QACpB,SAAS,EAAE,kBAAkB,GAAG,SAAS,CAAC;KAC3C;YASa,aAAa;CAW5B"}

package/dist/mcp/core/McpAppManager.js

@@ -1,82 +0,0 @@
-import { App, PostMessageTransport, } from "@modelcontextprotocol/ext-apps";
-import { print } from "@apollo/client/utilities";
-import { cacheAsync, promiseWithResolvers } from "../../utilities";
-/** @internal */
-export class McpAppManager {
-    app;
-    #toolName;
-    #toolMetadata;
-    #toolInput;
-    #hostContextCallbacks = new Set();
-    constructor(manifest) {
-        this.app = new App({ name: manifest.name, version: manifest.appVersion });
-    }
-    get toolName() {
-        return this.#toolName;
-    }
-    get toolMetadata() {
-        return this.#toolMetadata;
-    }
-    get toolInput() {
-        return this.#toolInput;
-    }
-    onHostContextChanged(cb) {
-        this.#hostContextCallbacks.add(cb);
-        return () => {
-            this.#hostContextCallbacks.delete(cb);
-        };
-    }
-    connect = cacheAsync(async () => {
-        let toolResult = promiseWithResolvers();
-        let toolInput = promiseWithResolvers();
-        this.app.ontoolresult = (params) => {
-            toolResult.resolve(params);
-        };
-        this.app.ontoolinput = (params) => {
-            toolInput.resolve(params);
-        };
-        this.app.onhostcontextchanged = (params) => {
-            this.#hostContextCallbacks.forEach((cb) => cb(params));
-        };
-        await this.connectToHost();
-        const { structuredContent, _meta } = await toolResult.promise;
-        const { arguments: args } = await toolInput.promise;
-        // Some hosts do not provide toolInfo in the ui/initialize response, so we
-        // fallback to `_meta.toolName` provided by Apollo MCP server if the value
-        // is not available.
-        this.#toolName =
-            this.app.getHostContext()?.toolInfo?.tool.name ??
-                _meta?.toolName ??
-                // Some hosts do not forward `_meta` nor do they provide `toolInfo`. Our
-                // MCP server provides `toolName` in `structuredContent` as a workaround
-                // that we can use if all else fails
-                structuredContent.toolName;
-        this.#toolMetadata = _meta;
-        this.#toolInput = args;
-        return {
-            ...structuredContent,
-            toolName: this.toolName,
-            args,
-        };
-    });
-    close() {
-        return this.app.close();
-    }
-    async executeQuery({ query, variables, }) {
-        const result = (await this.app.callServerTool({
-            name: "execute",
-            arguments: { query: print(query), variables },
-        }));
-        return result.structuredContent;
-    }
-    async connectToHost() {
-        try {
-            return await this.app.connect(new PostMessageTransport(window.parent, window.parent));
-        }
-        catch (e) {
-            const error = e instanceof Error ? e : new Error("Failed to connect");
-            throw error;
-        }
-    }
-}
-//# sourceMappingURL=McpAppManager.js.map

package/dist/mcp/core/McpAppManager.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"McpAppManager.js","sourceRoot":"","sources":["../../../src/mcp/core/McpAppManager.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,GAAG,EACH,oBAAoB,GAErB,MAAM,gCAAgC,CAAC;AAKxC,OAAO,EAAE,KAAK,EAAE,MAAM,0BAA0B,CAAC;AACjD,OAAO,EAAE,UAAU,EAAE,oBAAoB,EAAE,MAAM,iBAAiB,CAAC;AAOnE,gBAAgB;AAChB,MAAM,OAAO,aAAa;IACf,GAAG,CAAM;IAElB,SAAS,CAAqB;IAC9B,aAAa,CAA0D;IACvE,UAAU,CAAsC;IAEhD,qBAAqB,GAAG,IAAI,GAAG,EAE5B,CAAC;IAEJ,YAAY,QAA6B;QACvC,IAAI,CAAC,GAAG,GAAG,IAAI,GAAG,CAAC,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,EAAE,OAAO,EAAE,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC;IAC5E,CAAC;IAED,IAAI,QAAQ;QACV,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;IAED,IAAI,YAAY;QACd,OAAO,IAAI,CAAC,aAAa,CAAC;IAC5B,CAAC;IAED,IAAI,SAAS;QACX,OAAO,IAAI,CAAC,UAAU,CAAC;IACzB,CAAC;IAED,oBAAoB,CAClB,EAAmE;QAEnE,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAEnC,OAAO,GAAG,EAAE;YACV,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QACxC,CAAC,CAAC;IACJ,CAAC;IAED,OAAO,GAAG,UAAU,CAAC,KAAK,IAAI,EAAE;QAC9B,IAAI,UAAU,GAAG,oBAAoB,EAAsC,CAAC;QAC5E,IAAI,SAAS,GAAG,oBAAoB,EAAqC,CAAC;QAE1E,IAAI,CAAC,GAAG,CAAC,YAAY,GAAG,CAAC,MAAM,EAAE,EAAE;YACjC,UAAU,CAAC,OAAO,CAChB,MAAuD,CACxD,CAAC;QACJ,CAAC,CAAC;QAEF,IAAI,CAAC,GAAG,CAAC,WAAW,GAAG,CAAC,MAAM,EAAE,EAAE;YAChC,SAAS,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAC5B,CAAC,CAAC;QAEF,IAAI,CAAC,GAAG,CAAC,oBAAoB,GAAG,CAAC,MAAM,EAAE,EAAE;YACzC,IAAI,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC;QACzD,CAAC,CAAC;QAEF,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;QAE3B,MAAM,EAAE,iBAAiB,EAAE,KAAK,EAAE,GAAG,MAAM,UAAU,CAAC,OAAO,CAAC;QAC9D,MAAM,EAAE,SAAS,EAAE,IAAI,EAAE,GAAG,MAAM,SAAS,CAAC,OAAO,CAAC;QAEpD,0EAA0E;QAC1E,0EAA0E;QAC1E,oBAAoB;QACpB,IAAI,CAAC,SAAS;YACZ,IAAI,CAAC,GAAG,CAAC,cAAc,EAAE,EAAE,QAAQ,EAAE,IAAI,CAAC,IAAI;gBAC9C,KAAK,EAAE,QAAQ;gBACf,wEAAwE;gBACxE,wEAAwE;gBACxE,oCAAoC;gBACpC,iBAAiB,CAAC,QAAQ,CAAC;QAC7B,IAAI,CAAC,aAAa,GAAG,KAAK,CAAC;QAC3B,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;QAEvB,OAAO;YACL,GAAG,iBAAiB;YACpB,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,IAAI;SACL,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,KAAK;QACH,OAAO,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC;IAC1B,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,EACjB,KAAK,EACL,SAAS,GAIV;QACC,MAAM,MAAM,GAAG,CAAC,MAAM,IAAI,CAAC,GAAG,CAAC,cAAc,CAAC;YAC5C,IAAI,EAAE,SAAS;YACf,SAAS,EAAE,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,CAAC,EAAE,SAAS,EAAE;SAC9C,CAAC,CAA+B,CAAC;QAElC,OAAO,MAAM,CAAC,iBAAiB,CAAC;IAClC,CAAC;IAEO,KAAK,CAAC,aAAa;QACzB,IAAI,CAAC;YACH,OAAO,MAAM,IAAI,CAAC,GAAG,CAAC,OAAO,CAC3B,IAAI,oBAAoB,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,CACvD,CAAC;QACJ,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,MAAM,KAAK,GAAG,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAAC;YAEtE,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;CACF","sourcesContent":["import {\n  App,\n  PostMessageTransport,\n  type McpUiHostContextChangedNotification,\n} from \"@modelcontextprotocol/ext-apps\";\nimport type { CallToolResult } from \"@modelcontextprotocol/sdk/types.js\";\nimport type { ApplicationManifest } from \"../../types/application-manifest\";\nimport type { FormattedExecutionResult } from \"graphql\";\nimport type { DocumentNode, OperationVariables } from \"@apollo/client\";\nimport { print } from \"@apollo/client/utilities\";\nimport { cacheAsync, promiseWithResolvers } from \"../../utilities\";\nimport type { ApolloMcpServerApps } from \"../../core/types\";\n\ntype ExecuteQueryCallToolResult = Omit<CallToolResult, \"structuredContent\"> & {\n  structuredContent: FormattedExecutionResult;\n};\n\n/** @internal */\nexport class McpAppManager {\n  readonly app: App;\n\n  #toolName: string | undefined;\n  #toolMetadata: ApolloMcpServerApps.CallToolResult[\"_meta\"] | undefined;\n  #toolInput: Record<string, unknown> | undefined;\n\n  #hostContextCallbacks = new Set<\n    (params: McpUiHostContextChangedNotification[\"params\"]) => void\n  >();\n\n  constructor(manifest: ApplicationManifest) {\n    this.app = new App({ name: manifest.name, version: manifest.appVersion });\n  }\n\n  get toolName() {\n    return this.#toolName;\n  }\n\n  get toolMetadata() {\n    return this.#toolMetadata;\n  }\n\n  get toolInput() {\n    return this.#toolInput;\n  }\n\n  onHostContextChanged(\n    cb: (params: McpUiHostContextChangedNotification[\"params\"]) => void\n  ) {\n    this.#hostContextCallbacks.add(cb);\n\n    return () => {\n      this.#hostContextCallbacks.delete(cb);\n    };\n  }\n\n  connect = cacheAsync(async () => {\n    let toolResult = promiseWithResolvers<ApolloMcpServerApps.CallToolResult>();\n    let toolInput = promiseWithResolvers<Parameters<App[\"ontoolinput\"]>[0]>();\n\n    this.app.ontoolresult = (params) => {\n      toolResult.resolve(\n        params as unknown as ApolloMcpServerApps.CallToolResult\n      );\n    };\n\n    this.app.ontoolinput = (params) => {\n      toolInput.resolve(params);\n    };\n\n    this.app.onhostcontextchanged = (params) => {\n      this.#hostContextCallbacks.forEach((cb) => cb(params));\n    };\n\n    await this.connectToHost();\n\n    const { structuredContent, _meta } = await toolResult.promise;\n    const { arguments: args } = await toolInput.promise;\n\n    // Some hosts do not provide toolInfo in the ui/initialize response, so we\n    // fallback to `_meta.toolName` provided by Apollo MCP server if the value\n    // is not available.\n    this.#toolName =\n      this.app.getHostContext()?.toolInfo?.tool.name ??\n      _meta?.toolName ??\n      // Some hosts do not forward `_meta` nor do they provide `toolInfo`. Our\n      // MCP server provides `toolName` in `structuredContent` as a workaround\n      // that we can use if all else fails\n      structuredContent.toolName;\n    this.#toolMetadata = _meta;\n    this.#toolInput = args;\n\n    return {\n      ...structuredContent,\n      toolName: this.toolName,\n      args,\n    };\n  });\n\n  close() {\n    return this.app.close();\n  }\n\n  async executeQuery({\n    query,\n    variables,\n  }: {\n    query: DocumentNode;\n    variables: OperationVariables | undefined;\n  }) {\n    const result = (await this.app.callServerTool({\n      name: \"execute\",\n      arguments: { query: print(query), variables },\n    })) as ExecuteQueryCallToolResult;\n\n    return result.structuredContent;\n  }\n\n  private async connectToHost() {\n    try {\n      return await this.app.connect(\n        new PostMessageTransport(window.parent, window.parent)\n      );\n    } catch (e) {\n      const error = e instanceof Error ? e : new Error(\"Failed to connect\");\n\n      throw error;\n    }\n  }\n}\n"]}

package/dist/mcp/link/ToolCallLink.d.ts

@@ -1,28 +0,0 @@
-import { ApolloLink, Observable } from "@apollo/client";
-/**
- * A terminating link that sends a GraphQL request through an agent tool call.
- * When providing a custom link chain to `ApolloClient`, `ApolloClient` will
- * validate that the terminating link is an instance of this link.
- *
- * @example Providing a custom link chain
- *
- * ```ts
- * import { ApolloLink } from "@apollo/client";
- * import { ApolloClient, ToolCallLink } from "@apollo/client-ai-apps";
- *
- * const link = ApolloLink.from([
- *   ...otherLinks,
- *   new ToolCallLink()
- * ]);
- *
- * const client = new ApolloClient({
- *   link,
- *   // ...
- * });
- * ```
- */
-export declare class ToolCallLink extends ApolloLink {
-    readonly name = "ToolCallLink";
-    request(operation: ApolloLink.Operation): Observable<ApolloLink.Result>;
-}
-//# sourceMappingURL=ToolCallLink.d.ts.map

package/dist/mcp/link/ToolCallLink.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"ToolCallLink.d.ts","sourceRoot":"","sources":["../../../src/mcp/link/ToolCallLink.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAIxD;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,qBAAa,YAAa,SAAQ,UAAU;IAC1C,QAAQ,CAAC,IAAI,kBAAkB;IAE/B,OAAO,CAAC,SAAS,EAAE,UAAU,CAAC,SAAS,GAAG,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC;CAUxE"}

package/dist/mcp/link/ToolCallLink.js

@@ -1,35 +0,0 @@
-import { ApolloLink, Observable } from "@apollo/client";
-import { from } from "rxjs";
-/**
- * A terminating link that sends a GraphQL request through an agent tool call.
- * When providing a custom link chain to `ApolloClient`, `ApolloClient` will
- * validate that the terminating link is an instance of this link.
- *
- * @example Providing a custom link chain
- *
- * ```ts
- * import { ApolloLink } from "@apollo/client";
- * import { ApolloClient, ToolCallLink } from "@apollo/client-ai-apps";
- *
- * const link = ApolloLink.from([
- *   ...otherLinks,
- *   new ToolCallLink()
- * ]);
- *
- * const client = new ApolloClient({
- *   link,
- *   // ...
- * });
- * ```
- */
-export class ToolCallLink extends ApolloLink {
-    name = "ToolCallLink";
-    request(operation) {
-        const client = operation.client;
-        return from(client["appManager"].executeQuery({
-            query: operation.query,
-            variables: operation.variables,
-        }));
-    }
-}
-//# sourceMappingURL=ToolCallLink.js.map

package/dist/mcp/link/ToolCallLink.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"ToolCallLink.js","sourceRoot":"","sources":["../../../src/mcp/link/ToolCallLink.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AACxD,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AAG5B;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,MAAM,OAAO,YAAa,SAAQ,UAAU;IACjC,IAAI,GAAG,cAAc,CAAC;IAE/B,OAAO,CAAC,SAA+B;QACrC,MAAM,MAAM,GAAG,SAAS,CAAC,MAAyB,CAAC;QAEnD,OAAO,IAAI,CACT,MAAM,CAAC,YAAY,CAAC,CAAC,YAAY,CAAC;YAChC,KAAK,EAAE,SAAS,CAAC,KAAK;YACtB,SAAS,EAAE,SAAS,CAAC,SAAS;SAC/B,CAAC,CACH,CAAC;IACJ,CAAC;CACF","sourcesContent":["import { ApolloLink, Observable } from \"@apollo/client\";\nimport { from } from \"rxjs\";\nimport type { ApolloClient as McpApolloClient } from \"../core/ApolloClient\";\n\n/**\n * A terminating link that sends a GraphQL request through an agent tool call.\n * When providing a custom link chain to `ApolloClient`, `ApolloClient` will\n * validate that the terminating link is an instance of this link.\n *\n * @example Providing a custom link chain\n *\n * ```ts\n * import { ApolloLink } from \"@apollo/client\";\n * import { ApolloClient, ToolCallLink } from \"@apollo/client-ai-apps\";\n *\n * const link = ApolloLink.from([\n *   ...otherLinks,\n *   new ToolCallLink()\n * ]);\n *\n * const client = new ApolloClient({\n *   link,\n *   // ...\n * });\n * ```\n */\nexport class ToolCallLink extends ApolloLink {\n  readonly name = \"ToolCallLink\";\n\n  request(operation: ApolloLink.Operation): Observable<ApolloLink.Result> {\n    const client = operation.client as McpApolloClient;\n\n    return from(\n      client[\"appManager\"].executeQuery({\n        query: operation.query,\n        variables: operation.variables,\n      })\n    );\n  }\n}\n"]}

package/dist/mcp/react/hooks/createHydrationUtils.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"createHydrationUtils.d.ts","sourceRoot":"","sources":["../../../../src/mcp/react/hooks/createHydrationUtils.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EACV,YAAY,EACZ,kBAAkB,EAClB,iBAAiB,EAClB,MAAM,gBAAgB,CAAC;AAIxB,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,4BAA4B,CAAC;AAQ3D,KAAK,sBAAsB,CAAC,UAAU,IAAI;KACvC,CAAC,IAAI,MAAM,UAAU,GAAG,UAAU,CAAC,CAAC,CAAC,GAAG,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;CACjE,CAAC;AAEF,KAAK,cAAc,CAAC,UAAU,EAAE,KAAK,IAAI;KACtC,CAAC,IAAI,MAAM,UAAU,IAAI,CAAC,SAAS,MAAM,KAAK,GAC7C,KAAK,CAAC,CAAC,CAAC,SAAS,QAAQ,CAAC,GAAG,CAAC,GAC5B,KAAK,GACL,CAAC,GACH,CAAC,GAAG,CAAC,SAAS,MAAM,UAAU,GAAG,UAAU,CAAC,CAAC,CAAC,GAAG,KAAK;CACzD,CAAC;AAEF,KAAK,YAAY,CAAC,MAAM,IAAI,CAC1B,MAAM,EAAE,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,MAAM,KAAK,OAAO,CAAC,MAAM,CAAC,CAAC,KAC1D,IAAI,CAAC;AAEV,oBAAoB;AACpB,wBAAgB,oBAAoB,CAClC,UAAU,SAAS,kBAAkB,GAAG,kBAAkB,EAC1D,QAAQ,EAAE,iBAAiB,CAAC,GAAG,EAAE,UAAU,CAAC,GAAG,YAAY;2BAKzD,eAAe,SAAS,sBAAsB,CAAC,UAAU,CAAC,aAE/C,eAAe,GACxB,MAAM,CAAC,OAAO,CAAC,MAAM,eAAe,EAAE,MAAM,UAAU,CAAC,EAAE,KAAK,CAAC,KAChE,CACD,SAAS,EAAE,UAAU,EACrB,YAAY,EAAE,YAAY,CAAC,cAAc,CAAC,UAAU,EAAE,eAAe,CAAC,CAAC,CACxE;EAqIF"}

package/dist/mcp/react/hooks/createHydrationUtils.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"createHydrationUtils.js","sourceRoot":"","sources":["../../../../src/mcp/react/hooks/createHydrationUtils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,WAAW,EAAE,MAAM,EAAE,OAAO,EAAE,eAAe,EAAE,MAAM,OAAO,CAAC;AAMhF,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAC/C,OAAO,EAAE,UAAU,EAAE,MAAM,4BAA4B,CAAC;AAExD,OAAO,EAAE,KAAK,EAAE,MAAM,eAAe,CAAC;AACtC,OAAO,EAAE,OAAO,EAAE,MAAM,sCAAsC,CAAC;AAC/D,OAAO,EACL,wBAAwB,EACxB,4BAA4B,GAC7B,MAAM,6BAA6B,CAAC;AAkBrC,oBAAoB;AACpB,MAAM,UAAU,oBAAoB,CAElC,QAA2D;IAC3D,MAAM,iBAAiB,GAAG,wBAAwB,CAAC,QAAQ,CAAC,CAAC;IAC7D,MAAM,aAAa,GAAG,4BAA4B,CAAC,QAAQ,CAAC,CAAC;IAE7D,SAAS,oBAAoB,CAG3B,SACiE;QAKjE,MAAM,MAAM,GAAG,eAAe,EAAE,CAAC;QACjC,MAAM,QAAQ,GAAG,WAAW,EAAE,CAAC;QAC/B,MAAM,CAAC,SAAS,CAAC,GAAG,QAAQ,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAErD,MAAM,WAAW,GACf,SAAS,KAAK,SAAS;YACvB,QAAQ,KAAK,SAAS;YACtB,iBAAiB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAElC,MAAM,CAAC,SAAS,EAAE,YAAY,CAAC,GAAG,QAAQ,CAA0B,GAAG,EAAE;YACvE,MAAM,MAAM,GAA4B,EAAE,CAAC;YAE3C,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CACvC,WAAW,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CACpC,EAAE,CAAC;gBACF,IAAI,aAAa,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;oBACjD,MAAM,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;gBACtB,CAAC;YACH,CAAC;YAED,OAAO,MAAM,CAAC;QAChB,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,mBAAmB,CAAC,GAAG,QAAQ,CAA0B,GAAG,EAAE;YACnE,MAAM,MAAM,GAA4B,EAAE,CAAC;YAE3C,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,CAAC;gBACrD,IAAI,aAAa,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;oBAChD,MAAM,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC,KAAK,CAAC;gBAC5B,CAAC;YACH,CAAC;YAED,OAAO,MAAM,CAAC;QAChB,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,YAAY,EAAE,eAAe,CAAC,GAAG,QAAQ,CAAC,GAAG,EAAE;YACpD,MAAM,MAAM,GAA4B,EAAE,CAAC;YAE3C,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,mBAAmB,CAAC,EAAE,CAAC;gBAC/D,IAAI,WAAW,IAAI,GAAG,IAAI,SAAS,EAAE,CAAC;oBACpC,MAAM,CAAC,GAAG,CAAC,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC;gBAC/B,CAAC;qBAAM,IAAI,CAAC,WAAW,EAAE,CAAC;oBACxB,MAAM,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;gBACtB,CAAC;YACH,CAAC;YAED,OAAO,MAAM,CAAC;QAChB,CAAC,CAAC,CAAC;QAEH,MAAM,cAAc,GAAG,MAAM,CAAC,IAAI,GAAG,EAAU,CAAC,CAAC;QACjD,MAAM,gBAAgB,GAA4B,EAAE,CAAC;QAErD,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,CAAC;YACrD,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC;gBAAE,SAAS;YAE5D,MAAM,UAAU,GACd,cAAc,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC;gBAC/B,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,EAAE,mBAAmB,CAAC,GAAG,CAAC,CAAC,CAAC;YAEhD,IAAI,WAAW,IAAI,CAAC,UAAU,EAAE,CAAC;gBAC/B,IAAI,GAAG,IAAI,SAAS,EAAE,CAAC;oBACrB,gBAAgB,CAAC,GAAG,CAAC,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC;gBACzC,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,gBAAgB,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC,KAAK,CAAC;YACtC,CAAC;QACH,CAAC;QAED,IAAI,CAAC,KAAK,CAAC,gBAAgB,EAAE,YAAY,CAAC,EAAE,CAAC;YAC3C,eAAe,CAAC,gBAAgB,CAAC,CAAC;QACpC,CAAC;QAED,0EAA0E;QAC1E,wEAAwE;QACxE,sEAAsE;QACtE,yCAAyC;QACzC,eAAe,CAAC,GAAG,EAAE;YACnB,IAAI,WAAW,EAAE,CAAC;gBAChB,MAAM,CAAC,cAAc,EAAE,CAAC;YAC1B,CAAC;YACD,uDAAuD;QACzD,CAAC,EAAE,EAAE,CAAC,CAAC;QAEP,eAAe,CAAC,GAAG,EAAE;YACnB,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,CAAC;gBACrD,IACE,aAAa,CAAC,GAAG,CAAC,GAAG,CAAC;oBACtB,UAAU,CAAC,KAAK,CAAC;oBACjB,CAAC,cAAc,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC;oBAChC,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,EAAE,mBAAmB,CAAC,GAAG,CAAC,CAAC,EAC7C,CAAC;oBACD,cAAc,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;gBAClC,CAAC;YACH,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,MAAM,iBAAiB,GAAG,OAAO,CAAC,GAAG,EAAE;YACrC,OAAO,EAAE,GAAG,SAAS,EAAE,GAAG,YAAY,EAAgB,CAAC;QACzD,CAAC,EAAE,CAAC,SAAS,EAAE,YAAY,CAAC,CAAC,CAAC;QAE9B,MAAM,YAAY,GAAG,WAAW,CAE9B,CAAC,MAAM,EAAE,EAAE;YACX,YAAY,CAAC,CAAC,IAAI,EAAE,EAAE;gBACpB,MAAM,OAAO,GACX,OAAO,MAAM,KAAK,UAAU,CAAC,CAAC,CAAC,MAAM,CAAC,IAAW,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;gBAE9D,MAAM,QAAQ,GAAG,MAAM,CAAC,WAAW,CACjC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE;oBACvC,IAAI,GAAG,IAAI,mBAAmB,EAAE,CAAC;wBAC/B,IAAI,OAAO,EAAE,CAAC;4BACZ,OAAO,CAAC,IAAI,CACV,uCAAuC,GAAG,sBAAsB;gCAC9D,oDAAoD,CACvD,CAAC;wBACJ,CAAC;wBACD,OAAO,KAAK,CAAC;oBACf,CAAC;oBACD,OAAO,IAAI,CAAC;gBACd,CAAC,CAAC,CACH,CAAC;gBAEF,IAAI,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,MAAM,KAAK,CAAC;oBAAE,OAAO,IAAI,CAAC;gBAEpD,OAAO,EAAE,GAAG,IAAI,EAAE,GAAG,QAAQ,EAAE,CAAC;YAClC,CAAC,CAAC,CAAC;QACL,CAAC,EAAE,EAAE,CAAC,CAAC;QAEP,OAAO,CAAC,iBAAiB,EAAE,YAAY,CAAC,CAAC;IAC3C,CAAC;IAED,OAAO,EAAE,oBAAoB,EAAE,CAAC;AAClC,CAAC","sourcesContent":["import { useState, useCallback, useRef, useMemo, useLayoutEffect } from \"react\";\nimport type {\n  DocumentNode,\n  OperationVariables,\n  TypedDocumentNode,\n} from \"@apollo/client\";\nimport { useApolloClient } from \"./useApolloClient.js\";\nimport { useToolName } from \"./useToolName.js\";\nimport { isReactive } from \"../../../react/reactive.js\";\nimport type { Reactive } from \"../../../react/reactive.js\";\nimport { equal } from \"@wry/equality\";\nimport { __DEV__ } from \"@apollo/client/utilities/environment\";\nimport {\n  getToolNamesFromDocument,\n  getVariableNamesFromDocument,\n} from \"../../../utilities/index.js\";\n\ntype HydratedVariablesInput<TVariables> = {\n  [K in keyof TVariables]: TVariables[K] | Reactive<TVariables[K]>;\n};\n\ntype StateVariables<TVariables, Input> = {\n  [K in keyof TVariables as K extends keyof Input ?\n    Input[K] extends Reactive<any> ?\n      never\n    : K\n  : K]: K extends keyof TVariables ? TVariables[K] : never;\n};\n\ntype SetVariables<TState> = (\n  update: Partial<TState> | ((prev: TState) => Partial<TState>)\n) => void;\n\n/** @experimental */\nexport function createHydrationUtils<\n  TVariables extends OperationVariables = OperationVariables,\n>(document: TypedDocumentNode<any, TVariables> | DocumentNode) {\n  const documentToolNames = getToolNamesFromDocument(document);\n  const variableNames = getVariableNamesFromDocument(document);\n\n  function useHydratedVariables<\n    TInputVariables extends HydratedVariablesInput<TVariables>,\n  >(\n    variables: TInputVariables &\n      Record<Exclude<keyof TInputVariables, keyof TVariables>, never>\n  ): [\n    variables: TVariables,\n    setVariables: SetVariables<StateVariables<TVariables, TInputVariables>>,\n  ] {\n    const client = useApolloClient();\n    const toolName = useToolName();\n    const [toolInput] = useState(() => client.toolInput);\n\n    const toolMatches =\n      toolInput !== undefined &&\n      toolName !== undefined &&\n      documentToolNames.has(toolName);\n\n    const [stateVars, setStateVars] = useState<Record<string, unknown>>(() => {\n      const values: Record<string, unknown> = {};\n\n      for (const [key, value] of Object.entries(\n        toolMatches ? toolInput : variables\n      )) {\n        if (variableNames.has(key) && !isReactive(value)) {\n          values[key] = value;\n        }\n      }\n\n      return values;\n    });\n\n    const [initialReactiveVars] = useState<Record<string, unknown>>(() => {\n      const values: Record<string, unknown> = {};\n\n      for (const [key, value] of Object.entries(variables)) {\n        if (variableNames.has(key) && isReactive(value)) {\n          values[key] = value.value;\n        }\n      }\n\n      return values;\n    });\n\n    const [reactiveVars, setReactiveVars] = useState(() => {\n      const values: Record<string, unknown> = {};\n\n      for (const [key, value] of Object.entries(initialReactiveVars)) {\n        if (toolMatches && key in toolInput) {\n          values[key] = toolInput[key];\n        } else if (!toolMatches) {\n          values[key] = value;\n        }\n      }\n\n      return values;\n    });\n\n    const changedKeysRef = useRef(new Set<string>());\n    const nextReactiveVars: Record<string, unknown> = {};\n\n    for (const [key, value] of Object.entries(variables)) {\n      if (!variableNames.has(key) || !isReactive(value)) continue;\n\n      const hasChanged =\n        changedKeysRef.current.has(key) ||\n        !equal(value.value, initialReactiveVars[key]);\n\n      if (toolMatches && !hasChanged) {\n        if (key in toolInput) {\n          nextReactiveVars[key] = toolInput[key];\n        }\n      } else {\n        nextReactiveVars[key] = value.value;\n      }\n    }\n\n    if (!equal(nextReactiveVars, reactiveVars)) {\n      setReactiveVars(nextReactiveVars);\n    }\n\n    // Clear the tool input after first mount so that remounting the component\n    // uses the user-provided variables rather than the hydrated tool input.\n    // This runs once on mount; watchQuery also clears it when useQuery is\n    // present, so both paths are idempotent.\n    useLayoutEffect(() => {\n      if (toolMatches) {\n        client.clearToolInput();\n      }\n      // eslint-disable-next-line react-hooks/exhaustive-deps\n    }, []);\n\n    useLayoutEffect(() => {\n      for (const [key, value] of Object.entries(variables)) {\n        if (\n          variableNames.has(key) &&\n          isReactive(value) &&\n          !changedKeysRef.current.has(key) &&\n          !equal(value.value, initialReactiveVars[key])\n        ) {\n          changedKeysRef.current.add(key);\n        }\n      }\n    });\n\n    const resolvedVariables = useMemo(() => {\n      return { ...stateVars, ...reactiveVars } as TVariables;\n    }, [stateVars, reactiveVars]);\n\n    const setVariables = useCallback<\n      SetVariables<StateVariables<TVariables, TInputVariables>>\n    >((update) => {\n      setStateVars((prev) => {\n        const updates =\n          typeof update === \"function\" ? update(prev as any) : update;\n\n        const filtered = Object.fromEntries(\n          Object.entries(updates).filter(([key]) => {\n            if (key in initialReactiveVars) {\n              if (__DEV__) {\n                console.warn(\n                  `Attempted to set reactive variable \"${key}\" via setVariables. ` +\n                    `Reactive variables are read-only and are ignored. `\n                );\n              }\n              return false;\n            }\n            return true;\n          })\n        );\n\n        if (Object.keys(filtered).length === 0) return prev;\n\n        return { ...prev, ...filtered };\n      });\n    }, []);\n\n    return [resolvedVariables, setVariables];\n  }\n\n  return { useHydratedVariables };\n}\n"]}

package/dist/mcp/react/hooks/useApolloClient.d.ts

@@ -1,3 +0,0 @@
-import { ApolloClient } from "../../core/ApolloClient.js";
-export declare function useApolloClient(override?: ApolloClient): ApolloClient;
-//# sourceMappingURL=useApolloClient.d.ts.map

package/dist/mcp/react/hooks/useApolloClient.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"useApolloClient.d.ts","sourceRoot":"","sources":["../../../../src/mcp/react/hooks/useApolloClient.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAC;AAG1D,wBAAgB,eAAe,CAAC,QAAQ,CAAC,EAAE,YAAY,gBAStD"}

package/dist/mcp/react/hooks/useApolloClient.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"useApolloClient.js","sourceRoot":"","sources":["../../../../src/mcp/react/hooks/useApolloClient.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,IAAI,mBAAmB,EAAE,MAAM,sBAAsB,CAAC;AAC9E,OAAO,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAC;AAC1D,OAAO,EAAE,cAAc,EAAE,SAAS,EAAE,MAAM,6BAA6B,CAAC;AAExE,MAAM,UAAU,eAAe,CAAC,QAAuB;IACrD,MAAM,MAAM,GAAG,mBAAmB,CAAC,QAAQ,CAAiB,CAAC;IAE7D,SAAS,CACP,MAAM,CAAC,cAAc,CAAC,EACtB,+MAA+M,CAChN,CAAC;IAEF,OAAO,MAAM,CAAC;AAChB,CAAC","sourcesContent":["import { useApolloClient as useBaseApolloClient } from \"@apollo/client/react\";\nimport { ApolloClient } from \"../../core/ApolloClient.js\";\nimport { aiClientSymbol, invariant } from \"../../../utilities/index.js\";\n\nexport function useApolloClient(override?: ApolloClient) {\n  const client = useBaseApolloClient(override) as ApolloClient;\n\n  invariant(\n    client[aiClientSymbol],\n    'The \"client\" instance provided to <ApolloProvider /> is the wrong instance. You might have imported `ApolloClient` from `@apollo/client`. Please import `ApolloClient` from `@apollo/client-ai-apps` instead.'\n  );\n\n  return client;\n}\n"]}

package/dist/mcp/react/hooks/useApp.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"useApp.d.ts","sourceRoot":"","sources":["../../../../src/mcp/react/hooks/useApp.ts"],"names":[],"mappings":"AAEA,wBAAgB,MAAM,iDAErB"}

package/dist/mcp/react/hooks/useApp.js

@@ -1,5 +0,0 @@
-import { useApolloClient } from "./useApolloClient.js";
-export function useApp() {
-    return useApolloClient()["appManager"].app;
-}
-//# sourceMappingURL=useApp.js.map

package/dist/mcp/react/hooks/useApp.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"useApp.js","sourceRoot":"","sources":["../../../../src/mcp/react/hooks/useApp.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AAEvD,MAAM,UAAU,MAAM;IACpB,OAAO,eAAe,EAAE,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC;AAC7C,CAAC","sourcesContent":["import { useApolloClient } from \"./useApolloClient.js\";\n\nexport function useApp() {\n  return useApolloClient()[\"appManager\"].app;\n}\n"]}

package/dist/mcp/react/hooks/useHostContext.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"useHostContext.d.ts","sourceRoot":"","sources":["../../../../src/mcp/react/hooks/useHostContext.ts"],"names":[],"mappings":"AAGA,wBAAgB,cAAc,0EAU7B"}

package/dist/mcp/react/hooks/useHostContext.js

@@ -1,7 +0,0 @@
-import { useCallback, useSyncExternalStore } from "react";
-import { useApolloClient } from "./useApolloClient";
-export function useHostContext() {
-    const appManager = useApolloClient()["appManager"];
-    return useSyncExternalStore(useCallback((update) => appManager.onHostContextChanged(update), [appManager]), () => appManager.app.getHostContext());
-}
-//# sourceMappingURL=useHostContext.js.map

package/dist/mcp/react/hooks/useHostContext.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"useHostContext.js","sourceRoot":"","sources":["../../../../src/mcp/react/hooks/useHostContext.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,oBAAoB,EAAE,MAAM,OAAO,CAAC;AAC1D,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAEpD,MAAM,UAAU,cAAc;IAC5B,MAAM,UAAU,GAAG,eAAe,EAAE,CAAC,YAAY,CAAC,CAAC;IAEnD,OAAO,oBAAoB,CACzB,WAAW,CACT,CAAC,MAAM,EAAE,EAAE,CAAC,UAAU,CAAC,oBAAoB,CAAC,MAAM,CAAC,EACnD,CAAC,UAAU,CAAC,CACb,EACD,GAAG,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC,cAAc,EAAE,CACtC,CAAC;AACJ,CAAC","sourcesContent":["import { useCallback, useSyncExternalStore } from \"react\";\nimport { useApolloClient } from \"./useApolloClient\";\n\nexport function useHostContext() {\n  const appManager = useApolloClient()[\"appManager\"];\n\n  return useSyncExternalStore(\n    useCallback(\n      (update) => appManager.onHostContextChanged(update),\n      [appManager]\n    ),\n    () => appManager.app.getHostContext()\n  );\n}\n"]}

package/dist/mcp/react/hooks/useToolInfo.d.ts

@@ -1,3 +0,0 @@
-import type { ToolInfo } from "../../../core/typeRegistration.js";
-export declare function useToolInfo(): ToolInfo | undefined;
-//# sourceMappingURL=useToolInfo.d.ts.map

package/dist/mcp/react/hooks/useToolInfo.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"useToolInfo.d.ts","sourceRoot":"","sources":["../../../../src/mcp/react/hooks/useToolInfo.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,mCAAmC,CAAC;AAGlE,wBAAgB,WAAW,IAAI,QAAQ,GAAG,SAAS,CASlD"}

package/dist/mcp/react/hooks/useToolInfo.js

@@ -1,10 +0,0 @@
-import { useApolloClient } from "./useApolloClient.js";
-export function useToolInfo() {
-    const appManager = useApolloClient()["appManager"];
-    const toolName = appManager.toolName;
-    if (!toolName) {
-        return;
-    }
-    return { toolName, toolInput: appManager.toolInput };
-}
-//# sourceMappingURL=useToolInfo.js.map

package/dist/mcp/react/hooks/useToolInfo.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"useToolInfo.js","sourceRoot":"","sources":["../../../../src/mcp/react/hooks/useToolInfo.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AAEvD,MAAM,UAAU,WAAW;IACzB,MAAM,UAAU,GAAG,eAAe,EAAE,CAAC,YAAY,CAAC,CAAC;IACnD,MAAM,QAAQ,GAAG,UAAU,CAAC,QAAQ,CAAC;IAErC,IAAI,CAAC,QAAQ,EAAE,CAAC;QACd,OAAO;IACT,CAAC;IAED,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,UAAU,CAAC,SAAS,EAAE,CAAC;AACvD,CAAC","sourcesContent":["import type { ToolInfo } from \"../../../core/typeRegistration.js\";\nimport { useApolloClient } from \"./useApolloClient.js\";\n\nexport function useToolInfo(): ToolInfo | undefined {\n  const appManager = useApolloClient()[\"appManager\"];\n  const toolName = appManager.toolName;\n\n  if (!toolName) {\n    return;\n  }\n\n  return { toolName, toolInput: appManager.toolInput };\n}\n"]}

package/dist/mcp/react/hooks/useToolInput.d.ts

@@ -1,7 +0,0 @@
-import type { ToolInput } from "../../../core/typeRegistration.js";
-/**
- * @deprecated Please use the `useToolInfo` hook. `useToolInput` will be removed
- * in the next major version.
- */
-export declare function useToolInput(): ToolInput | undefined;
-//# sourceMappingURL=useToolInput.d.ts.map

package/dist/mcp/react/hooks/useToolInput.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"useToolInput.d.ts","sourceRoot":"","sources":["../../../../src/mcp/react/hooks/useToolInput.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,mCAAmC,CAAC;AAGnE;;;GAGG;AACH,wBAAgB,YAAY,IAAI,SAAS,GAAG,SAAS,CAEpD"}

package/dist/mcp/react/hooks/useToolInput.js

@@ -1,9 +0,0 @@
-import { useApolloClient } from "./useApolloClient.js";
-/**
- * @deprecated Please use the `useToolInfo` hook. `useToolInput` will be removed
- * in the next major version.
- */
-export function useToolInput() {
-    return useApolloClient()["appManager"].toolInput;
-}
-//# sourceMappingURL=useToolInput.js.map

package/dist/mcp/react/hooks/useToolInput.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"useToolInput.js","sourceRoot":"","sources":["../../../../src/mcp/react/hooks/useToolInput.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AAEvD;;;GAGG;AACH,MAAM,UAAU,YAAY;IAC1B,OAAO,eAAe,EAAE,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC;AACnD,CAAC","sourcesContent":["import type { ToolInput } from \"../../../core/typeRegistration.js\";\nimport { useApolloClient } from \"./useApolloClient.js\";\n\n/**\n * @deprecated Please use the `useToolInfo` hook. `useToolInput` will be removed\n * in the next major version.\n */\nexport function useToolInput(): ToolInput | undefined {\n  return useApolloClient()[\"appManager\"].toolInput;\n}\n"]}

package/dist/mcp/react/hooks/useToolMetadata.d.ts

@@ -1,2 +0,0 @@
-export declare function useToolMetadata(): import("../../../core/types.js").ApolloMcpServerApps.Meta | undefined;
-//# sourceMappingURL=useToolMetadata.d.ts.map

package/dist/mcp/react/hooks/useToolMetadata.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"useToolMetadata.d.ts","sourceRoot":"","sources":["../../../../src/mcp/react/hooks/useToolMetadata.ts"],"names":[],"mappings":"AAEA,wBAAgB,eAAe,0EAE9B"}