integrate-sdk 0.9.56 → 0.9.61

package/dist/ai/index.js

@@ -4060,6 +4060,15 @@ async function tryGetProviderTokens(manualTokens) {
 }
 
 // utils.ts
+function toEnabledToolsAsyncOptions(options) {
+  if (!options)
+    return;
+  const { integrationIds, connectedOnly, context, fetchConcurrency } = options;
+  if (integrationIds === undefined && connectedOnly === undefined && context === undefined && fetchConcurrency === undefined) {
+    return;
+  }
+  return { integrationIds, connectedOnly, context, fetchConcurrency };
+}
 function getProviderForTool(client, toolName) {
   return client.getProviderForTool?.(toolName);
 }
@@ -5141,7 +5150,7 @@ async function handleAnthropicToolCalls(client, messageContent, options) {
   const getCodeModeTool = async () => {
     if (cachedCodeModeTool)
       return cachedCodeModeTool;
-    const mcpTools = await client.getEnabledToolsAsync();
+    const mcpTools = await client.getEnabledToolsAsync(toEnabledToolsAsyncOptions(options));
     cachedCodeModeTool = buildCodeModeTool(client, {
       tools: mcpTools,
       providerTokens: options?.providerTokens,
@@ -5192,7 +5201,7 @@ async function getAnthropicTools(client, options) {
   }
   const finalOptions = providerTokens ? { ...options, providerTokens } : options;
   await ensureClientConnected(client);
-  const mcpTools = await client.getEnabledToolsAsync();
+  const mcpTools = await client.getEnabledToolsAsync(toEnabledToolsAsyncOptions(options));
   let effectiveMode;
   if (options?.mode !== undefined) {
     effectiveMode = options.mode;
@@ -5359,7 +5368,7 @@ async function executeGoogleFunctionCalls(client, functionCalls, options) {
   const getCodeModeTool = async () => {
     if (cachedCodeModeTool)
       return cachedCodeModeTool;
-    const mcpTools = await client.getEnabledToolsAsync();
+    const mcpTools = await client.getEnabledToolsAsync(toEnabledToolsAsyncOptions(options));
     cachedCodeModeTool = buildCodeModeTool(client, {
       tools: mcpTools,
       providerTokens: finalOptions?.providerTokens,
@@ -5398,7 +5407,7 @@ async function getGoogleTools(client, options) {
   }
   const finalOptions = providerTokens ? { ...options, providerTokens } : options;
   await ensureClientConnected(client);
-  const mcpTools = await client.getEnabledToolsAsync();
+  const mcpTools = await client.getEnabledToolsAsync(toEnabledToolsAsyncOptions(options));
   let effectiveMode;
   if (options?.mode !== undefined) {
     effectiveMode = options.mode;
@@ -5525,6 +5534,69 @@ function convertJsonSchemaToGoogleSchema(jsonSchema, TypeEnum) {
   return result;
 }
 
+// ../database/token-store.ts
+var USABLE_ACCESS_TOKEN_BUFFER_MS = 30 * 1000;
+var MIN_MEANINGFUL_EXPIRY_MS = Date.UTC(2000, 0, 1);
+async function listConnectedProviders(configuredIntegrationIds, getProviderToken, context) {
+  if (!context.userId || configuredIntegrationIds.length === 0) {
+    return [];
+  }
+  const connected = [];
+  await Promise.all(configuredIntegrationIds.map(async (provider) => {
+    try {
+      const token = await getProviderToken(provider, undefined, context);
+      if (token?.accessToken || typeof token?.refreshToken === "string" && token.refreshToken.length > 0) {
+        connected.push(provider);
+      }
+    } catch {}
+  }));
+  return connected.sort();
+}
+
+// tool-cache.ts
+function buildToolDiscoveryCacheKey(userId, connectedIntegrationIds) {
+  const sorted = [...connectedIntegrationIds].sort().join(",");
+  return `${userId}:${sorted}`;
+}
+async function resolveToolDiscoveryCacheKey(client, context, options) {
+  const userId = context.userId;
+  if (!userId)
+    return;
+  const configuredIds = (client.integrations ?? []).map((i) => i.id);
+  let targetIds;
+  if (options?.integrationIds?.length) {
+    targetIds = options.integrationIds;
+  } else if (options?.connectedOnly) {
+    targetIds = await listConnectedProviders(configuredIds, (provider, email, ctx) => client.getProviderToken(provider, email, ctx), context);
+  } else {
+    targetIds = configuredIds;
+  }
+  return buildToolDiscoveryCacheKey(userId, targetIds);
+}
+function stubsFromTools(tools) {
+  return tools.map((t) => ({
+    name: t.name,
+    description: t.description,
+    inputSchema: t.inputSchema
+  }));
+}
+function applyToolDiscoveryCache(client, stubs) {
+  client.hydrateToolCache(stubs);
+}
+async function persistToolDiscoveryCache(client, cache, cacheKey, ttlMs) {
+  const tools = client.getAvailableTools();
+  if (tools.length === 0)
+    return;
+  await cache.set(cacheKey, stubsFromTools(tools), ttlMs);
+}
+async function warmToolDiscoveryFromCache(client, cache, cacheKey) {
+  const stubs = await cache.get(cacheKey);
+  if (!stubs || stubs.length === 0)
+    return false;
+  applyToolDiscoveryCache(client, stubs);
+  return true;
+}
+
 // vercel-ai.ts
 function convertMCPToolToVercelAI(mcpTool, client, options) {
   return {
@@ -5547,7 +5619,21 @@ async function getVercelAITools(client, options) {
   }
   const finalOptions = providerTokens ? { ...options, providerTokens } : options;
   await ensureClientConnected(client);
-  const mcpTools = await client.getEnabledToolsAsync();
+  const cacheConfig = options?.cache;
+  let cacheKey;
+  if (cacheConfig?.cache && options?.context) {
+    cacheKey = await resolveToolDiscoveryCacheKey(client, options.context, {
+      integrationIds: options.integrationIds,
+      connectedOnly: options.connectedOnly
+    });
+    if (cacheKey) {
+      await warmToolDiscoveryFromCache(client, cacheConfig.cache, cacheKey);
+    }
+  }
+  const mcpTools = await client.getEnabledToolsAsync(toEnabledToolsAsyncOptions(options));
+  if (cacheConfig?.cache && cacheKey) {
+    await persistToolDiscoveryCache(client, cacheConfig.cache, cacheKey, cacheConfig.ttlMs);
+  }
   const vercelTools = {};
   let effectiveMode;
   if (options?.mode !== undefined) {
@@ -5616,7 +5702,7 @@ async function getOpenAITools(client, options) {
   }
   const finalOptions = providerTokens ? { ...options, providerTokens } : options;
   await ensureClientConnected(client);
-  const mcpTools = await client.getEnabledToolsAsync();
+  const mcpTools = await client.getEnabledToolsAsync(toEnabledToolsAsyncOptions(options));
   let effectiveMode;
   if (options?.mode !== undefined) {
     effectiveMode = options.mode;
@@ -5678,7 +5764,7 @@ async function handleOpenAIToolCalls(client, toolCalls, options) {
   const getCodeModeTool = async () => {
     if (cachedCodeModeTool)
       return cachedCodeModeTool;
-    const mcpTools = await client.getEnabledToolsAsync();
+    const mcpTools = await client.getEnabledToolsAsync(toEnabledToolsAsyncOptions(options));
     cachedCodeModeTool = buildCodeModeTool(client, {
       tools: mcpTools,
       providerTokens: options?.providerTokens,

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

@@ -1 +1 @@
-{"version":3,"file":"openai.d.ts","sourceRoot":"","sources":["../../../src/ai/openai.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AAE9C,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AACrD,OAAO,EAAkE,KAAK,cAAc,EAAE,MAAM,YAAY,CAAC;AAUjH,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAErC;;;GAGG;AACH,MAAM,WAAW,UAAU;IACzB,IAAI,EAAE,UAAU,CAAC;IACjB,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,EAAE;QACV,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;KACxB,GAAG,IAAI,CAAC;IACT,MAAM,EAAE,OAAO,GAAG,IAAI,CAAC;IACvB,WAAW,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;CAC7B;AAED;;GAEG;AACH,MAAM,WAAW,kBAAmB,SAAQ,cAAc;IACxD;;;OAGG;IACH,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,kDAAkD;IAClD,OAAO,CAAC,EAAE,UAAU,CAAC;IACrB;;;;;;;;;OASG;IACH,IAAI,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC;CACzB;AAiCD;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAsCG;AACH,wBAAsB,cAAc,CAClC,MAAM,EAAE,SAAS,CAAC,GAAG,CAAC,EACtB,OAAO,CAAC,EAAE,kBAAkB,GAC3B,OAAO,CAAC,UAAU,EAAE,CAAC,CAiFvB;AA6GD;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA0DG;AACH,wBAAsB,oBAAoB,CACxC,MAAM,EAAE,SAAS,CAAC,GAAG,CAAC,EACtB,QAAQ,EAAE;IAAE,MAAM,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAA,CAAC,GAAG,EAAE,MAAM,GAAG,GAAG,CAAA;KAAE,CAAC,CAAA;CAAE,EAChE,OAAO,CAAC,EAAE,kBAAkB,GAC3B,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,iBAAiB,CAAC,kBAAkB,EAAE,GAAG;IAAE,MAAM,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAA,CAAC,GAAG,EAAE,MAAM,GAAG,GAAG,CAAA;KAAE,CAAC,CAAA;CAAE,CAAC,CAyB3H"}
+{"version":3,"file":"openai.d.ts","sourceRoot":"","sources":["../../../src/ai/openai.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AAE9C,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AACrD,OAAO,EAAkE,KAAK,cAAc,EAA8B,MAAM,YAAY,CAAC;AAU7I,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAErC;;;GAGG;AACH,MAAM,WAAW,UAAU;IACzB,IAAI,EAAE,UAAU,CAAC;IACjB,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,EAAE;QACV,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;KACxB,GAAG,IAAI,CAAC;IACT,MAAM,EAAE,OAAO,GAAG,IAAI,CAAC;IACvB,WAAW,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;CAC7B;AAED;;GAEG;AACH,MAAM,WAAW,kBAAmB,SAAQ,cAAc;IACxD;;;OAGG;IACH,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,kDAAkD;IAClD,OAAO,CAAC,EAAE,UAAU,CAAC;IACrB;;;;;;;;;OASG;IACH,IAAI,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC;CACzB;AAiCD;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAsCG;AACH,wBAAsB,cAAc,CAClC,MAAM,EAAE,SAAS,CAAC,GAAG,CAAC,EACtB,OAAO,CAAC,EAAE,kBAAkB,GAC3B,OAAO,CAAC,UAAU,EAAE,CAAC,CAiFvB;AA6GD;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA0DG;AACH,wBAAsB,oBAAoB,CACxC,MAAM,EAAE,SAAS,CAAC,GAAG,CAAC,EACtB,QAAQ,EAAE;IAAE,MAAM,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAA,CAAC,GAAG,EAAE,MAAM,GAAG,GAAG,CAAA;KAAE,CAAC,CAAA;CAAE,EAChE,OAAO,CAAC,EAAE,kBAAkB,GAC3B,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,iBAAiB,CAAC,kBAAkB,EAAE,GAAG;IAAE,MAAM,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAA,CAAC,GAAG,EAAE,MAAM,GAAG,GAAG,CAAA;KAAE,CAAC,CAAA;CAAE,CAAC,CAyB3H"}

package/dist/ai/openai.js

@@ -4060,6 +4060,15 @@ async function tryGetProviderTokens(manualTokens) {
 }
 
 // utils.ts
+function toEnabledToolsAsyncOptions(options) {
+  if (!options)
+    return;
+  const { integrationIds, connectedOnly, context, fetchConcurrency } = options;
+  if (integrationIds === undefined && connectedOnly === undefined && context === undefined && fetchConcurrency === undefined) {
+    return;
+  }
+  return { integrationIds, connectedOnly, context, fetchConcurrency };
+}
 function getProviderForTool(client, toolName) {
   return client.getProviderForTool?.(toolName);
 }
@@ -5141,7 +5150,7 @@ async function getOpenAITools(client, options) {
   }
   const finalOptions = providerTokens ? { ...options, providerTokens } : options;
   await ensureClientConnected(client);
-  const mcpTools = await client.getEnabledToolsAsync();
+  const mcpTools = await client.getEnabledToolsAsync(toEnabledToolsAsyncOptions(options));
   let effectiveMode;
   if (options?.mode !== undefined) {
     effectiveMode = options.mode;
@@ -5203,7 +5212,7 @@ async function handleOpenAIToolCalls(client, toolCalls, options) {
   const getCodeModeTool = async () => {
     if (cachedCodeModeTool)
       return cachedCodeModeTool;
-    const mcpTools = await client.getEnabledToolsAsync();
+    const mcpTools = await client.getEnabledToolsAsync(toEnabledToolsAsyncOptions(options));
     cachedCodeModeTool = buildCodeModeTool(client, {
       tools: mcpTools,
       providerTokens: options?.providerTokens,

package/dist/ai/tool-cache.d.ts

@@ -0,0 +1,50 @@
+import type { MCPClient } from "../client.js";
+import type { MCPTool } from "../protocol/messages.js";
+import type { MCPContext } from "../config/types.js";
+/** Serializable MCP tool metadata for cross-request / Redis caching. */
+export type ToolMetadataStub = Pick<MCPTool, "name" | "description" | "inputSchema">;
+export interface ToolDiscoveryCacheAdapter {
+    get(key: string): Promise<ToolMetadataStub[] | null>;
+    set(key: string, stubs: ToolMetadataStub[], ttlMs?: number): Promise<void>;
+    /** Invalidate all keys for a user (prefix match) or a single key when exact. */
+    invalidate(userIdOrKey: string): Promise<void>;
+}
+export interface ToolDiscoveryCacheOptions {
+    cache: ToolDiscoveryCacheAdapter;
+    /** @default 5 minutes */
+    ttlMs?: number;
+}
+/**
+ * Build a cache key from user id and connected integration ids.
+ * Include connected providers so reconnecting OAuth invalidates stale tool lists.
+ */
+export declare function buildToolDiscoveryCacheKey(userId: string, connectedIntegrationIds: readonly string[]): string;
+/**
+ * In-memory tool discovery cache (single process). Use Redis in production serverless.
+ */
+export declare function createMemoryToolDiscoveryCache(): ToolDiscoveryCacheAdapter;
+/**
+ * Hook for database adapter `onTokenChange` — invalidates cached tool metadata for the user.
+ */
+export declare function createToolDiscoveryCacheInvalidator(cache: ToolDiscoveryCacheAdapter): (event: {
+    userId: string;
+}) => void;
+export declare function resolveToolDiscoveryCacheKey(client: MCPClient<any>, context: MCPContext, options?: {
+    integrationIds?: string[];
+    connectedOnly?: boolean;
+}): Promise<string | undefined>;
+export declare function stubsFromTools(tools: readonly MCPTool[]): ToolMetadataStub[];
+/**
+ * Hydrate the client tool cache from persisted stubs (serverless cold start).
+ */
+export declare function applyToolDiscoveryCache(client: MCPClient<any>, stubs: readonly ToolMetadataStub[]): void;
+/**
+ * Persist current client tool metadata to a cache adapter.
+ */
+export declare function persistToolDiscoveryCache(client: MCPClient<any>, cache: ToolDiscoveryCacheAdapter, cacheKey: string, ttlMs?: number): Promise<void>;
+/**
+ * Load stubs from cache into the client before tool discovery.
+ * @returns true when cache was hydrated
+ */
+export declare function warmToolDiscoveryFromCache(client: MCPClient<any>, cache: ToolDiscoveryCacheAdapter, cacheKey: string): Promise<boolean>;
+//# sourceMappingURL=tool-cache.d.ts.map

package/dist/ai/tool-cache.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-cache.d.ts","sourceRoot":"","sources":["../../../src/ai/tool-cache.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AAC9C,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,yBAAyB,CAAC;AACvD,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAGrD,wEAAwE;AACxE,MAAM,MAAM,gBAAgB,GAAG,IAAI,CAAC,OAAO,EAAE,MAAM,GAAG,aAAa,GAAG,aAAa,CAAC,CAAC;AAErF,MAAM,WAAW,yBAAyB;IACxC,GAAG,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,EAAE,GAAG,IAAI,CAAC,CAAC;IACrD,GAAG,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,gBAAgB,EAAE,EAAE,KAAK,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC3E,gFAAgF;IAChF,UAAU,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;CAChD;AAED,MAAM,WAAW,yBAAyB;IACxC,KAAK,EAAE,yBAAyB,CAAC;IACjC,yBAAyB;IACzB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED;;;GAGG;AACH,wBAAgB,0BAA0B,CACxC,MAAM,EAAE,MAAM,EACd,uBAAuB,EAAE,SAAS,MAAM,EAAE,GACzC,MAAM,CAGR;AAED;;GAEG;AACH,wBAAgB,8BAA8B,IAAI,yBAAyB,CAwB1E;AAED;;GAEG;AACH,wBAAgB,mCAAmC,CACjD,KAAK,EAAE,yBAAyB,GAC/B,CAAC,KAAK,EAAE;IAAE,MAAM,EAAE,MAAM,CAAA;CAAE,KAAK,IAAI,CAIrC;AAED,wBAAsB,4BAA4B,CAChD,MAAM,EAAE,SAAS,CAAC,GAAG,CAAC,EACtB,OAAO,EAAE,UAAU,EACnB,OAAO,CAAC,EAAE;IACR,cAAc,CAAC,EAAE,MAAM,EAAE,CAAC;IAC1B,aAAa,CAAC,EAAE,OAAO,CAAC;CACzB,GACA,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC,CAuB7B;AAED,wBAAgB,cAAc,CAAC,KAAK,EAAE,SAAS,OAAO,EAAE,GAAG,gBAAgB,EAAE,CAM5E;AAED;;GAEG;AACH,wBAAgB,uBAAuB,CACrC,MAAM,EAAE,SAAS,CAAC,GAAG,CAAC,EACtB,KAAK,EAAE,SAAS,gBAAgB,EAAE,GACjC,IAAI,CAEN;AAED;;GAEG;AACH,wBAAsB,yBAAyB,CAC7C,MAAM,EAAE,SAAS,CAAC,GAAG,CAAC,EACtB,KAAK,EAAE,yBAAyB,EAChC,QAAQ,EAAE,MAAM,EAChB,KAAK,CAAC,EAAE,MAAM,GACb,OAAO,CAAC,IAAI,CAAC,CAIf;AAED;;;GAGG;AACH,wBAAsB,0BAA0B,CAC9C,MAAM,EAAE,SAAS,CAAC,GAAG,CAAC,EACtB,KAAK,EAAE,yBAAyB,EAChC,QAAQ,EAAE,MAAM,GACf,OAAO,CAAC,OAAO,CAAC,CAKlB"}

package/dist/ai/utils.d.ts

@@ -5,6 +5,8 @@
  */
 import { z } from "zod";
 import type { MCPClient } from "../client.js";
+import type { MCPContext } from "../config/types.js";
+import type { EnabledToolsAsyncOptions } from "../config/types.js";
 /**
  * Options for AI provider tool conversions
  */
@@ -38,7 +40,31 @@ export interface AIToolsOptions {
      * ```
      */
     providerTokens?: Record<string, string>;
+    /**
+     * Limit tool discovery to specific integration IDs (e.g. `['github', 'gmail']`).
+     * Only `list_tools_by_integration` calls for these integrations are made.
+     */
+    integrationIds?: string[];
+    /**
+     * When true with `context.userId`, only fetch tools for integrations the user
+     * has OAuth tokens for (via `getProviderToken` / database adapter).
+     */
+    connectedOnly?: boolean;
+    /** User context for connected-only filtering and multi-tenant token lookup */
+    context?: MCPContext;
+    /**
+     * Max parallel `list_tools_by_integration` requests during cold discovery.
+     * @default 8
+     */
+    fetchConcurrency?: number;
+    /**
+     * Optional cache for tool metadata (serverless cold-start optimization).
+     * Pair with {@link createToolDiscoveryCacheInvalidator} on `onTokenChange`.
+     */
+    cache?: import("./tool-cache.js").ToolDiscoveryCacheOptions;
 }
+/** Map AI helper options to client tool-discovery options */
+export declare function toEnabledToolsAsyncOptions(options?: Pick<AIToolsOptions, "integrationIds" | "connectedOnly" | "context" | "fetchConcurrency">): EnabledToolsAsyncOptions | undefined;
 /**
  * Get the provider for a tool by checking which integration includes it
  * @internal

package/dist/ai/utils.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../../src/ai/utils.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AAE9C;;GAEG;AACH,MAAM,WAAW,cAAc;IAC7B;;;;;;;;;;;;;;;;;;;;;;;;;;;OA2BG;IACH,cAAc,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;CACzC;AAED;;;GAGG;AACH,wBAAgB,kBAAkB,CAAC,MAAM,EAAE,SAAS,CAAC,GAAG,CAAC,EAAE,QAAQ,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS,CAG/F;AAED;;;GAGG;AACH,wBAAgB,uBAAuB,CAAC,UAAU,EAAE,GAAG,GAAG,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,CAqFvE;AAED;;;;;GAKG;AACH,wBAAgB,eAAe,CAAC,MAAM,EAAE,GAAG,GAAG,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,CAoD7D;AAED;;;;GAIG;AACH,wBAAsB,oBAAoB,CACxC,MAAM,EAAE,SAAS,CAAC,GAAG,CAAC,EACtB,QAAQ,EAAE,MAAM,EAChB,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAC7B,OAAO,CAAC,EAAE,cAAc,GACvB,OAAO,CAAC,GAAG,CAAC,CAuCd;AAED;;;GAGG;AACH,wBAAsB,qBAAqB,CAAC,MAAM,EAAE,SAAS,CAAC,GAAG,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC,CAIjF;AAED;;;GAGG;AACH,OAAO,EAAE,iBAAiB,EAAE,oBAAoB,EAAE,MAAM,4BAA4B,CAAC"}
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../../src/ai/utils.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AAC9C,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AACrD,OAAO,KAAK,EAAE,wBAAwB,EAAE,MAAM,oBAAoB,CAAC;AAEnE;;GAEG;AACH,MAAM,WAAW,cAAc;IAC7B;;;;;;;;;;;;;;;;;;;;;;;;;;;OA2BG;IACH,cAAc,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACxC;;;OAGG;IACH,cAAc,CAAC,EAAE,MAAM,EAAE,CAAC;IAC1B;;;OAGG;IACH,aAAa,CAAC,EAAE,OAAO,CAAC;IACxB,8EAA8E;IAC9E,OAAO,CAAC,EAAE,UAAU,CAAC;IACrB;;;OAGG;IACH,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B;;;OAGG;IACH,KAAK,CAAC,EAAE,OAAO,iBAAiB,EAAE,yBAAyB,CAAC;CAC7D;AAED,6DAA6D;AAC7D,wBAAgB,0BAA0B,CACxC,OAAO,CAAC,EAAE,IAAI,CACZ,cAAc,EACd,gBAAgB,GAAG,eAAe,GAAG,SAAS,GAAG,kBAAkB,CACpE,GACA,wBAAwB,GAAG,SAAS,CAYtC;AAED;;;GAGG;AACH,wBAAgB,kBAAkB,CAAC,MAAM,EAAE,SAAS,CAAC,GAAG,CAAC,EAAE,QAAQ,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS,CAG/F;AAED;;;GAGG;AACH,wBAAgB,uBAAuB,CAAC,UAAU,EAAE,GAAG,GAAG,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,CAqFvE;AAED;;;;;GAKG;AACH,wBAAgB,eAAe,CAAC,MAAM,EAAE,GAAG,GAAG,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,CAoD7D;AAED;;;;GAIG;AACH,wBAAsB,oBAAoB,CACxC,MAAM,EAAE,SAAS,CAAC,GAAG,CAAC,EACtB,QAAQ,EAAE,MAAM,EAChB,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAC7B,OAAO,CAAC,EAAE,cAAc,GACvB,OAAO,CAAC,GAAG,CAAC,CAuCd;AAED;;;GAGG;AACH,wBAAsB,qBAAqB,CAAC,MAAM,EAAE,SAAS,CAAC,GAAG,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC,CAIjF;AAED;;;GAGG;AACH,OAAO,EAAE,iBAAiB,EAAE,oBAAoB,EAAE,MAAM,4BAA4B,CAAC"}

package/dist/ai/utils.js

@@ -4060,6 +4060,15 @@ async function tryGetProviderTokens(manualTokens) {
 }
 
 // utils.ts
+function toEnabledToolsAsyncOptions(options) {
+  if (!options)
+    return;
+  const { integrationIds, connectedOnly, context, fetchConcurrency } = options;
+  if (integrationIds === undefined && connectedOnly === undefined && context === undefined && fetchConcurrency === undefined) {
+    return;
+  }
+  return { integrationIds, connectedOnly, context, fetchConcurrency };
+}
 function getProviderForTool(client, toolName) {
   return client.getProviderForTool?.(toolName);
 }
@@ -4208,6 +4217,7 @@ async function ensureClientConnected(client) {
 }
 export {
   tryGetProviderTokens,
+  toEnabledToolsAsyncOptions,
   jsonSchemaToZod,
   jsonSchemaPropertyToZod,
   getProviderTokens,

package/dist/ai/vercel-ai.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"vercel-ai.d.ts","sourceRoot":"","sources":["../../../src/ai/vercel-ai.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AAE9C,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AACrD,OAAO,EAKL,KAAK,cAAc,EACpB,MAAM,YAAY,CAAC;AAUpB;;;GAGG;AACH,MAAM,WAAW,YAAY;IAC3B,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,WAAW,EAAE,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;IAC5B,OAAO,EAAE,CAAC,IAAI,EAAE,GAAG,EAAE,OAAO,CAAC,EAAE,GAAG,KAAK,OAAO,CAAC,GAAG,CAAC,CAAC;CACrD;AAED;;GAEG;AACH,MAAM,WAAW,oBAAqB,SAAQ,cAAc;IAC1D,kDAAkD;IAClD,OAAO,CAAC,EAAE,UAAU,CAAC;IACrB;;;;;;;;;;;OAWG;IACH,IAAI,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC;CACzB;AA+BD;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAmDG;AACH,wBAAsB,gBAAgB,CACpC,MAAM,EAAE,SAAS,CAAC,GAAG,CAAC,EACtB,OAAO,CAAC,EAAE,oBAAoB,gCAyE/B"}
+{"version":3,"file":"vercel-ai.d.ts","sourceRoot":"","sources":["../../../src/ai/vercel-ai.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AAE9C,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AACrD,OAAO,EAML,KAAK,cAAc,EACpB,MAAM,YAAY,CAAC;AAepB;;;GAGG;AACH,MAAM,WAAW,YAAY;IAC3B,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,WAAW,EAAE,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;IAC5B,OAAO,EAAE,CAAC,IAAI,EAAE,GAAG,EAAE,OAAO,CAAC,EAAE,GAAG,KAAK,OAAO,CAAC,GAAG,CAAC,CAAC;CACrD;AAED;;GAEG;AACH,MAAM,WAAW,oBAAqB,SAAQ,cAAc;IAC1D,kDAAkD;IAClD,OAAO,CAAC,EAAE,UAAU,CAAC;IACrB;;;;;;;;;;;OAWG;IACH,IAAI,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC;CACzB;AA+BD;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAmDG;AACH,wBAAsB,gBAAgB,CACpC,MAAM,EAAE,SAAS,CAAC,GAAG,CAAC,EACtB,OAAO,CAAC,EAAE,oBAAoB,gCA+F/B"}

package/dist/ai/vercel-ai.js

@@ -4060,6 +4060,15 @@ async function tryGetProviderTokens(manualTokens) {
 }
 
 // utils.ts
+function toEnabledToolsAsyncOptions(options) {
+  if (!options)
+    return;
+  const { integrationIds, connectedOnly, context, fetchConcurrency } = options;
+  if (integrationIds === undefined && connectedOnly === undefined && context === undefined && fetchConcurrency === undefined) {
+    return;
+  }
+  return { integrationIds, connectedOnly, context, fetchConcurrency };
+}
 function getProviderForTool(client, toolName) {
   return client.getProviderForTool?.(toolName);
 }
@@ -5120,6 +5129,69 @@ ${generated.compact}`;
   return { codeTool, typesTool };
 }
 
+// ../database/token-store.ts
+var USABLE_ACCESS_TOKEN_BUFFER_MS = 30 * 1000;
+var MIN_MEANINGFUL_EXPIRY_MS = Date.UTC(2000, 0, 1);
+async function listConnectedProviders(configuredIntegrationIds, getProviderToken, context) {
+  if (!context.userId || configuredIntegrationIds.length === 0) {
+    return [];
+  }
+  const connected = [];
+  await Promise.all(configuredIntegrationIds.map(async (provider) => {
+    try {
+      const token = await getProviderToken(provider, undefined, context);
+      if (token?.accessToken || typeof token?.refreshToken === "string" && token.refreshToken.length > 0) {
+        connected.push(provider);
+      }
+    } catch {}
+  }));
+  return connected.sort();
+}
+
+// tool-cache.ts
+function buildToolDiscoveryCacheKey(userId, connectedIntegrationIds) {
+  const sorted = [...connectedIntegrationIds].sort().join(",");
+  return `${userId}:${sorted}`;
+}
+async function resolveToolDiscoveryCacheKey(client, context, options) {
+  const userId = context.userId;
+  if (!userId)
+    return;
+  const configuredIds = (client.integrations ?? []).map((i) => i.id);
+  let targetIds;
+  if (options?.integrationIds?.length) {
+    targetIds = options.integrationIds;
+  } else if (options?.connectedOnly) {
+    targetIds = await listConnectedProviders(configuredIds, (provider, email, ctx) => client.getProviderToken(provider, email, ctx), context);
+  } else {
+    targetIds = configuredIds;
+  }
+  return buildToolDiscoveryCacheKey(userId, targetIds);
+}
+function stubsFromTools(tools) {
+  return tools.map((t) => ({
+    name: t.name,
+    description: t.description,
+    inputSchema: t.inputSchema
+  }));
+}
+function applyToolDiscoveryCache(client, stubs) {
+  client.hydrateToolCache(stubs);
+}
+async function persistToolDiscoveryCache(client, cache, cacheKey, ttlMs) {
+  const tools = client.getAvailableTools();
+  if (tools.length === 0)
+    return;
+  await cache.set(cacheKey, stubsFromTools(tools), ttlMs);
+}
+async function warmToolDiscoveryFromCache(client, cache, cacheKey) {
+  const stubs = await cache.get(cacheKey);
+  if (!stubs || stubs.length === 0)
+    return false;
+  applyToolDiscoveryCache(client, stubs);
+  return true;
+}
+
 // vercel-ai.ts
 function convertMCPToolToVercelAI(mcpTool, client, options) {
   return {
@@ -5142,7 +5214,21 @@ async function getVercelAITools(client, options) {
   }
   const finalOptions = providerTokens ? { ...options, providerTokens } : options;
   await ensureClientConnected(client);
-  const mcpTools = await client.getEnabledToolsAsync();
+  const cacheConfig = options?.cache;
+  let cacheKey;
+  if (cacheConfig?.cache && options?.context) {
+    cacheKey = await resolveToolDiscoveryCacheKey(client, options.context, {
+      integrationIds: options.integrationIds,
+      connectedOnly: options.connectedOnly
+    });
+    if (cacheKey) {
+      await warmToolDiscoveryFromCache(client, cacheConfig.cache, cacheKey);
+    }
+  }
+  const mcpTools = await client.getEnabledToolsAsync(toEnabledToolsAsyncOptions(options));
+  if (cacheConfig?.cache && cacheKey) {
+    await persistToolDiscoveryCache(client, cacheConfig.cache, cacheKey, cacheConfig.ttlMs);
+  }
   const vercelTools = {};
   let effectiveMode;
   if (options?.mode !== undefined) {

package/dist/database/index.d.ts

@@ -1,6 +1,6 @@
 export type { AccountIdentity, CreateTriggerInput, DatabaseDriver, DeleteDuplicateProviderTokensInput, DeleteProviderTokensInput, FlattenedTrigger, IntegrateAdapterHooks, IntegrateDatabaseAdapter, IntegrateDatabaseCallbacks, ListTriggersQuery, ProviderTokenRecord, TokenChangeEvent, TokenDatabaseDriver, TriggerDatabaseDriver, TriggerRecord, UpsertProviderTokenInput, } from "./types.js";
 export { createDatabaseAdapterCallbacks, createDatabaseAdapterFactory, } from "./factory.js";
-export { defaultResolveAccountIdentity, hasMeaningfulExpiresAt, hasUsableAccessToken, isLikelyUsableToken, normalizeAccountEmail, normalizeAccountEmailHint, normalizeAccountIdHint, normalizeAccountIdentifier, normalizeProviderTokenType, parseScopes, providerTokenRecordToData, selectProviderTokenRow, } from "./token-store.js";
+export { defaultResolveAccountIdentity, hasMeaningfulExpiresAt, hasUsableAccessToken, isLikelyUsableToken, normalizeAccountEmail, normalizeAccountEmailHint, normalizeAccountIdHint, normalizeAccountIdentifier, normalizeProviderTokenType, parseScopes, providerTokenRecordToData, selectProviderTokenRow, listConnectedProviders, listConnectedProvidersFromRows, } from "./token-store.js";
 export { flattenedTriggerToCreateInput, toDbSchedule, toDbTriggerUpdates, toSdkSchedule, toSdkTrigger, } from "./trigger-store.js";
 export { drizzleAdapter, drizzleAdapterCallbacks, createDrizzleDatabaseDriver, type DrizzleAdapterConfig, type DrizzleIntegrateSchema, type DrizzleProvider, } from "./adapters/drizzle.js";
 export { prismaAdapter, prismaAdapterCallbacks, createPrismaDatabaseDriver, type PrismaAdapterConfig, type PrismaIntegrateModelNames, type PrismaProvider, } from "./adapters/prisma.js";

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

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/database/index.ts"],"names":[],"mappings":"AAAA,YAAY,EACV,eAAe,EACf,kBAAkB,EAClB,cAAc,EACd,kCAAkC,EAClC,yBAAyB,EACzB,gBAAgB,EAChB,qBAAqB,EACrB,wBAAwB,EACxB,0BAA0B,EAC1B,iBAAiB,EACjB,mBAAmB,EACnB,gBAAgB,EAChB,mBAAmB,EACnB,qBAAqB,EACrB,aAAa,EACb,wBAAwB,GACzB,MAAM,YAAY,CAAC;AAEpB,OAAO,EACL,8BAA8B,EAC9B,4BAA4B,GAC7B,MAAM,cAAc,CAAC;AAEtB,OAAO,EACL,6BAA6B,EAC7B,sBAAsB,EACtB,oBAAoB,EACpB,mBAAmB,EACnB,qBAAqB,EACrB,yBAAyB,EACzB,sBAAsB,EACtB,0BAA0B,EAC1B,0BAA0B,EAC1B,WAAW,EACX,yBAAyB,EACzB,sBAAsB,GACvB,MAAM,kBAAkB,CAAC;AAE1B,OAAO,EACL,6BAA6B,EAC7B,YAAY,EACZ,kBAAkB,EAClB,aAAa,EACb,YAAY,GACb,MAAM,oBAAoB,CAAC;AAE5B,OAAO,EACL,cAAc,EACd,uBAAuB,EACvB,2BAA2B,EAC3B,KAAK,oBAAoB,EACzB,KAAK,sBAAsB,EAC3B,KAAK,eAAe,GACrB,MAAM,uBAAuB,CAAC;AAE/B,OAAO,EACL,aAAa,EACb,sBAAsB,EACtB,0BAA0B,EAC1B,KAAK,mBAAmB,EACxB,KAAK,yBAAyB,EAC9B,KAAK,cAAc,GACpB,MAAM,sBAAsB,CAAC;AAE9B,OAAO,EACL,cAAc,EACd,uBAAuB,EACvB,yBAAyB,EACzB,KAAK,kBAAkB,EACvB,KAAK,oBAAoB,GAC1B,MAAM,uBAAuB,CAAC;AAE/B,OAAO,EACL,sBAAsB,EACtB,gBAAgB,GACjB,MAAM,sBAAsB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/database/index.ts"],"names":[],"mappings":"AAAA,YAAY,EACV,eAAe,EACf,kBAAkB,EAClB,cAAc,EACd,kCAAkC,EAClC,yBAAyB,EACzB,gBAAgB,EAChB,qBAAqB,EACrB,wBAAwB,EACxB,0BAA0B,EAC1B,iBAAiB,EACjB,mBAAmB,EACnB,gBAAgB,EAChB,mBAAmB,EACnB,qBAAqB,EACrB,aAAa,EACb,wBAAwB,GACzB,MAAM,YAAY,CAAC;AAEpB,OAAO,EACL,8BAA8B,EAC9B,4BAA4B,GAC7B,MAAM,cAAc,CAAC;AAEtB,OAAO,EACL,6BAA6B,EAC7B,sBAAsB,EACtB,oBAAoB,EACpB,mBAAmB,EACnB,qBAAqB,EACrB,yBAAyB,EACzB,sBAAsB,EACtB,0BAA0B,EAC1B,0BAA0B,EAC1B,WAAW,EACX,yBAAyB,EACzB,sBAAsB,EACtB,sBAAsB,EACtB,8BAA8B,GAC/B,MAAM,kBAAkB,CAAC;AAE1B,OAAO,EACL,6BAA6B,EAC7B,YAAY,EACZ,kBAAkB,EAClB,aAAa,EACb,YAAY,GACb,MAAM,oBAAoB,CAAC;AAE5B,OAAO,EACL,cAAc,EACd,uBAAuB,EACvB,2BAA2B,EAC3B,KAAK,oBAAoB,EACzB,KAAK,sBAAsB,EAC3B,KAAK,eAAe,GACrB,MAAM,uBAAuB,CAAC;AAE/B,OAAO,EACL,aAAa,EACb,sBAAsB,EACtB,0BAA0B,EAC1B,KAAK,mBAAmB,EACxB,KAAK,yBAAyB,EAC9B,KAAK,cAAc,GACpB,MAAM,sBAAsB,CAAC;AAE9B,OAAO,EACL,cAAc,EACd,uBAAuB,EACvB,yBAAyB,EACzB,KAAK,kBAAkB,EACvB,KAAK,oBAAoB,GAC1B,MAAM,uBAAuB,CAAC;AAE/B,OAAO,EACL,sBAAsB,EACtB,gBAAgB,GACjB,MAAM,sBAAsB,CAAC"}

package/dist/database/index.js

@@ -113,6 +113,43 @@ function providerTokenRecordToData(row) {
     accountId: row.accountId ?? undefined
   };
 }
+function listConnectedProvidersFromRows(rows, configuredIntegrationIds) {
+  const allowed = configuredIntegrationIds ? new Set(configuredIntegrationIds) : undefined;
+  const rowsByProvider = new Map;
+  for (const row of rows) {
+    if (allowed && !allowed.has(row.provider))
+      continue;
+    const existing = rowsByProvider.get(row.provider);
+    if (existing) {
+      existing.push(row);
+    } else {
+      rowsByProvider.set(row.provider, [row]);
+    }
+  }
+  const connected = [];
+  for (const [provider, providerRows] of rowsByProvider.entries()) {
+    const selected = selectProviderTokenRow(providerRows);
+    if (selected && isLikelyUsableToken(selected)) {
+      connected.push(provider);
+    }
+  }
+  return connected.sort();
+}
+async function listConnectedProviders(configuredIntegrationIds, getProviderToken, context) {
+  if (!context.userId || configuredIntegrationIds.length === 0) {
+    return [];
+  }
+  const connected = [];
+  await Promise.all(configuredIntegrationIds.map(async (provider) => {
+    try {
+      const token = await getProviderToken(provider, undefined, context);
+      if (token?.accessToken || typeof token?.refreshToken === "string" && token.refreshToken.length > 0) {
+        connected.push(provider);
+      }
+    } catch {}
+  }));
+  return connected.sort();
+}
 function defaultResolveAccountIdentity(provider, tokenData, emailHint) {
   const accountEmail = normalizeAccountEmail(emailHint ?? tokenData.email ?? null);
   let accountId = normalizeAccountIdHint(tokenData.accountId ?? null);
@@ -1111,6 +1148,8 @@ export {
   normalizeAccountEmail,
   mongodbAdapterCallbacks,
   mongodbAdapter,
+  listConnectedProvidersFromRows,
+  listConnectedProviders,
   isLikelyUsableToken,
   integrateTrigger,
   integrateProviderToken,