@assistant-ui/ai-sdk 0.0.6 → 0.0.8

package/dist/transport/resumable.d.ts

@@ -1,38 +1,35 @@
-//#region src/transport/resumable.d.ts
 /** Response header used by the [Resumable Streams](/docs/guides/resumable-streams) server and client wiring. */
-declare const RESUMABLE_STREAM_ID_HEADER = "x-resumable-stream-id";
-type ResumableClientStorage = {
-  getStreamId(threadId?: string): string | null;
-  setStreamId(id: string, threadId?: string): void;
-  clear(threadId?: string): void;
-  /** Subscribes to stream id changes so automatic resume can react after mount. */
-  subscribe?(listener: () => void, threadId?: string): () => void;
+export declare const RESUMABLE_STREAM_ID_HEADER = "x-resumable-stream-id";
+export type ResumableClientStorage = {
+    getStreamId(threadId?: string): string | null;
+    setStreamId(id: string, threadId?: string): void;
+    clear(threadId?: string): void;
+    /** Subscribes to stream id changes so automatic resume can react after mount. */
+    subscribe?(listener: () => void, threadId?: string): () => void;
 };
 /** `sessionStorage`-backed storage for the pending resumable stream id. See the [Resumable Streams](/docs/guides/resumable-streams) guide for end-to-end wiring. */
-declare function createResumableSessionStorage(options?: {
-  /**
-   * Storage key for the pending stream id. A static string namespaces per route
-   * or chat surface. A getter is read lazily on every access, so the key can be
-   * derived from the active thread's identity; while the getter returns
-   * `undefined`, reads report no pending stream and writes are dropped, so a
-   * thread whose identity is not known yet never touches another thread's key.
-   *
-   * Under a remote thread list with more than one thread, scope the key per
-   * thread and create one storage instance per thread runtime rather than a
-   * single shared one. A shared key is written and cleared by whichever thread
-   * acts last, so one conversation's stream can resume inside another.
-   */
-  key?: string | (() => string | undefined);
+export declare function createResumableSessionStorage(options?: {
+    /**
+     * Storage key for the pending stream id. A static string namespaces per route
+     * or chat surface. A getter is read lazily on every access, so the key can be
+     * derived from the active thread's identity; while the getter returns
+     * `undefined`, reads report no pending stream and writes are dropped, so a
+     * thread whose identity is not known yet never touches another thread's key.
+     *
+     * Under a remote thread list with more than one thread, scope the key per
+     * thread and create one storage instance per thread runtime rather than a
+     * single shared one. A shared key is written and cleared by whichever thread
+     * acts last, so one conversation's stream can resume inside another.
+     */
+    key?: string | (() => string | undefined);
 }): ResumableClientStorage;
-type AssistantChatResumableOptions = {
-  storage: ResumableClientStorage;
-  resumeApi: string | ((streamId: string) => string);
-  /**
-   * Defaults to scanning for the AI SDK UIMessageStream `finish` marker.
-   * Cancellation never invokes this callback, only natural completion does.
-   */
-  isFinishEvent?: (chunk: Uint8Array, accumulator: string) => boolean;
+export type AssistantChatResumableOptions = {
+    storage: ResumableClientStorage;
+    resumeApi: string | ((streamId: string) => string);
+    /**
+     * Defaults to scanning for the AI SDK UIMessageStream `finish` marker.
+     * Cancellation never invokes this callback, only natural completion does.
+     */
+    isFinishEvent?: (chunk: Uint8Array, accumulator: string) => boolean;
 };
-//#endregion
-export { AssistantChatResumableOptions, RESUMABLE_STREAM_ID_HEADER, ResumableClientStorage, createResumableSessionStorage };
 //# sourceMappingURL=resumable.d.ts.map

package/dist/transport/resumable.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"resumable.d.ts","names":[],"sources":["../../src/transport/resumable.ts"],"mappings":";;cAKa;KAID;EACV,YAAY;EACZ,YAAY,YAAY;EACxB,MAAM;;EAEN,WAAW,sBAAsB;;;iBAkBnB,8BAA8B;;;;;;;;;;;;;EAa5C;IACE;KA0GQ;EACV,SAAS;EACT,sBAAsB;;;;;EAKtB,iBAAiB,OAAO,YAAY"}
+{"version":3,"file":"resumable.d.ts","sourceRoot":"","sources":["../../src/transport/resumable.ts"],"names":[],"mappings":"AAIA,gHAAgH;AAChH,eAAO,MAAM,0BAA0B,0BAAmC,CAAC;AAI3E,MAAM,MAAM,sBAAsB,GAAG;IACnC,WAAW,CAAC,QAAQ,CAAC,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI,CAAC;IAC9C,WAAW,CAAC,EAAE,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACjD,KAAK,CAAC,QAAQ,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC/B,iFAAiF;IACjF,SAAS,CAAC,CAAC,QAAQ,EAAE,MAAM,IAAI,EAAE,QAAQ,CAAC,EAAE,MAAM,GAAG,MAAM,IAAI,CAAC;CACjE,CAAC;AAgBF,oKAAoK;AACpK,wBAAgB,6BAA6B,CAAC,OAAO,CAAC,EAAE;IACtD;;;;;;;;;;;OAWG;IACH,GAAG,CAAC,EAAE,MAAM,GAAG,CAAC,MAAM,MAAM,GAAG,SAAS,CAAC,CAAC;CAC3C,GAAG,sBAAsB,CAwGzB;AAED,MAAM,MAAM,6BAA6B,GAAG;IAC1C,OAAO,EAAE,sBAAsB,CAAC;IAChC,SAAS,EAAE,MAAM,GAAG,CAAC,CAAC,QAAQ,EAAE,MAAM,KAAK,MAAM,CAAC,CAAC;IACnD;;;OAGG;IACH,aAAa,CAAC,EAAE,CAAC,KAAK,EAAE,UAAU,EAAE,WAAW,EAAE,MAAM,KAAK,OAAO,CAAC;CACrE,CAAC"}

package/dist/usage.d.ts

@@ -1,17 +1,16 @@
-/// <reference types="@assistant-ui/core/react" />
-//#region src/usage.d.ts
-type ThreadTokenUsage = {
-  totalTokens?: number;
-  inputTokens?: number;
-  outputTokens?: number;
-  reasoningTokens?: number;
-  cachedInputTokens?: number;
+/// <reference types="@assistant-ui/core/react" preserve="true" />
+export type ThreadTokenUsage = {
+    totalTokens?: number;
+    inputTokens?: number;
+    outputTokens?: number;
+    reasoningTokens?: number;
+    cachedInputTokens?: number;
 };
-interface TokenUsageExtractableMessage {
-  role?: string;
-  metadata?: unknown;
+export interface TokenUsageExtractableMessage {
+    role?: string;
+    metadata?: unknown;
 }
-declare function getThreadMessageTokenUsage(message: TokenUsageExtractableMessage | undefined): ThreadTokenUsage | undefined;
+export declare function getThreadMessageTokenUsage(message: TokenUsageExtractableMessage | undefined): ThreadTokenUsage | undefined;
 /**
  * Reads token usage from the newest assistant message that reports any.
  *
@@ -20,7 +19,5 @@ declare function getThreadMessageTokenUsage(message: TokenUsageExtractableMessag
  * other key the route returns into `metadata.custom`, which is where this hook
  * looks.
  */
-declare function useThreadTokenUsage(): ThreadTokenUsage | undefined;
-//#endregion
-export { ThreadTokenUsage, TokenUsageExtractableMessage, getThreadMessageTokenUsage, useThreadTokenUsage };
+export declare function useThreadTokenUsage(): ThreadTokenUsage | undefined;
 //# sourceMappingURL=usage.d.ts.map

package/dist/usage.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"usage.d.ts","names":[],"sources":["../src/usage.ts"],"mappings":";KAIY;EACV;EACA;EACA;EACA;EACA;;UAGe;EACf;EACA;;iBAuHc,2BACd,SAAS,2CACR;;;;;;;;;iBAsCa,uBAAuB"}
+{"version":3,"file":"usage.d.ts","sourceRoot":"","sources":["../src/usage.ts"],"names":[],"mappings":";AAIA,MAAM,MAAM,gBAAgB,GAAG;IAC7B,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,iBAAiB,CAAC,EAAE,MAAM,CAAC;CAC5B,CAAC;AAEF,MAAM,WAAW,4BAA4B;IAC3C,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,QAAQ,CAAC,EAAE,OAAO,CAAC;CACpB;AAsHD,wBAAgB,0BAA0B,CACxC,OAAO,EAAE,4BAA4B,GAAG,SAAS,GAChD,gBAAgB,GAAG,SAAS,CAa9B;AAiBD;;;;;;;GAOG;AACH,wBAAgB,mBAAmB,IAAI,gBAAgB,GAAG,SAAS,CAGlE"}

package/dist/usage.js.map

@@ -1 +1 @@
-{"version":3,"file":"usage.js","names":["useMemo","useAuiState","ThreadTokenUsage","totalTokens","inputTokens","outputTokens","reasoningTokens","cachedInputTokens","TokenUsageExtractableMessage","role","metadata","UsageRecord","Record","USAGE_KEYS","const","asRecord","value","Array","isArray","undefined","asPositiveTokenCount","Number","isFinite","computeTotalTokens","usage","normalizeUsage","record","result","hasFields","key","count","outputTokenDetails","inputTokenDetails","cacheReadTokens","withComputedTotal","usageFromSteps","steps","sums","present","stepsWithUsage","stepsWithComputableTotal","step","stepTotal","getThreadMessageTokenUsage","message","topLevelUsage","customUsage","custom","findLatestMessageWithUsage","messages","idx","length","useThreadTokenUsage","$","_c","msg","_temp","t0","s","thread"],"sources":["../src/usage.ts"],"sourcesContent":["/// <reference types=\"@assistant-ui/core/react\" />\nimport { useMemo } from \"react\";\nimport { useAuiState } from \"@assistant-ui/store\";\n\nexport type ThreadTokenUsage = {\n  totalTokens?: number;\n  inputTokens?: number;\n  outputTokens?: number;\n  reasoningTokens?: number;\n  cachedInputTokens?: number;\n};\n\nexport interface TokenUsageExtractableMessage {\n  role?: string;\n  metadata?: unknown;\n}\n\ntype UsageRecord = Record<string, unknown>;\n\nconst USAGE_KEYS = [\n  \"inputTokens\",\n  \"outputTokens\",\n  \"reasoningTokens\",\n  \"cachedInputTokens\",\n  \"totalTokens\",\n] as const satisfies (keyof ThreadTokenUsage)[];\n\nfunction asRecord(value: unknown): UsageRecord | undefined {\n  if (!value || typeof value !== \"object\" || Array.isArray(value))\n    return undefined;\n  return value as UsageRecord;\n}\n\nfunction asPositiveTokenCount(value: unknown): number | undefined {\n  if (typeof value !== \"number\" || !Number.isFinite(value) || value < 0) {\n    return undefined;\n  }\n  return value;\n}\n\nfunction computeTotalTokens(usage: ThreadTokenUsage): number | undefined {\n  if (usage.totalTokens !== undefined) return usage.totalTokens;\n  if (usage.inputTokens !== undefined && usage.outputTokens !== undefined) {\n    return usage.inputTokens + usage.outputTokens;\n  }\n  return undefined;\n}\n\nfunction normalizeUsage(value: unknown): ThreadTokenUsage | undefined {\n  const record = asRecord(value);\n  if (!record) return undefined;\n\n  const result: ThreadTokenUsage = {};\n  let hasFields = false;\n  for (const key of USAGE_KEYS) {\n    const count = asPositiveTokenCount(record[key]);\n    if (count !== undefined) {\n      result[key] = count;\n      hasFields = true;\n    }\n  }\n  // AI SDK v7 moved these under token detail objects; v6 kept them top-level.\n  if (result.reasoningTokens === undefined) {\n    const count = asPositiveTokenCount(\n      asRecord(record.outputTokenDetails)?.reasoningTokens,\n    );\n    if (count !== undefined) {\n      result.reasoningTokens = count;\n      hasFields = true;\n    }\n  }\n  if (result.cachedInputTokens === undefined) {\n    const count = asPositiveTokenCount(\n      asRecord(record.inputTokenDetails)?.cacheReadTokens,\n    );\n    if (count !== undefined) {\n      result.cachedInputTokens = count;\n      hasFields = true;\n    }\n  }\n  return hasFields ? result : undefined;\n}\n\nfunction withComputedTotal(\n  usage: ThreadTokenUsage,\n): ThreadTokenUsage | undefined {\n  const totalTokens = computeTotalTokens(usage);\n  return { ...usage, ...(totalTokens !== undefined && { totalTokens }) };\n}\n\nfunction usageFromSteps(value: unknown): ThreadTokenUsage | undefined {\n  const steps = Array.isArray(value) ? value : [];\n\n  const sums: Record<string, number> = {};\n  const present: Record<string, boolean> = {};\n  let stepsWithUsage = 0;\n  let stepsWithComputableTotal = 0;\n\n  for (const step of steps) {\n    const usage = normalizeUsage(asRecord(step)?.usage);\n    if (!usage) continue;\n    stepsWithUsage++;\n\n    const stepTotal = computeTotalTokens(usage);\n    if (stepTotal !== undefined) {\n      sums.totalTokens = (sums.totalTokens ?? 0) + stepTotal;\n      stepsWithComputableTotal++;\n    }\n\n    for (const key of USAGE_KEYS) {\n      if (key === \"totalTokens\") continue;\n      if (usage[key] !== undefined) {\n        sums[key] = (sums[key] ?? 0) + usage[key];\n        present[key] = true;\n      }\n    }\n  }\n\n  if (stepsWithUsage === 0) return undefined;\n\n  const result: ThreadTokenUsage = {};\n  if (stepsWithComputableTotal === stepsWithUsage) {\n    result.totalTokens = sums.totalTokens!;\n  }\n  for (const key of USAGE_KEYS) {\n    if (key === \"totalTokens\") continue;\n    if (present[key]) {\n      result[key] = sums[key]!;\n    }\n  }\n  return result;\n}\n\nexport function getThreadMessageTokenUsage(\n  message: TokenUsageExtractableMessage | undefined,\n): ThreadTokenUsage | undefined {\n  if (!message || message.role !== \"assistant\") return undefined;\n\n  const metadata = asRecord(message.metadata);\n  if (!metadata) return undefined;\n\n  const topLevelUsage = normalizeUsage(metadata.usage);\n  if (topLevelUsage) return withComputedTotal(topLevelUsage);\n\n  const customUsage = normalizeUsage(asRecord(metadata.custom)?.usage);\n  if (customUsage) return withComputedTotal(customUsage);\n\n  return usageFromSteps(metadata.steps);\n}\n\nfunction findLatestMessageWithUsage(\n  messages: readonly TokenUsageExtractableMessage[] | undefined,\n): TokenUsageExtractableMessage | undefined {\n  if (!messages) return undefined;\n\n  for (let idx = messages.length - 1; idx >= 0; idx -= 1) {\n    const message = messages[idx];\n    if (getThreadMessageTokenUsage(message)) {\n      return message;\n    }\n  }\n\n  return undefined;\n}\n\n/**\n * Reads token usage from the newest assistant message that reports any.\n *\n * A route attaches usage through the AI SDK's `messageMetadata` option. Because\n * a thread message carries a fixed metadata shape, the converter moves every\n * other key the route returns into `metadata.custom`, which is where this hook\n * looks.\n */\nexport function useThreadTokenUsage(): ThreadTokenUsage | undefined {\n  const msg = useAuiState((s) => findLatestMessageWithUsage(s.thread.messages));\n  return useMemo(() => getThreadMessageTokenUsage(msg), [msg]);\n}\n"],"mappings":";;;;AAmBA,MAAMa,aAAa;CACjB;CACA;CACA;CACA;CACA;AAAa;AAGf,SAASE,SAASC,OAAyC;CACzD,IAAI,CAACA,SAAS,OAAOA,UAAU,YAAYC,MAAMC,QAAQF,KAAK,GAC5D,OAAOG,KAAAA;CACT,OAAOH;AACT;AAEA,SAASI,qBAAqBJ,OAAoC;CAChE,IAAI,OAAOA,UAAU,YAAY,CAACK,OAAOC,SAASN,KAAK,KAAKA,QAAQ,GAClE;CAEF,OAAOA;AACT;AAEA,SAASO,mBAAmBC,OAA6C;CACvE,IAAIA,MAAMrB,gBAAgBgB,KAAAA,GAAW,OAAOK,MAAMrB;CAClD,IAAIqB,MAAMpB,gBAAgBe,KAAAA,KAAaK,MAAMnB,iBAAiBc,KAAAA,GAC5D,OAAOK,MAAMpB,cAAcoB,MAAMnB;AAGrC;AAEA,SAASoB,eAAeT,OAA8C;CACpE,MAAMU,SAASX,SAASC,KAAK;CAC7B,IAAI,CAACU,QAAQ,OAAOP,KAAAA;CAEpB,MAAMQ,SAA2B,CAAC;CAClC,IAAIC,YAAY;CAChB,KAAK,MAAMC,OAAOhB,YAAY;EAC5B,MAAMiB,QAAQV,qBAAqBM,OAAOG,IAAI;EAC9C,IAAIC,UAAUX,KAAAA,GAAW;GACvBQ,OAAOE,OAAOC;GACdF,YAAY;EACd;CACF;CAEA,IAAID,OAAOrB,oBAAoBa,KAAAA,GAAW;EACxC,MAAMW,QAAQV,qBACZL,SAASW,OAAOK,kBAAkB,CAAC,EAAEzB,eACvC;EACA,IAAIwB,UAAUX,KAAAA,GAAW;GACvBQ,OAAOrB,kBAAkBwB;GACzBF,YAAY;EACd;CACF;CACA,IAAID,OAAOpB,sBAAsBY,KAAAA,GAAW;EAC1C,MAAMW,QAAQV,qBACZL,SAASW,OAAOM,iBAAiB,CAAC,EAAEC,eACtC;EACA,IAAIH,UAAUX,KAAAA,GAAW;GACvBQ,OAAOpB,oBAAoBuB;GAC3BF,YAAY;EACd;CACF;CACA,OAAOA,YAAYD,SAASR,KAAAA;AAC9B;AAEA,SAASe,kBACPV,OAC8B;CAC9B,MAAMrB,cAAcoB,mBAAmBC,KAAK;CAC5C,OAAO;EAAE,GAAGA;EAAO,GAAIrB,gBAAgBgB,KAAAA,KAAa,EAAEhB,YAAY;CAAG;AACvE;AAEA,SAASgC,eAAenB,OAA8C;CACpE,MAAMoB,QAAQnB,MAAMC,QAAQF,KAAK,IAAIA,QAAQ,CAAA;CAE7C,MAAMqB,OAA+B,CAAC;CACtC,MAAMC,UAAmC,CAAC;CAC1C,IAAIC,iBAAiB;CACrB,IAAIC,2BAA2B;CAE/B,KAAK,MAAMC,QAAQL,OAAO;EACxB,MAAMZ,QAAQC,eAAeV,SAAS0B,IAAI,CAAC,EAAEjB,KAAK;EAClD,IAAI,CAACA,OAAO;EACZe;EAEA,MAAMG,YAAYnB,mBAAmBC,KAAK;EAC1C,IAAIkB,cAAcvB,KAAAA,GAAW;GAC3BkB,KAAKlC,eAAekC,KAAKlC,eAAe,KAAKuC;GAC7CF;EACF;EAEA,KAAK,MAAMX,OAAOhB,YAAY;GAC5B,IAAIgB,QAAQ,eAAe;GAC3B,IAAIL,MAAMK,SAASV,KAAAA,GAAW;IAC5BkB,KAAKR,QAAQQ,KAAKR,QAAQ,KAAKL,MAAMK;IACrCS,QAAQT,OAAO;GACjB;EACF;CACF;CAEA,IAAIU,mBAAmB,GAAG,OAAOpB,KAAAA;CAEjC,MAAMQ,SAA2B,CAAC;CAClC,IAAIa,6BAA6BD,gBAC/BZ,OAAOxB,cAAckC,KAAKlC;CAE5B,KAAK,MAAM0B,OAAOhB,YAAY;EAC5B,IAAIgB,QAAQ,eAAe;EAC3B,IAAIS,QAAQT,MACVF,OAAOE,OAAOQ,KAAKR;CAEvB;CACA,OAAOF;AACT;AAEA,SAAgBgB,2BACdC,SAC8B;CAC9B,IAAI,CAACA,WAAWA,QAAQnC,SAAS,aAAa,OAAOU,KAAAA;CAErD,MAAMT,WAAWK,SAAS6B,QAAQlC,QAAQ;CAC1C,IAAI,CAACA,UAAU,OAAOS,KAAAA;CAEtB,MAAM0B,gBAAgBpB,eAAef,SAASc,KAAK;CACnD,IAAIqB,eAAe,OAAOX,kBAAkBW,aAAa;CAEzD,MAAMC,cAAcrB,eAAeV,SAASL,SAASqC,MAAM,CAAC,EAAEvB,KAAK;CACnE,IAAIsB,aAAa,OAAOZ,kBAAkBY,WAAW;CAErD,OAAOX,eAAezB,SAAS0B,KAAK;AACtC;AAEA,SAASY,2BACPC,UAC0C;CAC1C,IAAI,CAACA,UAAU,OAAO9B,KAAAA;CAEtB,KAAK,IAAI+B,MAAMD,SAASE,SAAS,GAAGD,OAAO,GAAGA,OAAO,GAAG;EACtD,MAAMN,UAAUK,SAASC;EACzB,IAAIP,2BAA2BC,OAAO,GACpC,OAAOA;CAEX;AAGF;;;;;;;;;AAUA,SAAOQ,sBAAA;CAAA,MAAAC,IAAAC,EAAA,CAAA;CACL,MAAAC,MAAYtD,YAAYuD,KAAoD;CAAE,IAAAC;CAAA,IAAAJ,EAAA,OAAAE,KAAA;EACzDE,KAAAd,2BAA2BY,GAAG;EAACF,EAAA,KAAAE;EAAAF,EAAA,KAAAI;CAAA,OAAAA,KAAAJ,EAAA;CAAA,OAA/BI;AAA+B;AAF/C,SAAAD,MAAAE,GAAA;CAAA,OAC0BV,2BAA2BU,EAACC,OAAOV,QAAS;AAAC"}
+{"version":3,"file":"usage.js","names":["useMemo","useAuiState","ThreadTokenUsage","totalTokens","inputTokens","outputTokens","reasoningTokens","cachedInputTokens","TokenUsageExtractableMessage","role","metadata","UsageRecord","Record","USAGE_KEYS","const","asRecord","value","Array","isArray","undefined","asPositiveTokenCount","Number","isFinite","computeTotalTokens","usage","normalizeUsage","record","result","hasFields","key","count","outputTokenDetails","inputTokenDetails","cacheReadTokens","withComputedTotal","usageFromSteps","steps","sums","present","stepsWithUsage","stepsWithComputableTotal","step","stepTotal","getThreadMessageTokenUsage","message","topLevelUsage","customUsage","custom","findLatestMessageWithUsage","messages","idx","length","useThreadTokenUsage","$","_c","msg","_temp","t0","s","thread"],"sources":["../src/usage.ts"],"sourcesContent":["/// <reference types=\"@assistant-ui/core/react\" preserve=\"true\" />\nimport { useMemo } from \"react\";\nimport { useAuiState } from \"@assistant-ui/store\";\n\nexport type ThreadTokenUsage = {\n  totalTokens?: number;\n  inputTokens?: number;\n  outputTokens?: number;\n  reasoningTokens?: number;\n  cachedInputTokens?: number;\n};\n\nexport interface TokenUsageExtractableMessage {\n  role?: string;\n  metadata?: unknown;\n}\n\ntype UsageRecord = Record<string, unknown>;\n\nconst USAGE_KEYS = [\n  \"inputTokens\",\n  \"outputTokens\",\n  \"reasoningTokens\",\n  \"cachedInputTokens\",\n  \"totalTokens\",\n] as const satisfies (keyof ThreadTokenUsage)[];\n\nfunction asRecord(value: unknown): UsageRecord | undefined {\n  if (!value || typeof value !== \"object\" || Array.isArray(value))\n    return undefined;\n  return value as UsageRecord;\n}\n\nfunction asPositiveTokenCount(value: unknown): number | undefined {\n  if (typeof value !== \"number\" || !Number.isFinite(value) || value < 0) {\n    return undefined;\n  }\n  return value;\n}\n\nfunction computeTotalTokens(usage: ThreadTokenUsage): number | undefined {\n  if (usage.totalTokens !== undefined) return usage.totalTokens;\n  if (usage.inputTokens !== undefined && usage.outputTokens !== undefined) {\n    return usage.inputTokens + usage.outputTokens;\n  }\n  return undefined;\n}\n\nfunction normalizeUsage(value: unknown): ThreadTokenUsage | undefined {\n  const record = asRecord(value);\n  if (!record) return undefined;\n\n  const result: ThreadTokenUsage = {};\n  let hasFields = false;\n  for (const key of USAGE_KEYS) {\n    const count = asPositiveTokenCount(record[key]);\n    if (count !== undefined) {\n      result[key] = count;\n      hasFields = true;\n    }\n  }\n  // AI SDK v7 moved these under token detail objects; v6 kept them top-level.\n  if (result.reasoningTokens === undefined) {\n    const count = asPositiveTokenCount(\n      asRecord(record.outputTokenDetails)?.reasoningTokens,\n    );\n    if (count !== undefined) {\n      result.reasoningTokens = count;\n      hasFields = true;\n    }\n  }\n  if (result.cachedInputTokens === undefined) {\n    const count = asPositiveTokenCount(\n      asRecord(record.inputTokenDetails)?.cacheReadTokens,\n    );\n    if (count !== undefined) {\n      result.cachedInputTokens = count;\n      hasFields = true;\n    }\n  }\n  return hasFields ? result : undefined;\n}\n\nfunction withComputedTotal(\n  usage: ThreadTokenUsage,\n): ThreadTokenUsage | undefined {\n  const totalTokens = computeTotalTokens(usage);\n  return { ...usage, ...(totalTokens !== undefined && { totalTokens }) };\n}\n\nfunction usageFromSteps(value: unknown): ThreadTokenUsage | undefined {\n  const steps = Array.isArray(value) ? value : [];\n\n  const sums: Record<string, number> = {};\n  const present: Record<string, boolean> = {};\n  let stepsWithUsage = 0;\n  let stepsWithComputableTotal = 0;\n\n  for (const step of steps) {\n    const usage = normalizeUsage(asRecord(step)?.usage);\n    if (!usage) continue;\n    stepsWithUsage++;\n\n    const stepTotal = computeTotalTokens(usage);\n    if (stepTotal !== undefined) {\n      sums.totalTokens = (sums.totalTokens ?? 0) + stepTotal;\n      stepsWithComputableTotal++;\n    }\n\n    for (const key of USAGE_KEYS) {\n      if (key === \"totalTokens\") continue;\n      if (usage[key] !== undefined) {\n        sums[key] = (sums[key] ?? 0) + usage[key];\n        present[key] = true;\n      }\n    }\n  }\n\n  if (stepsWithUsage === 0) return undefined;\n\n  const result: ThreadTokenUsage = {};\n  if (stepsWithComputableTotal === stepsWithUsage) {\n    result.totalTokens = sums.totalTokens!;\n  }\n  for (const key of USAGE_KEYS) {\n    if (key === \"totalTokens\") continue;\n    if (present[key]) {\n      result[key] = sums[key]!;\n    }\n  }\n  return result;\n}\n\nexport function getThreadMessageTokenUsage(\n  message: TokenUsageExtractableMessage | undefined,\n): ThreadTokenUsage | undefined {\n  if (!message || message.role !== \"assistant\") return undefined;\n\n  const metadata = asRecord(message.metadata);\n  if (!metadata) return undefined;\n\n  const topLevelUsage = normalizeUsage(metadata.usage);\n  if (topLevelUsage) return withComputedTotal(topLevelUsage);\n\n  const customUsage = normalizeUsage(asRecord(metadata.custom)?.usage);\n  if (customUsage) return withComputedTotal(customUsage);\n\n  return usageFromSteps(metadata.steps);\n}\n\nfunction findLatestMessageWithUsage(\n  messages: readonly TokenUsageExtractableMessage[] | undefined,\n): TokenUsageExtractableMessage | undefined {\n  if (!messages) return undefined;\n\n  for (let idx = messages.length - 1; idx >= 0; idx -= 1) {\n    const message = messages[idx];\n    if (getThreadMessageTokenUsage(message)) {\n      return message;\n    }\n  }\n\n  return undefined;\n}\n\n/**\n * Reads token usage from the newest assistant message that reports any.\n *\n * A route attaches usage through the AI SDK's `messageMetadata` option. Because\n * a thread message carries a fixed metadata shape, the converter moves every\n * other key the route returns into `metadata.custom`, which is where this hook\n * looks.\n */\nexport function useThreadTokenUsage(): ThreadTokenUsage | undefined {\n  const msg = useAuiState((s) => findLatestMessageWithUsage(s.thread.messages));\n  return useMemo(() => getThreadMessageTokenUsage(msg), [msg]);\n}\n"],"mappings":";;;;AAmBA,MAAMa,aAAa;CACjB;CACA;CACA;CACA;CACA;AAAa;AAGf,SAASE,SAASC,OAAyC;CACzD,IAAI,CAACA,SAAS,OAAOA,UAAU,YAAYC,MAAMC,QAAQF,KAAK,GAC5D,OAAOG,KAAAA;CACT,OAAOH;AACT;AAEA,SAASI,qBAAqBJ,OAAoC;CAChE,IAAI,OAAOA,UAAU,YAAY,CAACK,OAAOC,SAASN,KAAK,KAAKA,QAAQ,GAClE;CAEF,OAAOA;AACT;AAEA,SAASO,mBAAmBC,OAA6C;CACvE,IAAIA,MAAMrB,gBAAgBgB,KAAAA,GAAW,OAAOK,MAAMrB;CAClD,IAAIqB,MAAMpB,gBAAgBe,KAAAA,KAAaK,MAAMnB,iBAAiBc,KAAAA,GAC5D,OAAOK,MAAMpB,cAAcoB,MAAMnB;AAGrC;AAEA,SAASoB,eAAeT,OAA8C;CACpE,MAAMU,SAASX,SAASC,KAAK;CAC7B,IAAI,CAACU,QAAQ,OAAOP,KAAAA;CAEpB,MAAMQ,SAA2B,CAAC;CAClC,IAAIC,YAAY;CAChB,KAAK,MAAMC,OAAOhB,YAAY;EAC5B,MAAMiB,QAAQV,qBAAqBM,OAAOG,IAAI;EAC9C,IAAIC,UAAUX,KAAAA,GAAW;GACvBQ,OAAOE,OAAOC;GACdF,YAAY;EACd;CACF;CAEA,IAAID,OAAOrB,oBAAoBa,KAAAA,GAAW;EACxC,MAAMW,QAAQV,qBACZL,SAASW,OAAOK,kBAAkB,CAAC,EAAEzB,eACvC;EACA,IAAIwB,UAAUX,KAAAA,GAAW;GACvBQ,OAAOrB,kBAAkBwB;GACzBF,YAAY;EACd;CACF;CACA,IAAID,OAAOpB,sBAAsBY,KAAAA,GAAW;EAC1C,MAAMW,QAAQV,qBACZL,SAASW,OAAOM,iBAAiB,CAAC,EAAEC,eACtC;EACA,IAAIH,UAAUX,KAAAA,GAAW;GACvBQ,OAAOpB,oBAAoBuB;GAC3BF,YAAY;EACd;CACF;CACA,OAAOA,YAAYD,SAASR,KAAAA;AAC9B;AAEA,SAASe,kBACPV,OAC8B;CAC9B,MAAMrB,cAAcoB,mBAAmBC,KAAK;CAC5C,OAAO;EAAE,GAAGA;EAAO,GAAIrB,gBAAgBgB,KAAAA,KAAa,EAAEhB,YAAY;CAAG;AACvE;AAEA,SAASgC,eAAenB,OAA8C;CACpE,MAAMoB,QAAQnB,MAAMC,QAAQF,KAAK,IAAIA,QAAQ,CAAA;CAE7C,MAAMqB,OAA+B,CAAC;CACtC,MAAMC,UAAmC,CAAC;CAC1C,IAAIC,iBAAiB;CACrB,IAAIC,2BAA2B;CAE/B,KAAK,MAAMC,QAAQL,OAAO;EACxB,MAAMZ,QAAQC,eAAeV,SAAS0B,IAAI,CAAC,EAAEjB,KAAK;EAClD,IAAI,CAACA,OAAO;EACZe;EAEA,MAAMG,YAAYnB,mBAAmBC,KAAK;EAC1C,IAAIkB,cAAcvB,KAAAA,GAAW;GAC3BkB,KAAKlC,eAAekC,KAAKlC,eAAe,KAAKuC;GAC7CF;EACF;EAEA,KAAK,MAAMX,OAAOhB,YAAY;GAC5B,IAAIgB,QAAQ,eAAe;GAC3B,IAAIL,MAAMK,SAASV,KAAAA,GAAW;IAC5BkB,KAAKR,QAAQQ,KAAKR,QAAQ,KAAKL,MAAMK;IACrCS,QAAQT,OAAO;GACjB;EACF;CACF;CAEA,IAAIU,mBAAmB,GAAG,OAAOpB,KAAAA;CAEjC,MAAMQ,SAA2B,CAAC;CAClC,IAAIa,6BAA6BD,gBAC/BZ,OAAOxB,cAAckC,KAAKlC;CAE5B,KAAK,MAAM0B,OAAOhB,YAAY;EAC5B,IAAIgB,QAAQ,eAAe;EAC3B,IAAIS,QAAQT,MACVF,OAAOE,OAAOQ,KAAKR;CAEvB;CACA,OAAOF;AACT;AAEA,SAAgBgB,2BACdC,SAC8B;CAC9B,IAAI,CAACA,WAAWA,QAAQnC,SAAS,aAAa,OAAOU,KAAAA;CAErD,MAAMT,WAAWK,SAAS6B,QAAQlC,QAAQ;CAC1C,IAAI,CAACA,UAAU,OAAOS,KAAAA;CAEtB,MAAM0B,gBAAgBpB,eAAef,SAASc,KAAK;CACnD,IAAIqB,eAAe,OAAOX,kBAAkBW,aAAa;CAEzD,MAAMC,cAAcrB,eAAeV,SAASL,SAASqC,MAAM,CAAC,EAAEvB,KAAK;CACnE,IAAIsB,aAAa,OAAOZ,kBAAkBY,WAAW;CAErD,OAAOX,eAAezB,SAAS0B,KAAK;AACtC;AAEA,SAASY,2BACPC,UAC0C;CAC1C,IAAI,CAACA,UAAU,OAAO9B,KAAAA;CAEtB,KAAK,IAAI+B,MAAMD,SAASE,SAAS,GAAGD,OAAO,GAAGA,OAAO,GAAG;EACtD,MAAMN,UAAUK,SAASC;EACzB,IAAIP,2BAA2BC,OAAO,GACpC,OAAOA;CAEX;AAGF;;;;;;;;;AAUA,SAAOQ,sBAAA;CAAA,MAAAC,IAAAC,EAAA,CAAA;CACL,MAAAC,MAAYtD,YAAYuD,KAAoD;CAAE,IAAAC;CAAA,IAAAJ,EAAA,OAAAE,KAAA;EACzDE,KAAAd,2BAA2BY,GAAG;EAACF,EAAA,KAAAE;EAAAF,EAAA,KAAAI;CAAA,OAAAA,KAAAJ,EAAA;CAAA,OAA/BI;AAA+B;AAF/C,SAAAD,MAAAE,GAAA;CAAA,OAC0BV,2BAA2BU,EAACC,OAAOV,QAAS;AAAC"}

package/dist/utils/getVercelAIMessages.d.ts

@@ -1,7 +1,4 @@
-import { UIMessage } from "ai";
-import { ThreadMessage } from "@assistant-ui/core";
-//#region src/utils/getVercelAIMessages.d.ts
-declare const getVercelAIMessages: <UI_MESSAGE extends UIMessage = UIMessage>(message: ThreadMessage) => UI_MESSAGE[];
-//#endregion
-export { getVercelAIMessages };
+import { type ThreadMessage } from "@assistant-ui/core";
+import type { UIMessage } from "ai";
+export declare const getVercelAIMessages: <UI_MESSAGE extends UIMessage = UIMessage>(message: ThreadMessage) => UI_MESSAGE[];
 //# sourceMappingURL=getVercelAIMessages.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"getVercelAIMessages.d.ts","names":[],"sources":["../../src/utils/getVercelAIMessages.ts"],"mappings":";;;cAMa,sBAAuB,mBAAmB,YAAY,WACjE,SAAS,kBAEmC"}
+{"version":3,"file":"getVercelAIMessages.d.ts","sourceRoot":"","sources":["../../src/utils/getVercelAIMessages.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,KAAK,aAAa,EACnB,MAAM,oBAAoB,CAAC;AAC5B,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,IAAI,CAAC;AAEpC,eAAO,MAAM,mBAAmB,GAAI,UAAU,SAAS,SAAS,GAAG,SAAS,EAC1E,SAAS,aAAa,KAEsB,UAAU,EACvD,CAAC"}

package/dist/utils/sliceMessagesUntil.d.ts

@@ -1,6 +1,3 @@
-import { UIMessage } from "ai";
-//#region src/utils/sliceMessagesUntil.d.ts
-declare const sliceMessagesUntil: <UI_MESSAGE extends UIMessage = UIMessage>(messages: UI_MESSAGE[], messageId: string | null) => UI_MESSAGE[];
-//#endregion
-export { sliceMessagesUntil };
+import type { UIMessage } from "ai";
+export declare const sliceMessagesUntil: <UI_MESSAGE extends UIMessage = UIMessage>(messages: UI_MESSAGE[], messageId: string | null) => UI_MESSAGE[];
 //# sourceMappingURL=sliceMessagesUntil.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"sliceMessagesUntil.d.ts","names":[],"sources":["../../src/utils/sliceMessagesUntil.ts"],"mappings":";;cAEa,qBAAsB,mBAAmB,YAAY,WAChE,UAAU,cACV,6BAAwB"}
+{"version":3,"file":"sliceMessagesUntil.d.ts","sourceRoot":"","sources":["../../src/utils/sliceMessagesUntil.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,IAAI,CAAC;AAEpC,eAAO,MAAM,kBAAkB,GAAI,UAAU,SAAS,SAAS,GAAG,SAAS,EACzE,UAAU,UAAU,EAAE,EACtB,WAAW,MAAM,GAAG,IAAI,iBAezB,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@assistant-ui/ai-sdk",
-  "version": "0.0.6",
+  "version": "0.0.8",
   "description": "Vercel AI SDK adapter for assistant-ui",
   "keywords": [
     "ai-sdk",
@@ -46,17 +46,17 @@
   ],
   "sideEffects": false,
   "dependencies": {
-    "@ai-sdk/mcp": "^2.0.45",
-    "@ai-sdk/react": "^4.0.96",
-    "@assistant-ui/core": "^0.3.19",
-    "@assistant-ui/store": "^0.3.13",
-    "@assistant-ui/tap": "^0.9.17",
-    "ai": "^7.0.93",
-    "assistant-cloud": "^0.2.1",
-    "assistant-stream": "^0.3.43"
+    "@ai-sdk/mcp": "^2.0.49",
+    "@ai-sdk/react": "^4.0.104",
+    "@assistant-ui/core": "^0.3.21",
+    "@assistant-ui/store": "^0.3.15",
+    "@assistant-ui/tap": "^0.9.19",
+    "assistant-cloud": "^0.2.3",
+    "assistant-stream": "^0.3.45"
   },
   "peerDependencies": {
     "@types/react": "*",
+    "ai": "^7.0.101",
     "react": "^18 || ^19"
   },
   "peerDependenciesMeta": {
@@ -68,14 +68,15 @@
     }
   },
   "devDependencies": {
-    "@assistant-ui/x-buildutils": "0.0.27",
+    "@assistant-ui/x-buildutils": "0.0.29",
     "@testing-library/react": "^16.3.3",
     "@types/json-schema": "^7.0.15",
-    "@types/react": "^19.2.18",
-    "@types/react-dom": "^19.2.7",
+    "@types/react": "^19.3.0",
+    "@types/react-dom": "^19.3.0",
+    "ai": "^7.0.101",
     "jsdom": "^30.0.1",
-    "react": "^19.2.8",
-    "vitest": "^5.0.0"
+    "react": "^19.3.0",
+    "vitest": "^5.0.1"
   },
   "publishConfig": {
     "access": "public",
@@ -93,6 +94,7 @@
   "scripts": {
     "build": "aui-build",
     "test": "vitest run",
+    "typecheck": "tsc --noEmit && tsc --noEmit -p tsconfig.test.json",
     "test:react-compiler": "vitest run --config vitest.react-compiler.config.ts",
     "test:standalone": "vitest run --project standalone",
     "test:watch": "vitest"

package/src/adapters/vercelAttachmentAdapter.test.ts

@@ -26,4 +26,27 @@ describe("vercelAttachmentAdapter", () => {
     if (part?.type !== "file") throw new Error("expected file content part");
     expect(part.data).toBe("data:text/plain;base64,aGVsbG8=");
   });
+
+  it("aborts a file send via the shared core encoder", async () => {
+    globalThis.FileReader = undefined as unknown as typeof FileReader;
+
+    const file = new File(["hello"], "a.txt", { type: "text/plain" });
+    const controller = new AbortController();
+    controller.abort();
+
+    await expect(
+      vercelAttachmentAdapter.send(
+        {
+          id: "1",
+          type: "file",
+          name: file.name,
+          file,
+          contentType: file.type,
+          content: [],
+          status: { type: "requires-action", reason: "composer-send" },
+        },
+        { signal: controller.signal },
+      ),
+    ).rejects.toBe(controller.signal.reason);
+  });
 });

package/src/adapters/vercelAttachmentAdapter.ts

@@ -15,7 +15,7 @@ export const vercelAttachmentAdapter: AttachmentAdapter = {
       status: { type: "requires-action", reason: "composer-send" },
     };
   },
-  async send(attachment) {
+  async send(attachment, options) {
     // noop
     return {
       ...attachment,
@@ -25,7 +25,7 @@ export const vercelAttachmentAdapter: AttachmentAdapter = {
           type: "file",
           mimeType: attachment.contentType ?? "",
           filename: attachment.name,
-          data: await getFileDataURL(attachment.file),
+          data: await getFileDataURL(attachment.file, options),
         },
       ],
     };