@genkit-ai/vercel-ai 0.2.0-rc.0

package/lib/mapping.mjs

@@ -0,0 +1,247 @@
+function mapRole(role) {
+  switch (role) {
+    case "assistant":
+      return "model";
+    case "user":
+      return "user";
+    case "system":
+      return "system";
+    default:
+      return "user";
+  }
+}
+function mapGenkitRole(role) {
+  switch (role) {
+    case "model":
+      return "assistant";
+    case "system":
+      return "system";
+    case "user":
+    default:
+      return "user";
+  }
+}
+function extractPartMetadata(part) {
+  const md = part?.metadata;
+  if (md && typeof md === "object" && !Array.isArray(md)) {
+    return md;
+  }
+  return void 0;
+}
+function mapUIPartToGenkit(part) {
+  const metadata = extractPartMetadata(part);
+  const withMetadata = (p) => metadata ? { ...p, metadata } : p;
+  if (part.type === "text") {
+    return [withMetadata({ text: part.text })];
+  }
+  if (part.type === "reasoning") {
+    const reasoning = part.text;
+    return typeof reasoning === "string" && reasoning !== "" ? [withMetadata({ reasoning })] : [];
+  }
+  if (part.type === "file") {
+    return [
+      withMetadata({
+        media: {
+          url: part.url,
+          contentType: part.mediaType
+        }
+      })
+    ];
+  }
+  if (part.type === "dynamic-tool" || part.type.startsWith("tool-")) {
+    const p = part;
+    const toolCallId = typeof p.toolCallId === "string" ? p.toolCallId : crypto.randomUUID();
+    const toolName = typeof p.toolName === "string" && p.toolName ? p.toolName : part.type.startsWith("tool-") ? part.type.slice("tool-".length) : "unknown";
+    const parts = [
+      withMetadata({
+        toolRequest: {
+          ref: toolCallId,
+          name: toolName,
+          input: p.input
+        }
+      })
+    ];
+    if (p.state === "output-available" && p.output !== void 0) {
+      parts.push(
+        withMetadata({
+          toolResponse: {
+            ref: toolCallId,
+            name: toolName,
+            output: p.output
+          }
+        })
+      );
+    }
+    return parts;
+  }
+  return [];
+}
+function mapUIMessageToGenkit(msg) {
+  const role = mapRole(msg.role);
+  const genkitParts = msg.parts.flatMap(mapUIPartToGenkit);
+  return { role, content: genkitParts };
+}
+function mapGenkitPartToUI(part, responsesByRef) {
+  const metadata = part.metadata && typeof part.metadata === "object" ? part.metadata : void 0;
+  const withMetadata = (p) => metadata ? { ...p, metadata } : p;
+  if (typeof part.text === "string") {
+    return [withMetadata({ type: "text", text: part.text })];
+  }
+  if (typeof part.reasoning === "string") {
+    return part.reasoning !== "" ? [withMetadata({ type: "reasoning", text: part.reasoning })] : [];
+  }
+  if (part.media) {
+    return [
+      withMetadata({
+        type: "file",
+        url: part.media.url,
+        mediaType: part.media.contentType
+      })
+    ];
+  }
+  if (part.toolRequest) {
+    const tr = part.toolRequest;
+    const toolCallId = tr.ref ?? crypto.randomUUID();
+    const base = {
+      type: `tool-${tr.name}`,
+      toolCallId,
+      input: tr.input
+    };
+    if (responsesByRef?.has(toolCallId)) {
+      return [
+        withMetadata({
+          ...base,
+          state: "output-available",
+          output: responsesByRef.get(toolCallId)
+        })
+      ];
+    }
+    return [withMetadata({ ...base, state: "input-available" })];
+  }
+  return [];
+}
+function mapGenkitMessageToUI(msg, responsesByRef, id) {
+  const parts = (msg.content ?? []).flatMap(
+    (part) => mapGenkitPartToUI(part, responsesByRef)
+  );
+  return {
+    id: id ?? `restored-${crypto.randomUUID()}`,
+    role: mapGenkitRole(msg.role),
+    parts
+  };
+}
+function messagesFromSnapshot(messages) {
+  const responsesByRef = /* @__PURE__ */ new Map();
+  for (const msg of messages) {
+    for (const part of msg.content ?? []) {
+      if (part.toolResponse?.ref) {
+        responsesByRef.set(part.toolResponse.ref, part.toolResponse.output);
+      }
+    }
+  }
+  const result = [];
+  let index = 0;
+  for (const msg of messages) {
+    if (msg.role === "tool") continue;
+    const ui = mapGenkitMessageToUI(msg, responsesByRef, `restored-${index}`);
+    index++;
+    if (ui.parts.length > 0) result.push(ui);
+  }
+  return result;
+}
+const RESTART_MARKER = "__genkitVercelAiRestart__";
+function restartInterrupt(metadata) {
+  return metadata === void 0 ? { [RESTART_MARKER]: true } : { [RESTART_MARKER]: true, metadata };
+}
+function asRestartInterrupt(output) {
+  if (output && typeof output === "object" && output[RESTART_MARKER] === true) {
+    return output;
+  }
+  return void 0;
+}
+function isEmptyUserMessage(msg) {
+  return !msg.parts.some((part) => {
+    if (part.type === "text") {
+      return typeof part.text === "string" && part.text.trim() !== "";
+    }
+    if (part.type === "file") {
+      return true;
+    }
+    return false;
+  });
+}
+function resolvedToolsFromMessage(msg) {
+  const seen = /* @__PURE__ */ new Set();
+  const results = [];
+  for (const part of msg.parts) {
+    const p = part;
+    const isTool = part.type === "dynamic-tool" || part.type.startsWith("tool-");
+    if (isTool && typeof p.toolCallId === "string" && p.toolCallId && !seen.has(p.toolCallId) && p.state === "output-available" && p.output !== void 0) {
+      const toolName = typeof p.toolName === "string" && p.toolName ? p.toolName : part.type.startsWith("tool-") ? part.type.slice("tool-".length) : "unknown";
+      seen.add(p.toolCallId);
+      results.push({
+        toolCallId: p.toolCallId,
+        toolName,
+        input: p.input,
+        result: p.output
+      });
+    }
+  }
+  return results;
+}
+function extractResolvedToolResults(messages) {
+  const seen = /* @__PURE__ */ new Set();
+  const results = [];
+  for (let i = messages.length - 1; i >= 0; i--) {
+    const msg = messages[i];
+    if (msg.role !== "assistant") continue;
+    for (const tr of resolvedToolsFromMessage(msg)) {
+      if (seen.has(tr.toolCallId)) continue;
+      seen.add(tr.toolCallId);
+      results.push(tr);
+    }
+  }
+  return results;
+}
+function currentTurnResolvedTools(messages) {
+  let lastAssistantIdx = -1;
+  for (let i = messages.length - 1; i >= 0; i--) {
+    if (messages[i].role === "assistant") {
+      lastAssistantIdx = i;
+      break;
+    }
+  }
+  if (lastAssistantIdx === -1) return [];
+  for (let i = lastAssistantIdx + 1; i < messages.length; i++) {
+    if (messages[i].role === "user" && !isEmptyUserMessage(messages[i])) {
+      return [];
+    }
+  }
+  return resolvedToolsFromMessage(messages[lastAssistantIdx]);
+}
+function findLastUserMessage(messages) {
+  for (let i = messages.length - 1; i >= 0; i--) {
+    if (messages[i].role === "user") return messages[i];
+  }
+  return void 0;
+}
+function findLastNonEmptyUserMessage(messages) {
+  for (let i = messages.length - 1; i >= 0; i--) {
+    const msg = messages[i];
+    if (msg.role === "user" && !isEmptyUserMessage(msg)) return msg;
+  }
+  return void 0;
+}
+export {
+  asRestartInterrupt,
+  currentTurnResolvedTools,
+  extractResolvedToolResults,
+  findLastNonEmptyUserMessage,
+  findLastUserMessage,
+  mapGenkitMessageToUI,
+  mapUIMessageToGenkit,
+  mapUIPartToGenkit,
+  messagesFromSnapshot,
+  restartInterrupt
+};
+//# sourceMappingURL=mapping.mjs.map

package/lib/mapping.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/mapping.ts"],"sourcesContent":["/**\n * Copyright 2026 Google LLC\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport type { UIMessage } from 'ai';\nimport type { MessageData, Part } from 'genkit';\n\n// ---------------------------------------------------------------------------\n// UIMessage → Genkit\n// ---------------------------------------------------------------------------\n\n/**\n * Maps a Vercel AI SDK role to a Genkit role.\n */\nfunction mapRole(role: UIMessage['role']): MessageData['role'] {\n  switch (role) {\n    case 'assistant':\n      return 'model';\n    case 'user':\n      return 'user';\n    case 'system':\n      return 'system';\n    default:\n      // Unknown roles fall back to 'user' so content is not silently dropped.\n      return 'user';\n  }\n}\n\n/**\n * Maps a Genkit role to a Vercel AI SDK role.\n */\nfunction mapGenkitRole(role: MessageData['role']): UIMessage['role'] {\n  switch (role) {\n    case 'model':\n      return 'assistant';\n    case 'system':\n      return 'system';\n    case 'user':\n    default:\n      return 'user';\n  }\n}\n\n/**\n * Extracts a plain-object `metadata` value from an AI SDK UI part, if present.\n *\n * AI SDK part types don't all formally declare a `metadata` field, but the\n * transport (and middleware-aware UIs) may attach one. We read it defensively\n * so Genkit part metadata survives a UI → Genkit round-trip.\n */\nfunction extractPartMetadata(\n  part: unknown\n): Record<string, unknown> | undefined {\n  const md = (part as { metadata?: unknown } | null)?.metadata;\n  if (md && typeof md === 'object' && !Array.isArray(md)) {\n    return md as Record<string, unknown>;\n  }\n  return undefined;\n}\n\n/**\n * Maps a single Vercel AI SDK v6 UIMessagePart to zero or more Genkit Parts.\n *\n * Handles `text`, `file`, and tool parts (both `ToolUIPart` with\n * `type: 'tool-<name>'` and `DynamicToolUIPart` with `type: 'dynamic-tool'`).\n *\n * A tool part with state `output-available` produces both a `toolRequest`\n * and a `toolResponse` part (matching Genkit's conversation model).\n */\nexport function mapUIPartToGenkit(part: UIMessage['parts'][number]): Part[] {\n  // Genkit part metadata that may have been attached to the UI part (e.g. by\n  // a middleware-aware UI or a previous Genkit → UI mapping). Preserved so it\n  // survives a UI → Genkit round-trip.\n  const metadata = extractPartMetadata(part);\n  const withMetadata = <T extends Part>(p: T): T =>\n    metadata ? ({ ...p, metadata } as T) : p;\n\n  if (part.type === 'text') {\n    return [withMetadata({ text: part.text })];\n  }\n\n  if (part.type === 'reasoning') {\n    // Map AI SDK reasoning parts to Genkit reasoning parts so the model's\n    // thinking is preserved across turns (e.g. when replaying history).\n    // Empty reasoning is dropped to avoid emitting stray, contentless\n    // reasoning blocks (mirrors the streaming side in client.ts, which also\n    // skips `reasoning === ''`).\n    const reasoning = (part as { text?: unknown }).text;\n    return typeof reasoning === 'string' && reasoning !== ''\n      ? [withMetadata({ reasoning })]\n      : [];\n  }\n\n  if (part.type === 'file') {\n    return [\n      withMetadata({\n        media: {\n          url: part.url,\n          contentType: part.mediaType,\n        },\n      }),\n    ];\n  }\n\n  // Handle v6 tool parts:\n  //   ToolUIPart:    type: 'tool-<name>', toolCallId, state, input?, output?\n  //   DynamicToolUIPart: type: 'dynamic-tool', toolName, toolCallId, state, input?, output?\n  //\n  // Use runtime property checks via Record cast since TypeScript can't\n  // narrow on prefix match for template literal types.\n  if (part.type === 'dynamic-tool' || part.type.startsWith('tool-')) {\n    const p = part as unknown as Record<string, unknown>;\n    const toolCallId =\n      typeof p.toolCallId === 'string' ? p.toolCallId : crypto.randomUUID();\n    const toolName =\n      typeof p.toolName === 'string' && p.toolName\n        ? p.toolName\n        : part.type.startsWith('tool-')\n          ? part.type.slice('tool-'.length)\n          : 'unknown';\n\n    const parts: Part[] = [\n      withMetadata({\n        toolRequest: {\n          ref: toolCallId,\n          name: toolName,\n          input: p.input,\n        },\n      }),\n    ];\n\n    if (p.state === 'output-available' && p.output !== undefined) {\n      parts.push(\n        withMetadata({\n          toolResponse: {\n            ref: toolCallId,\n            name: toolName,\n            output: p.output,\n          },\n        })\n      );\n    }\n\n    return parts;\n  }\n\n  // Unsupported part types (source-*, step-start, data-*) are silently\n  // skipped — they don't have a direct Genkit equivalent.\n  return [];\n}\n\n/**\n * Maps a Vercel AI SDK UIMessage to a Genkit MessageData.\n */\nexport function mapUIMessageToGenkit(msg: UIMessage): MessageData {\n  const role = mapRole(msg.role);\n  const genkitParts: Part[] = msg.parts.flatMap(mapUIPartToGenkit);\n  return { role, content: genkitParts };\n}\n\n// ---------------------------------------------------------------------------\n// Genkit → UIMessage (session restore)\n// ---------------------------------------------------------------------------\n\n/**\n * Maps a single Genkit {@link Part} to zero or more AI SDK UI parts.\n *\n * `toolResponse` outputs live in separate (`tool`-role) Genkit messages, so a\n * `toolRequest` is paired with its response via `responsesByRef` to produce a\n * single AI SDK tool part: `output-available` when the response is known, or\n * `input-available` when it is still pending (e.g. an unresolved interrupt).\n *\n * Standalone `toolResponse` parts are skipped here — they're merged into the\n * matching request's tool part instead of producing a separate one.\n */\nfunction mapGenkitPartToUI(\n  part: Part,\n  responsesByRef?: Map<string, unknown>\n): UIMessage['parts'] {\n  // Preserve Genkit part metadata on the UI part so it survives a\n  // Genkit → UI → Genkit round-trip (read back by extractPartMetadata).\n  const metadata =\n    part.metadata && typeof part.metadata === 'object'\n      ? (part.metadata as Record<string, unknown>)\n      : undefined;\n  const withMetadata = (\n    p: Record<string, unknown>\n  ): UIMessage['parts'][number] =>\n    (metadata ? { ...p, metadata } : p) as UIMessage['parts'][number];\n\n  if (typeof part.text === 'string') {\n    return [withMetadata({ type: 'text', text: part.text })];\n  }\n\n  if (typeof part.reasoning === 'string') {\n    return part.reasoning !== ''\n      ? [withMetadata({ type: 'reasoning', text: part.reasoning })]\n      : [];\n  }\n\n  if (part.media) {\n    return [\n      withMetadata({\n        type: 'file',\n        url: part.media.url,\n        mediaType: part.media.contentType,\n      }),\n    ];\n  }\n\n  if (part.toolRequest) {\n    const tr = part.toolRequest;\n    const toolCallId = tr.ref ?? crypto.randomUUID();\n    const base: Record<string, unknown> = {\n      type: `tool-${tr.name}`,\n      toolCallId,\n      input: tr.input,\n    };\n    if (responsesByRef?.has(toolCallId)) {\n      return [\n        withMetadata({\n          ...base,\n          state: 'output-available',\n          output: responsesByRef.get(toolCallId),\n        }),\n      ];\n    }\n    return [withMetadata({ ...base, state: 'input-available' })];\n  }\n\n  // Standalone toolResponse (paired with a request above) or any other part\n  // type without a direct UI equivalent is skipped.\n  return [];\n}\n\n/**\n * Maps a single Genkit {@link MessageData} to an AI SDK {@link UIMessage}.\n *\n * Pass `responsesByRef` (built across the whole conversation) so each\n * `toolRequest` can be paired with the `toolResponse` that resolved it, even\n * though they live in different messages. When omitted, all tool parts are\n * emitted in the `input-available` state.\n */\nexport function mapGenkitMessageToUI(\n  msg: MessageData,\n  responsesByRef?: Map<string, unknown>,\n  id?: string\n): UIMessage {\n  const parts = (msg.content ?? []).flatMap((part) =>\n    mapGenkitPartToUI(part, responsesByRef)\n  );\n  return {\n    id: id ?? `restored-${crypto.randomUUID()}`,\n    role: mapGenkitRole(msg.role),\n    parts,\n  };\n}\n\n/**\n * Converts the `messages` of a Genkit `SessionSnapshot` (an array of\n * {@link MessageData}) into AI SDK {@link UIMessage}s suitable for seeding\n * `useChat` (via its `messages` option or `setMessages`) to rehydrate a\n * previous conversation after a reload.\n *\n * Tool requests and their responses (which Genkit stores in separate\n * `model`/`tool` messages) are merged into single AI SDK tool parts, so the\n * UI renders each tool call as one element with both its input and output.\n * `tool`-role messages are therefore omitted from the result. Unresolved tool\n * requests (e.g. a pending interrupt) are emitted in the `input-available`\n * state.\n *\n * Conversation state is server-managed by `sessionId` (the `useChat` `id`), so\n * to resume the *next* turn you only need to reuse the same chat `id` — there\n * is no snapshot to restore on the client.\n *\n * @example\n * ```ts\n * const snapshot = await runFlow({ url: '/api/chat/weather/state', input: sessionId });\n * const messages = messagesFromSnapshot(snapshot.state.messages);\n * // pass `messages` to useChat({ id: sessionId, messages })\n * ```\n */\nexport function messagesFromSnapshot(messages: MessageData[]): UIMessage[] {\n  // Collect every tool response by ref so a request in one message can be\n  // paired with its response in another (Genkit stores them separately).\n  const responsesByRef = new Map<string, unknown>();\n  for (const msg of messages) {\n    for (const part of msg.content ?? []) {\n      if (part.toolResponse?.ref) {\n        responsesByRef.set(part.toolResponse.ref, part.toolResponse.output);\n      }\n    }\n  }\n\n  const result: UIMessage[] = [];\n  let index = 0;\n  for (const msg of messages) {\n    // tool-role messages only carry responses, which are merged into the\n    // matching request's tool part — skip them.\n    if (msg.role === 'tool') continue;\n    const ui = mapGenkitMessageToUI(msg, responsesByRef, `restored-${index}`);\n    index++;\n    // Drop messages that produced no renderable parts.\n    if (ui.parts.length > 0) result.push(ui);\n  }\n  return result;\n}\n\n// ---------------------------------------------------------------------------\n// Interrupt restart support\n// ---------------------------------------------------------------------------\n\n/**\n * Marker placed on a tool output to signal that the user chose to *restart*\n * (re-run) an interrupted tool rather than supply a final response.\n *\n * `Symbol.for` is intentionally avoided because tool outputs are serialized to\n * JSON before reaching the transport; a string marker survives that round-trip.\n */\nconst RESTART_MARKER = '__genkitVercelAiRestart__';\n\n/**\n * The serialized shape produced by {@link restartInterrupt}.\n */\nexport interface RestartInterruptOutput {\n  [RESTART_MARKER]: true;\n  /**\n   * Optional metadata surfaced to the tool as its `resumed` argument when it\n   * re-runs server-side. Defaults to `true` when omitted.\n   */\n  metadata?: unknown;\n}\n\n/**\n * Builds a tool output that instructs the Genkit agent to **restart** (re-run)\n * an interrupted tool instead of accepting a user-supplied response.\n *\n * Pass the result to the AI SDK's `addToolResult` as the tool `output`:\n *\n * ```tsx\n * addToolResult({\n *   tool: 'userApproval',\n *   toolCallId,\n *   output: restartInterrupt({ note: 'user revised the request' }),\n * });\n * ```\n *\n * The transport detects this marker and emits a `resume.restart` entry (with\n * the original tool input, which the server requires to match exactly) instead\n * of a `resume.respond` entry. Any `metadata` is passed through to the tool as\n * its `resumed` value.\n */\nexport function restartInterrupt(metadata?: unknown): RestartInterruptOutput {\n  return metadata === undefined\n    ? { [RESTART_MARKER]: true }\n    : { [RESTART_MARKER]: true, metadata };\n}\n\n/**\n * Returns the restart payload if `output` was produced by\n * {@link restartInterrupt}, otherwise `undefined`.\n */\nexport function asRestartInterrupt(\n  output: unknown\n): RestartInterruptOutput | undefined {\n  if (\n    output &&\n    typeof output === 'object' &&\n    (output as Record<string, unknown>)[RESTART_MARKER] === true\n  ) {\n    return output as RestartInterruptOutput;\n  }\n  return undefined;\n}\n\n// ---------------------------------------------------------------------------\n// Helpers\n// ---------------------------------------------------------------------------\n\n/**\n * A resolved tool invocation extracted from the frontend message history.\n */\nexport interface ResolvedToolResult {\n  toolCallId: string;\n  toolName: string;\n  /** The original tool input (needed to build a `resume.restart`). */\n  input: unknown;\n  /** The user-supplied output (or a {@link restartInterrupt} marker). */\n  result: unknown;\n}\n\n/**\n * Returns true if a user message carries no meaningful content — i.e. it has\n * no non-empty text and no file parts. The AI SDK's `sendMessage({ text: '' })`\n * (used to nudge a resume) produces such a \"phantom\" user message, which must\n * not be treated as a real user turn.\n */\nfunction isEmptyUserMessage(msg: UIMessage): boolean {\n  return !msg.parts.some((part) => {\n    if (part.type === 'text') {\n      return typeof part.text === 'string' && part.text.trim() !== '';\n    }\n    if (part.type === 'file') {\n      return true;\n    }\n    return false;\n  });\n}\n\n/**\n * Extracts the resolved tool invocations carried by a single UIMessage.\n *\n * Supports the v6 per-tool format (`type === 'tool-<toolName>'`) as well as\n * dynamic tools (`type === 'dynamic-tool'`, with an explicit `toolName`),\n * matching `state === 'output-available'`. De-duplicates by `toolCallId`.\n */\nfunction resolvedToolsFromMessage(msg: UIMessage): ResolvedToolResult[] {\n  const seen = new Set<string>();\n  const results: ResolvedToolResult[] = [];\n  for (const part of msg.parts) {\n    const p = part as unknown as Record<string, unknown>;\n    const isTool =\n      part.type === 'dynamic-tool' || part.type.startsWith('tool-');\n    if (\n      isTool &&\n      typeof p.toolCallId === 'string' &&\n      p.toolCallId &&\n      !seen.has(p.toolCallId) &&\n      p.state === 'output-available' &&\n      p.output !== undefined\n    ) {\n      const toolName =\n        typeof p.toolName === 'string' && p.toolName\n          ? p.toolName\n          : part.type.startsWith('tool-')\n            ? part.type.slice('tool-'.length)\n            : 'unknown';\n      seen.add(p.toolCallId);\n      results.push({\n        toolCallId: p.toolCallId,\n        toolName,\n        input: p.input,\n        result: p.output,\n      });\n    }\n  }\n  return results;\n}\n\n/**\n * Extracts resolved tool invocations from a UIMessage array.\n *\n * Supports the v6 per-tool format (`type === 'tool-<toolName>'`) as well as\n * dynamic tools (`type === 'dynamic-tool'`, with an explicit `toolName`),\n * matching `state === 'output-available'`.\n *\n * Scans **all** messages (most-recent first) and returns every resolved tool\n * result, de-duplicated by `toolCallId` (keeping the most recent).\n *\n * Prefer {@link currentTurnResolvedTools} for interrupt-resume detection — it\n * scopes the scan to the most recent turn so prior turns' (already-completed)\n * tool calls are not re-sent as a resume payload.\n */\nexport function extractResolvedToolResults(\n  messages: UIMessage[]\n): ResolvedToolResult[] {\n  const seen = new Set<string>();\n  const results: ResolvedToolResult[] = [];\n\n  // Walk backwards so the most recent resolution for a given toolCallId wins.\n  for (let i = messages.length - 1; i >= 0; i--) {\n    const msg = messages[i];\n    if (msg.role !== 'assistant') continue;\n    for (const tr of resolvedToolsFromMessage(msg)) {\n      if (seen.has(tr.toolCallId)) continue;\n      seen.add(tr.toolCallId);\n      results.push(tr);\n    }\n  }\n\n  return results;\n}\n\n/**\n * Returns the resolved tool invocations that constitute an interrupt resume\n * for the *current* turn, or an empty array if this is a fresh user turn.\n *\n * An interrupt resume is identified by the last assistant message carrying\n * resolved tool outputs (e.g. supplied via `addToolResult`) with no *non-empty*\n * user message after it. A real user message following the assistant turn means\n * the user started a new turn, so no resume is performed.\n *\n * This deliberately scopes detection to the most recent turn: tool calls from\n * earlier turns are already part of the server-side session state and must not\n * be replayed as a resume payload.\n */\nexport function currentTurnResolvedTools(\n  messages: UIMessage[]\n): ResolvedToolResult[] {\n  // Locate the most recent assistant message.\n  let lastAssistantIdx = -1;\n  for (let i = messages.length - 1; i >= 0; i--) {\n    if (messages[i].role === 'assistant') {\n      lastAssistantIdx = i;\n      break;\n    }\n  }\n  if (lastAssistantIdx === -1) return [];\n\n  // A non-empty user message after the assistant turn means a fresh user turn.\n  for (let i = lastAssistantIdx + 1; i < messages.length; i++) {\n    if (messages[i].role === 'user' && !isEmptyUserMessage(messages[i])) {\n      return [];\n    }\n  }\n\n  return resolvedToolsFromMessage(messages[lastAssistantIdx]);\n}\n\n/**\n * Finds the last user message in a UIMessage array.\n */\nexport function findLastUserMessage(\n  messages: UIMessage[]\n): UIMessage | undefined {\n  for (let i = messages.length - 1; i >= 0; i--) {\n    if (messages[i].role === 'user') return messages[i];\n  }\n  return undefined;\n}\n\n/**\n * Finds the last *non-empty* user message in a UIMessage array, skipping\n * phantom empty messages produced by `sendMessage({ text: '' })`.\n */\nexport function findLastNonEmptyUserMessage(\n  messages: UIMessage[]\n): UIMessage | undefined {\n  for (let i = messages.length - 1; i >= 0; i--) {\n    const msg = messages[i];\n    if (msg.role === 'user' && !isEmptyUserMessage(msg)) return msg;\n  }\n  return undefined;\n}\n"],"mappings":"AA0BA,SAAS,QAAQ,MAA8C;AAC7D,UAAQ,MAAM;AAAA,IACZ,KAAK;AACH,aAAO;AAAA,IACT,KAAK;AACH,aAAO;AAAA,IACT,KAAK;AACH,aAAO;AAAA,IACT;AAEE,aAAO;AAAA,EACX;AACF;AAKA,SAAS,cAAc,MAA8C;AACnE,UAAQ,MAAM;AAAA,IACZ,KAAK;AACH,aAAO;AAAA,IACT,KAAK;AACH,aAAO;AAAA,IACT,KAAK;AAAA,IACL;AACE,aAAO;AAAA,EACX;AACF;AASA,SAAS,oBACP,MACqC;AACrC,QAAM,KAAM,MAAwC;AACpD,MAAI,MAAM,OAAO,OAAO,YAAY,CAAC,MAAM,QAAQ,EAAE,GAAG;AACtD,WAAO;AAAA,EACT;AACA,SAAO;AACT;AAWO,SAAS,kBAAkB,MAA0C;AAI1E,QAAM,WAAW,oBAAoB,IAAI;AACzC,QAAM,eAAe,CAAiB,MACpC,WAAY,EAAE,GAAG,GAAG,SAAS,IAAU;AAEzC,MAAI,KAAK,SAAS,QAAQ;AACxB,WAAO,CAAC,aAAa,EAAE,MAAM,KAAK,KAAK,CAAC,CAAC;AAAA,EAC3C;AAEA,MAAI,KAAK,SAAS,aAAa;AAM7B,UAAM,YAAa,KAA4B;AAC/C,WAAO,OAAO,cAAc,YAAY,cAAc,KAClD,CAAC,aAAa,EAAE,UAAU,CAAC,CAAC,IAC5B,CAAC;AAAA,EACP;AAEA,MAAI,KAAK,SAAS,QAAQ;AACxB,WAAO;AAAA,MACL,aAAa;AAAA,QACX,OAAO;AAAA,UACL,KAAK,KAAK;AAAA,UACV,aAAa,KAAK;AAAA,QACpB;AAAA,MACF,CAAC;AAAA,IACH;AAAA,EACF;AAQA,MAAI,KAAK,SAAS,kBAAkB,KAAK,KAAK,WAAW,OAAO,GAAG;AACjE,UAAM,IAAI;AACV,UAAM,aACJ,OAAO,EAAE,eAAe,WAAW,EAAE,aAAa,OAAO,WAAW;AACtE,UAAM,WACJ,OAAO,EAAE,aAAa,YAAY,EAAE,WAChC,EAAE,WACF,KAAK,KAAK,WAAW,OAAO,IAC1B,KAAK,KAAK,MAAM,QAAQ,MAAM,IAC9B;AAER,UAAM,QAAgB;AAAA,MACpB,aAAa;AAAA,QACX,aAAa;AAAA,UACX,KAAK;AAAA,UACL,MAAM;AAAA,UACN,OAAO,EAAE;AAAA,QACX;AAAA,MACF,CAAC;AAAA,IACH;AAEA,QAAI,EAAE,UAAU,sBAAsB,EAAE,WAAW,QAAW;AAC5D,YAAM;AAAA,QACJ,aAAa;AAAA,UACX,cAAc;AAAA,YACZ,KAAK;AAAA,YACL,MAAM;AAAA,YACN,QAAQ,EAAE;AAAA,UACZ;AAAA,QACF,CAAC;AAAA,MACH;AAAA,IACF;AAEA,WAAO;AAAA,EACT;AAIA,SAAO,CAAC;AACV;AAKO,SAAS,qBAAqB,KAA6B;AAChE,QAAM,OAAO,QAAQ,IAAI,IAAI;AAC7B,QAAM,cAAsB,IAAI,MAAM,QAAQ,iBAAiB;AAC/D,SAAO,EAAE,MAAM,SAAS,YAAY;AACtC;AAiBA,SAAS,kBACP,MACA,gBACoB;AAGpB,QAAM,WACJ,KAAK,YAAY,OAAO,KAAK,aAAa,WACrC,KAAK,WACN;AACN,QAAM,eAAe,CACnB,MAEC,WAAW,EAAE,GAAG,GAAG,SAAS,IAAI;AAEnC,MAAI,OAAO,KAAK,SAAS,UAAU;AACjC,WAAO,CAAC,aAAa,EAAE,MAAM,QAAQ,MAAM,KAAK,KAAK,CAAC,CAAC;AAAA,EACzD;AAEA,MAAI,OAAO,KAAK,cAAc,UAAU;AACtC,WAAO,KAAK,cAAc,KACtB,CAAC,aAAa,EAAE,MAAM,aAAa,MAAM,KAAK,UAAU,CAAC,CAAC,IAC1D,CAAC;AAAA,EACP;AAEA,MAAI,KAAK,OAAO;AACd,WAAO;AAAA,MACL,aAAa;AAAA,QACX,MAAM;AAAA,QACN,KAAK,KAAK,MAAM;AAAA,QAChB,WAAW,KAAK,MAAM;AAAA,MACxB,CAAC;AAAA,IACH;AAAA,EACF;AAEA,MAAI,KAAK,aAAa;AACpB,UAAM,KAAK,KAAK;AAChB,UAAM,aAAa,GAAG,OAAO,OAAO,WAAW;AAC/C,UAAM,OAAgC;AAAA,MACpC,MAAM,QAAQ,GAAG,IAAI;AAAA,MACrB;AAAA,MACA,OAAO,GAAG;AAAA,IACZ;AACA,QAAI,gBAAgB,IAAI,UAAU,GAAG;AACnC,aAAO;AAAA,QACL,aAAa;AAAA,UACX,GAAG;AAAA,UACH,OAAO;AAAA,UACP,QAAQ,eAAe,IAAI,UAAU;AAAA,QACvC,CAAC;AAAA,MACH;AAAA,IACF;AACA,WAAO,CAAC,aAAa,EAAE,GAAG,MAAM,OAAO,kBAAkB,CAAC,CAAC;AAAA,EAC7D;AAIA,SAAO,CAAC;AACV;AAUO,SAAS,qBACd,KACA,gBACA,IACW;AACX,QAAM,SAAS,IAAI,WAAW,CAAC,GAAG;AAAA,IAAQ,CAAC,SACzC,kBAAkB,MAAM,cAAc;AAAA,EACxC;AACA,SAAO;AAAA,IACL,IAAI,MAAM,YAAY,OAAO,WAAW,CAAC;AAAA,IACzC,MAAM,cAAc,IAAI,IAAI;AAAA,IAC5B;AAAA,EACF;AACF;AA0BO,SAAS,qBAAqB,UAAsC;AAGzE,QAAM,iBAAiB,oBAAI,IAAqB;AAChD,aAAW,OAAO,UAAU;AAC1B,eAAW,QAAQ,IAAI,WAAW,CAAC,GAAG;AACpC,UAAI,KAAK,cAAc,KAAK;AAC1B,uBAAe,IAAI,KAAK,aAAa,KAAK,KAAK,aAAa,MAAM;AAAA,MACpE;AAAA,IACF;AAAA,EACF;AAEA,QAAM,SAAsB,CAAC;AAC7B,MAAI,QAAQ;AACZ,aAAW,OAAO,UAAU;AAG1B,QAAI,IAAI,SAAS,OAAQ;AACzB,UAAM,KAAK,qBAAqB,KAAK,gBAAgB,YAAY,KAAK,EAAE;AACxE;AAEA,QAAI,GAAG,MAAM,SAAS,EAAG,QAAO,KAAK,EAAE;AAAA,EACzC;AACA,SAAO;AACT;AAaA,MAAM,iBAAiB;AAiChB,SAAS,iBAAiB,UAA4C;AAC3E,SAAO,aAAa,SAChB,EAAE,CAAC,cAAc,GAAG,KAAK,IACzB,EAAE,CAAC,cAAc,GAAG,MAAM,SAAS;AACzC;AAMO,SAAS,mBACd,QACoC;AACpC,MACE,UACA,OAAO,WAAW,YACjB,OAAmC,cAAc,MAAM,MACxD;AACA,WAAO;AAAA,EACT;AACA,SAAO;AACT;AAwBA,SAAS,mBAAmB,KAAyB;AACnD,SAAO,CAAC,IAAI,MAAM,KAAK,CAAC,SAAS;AAC/B,QAAI,KAAK,SAAS,QAAQ;AACxB,aAAO,OAAO,KAAK,SAAS,YAAY,KAAK,KAAK,KAAK,MAAM;AAAA,IAC/D;AACA,QAAI,KAAK,SAAS,QAAQ;AACxB,aAAO;AAAA,IACT;AACA,WAAO;AAAA,EACT,CAAC;AACH;AASA,SAAS,yBAAyB,KAAsC;AACtE,QAAM,OAAO,oBAAI,IAAY;AAC7B,QAAM,UAAgC,CAAC;AACvC,aAAW,QAAQ,IAAI,OAAO;AAC5B,UAAM,IAAI;AACV,UAAM,SACJ,KAAK,SAAS,kBAAkB,KAAK,KAAK,WAAW,OAAO;AAC9D,QACE,UACA,OAAO,EAAE,eAAe,YACxB,EAAE,cACF,CAAC,KAAK,IAAI,EAAE,UAAU,KACtB,EAAE,UAAU,sBACZ,EAAE,WAAW,QACb;AACA,YAAM,WACJ,OAAO,EAAE,aAAa,YAAY,EAAE,WAChC,EAAE,WACF,KAAK,KAAK,WAAW,OAAO,IAC1B,KAAK,KAAK,MAAM,QAAQ,MAAM,IAC9B;AACR,WAAK,IAAI,EAAE,UAAU;AACrB,cAAQ,KAAK;AAAA,QACX,YAAY,EAAE;AAAA,QACd;AAAA,QACA,OAAO,EAAE;AAAA,QACT,QAAQ,EAAE;AAAA,MACZ,CAAC;AAAA,IACH;AAAA,EACF;AACA,SAAO;AACT;AAgBO,SAAS,2BACd,UACsB;AACtB,QAAM,OAAO,oBAAI,IAAY;AAC7B,QAAM,UAAgC,CAAC;AAGvC,WAAS,IAAI,SAAS,SAAS,GAAG,KAAK,GAAG,KAAK;AAC7C,UAAM,MAAM,SAAS,CAAC;AACtB,QAAI,IAAI,SAAS,YAAa;AAC9B,eAAW,MAAM,yBAAyB,GAAG,GAAG;AAC9C,UAAI,KAAK,IAAI,GAAG,UAAU,EAAG;AAC7B,WAAK,IAAI,GAAG,UAAU;AACtB,cAAQ,KAAK,EAAE;AAAA,IACjB;AAAA,EACF;AAEA,SAAO;AACT;AAeO,SAAS,yBACd,UACsB;AAEtB,MAAI,mBAAmB;AACvB,WAAS,IAAI,SAAS,SAAS,GAAG,KAAK,GAAG,KAAK;AAC7C,QAAI,SAAS,CAAC,EAAE,SAAS,aAAa;AACpC,yBAAmB;AACnB;AAAA,IACF;AAAA,EACF;AACA,MAAI,qBAAqB,GAAI,QAAO,CAAC;AAGrC,WAAS,IAAI,mBAAmB,GAAG,IAAI,SAAS,QAAQ,KAAK;AAC3D,QAAI,SAAS,CAAC,EAAE,SAAS,UAAU,CAAC,mBAAmB,SAAS,CAAC,CAAC,GAAG;AACnE,aAAO,CAAC;AAAA,IACV;AAAA,EACF;AAEA,SAAO,yBAAyB,SAAS,gBAAgB,CAAC;AAC5D;AAKO,SAAS,oBACd,UACuB;AACvB,WAAS,IAAI,SAAS,SAAS,GAAG,KAAK,GAAG,KAAK;AAC7C,QAAI,SAAS,CAAC,EAAE,SAAS,OAAQ,QAAO,SAAS,CAAC;AAAA,EACpD;AACA,SAAO;AACT;AAMO,SAAS,4BACd,UACuB;AACvB,WAAS,IAAI,SAAS,SAAS,GAAG,KAAK,GAAG,KAAK;AAC7C,UAAM,MAAM,SAAS,CAAC;AACtB,QAAI,IAAI,SAAS,UAAU,CAAC,mBAAmB,GAAG,EAAG,QAAO;AAAA,EAC9D;AACA,SAAO;AACT;","names":[]}

package/package.json

@@ -0,0 +1,94 @@
+{
+  "name": "@genkit-ai/vercel-ai",
+  "version": "0.2.0-rc.0",
+  "description": "A Genkit plugin providing a Vercel AI SDK ChatTransport for connecting useChat to Genkit Agents.",
+  "keywords": [
+    "genkit",
+    "genkit-plugin",
+    "vercel",
+    "vercel-ai",
+    "ai-sdk",
+    "useChat",
+    "ChatTransport",
+    "streaming",
+    "agents",
+    "ai",
+    "genai",
+    "generative-ai"
+  ],
+  "homepage": "https://github.com/genkit-ai/genkit/tree/main/js/plugins/vercel-ai#readme",
+  "bugs": {
+    "url": "https://github.com/genkit-ai/genkit/issues"
+  },
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/genkit-ai/genkit.git",
+    "directory": "js/plugins/vercel-ai"
+  },
+  "license": "Apache-2.0",
+  "author": "genkit",
+  "type": "commonjs",
+  "main": "./lib/index.js",
+  "types": "./lib/index.d.ts",
+  "exports": {
+    ".": {
+      "require": "./lib/index.js",
+      "import": "./lib/index.mjs",
+      "types": "./lib/index.d.ts",
+      "default": "./lib/index.js"
+    },
+    "./client": {
+      "require": "./lib/client.js",
+      "import": "./lib/client.mjs",
+      "types": "./lib/client.d.ts",
+      "default": "./lib/client.js"
+    }
+  },
+  "typesVersions": {
+    "*": {
+      "client": [
+        "lib/client"
+      ]
+    }
+  },
+  "files": [
+    "lib",
+    "README.md",
+    "LICENSE"
+  ],
+  "peerDependencies": {
+    "@ai-sdk/react": "^3.0.0",
+    "ai": "^6.0.0",
+    "genkit": "^1.39.0-rc.0"
+  },
+  "peerDependenciesMeta": {
+    "@ai-sdk/react": {
+      "optional": true
+    }
+  },
+  "devDependencies": {
+    "@ai-sdk/react": "^3.0.0",
+    "ai": "^6.0.192",
+    "@types/express": "^4.17.21",
+    "@types/node": "^20.11.16",
+    "express": "^4.21.1",
+    "get-port": "^5.1.0",
+    "npm-run-all": "^4.1.5",
+    "rimraf": "^6.0.1",
+    "tsup": "^8.3.5",
+    "tsx": "^4.19.2",
+    "typescript": "^5.9.3",
+    "@genkit-ai/express": "1.39.0-rc.0"
+  },
+  "engines": {
+    "node": ">=20"
+  },
+  "scripts": {
+    "check": "tsc",
+    "compile": "tsup-node",
+    "build:clean": "rimraf ./lib",
+    "build": "npm-run-all build:clean check compile",
+    "build:watch": "tsup-node --watch",
+    "test": "tsx --test ./tests/*_test.ts"
+  }
+}