@apollo/client-ai-apps 0.6.4 → 0.7.0

package/dist/core/McpAppManager.d.ts

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

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

@@ -0,0 +1 @@
+{"version":3,"file":"McpAppManager.d.ts","sourceRoot":"","sources":["../../src/core/McpAppManager.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,GAAG,EACH,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,YAAY,CAAC;AAOtD,UAAU,mBAAmB;IAC3B,iBAAiB,EAAE,mBAAmB,CAAC,iBAAiB,CAAC;IACzD,SAAS,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,CAAC;IAC/C,KAAK,EAAE,mBAAmB,CAAC,IAAI,GAAG,SAAS,CAAC;CAC7C;AAED,MAAM,MAAM,2BAA2B,GAAG,CACxC,GAAG,EAAE,GAAG,KACL,OAAO,CAAC,mBAAmB,CAAC,CAAC;AAElC,gBAAgB;AAChB,qBAAa,aAAa;;IACxB,QAAQ,CAAC,GAAG,EAAE,GAAG,CAAC;gBAShB,QAAQ,EAAE,mBAAmB,EAC7B,aAAa,EAAE,2BAA2B;IAM5C,oBAAoB,CAClB,EAAE,EAAE,CAAC,MAAM,EAAE,mCAAmC,CAAC,QAAQ,CAAC,KAAK,IAAI;IASrE,OAAO;;;;;;;;;;;;MA2BJ;IAEH,KAAK;IAIC,YAAY,CAAC,EACjB,KAAK,EACL,SAAS,GACV,EAAE;QACD,KAAK,EAAE,YAAY,CAAC;QACpB,SAAS,EAAE,kBAAkB,GAAG,SAAS,CAAC;KAC3C;;;;;CAWF"}

package/dist/core/McpAppManager.js

@@ -0,0 +1,56 @@
+import { App, } from "@modelcontextprotocol/ext-apps";
+import { print } from "@apollo/client/utilities";
+import { cacheAsync } from "../utilities/index.js";
+/** @internal */
+export class McpAppManager {
+    app;
+    #connectToHost;
+    #hostContextCallbacks = new Set();
+    constructor(manifest, connectToHost) {
+        this.app = new App({ name: manifest.name, version: manifest.appVersion });
+        this.#connectToHost = connectToHost;
+    }
+    onHostContextChanged(cb) {
+        this.#hostContextCallbacks.add(cb);
+        return () => {
+            this.#hostContextCallbacks.delete(cb);
+        };
+    }
+    connect = cacheAsync(async () => {
+        this.app.onhostcontextchanged = (params) => {
+            this.#hostContextCallbacks.forEach((cb) => cb(params));
+        };
+        const { structuredContent, _meta, toolInput } = await this.#connectToHost(this.app);
+        return {
+            structuredContent: {
+                ...structuredContent,
+                ..._meta?.structuredContent,
+            },
+            toolInput,
+            // 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.
+            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,
+            _meta,
+        };
+    });
+    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,
+            ...result._meta?.structuredContent,
+        };
+    }
+}
+//# sourceMappingURL=McpAppManager.js.map

package/dist/core/McpAppManager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"McpAppManager.js","sourceRoot":"","sources":["../../src/core/McpAppManager.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,GAAG,GAEJ,MAAM,gCAAgC,CAAC;AAKxC,OAAO,EAAE,KAAK,EAAE,MAAM,0BAA0B,CAAC;AACjD,OAAO,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AAkBnD,gBAAgB;AAChB,MAAM,OAAO,aAAa;IACf,GAAG,CAAM;IAElB,cAAc,CAA8B;IAE5C,qBAAqB,GAAG,IAAI,GAAG,EAE5B,CAAC;IAEJ,YACE,QAA6B,EAC7B,aAA0C;QAE1C,IAAI,CAAC,GAAG,GAAG,IAAI,GAAG,CAAC,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,EAAE,OAAO,EAAE,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC;QAC1E,IAAI,CAAC,cAAc,GAAG,aAAa,CAAC;IACtC,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,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,EAAE,iBAAiB,EAAE,KAAK,EAAE,SAAS,EAAE,GAAG,MAAM,IAAI,CAAC,cAAc,CACvE,IAAI,CAAC,GAAG,CACT,CAAC;QAEF,OAAO;YACL,iBAAiB,EAAE;gBACjB,GAAG,iBAAiB;gBACpB,GAAG,KAAK,EAAE,iBAAiB;aAC5B;YACD,SAAS;YACT,0EAA0E;YAC1E,0EAA0E;YAC1E,oBAAoB;YACpB,QAAQ,EACN,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;YAC5B,KAAK;SACN,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;YACL,GAAG,MAAM,CAAC,iBAAiB;YAC3B,GAAG,MAAM,CAAC,KAAK,EAAE,iBAAiB;SACnC,CAAC;IACJ,CAAC;CACF","sourcesContent":["import {\n  App,\n  type McpUiHostContextChangedNotification,\n} from \"@modelcontextprotocol/ext-apps\";\nimport type { CallToolResult } from \"@modelcontextprotocol/sdk/types.js\";\nimport type { ApplicationManifest } from \"../types/application-manifest.js\";\nimport type { FormattedExecutionResult } from \"graphql\";\nimport type { DocumentNode, OperationVariables } from \"@apollo/client\";\nimport { print } from \"@apollo/client/utilities\";\nimport { cacheAsync } from \"../utilities/index.js\";\nimport type { ApolloMcpServerApps } from \"./types.js\";\n\ntype ExecuteQueryCallToolResult = Omit<CallToolResult, \"structuredContent\"> & {\n  structuredContent: FormattedExecutionResult;\n  _meta?: { structuredContent?: FormattedExecutionResult };\n};\n\ninterface ConnectToHostResult {\n  structuredContent: ApolloMcpServerApps.StructuredContent;\n  toolInput: Record<string, unknown> | undefined;\n  _meta: ApolloMcpServerApps.Meta | undefined;\n}\n\nexport type ConnectToHostImplementation = (\n  app: App\n) => Promise<ConnectToHostResult>;\n\n/** @internal */\nexport class McpAppManager {\n  readonly app: App;\n\n  #connectToHost: ConnectToHostImplementation;\n\n  #hostContextCallbacks = new Set<\n    (params: McpUiHostContextChangedNotification[\"params\"]) => void\n  >();\n\n  constructor(\n    manifest: ApplicationManifest,\n    connectToHost: ConnectToHostImplementation\n  ) {\n    this.app = new App({ name: manifest.name, version: manifest.appVersion });\n    this.#connectToHost = connectToHost;\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    this.app.onhostcontextchanged = (params) => {\n      this.#hostContextCallbacks.forEach((cb) => cb(params));\n    };\n\n    const { structuredContent, _meta, toolInput } = await this.#connectToHost(\n      this.app\n    );\n\n    return {\n      structuredContent: {\n        ...structuredContent,\n        ..._meta?.structuredContent,\n      },\n      toolInput,\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      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      _meta,\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 {\n      ...result.structuredContent,\n      ...result._meta?.structuredContent,\n    };\n  }\n}\n"]}

package/dist/core/typeRegistration.d.ts

@@ -1,22 +1,8 @@
 export interface Register {
 }
-/**
- * @deprecated Please use the `ToolInfo` type instead. `ToolName` will be
- * removed in the next major version.
- */
-export type ToolName = Register extends {
-    toolName: infer T extends string;
-} ? T : string;
 type RegisteredToolInputs = Register extends {
     toolInputs: infer T extends Record<string, unknown>;
 } ? T : never;
-/**
- * @deprecated Please use the `ToolInfo` type instead. `ToolInput` will be
- * removed in the next major version.
- */
-export type ToolInput = [
-    RegisteredToolInputs
-] extends [never] ? Record<string, unknown> : RegisteredToolInputs[keyof RegisteredToolInputs];
 type ToolInfoFromInputs<T extends Record<string, unknown>> = {
     [K in keyof T]: {
         toolName: K;

package/dist/core/typeRegistration.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"typeRegistration.d.ts","sourceRoot":"","sources":["../../src/core/typeRegistration.ts"],"names":[],"mappings":"AAAA,MAAM,WAAW,QAAQ;CAAG;AAE5B;;;GAGG;AACH,MAAM,MAAM,QAAQ,GAClB,QAAQ,SAAS;IAAE,QAAQ,EAAE,MAAM,CAAC,SAAS,MAAM,CAAA;CAAE,GAAG,CAAC,GAAG,MAAM,CAAC;AAErE,KAAK,oBAAoB,GACvB,QAAQ,SAAS;IAAE,UAAU,EAAE,MAAM,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;CAAE,GAAG,CAAC,GAC1E,KAAK,CAAC;AAEV;;;GAGG;AACH,MAAM,MAAM,SAAS,GACnB;IAAC,oBAAoB;CAAC,SAAS,CAAC,KAAK,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAC9D,oBAAoB,CAAC,MAAM,oBAAoB,CAAC,CAAC;AAErD,KAAK,kBAAkB,CAAC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,IAAI;KAC1D,CAAC,IAAI,MAAM,CAAC,GAAG;QACd,QAAQ,EAAE,CAAC,CAAC;QACZ,SAAS,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,SAAS,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;KACzE;CACF,CAAC,MAAM,CAAC,CAAC,CAAC;AAEX,MAAM,MAAM,QAAQ,GAClB;IAAC,oBAAoB;CAAC,SAAS,CAAC,KAAK,CAAC,GACpC;IAAE,QAAQ,EAAE,MAAM,CAAC;IAAC,SAAS,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,CAAA;CAAE,GACpE,kBAAkB,CAAC,oBAAoB,CAAC,CAAC"}
+{"version":3,"file":"typeRegistration.d.ts","sourceRoot":"","sources":["../../src/core/typeRegistration.ts"],"names":[],"mappings":"AAAA,MAAM,WAAW,QAAQ;CAAG;AAE5B,KAAK,oBAAoB,GACvB,QAAQ,SAAS;IAAE,UAAU,EAAE,MAAM,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;CAAE,GAAG,CAAC,GAC1E,KAAK,CAAC;AAEV,KAAK,kBAAkB,CAAC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,IAAI;KAC1D,CAAC,IAAI,MAAM,CAAC,GAAG;QACd,QAAQ,EAAE,CAAC,CAAC;QACZ,SAAS,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,SAAS,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;KACzE;CACF,CAAC,MAAM,CAAC,CAAC,CAAC;AAEX,MAAM,MAAM,QAAQ,GAClB;IAAC,oBAAoB;CAAC,SAAS,CAAC,KAAK,CAAC,GACpC;IAAE,QAAQ,EAAE,MAAM,CAAC;IAAC,SAAS,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,CAAA;CAAE,GACpE,kBAAkB,CAAC,oBAAoB,CAAC,CAAC"}

package/dist/core/typeRegistration.js.map

@@ -1 +1 @@
-{"version":3,"file":"typeRegistration.js","sourceRoot":"","sources":["../../src/core/typeRegistration.ts"],"names":[],"mappings":"","sourcesContent":["export interface Register {}\n\n/**\n * @deprecated Please use the `ToolInfo` type instead. `ToolName` will be\n * removed in the next major version.\n */\nexport type ToolName =\n  Register extends { toolName: infer T extends string } ? T : string;\n\ntype RegisteredToolInputs =\n  Register extends { toolInputs: infer T extends Record<string, unknown> } ? T\n  : never;\n\n/**\n * @deprecated Please use the `ToolInfo` type instead. `ToolInput` will be\n * removed in the next major version.\n */\nexport type ToolInput =\n  [RegisteredToolInputs] extends [never] ? Record<string, unknown>\n  : RegisteredToolInputs[keyof RegisteredToolInputs];\n\ntype ToolInfoFromInputs<T extends Record<string, unknown>> = {\n  [K in keyof T]: {\n    toolName: K;\n    toolInput: T[K] extends Record<string, never> ? T[K] | undefined : T[K];\n  };\n}[keyof T];\n\nexport type ToolInfo =\n  [RegisteredToolInputs] extends [never] ?\n    { toolName: string; toolInput: Record<string, unknown> | undefined }\n  : ToolInfoFromInputs<RegisteredToolInputs>;\n"]}
+{"version":3,"file":"typeRegistration.js","sourceRoot":"","sources":["../../src/core/typeRegistration.ts"],"names":[],"mappings":"","sourcesContent":["export interface Register {}\n\ntype RegisteredToolInputs =\n  Register extends { toolInputs: infer T extends Record<string, unknown> } ? T\n  : never;\n\ntype ToolInfoFromInputs<T extends Record<string, unknown>> = {\n  [K in keyof T]: {\n    toolName: K;\n    toolInput: T[K] extends Record<string, never> ? T[K] | undefined : T[K];\n  };\n}[keyof T];\n\nexport type ToolInfo =\n  [RegisteredToolInputs] extends [never] ?\n    { toolName: string; toolInput: Record<string, unknown> | undefined }\n  : ToolInfoFromInputs<RegisteredToolInputs>;\n"]}

package/dist/core/types.d.ts

@@ -3,10 +3,11 @@ import type { FormattedExecutionResult } from "graphql";
 export declare namespace ApolloMcpServerApps {
     interface Meta {
         toolName: string;
+        structuredContent?: ApolloMcpServerApps.StructuredContent;
         [x: string]: unknown;
     }
     interface StructuredContent {
-        result: FormattedExecutionResult;
+        result?: FormattedExecutionResult;
         prefetch?: Record<string, FormattedExecutionResult>;
         toolName?: string;
         [x: string]: unknown;

package/dist/core/types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/core/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,IAAI,iBAAiB,EAAE,MAAM,oCAAoC,CAAC;AAC9F,OAAO,KAAK,EAAE,wBAAwB,EAAE,MAAM,SAAS,CAAC;AAExD,yBAAiB,mBAAmB,CAAC;IACnC,UAAiB,IAAI;QACnB,QAAQ,EAAE,MAAM,CAAC;QACjB,CAAC,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC;KACtB;IAED,UAAiB,iBAAiB;QAChC,MAAM,EAAE,wBAAwB,CAAC;QACjC,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,wBAAwB,CAAC,CAAC;QACpD,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,CAAC,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC;KACtB;IAED,UAAiB,cAAe,SAAQ,iBAAiB;QACvD,KAAK,CAAC,EAAE,mBAAmB,CAAC,IAAI,CAAC;QACjC,iBAAiB,EAAE,mBAAmB,CAAC,iBAAiB,CAAC;KAC1D;CACF"}
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/core/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,IAAI,iBAAiB,EAAE,MAAM,oCAAoC,CAAC;AAC9F,OAAO,KAAK,EAAE,wBAAwB,EAAE,MAAM,SAAS,CAAC;AAExD,yBAAiB,mBAAmB,CAAC;IACnC,UAAiB,IAAI;QACnB,QAAQ,EAAE,MAAM,CAAC;QACjB,iBAAiB,CAAC,EAAE,mBAAmB,CAAC,iBAAiB,CAAC;QAC1D,CAAC,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC;KACtB;IAED,UAAiB,iBAAiB;QAChC,MAAM,CAAC,EAAE,wBAAwB,CAAC;QAClC,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,wBAAwB,CAAC,CAAC;QACpD,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,CAAC,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC;KACtB;IAED,UAAiB,cAAe,SAAQ,iBAAiB;QACvD,KAAK,CAAC,EAAE,mBAAmB,CAAC,IAAI,CAAC;QACjC,iBAAiB,EAAE,mBAAmB,CAAC,iBAAiB,CAAC;KAC1D;CACF"}

package/dist/core/types.js.map

@@ -1 +1 @@
-{"version":3,"file":"types.js","sourceRoot":"","sources":["../../src/core/types.ts"],"names":[],"mappings":"","sourcesContent":["import type { CallToolResult as McpCallToolResult } from \"@modelcontextprotocol/sdk/types.js\";\nimport type { FormattedExecutionResult } from \"graphql\";\n\nexport namespace ApolloMcpServerApps {\n  export interface Meta {\n    toolName: string;\n    [x: string]: unknown;\n  }\n\n  export interface StructuredContent {\n    result: FormattedExecutionResult;\n    prefetch?: Record<string, FormattedExecutionResult>;\n    toolName?: string;\n    [x: string]: unknown;\n  }\n\n  export interface CallToolResult extends McpCallToolResult {\n    _meta?: ApolloMcpServerApps.Meta;\n    structuredContent: ApolloMcpServerApps.StructuredContent;\n  }\n}\n"]}
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../src/core/types.ts"],"names":[],"mappings":"","sourcesContent":["import type { CallToolResult as McpCallToolResult } from \"@modelcontextprotocol/sdk/types.js\";\nimport type { FormattedExecutionResult } from \"graphql\";\n\nexport namespace ApolloMcpServerApps {\n  export interface Meta {\n    toolName: string;\n    structuredContent?: ApolloMcpServerApps.StructuredContent;\n    [x: string]: unknown;\n  }\n\n  export interface StructuredContent {\n    result?: FormattedExecutionResult;\n    prefetch?: Record<string, FormattedExecutionResult>;\n    toolName?: string;\n    [x: string]: unknown;\n  }\n\n  export interface CallToolResult extends McpCallToolResult {\n    _meta?: ApolloMcpServerApps.Meta;\n    structuredContent: ApolloMcpServerApps.StructuredContent;\n  }\n}\n"]}

package/dist/index.d.ts

@@ -1,5 +1,5 @@
 export type { ApplicationManifest, ManifestOperation, ManifestTool, ManifestExtraInput, ManifestCsp, ManifestLabels, ManifestWidgetSettings, } from "./types/application-manifest.js";
-export type { Register, ToolInfo, ToolInput, ToolName, } from "./core/typeRegistration.js";
+export type { Register, ToolInfo } from "./core/typeRegistration.js";
 export { ApolloClient } from "./core/ApolloClient.js";
 export { ToolCallLink } from "./link/ToolCallLink.js";
 export { Platform } from "./core/Platform.js";

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,YAAY,EACV,mBAAmB,EACnB,iBAAiB,EACjB,YAAY,EACZ,kBAAkB,EAClB,WAAW,EACX,cAAc,EACd,sBAAsB,GACvB,MAAM,iCAAiC,CAAC;AAEzC,YAAY,EACV,QAAQ,EACR,QAAQ,EACR,SAAS,EACT,QAAQ,GACT,MAAM,4BAA4B,CAAC;AAEpC,OAAO,EAAE,YAAY,EAAE,MAAM,wBAAwB,CAAC;AACtD,OAAO,EAAE,YAAY,EAAE,MAAM,wBAAwB,CAAC;AAEtD,OAAO,EAAE,QAAQ,EAAE,MAAM,oBAAoB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,YAAY,EACV,mBAAmB,EACnB,iBAAiB,EACjB,YAAY,EACZ,kBAAkB,EAClB,WAAW,EACX,cAAc,EACd,sBAAsB,GACvB,MAAM,iCAAiC,CAAC;AAEzC,YAAY,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM,4BAA4B,CAAC;AAErE,OAAO,EAAE,YAAY,EAAE,MAAM,wBAAwB,CAAC;AACtD,OAAO,EAAE,YAAY,EAAE,MAAM,wBAAwB,CAAC;AAEtD,OAAO,EAAE,QAAQ,EAAE,MAAM,oBAAoB,CAAC"}

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAiBA,OAAO,EAAE,YAAY,EAAE,MAAM,wBAAwB,CAAC;AACtD,OAAO,EAAE,YAAY,EAAE,MAAM,wBAAwB,CAAC;AAEtD,OAAO,EAAE,QAAQ,EAAE,MAAM,oBAAoB,CAAC","sourcesContent":["export type {\n  ApplicationManifest,\n  ManifestOperation,\n  ManifestTool,\n  ManifestExtraInput,\n  ManifestCsp,\n  ManifestLabels,\n  ManifestWidgetSettings,\n} from \"./types/application-manifest.js\";\n\nexport type {\n  Register,\n  ToolInfo,\n  ToolInput,\n  ToolName,\n} from \"./core/typeRegistration.js\";\n\nexport { ApolloClient } from \"./core/ApolloClient.js\";\nexport { ToolCallLink } from \"./link/ToolCallLink.js\";\n\nexport { Platform } from \"./core/Platform.js\";\n"]}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAYA,OAAO,EAAE,YAAY,EAAE,MAAM,wBAAwB,CAAC;AACtD,OAAO,EAAE,YAAY,EAAE,MAAM,wBAAwB,CAAC;AAEtD,OAAO,EAAE,QAAQ,EAAE,MAAM,oBAAoB,CAAC","sourcesContent":["export type {\n  ApplicationManifest,\n  ManifestOperation,\n  ManifestTool,\n  ManifestExtraInput,\n  ManifestCsp,\n  ManifestLabels,\n  ManifestWidgetSettings,\n} from \"./types/application-manifest.js\";\n\nexport type { Register, ToolInfo } from \"./core/typeRegistration.js\";\n\nexport { ApolloClient } from \"./core/ApolloClient.js\";\nexport { ToolCallLink } from \"./link/ToolCallLink.js\";\n\nexport { Platform } from \"./core/Platform.js\";\n"]}

package/dist/index.mcp.d.ts

@@ -1,5 +1,4 @@
 export * from "./index.js";
 export { Platform } from "./mcp/core/Platform.js";
 export { ApolloClient } from "./mcp/core/ApolloClient.js";
-export { ToolCallLink } from "./mcp/link/ToolCallLink.js";
 //# sourceMappingURL=index.mcp.d.ts.map

package/dist/index.mcp.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.mcp.d.ts","sourceRoot":"","sources":["../src/index.mcp.ts"],"names":[],"mappings":"AAAA,cAAc,YAAY,CAAC;AAE3B,OAAO,EAAE,QAAQ,EAAE,MAAM,wBAAwB,CAAC;AAClD,OAAO,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAC;AAC1D,OAAO,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAC"}
+{"version":3,"file":"index.mcp.d.ts","sourceRoot":"","sources":["../src/index.mcp.ts"],"names":[],"mappings":"AAAA,cAAc,YAAY,CAAC;AAE3B,OAAO,EAAE,QAAQ,EAAE,MAAM,wBAAwB,CAAC;AAClD,OAAO,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAC"}

package/dist/index.mcp.js

@@ -1,5 +1,4 @@
 export * from "./index.js";
 export { Platform } from "./mcp/core/Platform.js";
 export { ApolloClient } from "./mcp/core/ApolloClient.js";
-export { ToolCallLink } from "./mcp/link/ToolCallLink.js";
 //# sourceMappingURL=index.mcp.js.map

package/dist/index.mcp.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.mcp.js","sourceRoot":"","sources":["../src/index.mcp.ts"],"names":[],"mappings":"AAAA,cAAc,YAAY,CAAC;AAE3B,OAAO,EAAE,QAAQ,EAAE,MAAM,wBAAwB,CAAC;AAClD,OAAO,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAC;AAC1D,OAAO,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAC","sourcesContent":["export * from \"./index.js\";\n\nexport { Platform } from \"./mcp/core/Platform.js\";\nexport { ApolloClient } from \"./mcp/core/ApolloClient.js\";\nexport { ToolCallLink } from \"./mcp/link/ToolCallLink.js\";\n"]}
+{"version":3,"file":"index.mcp.js","sourceRoot":"","sources":["../src/index.mcp.ts"],"names":[],"mappings":"AAAA,cAAc,YAAY,CAAC;AAE3B,OAAO,EAAE,QAAQ,EAAE,MAAM,wBAAwB,CAAC;AAClD,OAAO,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAC","sourcesContent":["export * from \"./index.js\";\n\nexport { Platform } from \"./mcp/core/Platform.js\";\nexport { ApolloClient } from \"./mcp/core/ApolloClient.js\";\n"]}

package/dist/index.openai.d.ts

@@ -1,5 +1,4 @@
 export * from "./index.js";
 export { Platform } from "./openai/core/Platform.js";
 export { ApolloClient } from "./openai/core/ApolloClient.js";
-export { ToolCallLink } from "./openai/link/ToolCallLink.js";
 //# sourceMappingURL=index.openai.d.ts.map

package/dist/index.openai.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.openai.d.ts","sourceRoot":"","sources":["../src/index.openai.ts"],"names":[],"mappings":"AAAA,cAAc,YAAY,CAAC;AAE3B,OAAO,EAAE,QAAQ,EAAE,MAAM,2BAA2B,CAAC;AACrD,OAAO,EAAE,YAAY,EAAE,MAAM,+BAA+B,CAAC;AAC7D,OAAO,EAAE,YAAY,EAAE,MAAM,+BAA+B,CAAC"}
+{"version":3,"file":"index.openai.d.ts","sourceRoot":"","sources":["../src/index.openai.ts"],"names":[],"mappings":"AAAA,cAAc,YAAY,CAAC;AAE3B,OAAO,EAAE,QAAQ,EAAE,MAAM,2BAA2B,CAAC;AACrD,OAAO,EAAE,YAAY,EAAE,MAAM,+BAA+B,CAAC"}

package/dist/index.openai.js

@@ -1,5 +1,4 @@
 export * from "./index.js";
 export { Platform } from "./openai/core/Platform.js";
 export { ApolloClient } from "./openai/core/ApolloClient.js";
-export { ToolCallLink } from "./openai/link/ToolCallLink.js";
 //# sourceMappingURL=index.openai.js.map

package/dist/index.openai.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.openai.js","sourceRoot":"","sources":["../src/index.openai.ts"],"names":[],"mappings":"AAAA,cAAc,YAAY,CAAC;AAE3B,OAAO,EAAE,QAAQ,EAAE,MAAM,2BAA2B,CAAC;AACrD,OAAO,EAAE,YAAY,EAAE,MAAM,+BAA+B,CAAC;AAC7D,OAAO,EAAE,YAAY,EAAE,MAAM,+BAA+B,CAAC","sourcesContent":["export * from \"./index.js\";\n\nexport { Platform } from \"./openai/core/Platform.js\";\nexport { ApolloClient } from \"./openai/core/ApolloClient.js\";\nexport { ToolCallLink } from \"./openai/link/ToolCallLink.js\";\n"]}
+{"version":3,"file":"index.openai.js","sourceRoot":"","sources":["../src/index.openai.ts"],"names":[],"mappings":"AAAA,cAAc,YAAY,CAAC;AAE3B,OAAO,EAAE,QAAQ,EAAE,MAAM,2BAA2B,CAAC;AACrD,OAAO,EAAE,YAAY,EAAE,MAAM,+BAA+B,CAAC","sourcesContent":["export * from \"./index.js\";\n\nexport { Platform } from \"./openai/core/Platform.js\";\nexport { ApolloClient } from \"./openai/core/ApolloClient.js\";\n"]}

package/dist/link/ToolCallLink.d.ts

@@ -22,6 +22,11 @@ import { ApolloLink } from "@apollo/client";
  * ```
  */
 export declare class ToolCallLink extends ApolloLink {
-    constructor();
+    readonly name = "ToolCallLink";
+    request(operation: ApolloLink.Operation): import("rxjs").Observable<{
+        errors?: ReadonlyArray<import("graphql").GraphQLFormattedError>;
+        data?: import("graphql/jsutils/ObjMap").ObjMap<unknown> | null;
+        extensions?: import("graphql/jsutils/ObjMap").ObjMap<unknown>;
+    }>;
 }
 //# sourceMappingURL=ToolCallLink.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"ToolCallLink.d.ts","sourceRoot":"","sources":["../../src/link/ToolCallLink.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAE5C;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,qBAAa,YAAa,SAAQ,UAAU;;CAQ3C"}
+{"version":3,"file":"ToolCallLink.d.ts","sourceRoot":"","sources":["../../src/link/ToolCallLink.ts"],"names":[],"mappings":"AAAA,OAAO,EAAgB,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAO1D;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,qBAAa,YAAa,SAAQ,UAAU;IAC1C,QAAQ,CAAC,IAAI,kBAAkB;IAC/B,OAAO,CAAC,SAAS,EAAE,UAAU,CAAC,SAAS;;;;;CAUxC"}

package/dist/link/ToolCallLink.js

@@ -1,4 +1,7 @@
-import { ApolloLink } from "@apollo/client";
+import { ApolloClient, ApolloLink } from "@apollo/client";
+import { __DEV__ } from "@apollo/client/utilities/environment";
+import { from } from "rxjs";
+import { aiClientSymbol, invariant } from "../utilities/index.js";
 /**
  * A terminating link that sends a GraphQL request through an agent tool call.
  * When providing a custom link chain to `ApolloClient`, `ApolloClient` will
@@ -22,9 +25,19 @@ import { ApolloLink } from "@apollo/client";
  * ```
  */
 export class ToolCallLink extends ApolloLink {
-    constructor() {
-        super();
-        throw new Error("Cannot construct a `ToolCallLink` from `@apollo/client-ai-apps` without export conditions. Please set export conditions or import from  the `/openai` or `/mcp` subpath directly.");
+    name = "ToolCallLink";
+    request(operation) {
+        const client = getPrivateAccess(operation.client);
+        return from(client.appManager.executeQuery({
+            query: operation.query,
+            variables: operation.variables,
+        }));
     }
 }
+function getPrivateAccess(client) {
+    if (__DEV__) {
+        invariant(client[aiClientSymbol], 'The "client" instance used with `ToolCallLink` is the wrong instance. You might have imported `ApolloClient` from `@apollo/client`. Please import `ApolloClient` from `@apollo/client-ai-apps` instead.');
+    }
+    return client;
+}
 //# sourceMappingURL=ToolCallLink.js.map

package/dist/link/ToolCallLink.js.map

@@ -1 +1 @@
-{"version":3,"file":"ToolCallLink.js","sourceRoot":"","sources":["../../src/link/ToolCallLink.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAE5C;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,MAAM,OAAO,YAAa,SAAQ,UAAU;IAC1C;QACE,KAAK,EAAE,CAAC;QAER,MAAM,IAAI,KAAK,CACb,mLAAmL,CACpL,CAAC;IACJ,CAAC;CACF","sourcesContent":["import { ApolloLink } from \"@apollo/client\";\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  constructor() {\n    super();\n\n    throw new Error(\n      \"Cannot construct a `ToolCallLink` from `@apollo/client-ai-apps` without export conditions. Please set export conditions or import from  the `/openai` or `/mcp` subpath directly.\"\n    );\n  }\n}\n"]}
+{"version":3,"file":"ToolCallLink.js","sourceRoot":"","sources":["../../src/link/ToolCallLink.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAC1D,OAAO,EAAE,OAAO,EAAE,MAAM,sCAAsC,CAAC;AAC/D,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AAG5B,OAAO,EAAE,cAAc,EAAE,SAAS,EAAE,MAAM,uBAAuB,CAAC;AAElE;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,MAAM,OAAO,YAAa,SAAQ,UAAU;IACjC,IAAI,GAAG,cAAc,CAAC;IAC/B,OAAO,CAAC,SAA+B;QACrC,MAAM,MAAM,GAAG,gBAAgB,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QAElD,OAAO,IAAI,CACT,MAAM,CAAC,UAAU,CAAC,YAAY,CAAC;YAC7B,KAAK,EAAE,SAAS,CAAC,KAAK;YACtB,SAAS,EAAE,SAAS,CAAC,SAAS;SAC/B,CAAC,CACH,CAAC;IACJ,CAAC;CACF;AAED,SAAS,gBAAgB,CACvB,MAAoB;IAEpB,IAAI,OAAO,EAAE,CAAC;QACZ,SAAS,CACN,MAAc,CAAC,cAAc,CAAC,EAC/B,yMAAyM,CAC1M,CAAC;IACJ,CAAC;IAED,OAAO,MAAa,CAAC;AACvB,CAAC","sourcesContent":["import { ApolloClient, ApolloLink } from \"@apollo/client\";\nimport { __DEV__ } from \"@apollo/client/utilities/environment\";\nimport { from } from \"rxjs\";\n\nimport type { McpAppManager } from \"../core/McpAppManager\";\nimport { aiClientSymbol, invariant } from \"../utilities/index.js\";\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  request(operation: ApolloLink.Operation) {\n    const client = getPrivateAccess(operation.client);\n\n    return from(\n      client.appManager.executeQuery({\n        query: operation.query,\n        variables: operation.variables,\n      })\n    );\n  }\n}\n\nfunction getPrivateAccess(\n  client: ApolloClient\n): ApolloClient & { appManager: McpAppManager } {\n  if (__DEV__) {\n    invariant(\n      (client as any)[aiClientSymbol],\n      'The \"client\" instance used with `ToolCallLink` is the wrong instance. You might have imported `ApolloClient` from `@apollo/client`. Please import `ApolloClient` from `@apollo/client-ai-apps` instead.'\n    );\n  }\n\n  return client as any;\n}\n"]}

package/dist/link/ToolHydrationLink.d.ts

@@ -0,0 +1,21 @@
+import { ApolloLink, Observable } from "@apollo/client";
+import type { OperationVariables } from "@apollo/client";
+import type { FormattedExecutionResult } from "graphql";
+import type { ManifestOperation } from "../types/application-manifest.js";
+/**
+ * @internal
+ * Holds requests until the client is fully hydrated after a tool call. It is
+ * particularly useful for `network-only` and `cache-and-network` fetch policies
+ * where the request makes it to the link chain because it prevents a followup
+ * tool call to the MCP server for the just fetched data from the MCP server.
+ */
+export declare class ToolHydrationLink extends ApolloLink {
+    #private;
+    hydrate(operation: ManifestOperation, { result, variables, }: {
+        result: FormattedExecutionResult;
+        variables: OperationVariables;
+    }): void;
+    complete(): void;
+    request(operation: ApolloLink.Operation, forward: ApolloLink.ForwardFunction): Observable<ApolloLink.Result>;
+}
+//# sourceMappingURL=ToolHydrationLink.d.ts.map

package/dist/link/ToolHydrationLink.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ToolHydrationLink.d.ts","sourceRoot":"","sources":["../../src/link/ToolHydrationLink.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AACxD,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AAEzD,OAAO,KAAK,EAAE,wBAAwB,EAAE,MAAM,SAAS,CAAC;AAExD,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,kCAAkC,CAAC;AAK1E;;;;;;GAMG;AACH,qBAAa,iBAAkB,SAAQ,UAAU;;IAK/C,OAAO,CACL,SAAS,EAAE,iBAAiB,EAC5B,EACE,MAAM,EACN,SAAS,GACV,EAAE;QAAE,MAAM,EAAE,wBAAwB,CAAC;QAAC,SAAS,EAAE,kBAAkB,CAAA;KAAE;IAQxE,QAAQ,IAAI,IAAI;IAQhB,OAAO,CACL,SAAS,EAAE,UAAU,CAAC,SAAS,EAC/B,OAAO,EAAE,UAAU,CAAC,eAAe,GAClC,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC;CAiCjC"}

package/dist/link/ToolHydrationLink.js

@@ -0,0 +1,57 @@
+import { ApolloLink, Observable } from "@apollo/client";
+import { canonicalStringify } from "@apollo/client/utilities";
+import { of } from "rxjs";
+/**
+ * @internal
+ * Holds requests until the client is fully hydrated after a tool call. It is
+ * particularly useful for `network-only` and `cache-and-network` fetch policies
+ * where the request makes it to the link chain because it prevents a followup
+ * tool call to the MCP server for the just fetched data from the MCP server.
+ */
+export class ToolHydrationLink extends ApolloLink {
+    #hydrated = false;
+    #pending = [];
+    #operations = new Map();
+    hydrate(operation, { result, variables, }) {
+        this.#operations.set(getKey({ operationName: operation.name, variables }), result);
+    }
+    complete() {
+        this.#hydrated = true;
+        for (const resolve of this.#pending.splice(0)) {
+            resolve();
+        }
+    }
+    request(operation, forward) {
+        const maybeSendToTerminatingLink = () => {
+            const key = getKey(operation);
+            const result = this.#operations.get(key);
+            if (result) {
+                this.#operations.delete(key);
+                return of(result);
+            }
+            return forward(operation);
+        };
+        if (this.#hydrated) {
+            return maybeSendToTerminatingLink();
+        }
+        return new Observable((observer) => {
+            let active = true;
+            const resolve = () => {
+                if (!active)
+                    return;
+                maybeSendToTerminatingLink().subscribe(observer);
+            };
+            this.#pending.push(resolve);
+            return () => {
+                active = false;
+                const idx = this.#pending.indexOf(resolve);
+                if (idx !== -1)
+                    this.#pending.splice(idx, 1);
+            };
+        });
+    }
+}
+function getKey({ operationName, variables, }) {
+    return `${operationName}:${canonicalStringify(variables ?? {})}`;
+}
+//# sourceMappingURL=ToolHydrationLink.js.map

package/dist/link/ToolHydrationLink.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ToolHydrationLink.js","sourceRoot":"","sources":["../../src/link/ToolHydrationLink.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAExD,OAAO,EAAE,kBAAkB,EAAE,MAAM,0BAA0B,CAAC;AAE9D,OAAO,EAAE,EAAE,EAAE,MAAM,MAAM,CAAC;AAM1B;;;;;;GAMG;AACH,MAAM,OAAO,iBAAkB,SAAQ,UAAU;IAC/C,SAAS,GAAG,KAAK,CAAC;IAClB,QAAQ,GAAgB,EAAE,CAAC;IAC3B,WAAW,GAAG,IAAI,GAAG,EAA0C,CAAC;IAEhE,OAAO,CACL,SAA4B,EAC5B,EACE,MAAM,EACN,SAAS,GAC2D;QAEtE,IAAI,CAAC,WAAW,CAAC,GAAG,CAClB,MAAM,CAAC,EAAE,aAAa,EAAE,SAAS,CAAC,IAAI,EAAE,SAAS,EAAE,CAAC,EACpD,MAAM,CACP,CAAC;IACJ,CAAC;IAED,QAAQ;QACN,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QAEtB,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC;YAC9C,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;IAED,OAAO,CACL,SAA+B,EAC/B,OAAmC;QAEnC,MAAM,0BAA0B,GAAG,GAAkC,EAAE;YACrE,MAAM,GAAG,GAAG,MAAM,CAAC,SAAS,CAAC,CAAC;YAC9B,MAAM,MAAM,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAEzC,IAAI,MAAM,EAAE,CAAC;gBACX,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;gBAC7B,OAAO,EAAE,CAAC,MAAM,CAAC,CAAC;YACpB,CAAC;YAED,OAAO,OAAO,CAAC,SAAS,CAAC,CAAC;QAC5B,CAAC,CAAC;QAEF,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,OAAO,0BAA0B,EAAE,CAAC;QACtC,CAAC;QAED,OAAO,IAAI,UAAU,CAAC,CAAC,QAAQ,EAAE,EAAE;YACjC,IAAI,MAAM,GAAG,IAAI,CAAC;YAElB,MAAM,OAAO,GAAG,GAAG,EAAE;gBACnB,IAAI,CAAC,MAAM;oBAAE,OAAO;gBACpB,0BAA0B,EAAE,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;YACnD,CAAC,CAAC;YACF,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAE5B,OAAO,GAAG,EAAE;gBACV,MAAM,GAAG,KAAK,CAAC;gBACf,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;gBAC3C,IAAI,GAAG,KAAK,CAAC,CAAC;oBAAE,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;YAC/C,CAAC,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC;CACF;AAED,SAAS,MAAM,CAAC,EACd,aAAa,EACb,SAAS,GAIV;IACC,OAAO,GAAG,aAAa,IAAI,kBAAkB,CAAC,SAAS,IAAI,EAAE,CAAC,EAAkB,CAAC;AACnF,CAAC","sourcesContent":["import { ApolloLink, Observable } from \"@apollo/client\";\nimport type { OperationVariables } from \"@apollo/client\";\nimport { canonicalStringify } from \"@apollo/client/utilities\";\nimport type { FormattedExecutionResult } from \"graphql\";\nimport { of } from \"rxjs\";\nimport type { ManifestOperation } from \"../types/application-manifest.js\";\n\ntype ResolveFn = () => void;\ntype OperationKey = string & { __type: \"PendingKey\" };\n\n/**\n * @internal\n * Holds requests until the client is fully hydrated after a tool call. It is\n * particularly useful for `network-only` and `cache-and-network` fetch policies\n * where the request makes it to the link chain because it prevents a followup\n * tool call to the MCP server for the just fetched data from the MCP server.\n */\nexport class ToolHydrationLink extends ApolloLink {\n  #hydrated = false;\n  #pending: ResolveFn[] = [];\n  #operations = new Map<OperationKey, FormattedExecutionResult>();\n\n  hydrate(\n    operation: ManifestOperation,\n    {\n      result,\n      variables,\n    }: { result: FormattedExecutionResult; variables: OperationVariables }\n  ) {\n    this.#operations.set(\n      getKey({ operationName: operation.name, variables }),\n      result\n    );\n  }\n\n  complete(): void {\n    this.#hydrated = true;\n\n    for (const resolve of this.#pending.splice(0)) {\n      resolve();\n    }\n  }\n\n  request(\n    operation: ApolloLink.Operation,\n    forward: ApolloLink.ForwardFunction\n  ): Observable<ApolloLink.Result> {\n    const maybeSendToTerminatingLink = (): Observable<ApolloLink.Result> => {\n      const key = getKey(operation);\n      const result = this.#operations.get(key);\n\n      if (result) {\n        this.#operations.delete(key);\n        return of(result);\n      }\n\n      return forward(operation);\n    };\n\n    if (this.#hydrated) {\n      return maybeSendToTerminatingLink();\n    }\n\n    return new Observable((observer) => {\n      let active = true;\n\n      const resolve = () => {\n        if (!active) return;\n        maybeSendToTerminatingLink().subscribe(observer);\n      };\n      this.#pending.push(resolve);\n\n      return () => {\n        active = false;\n        const idx = this.#pending.indexOf(resolve);\n        if (idx !== -1) this.#pending.splice(idx, 1);\n      };\n    });\n  }\n}\n\nfunction getKey({\n  operationName,\n  variables,\n}: {\n  operationName: string | undefined;\n  variables?: OperationVariables;\n}): OperationKey {\n  return `${operationName}:${canonicalStringify(variables ?? {})}` as OperationKey;\n}\n"]}

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

@@ -1,26 +1,9 @@
-import { ApolloClient as BaseApolloClient } from "@apollo/client";
-import type { WatchQueryOptions, ObservableQuery, OperationVariables } from "@apollo/client";
-import type { ApplicationManifest } from "../../types/application-manifest.js";
-import { aiClientSymbol } from "../../utilities/index.js";
+import { AbstractApolloClient } from "../../core/AbstractApolloClient.js";
 export declare namespace ApolloClient {
-    interface Options extends Omit<BaseApolloClient.Options, "link"> {
-        link?: BaseApolloClient.Options["link"];
-        manifest: ApplicationManifest;
+    interface Options extends AbstractApolloClient.Options {
     }
 }
-export declare class ApolloClient extends BaseApolloClient {
-    #private;
-    manifest: ApplicationManifest;
-    private readonly appManager;
-    /** @internal */
-    readonly [aiClientSymbol] = true;
+export declare class ApolloClient extends AbstractApolloClient {
     constructor(options: ApolloClient.Options);
-    stop(): void;
-    get toolInput(): Record<string, unknown> | undefined;
-    clearToolInput(): void;
-    watchQuery<T = any, TVariables extends OperationVariables = OperationVariables>(options: WatchQueryOptions<TVariables, T>): ObservableQuery<T, TVariables>;
-    connect: (() => Promise<void>) & {
-        reset: () => void;
-    };
 }
 //# sourceMappingURL=ApolloClient.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"ApolloClient.d.ts","sourceRoot":"","sources":["../../../src/mcp/core/ApolloClient.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,YAAY,IAAI,gBAAgB,EAEjC,MAAM,gBAAgB,CAAC;AACxB,OAAO,KAAK,EACV,iBAAiB,EACjB,eAAe,EACf,kBAAkB,EACnB,MAAM,gBAAgB,CAAC;AAKxB,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,qCAAqC,CAAC;AAE/E,OAAO,EACL,cAAc,EAMf,MAAM,0BAA0B,CAAC;AAGlC,MAAM,CAAC,OAAO,WAAW,YAAY,CAAC;IACpC,UAAiB,OAAQ,SAAQ,IAAI,CAAC,gBAAgB,CAAC,OAAO,EAAE,MAAM,CAAC;QACrE,IAAI,CAAC,EAAE,gBAAgB,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QACxC,QAAQ,EAAE,mBAAmB,CAAC;KAC/B;CACF;AAED,qBAAa,YAAa,SAAQ,gBAAgB;;IAChD,QAAQ,EAAE,mBAAmB,CAAC;IAC9B,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAgB;IAE3C,gBAAgB;IAChB,QAAQ,CAAC,CAAC,cAAc,CAAC,QAAQ;gBAIrB,OAAO,EAAE,YAAY,CAAC,OAAO;IA6BzC,IAAI;IAKJ,IAAI,SAAS,wCAEZ;IAED,cAAc;IAId,UAAU,CACR,CAAC,GAAG,GAAG,EACP,UAAU,SAAS,kBAAkB,GAAG,kBAAkB,EAC1D,OAAO,EAAE,iBAAiB,CAAC,UAAU,EAAE,CAAC,CAAC,GAAG,eAAe,CAAC,CAAC,EAAE,UAAU,CAAC;IAyC5E,OAAO;;MAsBJ;CACJ"}
+{"version":3,"file":"ApolloClient.d.ts","sourceRoot":"","sources":["../../../src/mcp/core/ApolloClient.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,oBAAoB,EAAE,MAAM,oCAAoC,CAAC;AAI1E,MAAM,CAAC,OAAO,WAAW,YAAY,CAAC;IACpC,UAAiB,OAAQ,SAAQ,oBAAoB,CAAC,OAAO;KAAG;CACjE;AAED,qBAAa,YAAa,SAAQ,oBAAoB;gBACxC,OAAO,EAAE,YAAY,CAAC,OAAO;CA6B1C"}

package/dist/mcp/core/ApolloClient.js

@@ -1,103 +1,25 @@
-import { ApolloLink, ApolloClient as BaseApolloClient, DocumentTransform, } from "@apollo/client";
-import { removeDirectivesFromDocument } from "@apollo/client/utilities/internal";
-import { parse, visit } from "graphql";
-import { equal } from "@wry/equality";
-import { __DEV__ } from "@apollo/client/utilities/environment";
-import { ToolCallLink } from "../link/ToolCallLink.js";
-import { aiClientSymbol, cacheAsync, getToolNamesFromDocument, getVariableNamesFromDocument, getVariablesForOperationFromToolInput, warnOnVariableMismatch, } from "../../utilities/index.js";
-import { McpAppManager } from "./McpAppManager.js";
-export class ApolloClient extends BaseApolloClient {
-    manifest;
-    appManager;
-    /** @internal */
-    [aiClientSymbol] = true;
-    #toolInput;
+import { AbstractApolloClient } from "../../core/AbstractApolloClient.js";
+import { connectToHost, promiseWithResolvers } from "../../utilities/index.js";
+export class ApolloClient extends AbstractApolloClient {
     constructor(options) {
-        const link = options.link ?? new ToolCallLink();
-        if (__DEV__) {
-            validateTerminatingLink(link);
-        }
-        super({
-            ...options,
-            link,
-            // Strip out the prefetch/tool directives so they don't get sent with the operation to the server
-            documentTransform: new DocumentTransform((document) => {
-                const serverDocument = removeDirectivesFromDocument([{ name: "prefetch" }, { name: "tool" }], document);
-                return visit(serverDocument, {
-                    OperationDefinition(node) {
-                        return { ...node, description: undefined };
-                    },
-                });
-            }).concat(options.documentTransform ?? DocumentTransform.identity()),
+        super(options, async (app) => {
+            const toolResult = promiseWithResolvers();
+            const toolInput = promiseWithResolvers();
+            app.ontoolresult = (params) => {
+                toolResult.resolve(params);
+            };
+            app.ontoolinput = (params) => {
+                toolInput.resolve(params);
+            };
+            await connectToHost(app);
+            const { structuredContent, _meta } = await toolResult.promise;
+            const { arguments: args } = await toolInput.promise;
+            return {
+                structuredContent,
+                toolInput: args,
+                _meta,
+            };
         });
-        this.manifest = options.manifest;
-        this.appManager = new McpAppManager(this.manifest);
     }
-    stop() {
-        super.stop();
-        this.appManager.close().catch(() => { });
-    }
-    get toolInput() {
-        return this.#toolInput;
-    }
-    clearToolInput() {
-        this.#toolInput = undefined;
-    }
-    watchQuery(options) {
-        if (__DEV__) {
-            const toolInput = this.#toolInput;
-            if (toolInput) {
-                const toolName = this.appManager.toolName;
-                const hasMatchingTool = !!toolName && getToolNamesFromDocument(options.query).has(toolName);
-                if (hasMatchingTool) {
-                    // Clear after first matching comparison so this only fires once and
-                    // remounting doesn't produce spurious warnings.
-                    this.#toolInput = undefined;
-                    const variableNames = getVariableNamesFromDocument(options.query);
-                    if (variableNames.size > 0) {
-                        const { variables } = options;
-                        const toolInputVariables = Object.entries(toolInput).filter(([key]) => variableNames.has(key));
-                        const hasToolInputMismatch = toolInputVariables.some(([key, value]) => !equal(value, variables?.[key]));
-                        if (hasToolInputMismatch) {
-                            warnOnVariableMismatch(options.query, Object.fromEntries(toolInputVariables), variables);
-                        }
-                    }
-                }
-            }
-        }
-        return super.watchQuery(options);
-    }
-    connect = cacheAsync(async () => {
-        const { prefetch, result, toolName, args } = await this.appManager.connect();
-        this.#toolInput = args;
-        this.manifest.operations.forEach((operation) => {
-            if (operation.prefetchID && prefetch?.[operation.prefetchID]) {
-                this.writeQuery({
-                    query: parse(operation.body),
-                    data: prefetch[operation.prefetchID].data,
-                });
-            }
-            if (operation.tools.find((tool) => tool.name === toolName)) {
-                this.writeQuery({
-                    query: parse(operation.body),
-                    data: result.data,
-                    variables: getVariablesForOperationFromToolInput(operation, args),
-                });
-            }
-        });
-    });
-}
-function validateTerminatingLink(link) {
-    let terminatingLink = link;
-    while (terminatingLink.right) {
-        terminatingLink = terminatingLink.right;
-    }
-    if (!isNamedLink(terminatingLink) ||
-        terminatingLink.name !== "ToolCallLink") {
-        throw new Error("The terminating link must be a `ToolCallLink`. If you are using a `split` link, ensure the `right` branch uses a `ToolCallLink` as the terminating link.");
-    }
-}
-function isNamedLink(link) {
-    return "name" in link;
 }
 //# sourceMappingURL=ApolloClient.js.map