content-agent 1.2.0 → 1.2.1

package/dist/react.d.ts

@@ -2,10 +2,10 @@ import { b as OutputFormat } from "./_chunks-dts/types.js";
 import { UIMessage } from "ai";
 import { Application, Context } from "@sanity/agent-types";
 import { UseChatHelpers, UseChatOptions } from "@ai-sdk/react";
-/** `useChat` options that {@link useContentAgentChat} owns; the rest pass through. */
-type ManagedChatOptions = 'id' | 'transport' | 'resume' | 'messages' | 'onError' | 'onFinish' | 'onData';
 /** A value, or a getter resolved at send time (avoids stale closures). */
 type Resolvable<T> = T | (() => T | undefined);
+/** `useChat` options that {@link useContentAgentChat} owns; the rest pass through. */
+type ManagedChatOptions = 'id' | 'transport' | 'resume' | 'messages' | 'onError' | 'onFinish' | 'onData';
 /**
  * Options for {@link useContentAgentChat}.
  *
@@ -34,12 +34,12 @@ interface UseContentAgentChatOptions<UI_MESSAGE extends UIMessage = UIMessage> {
   /** Identifies the creating client (e.g. `'dashboard'`). */
   clientId?: string;
   /**
-   * When true, reconnect to an in-flight stream once per thread (page reload
-   * or thread switch). The SDK's `resume: true` only fires on first mount, so
-   * this hook drives resume per (Chat x threadId) pair instead.
+   * When true, reconnect to an in-flight stream once per thread (page reload).
+   * The SDK's `resume: true` only fires on first mount, so this hook drives
+   * resume per (Chat x threadId) pair instead.
    */
   autoResume?: boolean;
-  /** Initial messages (e.g. server-rendered history). */
+  /** Initial messages for a thread's first mount (e.g. server-rendered history). */
   messages?: UI_MESSAGE[];
   onError?: (error: Error) => void;
   onFinish?: () => void;
@@ -48,9 +48,10 @@ interface UseContentAgentChatOptions<UI_MESSAGE extends UIMessage = UIMessage> {
     data: unknown;
   }) => void;
   /**
-   * Any other `useChat` options to forward (e.g. `onToolCall`, `generateId`,
-   * `experimental_throttle`). The options this hook manages (transport, id,
-   * resume, messages, the callbacks above) take precedence.
+   * Any other `useChat`/`Chat` options to forward (e.g. `onToolCall`,
+   * `generateId`, `sendAutomaticallyWhen`). The options this hook manages
+   * (transport, id, messages, the callbacks above) take precedence. These are
+   * read once when a thread's Chat is created.
    */
   chatOptions?: Omit<UseChatOptions<UI_MESSAGE>, ManagedChatOptions>;
 }
@@ -61,36 +62,39 @@ interface UseContentAgentChatOptions<UI_MESSAGE extends UIMessage = UIMessage> {
  */
 interface UseSanityChatResumeOptions {
   /**
-   * When true, reconnect to an in-flight stream once per thread (page reload
-   * or thread switch). The SDK's `resume: true` only fires on first mount, so
-   * this drives resume per (Chat x threadId) pair instead.
+   * When true, reconnect to an in-flight stream once per thread. The SDK's
+   * `resume: true` only fires on first mount, so this drives resume per
+   * (Chat x threadId) pair instead.
    */
   autoResume?: boolean;
 }
 /**
- * Add Sanity resume semantics to a `useChat` instance you own: reconnect to an
- * in-flight stream once per thread, and abort the previous Chat's stream when
- * the thread changes. Use this when you need full control of the `useChat`
- * call; {@link useContentAgentChat} composes it for the turnkey path.
+ * Reconnect to an in-flight stream once per (Chat x threadId), when idle.
+ *
+ * Does not stop the stream on thread change or unmount: per the AI SDK
+ * resumable-stream model, client-side aborts are disconnects that should stay
+ * resumable, not cancellations. Stopping is an explicit user action (call
+ * `chat.stop()` and/or a server stop endpoint), never route or unmount
+ * cleanup. See https://ai-sdk.dev/docs/troubleshooting/abort-breaks-resumable-streams.
  *
  * @category React
  *
  * @example
  * ```tsx
- * const transport = useMemo(() => createSanityChatTransport({ ... }), [...])
- * const chat = useChat({ id: threadId, transport, onToolCall })
+ * const chat = useChat({ chat: myThreadChat })
  * useSanityChatResume(chat, { autoResume: threadExistsOnBackend })
  * ```
  */
-declare function useSanityChatResume<UI_MESSAGE extends UIMessage>(chat: Pick<UseChatHelpers<UI_MESSAGE>, 'id' | 'status' | 'stop' | 'resumeStream'>, options?: UseSanityChatResumeOptions): void;
+declare function useSanityChatResume<UI_MESSAGE extends UIMessage>(chat: Pick<UseChatHelpers<UI_MESSAGE>, 'id' | 'status' | 'resumeStream'>, options?: UseSanityChatResumeOptions): void;
 /**
- * `useChat` wired to the Sanity Content Agent: transport with bearer auth and
- * transparent resume, per-turn body shaping (last message wire-shaped, plus
- * application/context/format/clientId), `x-request-id` idempotency, resume on
- * thread change, and stream cleanup when the thread changes.
+ * `useChat` wired to the Sanity Content Agent: bearer auth, per-turn body
+ * shaping (application/context/format/clientId), `x-request-id` idempotency,
+ * and resume on reload.
  *
- * Returns the standard `useChat` helpers, so messages, status, sendMessage,
- * stop, resumeStream, and setMessages all behave exactly as upstream.
+ * Holds one `Chat` instance per thread (the AI SDK's recommended pattern for
+ * multiple conversations), so switching threads mid-stream and back keeps the
+ * thread's messages and its still-running stream instead of rebuilding an empty
+ * Chat. Returns the standard `useChat` helpers.
  *
  * @category React
  *

package/dist/react.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"react.d.ts","names":[],"sources":["../src/react.ts"],"mappings":";;;;;KASK,kBAAA;;KAUA,UAAA,MAAgB,CAAA,UAAW,CAAA;;;AAVT;;;UAiBN,0BAAA,oBAA8C,SAAA,GAAY,SAAA;EAP3D;EASd,cAAA;EAT8B;EAW9B,QAAA;EAX+B;AAOjC;;;EASE,OAAA;EATyE;EAWzE,UAAA;EAIc;EAFd,KAAA;EAIU;EAFV,WAAA,GAAc,UAAA,CAAW,WAAA;EAcd;EAZX,OAAA,GAAU,UAAA,CAAW,OAAA;EAqBa;EAnBlC,MAAA,GAAS,YAAA;EAmBsC;EAjB/C,QAAA;EAiBkB;;;;;EAXlB,UAAA;EAvBA;EAyBA,QAAA,GAAW,UAAA;EACX,OAAA,IAAW,KAAA,EAAO,KAAA;EAClB,QAAA;EACA,MAAA,IAAU,QAAA;IAAY,IAAA;IAAc,IAAA;EAAA;EAf1B;;;;;EAqBV,WAAA,GAAc,IAAA,CAAK,cAAA,CAAe,UAAA,GAAa,kBAAA;AAAA;;;;;;UAQhC,0BAAA;EAdO;;;;;EAoBtB,UAAA;AAAA;;;;AANF;;;;;AAwBA;;;;;;;iBAAgB,mBAAA,oBAAuC,SAAA,CAAA,CACrD,IAAA,EAAM,IAAA,CAAK,cAAA,CAAe,UAAA,+CAC1B,OAAA,GAAU,0BAAA;;;;;;;;;;;;;AA8CZ;;;;;;;;;;iBAAgB,mBAAA,oBAAuC,SAAA,GAAY,SAAA,CAAA,CACjE,OAAA,EAAS,0BAAA,CAA2B,UAAA,IACnC,cAAA,CAAe,UAAA"}
+{"version":3,"file":"react.d.ts","names":[],"sources":["../src/_internal/create-thread-chat.ts","../src/react.ts"],"mappings":";;;;AASA;AAAA,KAAY,UAAA,MAAgB,CAAA,UAAW,CAAA;;KAG3B,kBAAA;AAHZ;;;;;AAAA,UCSiB,0BAAA,oBAA8C,SAAA,GAAY,SAAA;EDTpC;ECWrC,cAAA;EDXsC;ECatC,QAAA;EDV4B;;;;ECe5B,OAAA;EATe;EAWf,UAAA;EAXyC;EAazC,KAAA;EAbyE;EAezE,WAAA,GAAc,UAAA,CAAW,WAAA;EAAX;EAEd,OAAA,GAAU,UAAA,CAAW,OAAA;EAAX;EAEV,MAAA,GAAS,YAAA;EAUE;EARX,QAAA;EAkBkC;;;;;EAZlC,UAAA;EA3B0C;EA6B1C,QAAA,GAAW,UAAA;EACX,OAAA,IAAW,KAAA,EAAO,KAAA;EAClB,QAAA;EACA,MAAA,IAAU,QAAA;IAAY,IAAA;IAAc,IAAA;EAAA;EAjBpC;;;;;;EAwBA,WAAA,GAAc,IAAA,CAAK,cAAA,CAAe,UAAA,GAAa,kBAAA;AAAA;;;;;;UAQhC,0BAAA;EAjBJ;;;;;EAuBX,UAAA;AAAA;;;;;;;AANF;;;;;AA0BA;;;;;;iBAAgB,mBAAA,oBAAuC,SAAA,CAAA,CACrD,IAAA,EAAM,IAAA,CAAK,cAAA,CAAe,UAAA,sCAC1B,OAAA,GAAU,0BAAA;;;;;;;;;;;;;;AAsCZ;;;;;;;;;;iBAAgB,mBAAA,oBAAuC,SAAA,GAAY,SAAA,CAAA,CACjE,OAAA,EAAS,0BAAA,CAA2B,UAAA,IACnC,cAAA,CAAe,UAAA"}

package/dist/react.js

@@ -1,14 +1,76 @@
-import { useChat } from "@ai-sdk/react";
-import { useRef, useMemo, useEffect } from "react";
+import { Chat, useChat } from "@ai-sdk/react";
+import { useRef, useEffect } from "react";
 import { createSanityChatTransport } from "./chat-transport.js";
 import { buildSendBody } from "./messages.js";
-function useSanityChatResume(chat, options) {
-  const { autoResume = !1 } = options ?? {}, { id: threadId, status, stop, resumeStream } = chat;
-  useEffect(() => () => {
-    stop().catch(() => {
+function getOrCreateThreadChat(cache, threadId, create) {
+  const existing = cache.get(threadId);
+  if (existing)
+    return cache.delete(threadId), cache.set(threadId, existing), existing;
+  const created = create();
+  return cache.set(threadId, created), created;
+}
+function pruneThreadChats(cache, keepId, max = 25) {
+  for (const key of [...cache.keys()]) {
+    if (cache.size <= max) break;
+    if (key === keepId) continue;
+    const evicted = cache.get(key);
+    cache.delete(key), evicted?.stop().catch(() => {
     });
-  }, [stop]);
-  const resumeEvaluatedForRef = useRef(void 0);
+  }
+}
+function createThreadChat({
+  organizationId,
+  threadId,
+  baseURL,
+  apiVersion,
+  messages,
+  chatOptions,
+  latest
+}) {
+  const transport = createSanityChatTransport({
+    organizationId,
+    threadId,
+    baseURL,
+    apiVersion,
+    token: () => resolve(latest.current.token),
+    prepareSendMessagesRequest: ({ messages: turnMessages, body, headers }) => {
+      const current = latest.current;
+      return {
+        headers: withRequestId(headers, turnMessages),
+        body: buildSendBody({
+          messages: turnMessages,
+          body,
+          application: resolve(current.application),
+          context: resolve(current.context),
+          format: current.format,
+          clientId: current.clientId
+        })
+      };
+    }
+  });
+  return new Chat({
+    ...chatOptions,
+    id: threadId,
+    transport,
+    messages,
+    onError: (error) => latest.current.onError?.(error),
+    onFinish: () => latest.current.onFinish?.(),
+    onData: (dataPart) => latest.current.onData?.(dataPart)
+  });
+}
+function resolve(value) {
+  return typeof value == "function" ? value() : value;
+}
+function withRequestId(headers, messages) {
+  const merged = {};
+  new Headers(headers).forEach((value, key) => {
+    merged[key] = value;
+  });
+  const lastUser = [...messages].reverse().find((m) => m.role === "user");
+  return lastUser && (merged["x-request-id"] = lastUser.id), merged;
+}
+function useSanityChatResume(chat, options) {
+  const { autoResume = !1 } = options ?? {}, { id: threadId, status, resumeStream } = chat, resumeEvaluatedForRef = useRef(void 0);
   useEffect(() => {
     !autoResume || !threadId || status === "ready" && resumeEvaluatedForRef.current !== threadId && (resumeEvaluatedForRef.current = threadId, resumeStream());
   }, [autoResume, threadId, status, resumeStream]);
@@ -29,54 +91,38 @@ function useContentAgentChat(options) {
     onError,
     onFinish,
     onData
-  } = options, latest = useRef({ token, application, context, format, clientId });
-  latest.current = { token, application, context, format, clientId };
-  const transport = useMemo(() => {
-    if (!(!organizationId || !threadId))
-      return createSanityChatTransport({
-        organizationId,
-        threadId,
-        baseURL,
-        apiVersion,
-        token: () => resolve(latest.current.token),
-        prepareSendMessagesRequest: ({ messages: turnMessages, body, headers }) => {
-          const current = latest.current;
-          return {
-            headers: withRequestId(headers, turnMessages),
-            body: buildSendBody({
-              messages: turnMessages,
-              body,
-              application: resolve(current.application),
-              context: resolve(current.context),
-              format: current.format,
-              clientId: current.clientId
-            })
-          };
-        }
-      });
-  }, [organizationId, threadId, baseURL, apiVersion]), chat = useChat({
-    ...options.chatOptions,
-    id: threadId,
-    transport,
-    resume: !1,
-    messages,
+  } = options, latest = useRef({
+    token,
+    application,
+    context,
+    format,
+    clientId,
     onError,
     onFinish,
     onData
   });
+  latest.current = { token, application, context, format, clientId, onError, onFinish, onData };
+  const chatsRef = useRef(/* @__PURE__ */ new Map()), fallbackRef = useRef(null);
+  fallbackRef.current ??= new Chat({});
+  const threadChat = organizationId && threadId ? getOrCreateThreadChat(
+    chatsRef.current,
+    threadId,
+    () => createThreadChat({
+      organizationId,
+      threadId,
+      baseURL,
+      apiVersion,
+      messages,
+      chatOptions: options.chatOptions,
+      latest
+    })
+  ) : fallbackRef.current;
+  useEffect(() => {
+    pruneThreadChats(chatsRef.current, threadId);
+  }, [threadId]);
+  const chat = useChat({ chat: threadChat });
   return useSanityChatResume(chat, { autoResume }), chat;
 }
-function resolve(value) {
-  return typeof value == "function" ? value() : value;
-}
-function withRequestId(headers, messages) {
-  const merged = {};
-  new Headers(headers).forEach((value, key) => {
-    merged[key] = value;
-  });
-  const lastUser = [...messages].reverse().find((m) => m.role === "user");
-  return lastUser && (merged["x-request-id"] = lastUser.id), merged;
-}
 export {
   useContentAgentChat,
   useSanityChatResume

package/dist/react.js.map

@@ -1 +1 @@
-{"version":3,"file":"react.js","sources":["../src/react.ts"],"sourcesContent":["import { type UseChatHelpers, type UseChatOptions, useChat } from '@ai-sdk/react'\nimport type { Application, Context } from '@sanity/agent-types'\nimport type { UIMessage } from 'ai'\nimport { useEffect, useMemo, useRef } from 'react'\nimport { createSanityChatTransport } from './_internal/chat-transport'\nimport { buildSendBody } from './_internal/messages'\nimport type { OutputFormat } from './types'\n\n/** `useChat` options that {@link useContentAgentChat} owns; the rest pass through. */\ntype ManagedChatOptions =\n  | 'id'\n  | 'transport'\n  | 'resume'\n  | 'messages'\n  | 'onError'\n  | 'onFinish'\n  | 'onData'\n\n/** A value, or a getter resolved at send time (avoids stale closures). */\ntype Resolvable<T> = T | (() => T | undefined)\n\n/**\n * Options for {@link useContentAgentChat}.\n *\n * @category React\n */\nexport interface UseContentAgentChatOptions<UI_MESSAGE extends UIMessage = UIMessage> {\n  /** Sanity organization ID. When undefined, no transport is built. */\n  organizationId: string | undefined\n  /** Stable thread id. When undefined, no transport is built. */\n  threadId: string | undefined\n  /**\n   * Base URL for the Sanity API.\n   * @default 'https://api.sanity.io'\n   */\n  baseURL?: string\n  /** API version segment, e.g. `'v2026-01-12'`. */\n  apiVersion?: string\n  /** Bearer token, or a getter read at request time. */\n  token?: string | (() => string | undefined)\n  /** Target application; pass a getter to read it fresh on each turn. */\n  application?: Resolvable<Application>\n  /** Per-turn request context; pass a getter to read it fresh on each turn. */\n  context?: Resolvable<Context>\n  /** Directive output format (`'directives'` for custom client rendering). */\n  format?: OutputFormat\n  /** Identifies the creating client (e.g. `'dashboard'`). */\n  clientId?: string\n  /**\n   * When true, reconnect to an in-flight stream once per thread (page reload\n   * or thread switch). The SDK's `resume: true` only fires on first mount, so\n   * this hook drives resume per (Chat x threadId) pair instead.\n   */\n  autoResume?: boolean\n  /** Initial messages (e.g. server-rendered history). */\n  messages?: UI_MESSAGE[]\n  onError?: (error: Error) => void\n  onFinish?: () => void\n  onData?: (dataPart: { type: string; data: unknown }) => void\n  /**\n   * Any other `useChat` options to forward (e.g. `onToolCall`, `generateId`,\n   * `experimental_throttle`). The options this hook manages (transport, id,\n   * resume, messages, the callbacks above) take precedence.\n   */\n  chatOptions?: Omit<UseChatOptions<UI_MESSAGE>, ManagedChatOptions>\n}\n\n/**\n * Options for {@link useSanityChatResume}.\n *\n * @category React\n */\nexport interface UseSanityChatResumeOptions {\n  /**\n   * When true, reconnect to an in-flight stream once per thread (page reload\n   * or thread switch). The SDK's `resume: true` only fires on first mount, so\n   * this drives resume per (Chat x threadId) pair instead.\n   */\n  autoResume?: boolean\n}\n\n/**\n * Add Sanity resume semantics to a `useChat` instance you own: reconnect to an\n * in-flight stream once per thread, and abort the previous Chat's stream when\n * the thread changes. Use this when you need full control of the `useChat`\n * call; {@link useContentAgentChat} composes it for the turnkey path.\n *\n * @category React\n *\n * @example\n * ```tsx\n * const transport = useMemo(() => createSanityChatTransport({ ... }), [...])\n * const chat = useChat({ id: threadId, transport, onToolCall })\n * useSanityChatResume(chat, { autoResume: threadExistsOnBackend })\n * ```\n */\nexport function useSanityChatResume<UI_MESSAGE extends UIMessage>(\n  chat: Pick<UseChatHelpers<UI_MESSAGE>, 'id' | 'status' | 'stop' | 'resumeStream'>,\n  options?: UseSanityChatResumeOptions,\n): void {\n  const { autoResume = false } = options ?? {}\n  const { id: threadId, status, stop, resumeStream } = chat\n\n  // Stop the previous Chat when the thread changes. The SDK recreates the Chat\n  // on `id` change but leaves the old fetch running; the cleanup closure\n  // captures the prior `stop`, so it aborts the OLD Chat.\n  useEffect(() => {\n    return () => void stop().catch(() => {})\n  }, [stop])\n\n  // Resume any in-flight stream once per (Chat x threadId). Skipped unless idle\n  // so a concurrent send and resume don't collide.\n  const resumeEvaluatedForRef = useRef<string | undefined>(undefined)\n  useEffect(() => {\n    if (!autoResume || !threadId) return\n    if (status !== 'ready') return\n    if (resumeEvaluatedForRef.current === threadId) return\n    resumeEvaluatedForRef.current = threadId\n    void resumeStream()\n  }, [autoResume, threadId, status, resumeStream])\n}\n\n/**\n * `useChat` wired to the Sanity Content Agent: transport with bearer auth and\n * transparent resume, per-turn body shaping (last message wire-shaped, plus\n * application/context/format/clientId), `x-request-id` idempotency, resume on\n * thread change, and stream cleanup when the thread changes.\n *\n * Returns the standard `useChat` helpers, so messages, status, sendMessage,\n * stop, resumeStream, and setMessages all behave exactly as upstream.\n *\n * @category React\n *\n * @example\n * ```tsx\n * const chat = useContentAgentChat({\n *   organizationId,\n *   threadId,\n *   token: () => client.config().token,\n *   application,\n *   autoResume: threadExistsOnBackend,\n * })\n * ```\n */\nexport function useContentAgentChat<UI_MESSAGE extends UIMessage = UIMessage>(\n  options: UseContentAgentChatOptions<UI_MESSAGE>,\n): UseChatHelpers<UI_MESSAGE> {\n  const {\n    organizationId,\n    threadId,\n    baseURL,\n    apiVersion,\n    token,\n    application,\n    context,\n    format,\n    clientId,\n    autoResume = false,\n    messages,\n    onError,\n    onFinish,\n    onData,\n  } = options\n\n  // Per-turn fields are read fresh on each send via a latest-ref so the\n  // transport (and the Chat behind it) only rebuilds on URL identity, never\n  // when the token refreshes or the user switches application/context.\n  const latest = useRef({ token, application, context, format, clientId })\n  latest.current = { token, application, context, format, clientId }\n\n  const transport = useMemo(() => {\n    if (!organizationId || !threadId) return undefined\n\n    return createSanityChatTransport<UI_MESSAGE>({\n      organizationId,\n      threadId,\n      baseURL,\n      apiVersion,\n      token: () => resolve(latest.current.token),\n      prepareSendMessagesRequest: ({ messages: turnMessages, body, headers }) => {\n        const current = latest.current\n        return {\n          headers: withRequestId(headers, turnMessages),\n          body: buildSendBody({\n            messages: turnMessages,\n            body,\n            application: resolve(current.application),\n            context: resolve(current.context),\n            format: current.format,\n            clientId: current.clientId,\n          }),\n        }\n      },\n    })\n  }, [organizationId, threadId, baseURL, apiVersion])\n\n  const chat = useChat<UI_MESSAGE>({\n    ...options.chatOptions,\n    id: threadId,\n    transport,\n    resume: false,\n    messages,\n    onError,\n    onFinish,\n    onData,\n  })\n\n  useSanityChatResume(chat, { autoResume })\n\n  return chat\n}\n\nfunction resolve<T>(value: Resolvable<T> | undefined): T | undefined {\n  return typeof value === 'function' ? (value as () => T | undefined)() : value\n}\n\n/** Spread the SDK's base headers and stamp `x-request-id` from the last user\n *  turn so the optimistic SDK message and the server share one id. */\nfunction withRequestId(headers: HeadersInit | undefined, messages: UIMessage[]): HeadersInit {\n  const merged: Record<string, string> = {}\n  new Headers(headers).forEach((value, key) => {\n    merged[key] = value\n  })\n  const lastUser = [...messages].reverse().find((m) => m.role === 'user')\n  if (lastUser) merged['x-request-id'] = lastUser.id\n  return merged\n}\n"],"names":[],"mappings":";;;;AAgGO,SAAS,oBACd,MACA,SACM;AACN,QAAM,EAAE,aAAa,GAAA,IAAU,WAAW,CAAA,GACpC,EAAE,IAAI,UAAU,QAAQ,MAAM,iBAAiB;AAKrD,YAAU,MACD,MAAG;AAAQ,SAAA,EAAO,MAAM,MAAM;AAAA,IAAC,CAAC;AAAA,EAAA,GACtC,CAAC,IAAI,CAAC;AAIT,QAAM,wBAAwB,OAA2B,MAAS;AAClE,YAAU,MAAM;AACV,KAAC,cAAc,CAAC,YAChB,WAAW,WACX,sBAAsB,YAAY,aACtC,sBAAsB,UAAU,UAC3B,aAAA;AAAA,EACP,GAAG,CAAC,YAAY,UAAU,QAAQ,YAAY,CAAC;AACjD;AAwBO,SAAS,oBACd,SAC4B;AAC5B,QAAM;AAAA,IACJ;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA,aAAa;AAAA,IACb;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EAAA,IACE,SAKE,SAAS,OAAO,EAAE,OAAO,aAAa,SAAS,QAAQ,UAAU;AACvE,SAAO,UAAU,EAAE,OAAO,aAAa,SAAS,QAAQ,SAAA;AAExD,QAAM,YAAY,QAAQ,MAAM;AAC9B,QAAI,EAAA,CAAC,kBAAkB,CAAC;AAExB,aAAO,0BAAsC;AAAA,QAC3C;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA,OAAO,MAAM,QAAQ,OAAO,QAAQ,KAAK;AAAA,QACzC,4BAA4B,CAAC,EAAE,UAAU,cAAc,MAAM,cAAc;AACzE,gBAAM,UAAU,OAAO;AACvB,iBAAO;AAAA,YACL,SAAS,cAAc,SAAS,YAAY;AAAA,YAC5C,MAAM,cAAc;AAAA,cAClB,UAAU;AAAA,cACV;AAAA,cACA,aAAa,QAAQ,QAAQ,WAAW;AAAA,cACxC,SAAS,QAAQ,QAAQ,OAAO;AAAA,cAChC,QAAQ,QAAQ;AAAA,cAChB,UAAU,QAAQ;AAAA,YAAA,CACnB;AAAA,UAAA;AAAA,QAEL;AAAA,MAAA,CACD;AAAA,EACH,GAAG,CAAC,gBAAgB,UAAU,SAAS,UAAU,CAAC,GAE5C,OAAO,QAAoB;AAAA,IAC/B,GAAG,QAAQ;AAAA,IACX,IAAI;AAAA,IACJ;AAAA,IACA,QAAQ;AAAA,IACR;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EAAA,CACD;AAED,SAAA,oBAAoB,MAAM,EAAE,WAAA,CAAY,GAEjC;AACT;AAEA,SAAS,QAAW,OAAiD;AACnE,SAAO,OAAO,SAAU,aAAc,MAAA,IAAkC;AAC1E;AAIA,SAAS,cAAc,SAAkC,UAAoC;AAC3F,QAAM,SAAiC,CAAA;AACvC,MAAI,QAAQ,OAAO,EAAE,QAAQ,CAAC,OAAO,QAAQ;AAC3C,WAAO,GAAG,IAAI;AAAA,EAChB,CAAC;AACD,QAAM,WAAW,CAAC,GAAG,QAAQ,EAAE,QAAA,EAAU,KAAK,CAAC,MAAM,EAAE,SAAS,MAAM;AACtE,SAAI,aAAU,OAAO,cAAc,IAAI,SAAS,KACzC;AACT;"}
+{"version":3,"file":"react.js","sources":["../src/_internal/chat-cache.ts","../src/_internal/create-thread-chat.ts","../src/react.ts"],"sourcesContent":["import type { Chat } from '@ai-sdk/react'\nimport type { UIMessage } from 'ai'\n\n/** Cap on cached per-thread Chat instances, to bound per-session memory. */\nexport const MAX_CACHED_CHATS = 25\n\n/**\n * Get the cached Chat for a thread, creating it via `create` on a miss.\n *\n * This is the building block that lets a thread survive being switched away\n * from: `useChat` rebuilds an empty Chat whenever its `id` changes, so the\n * turnkey hook holds one Chat per thread here and hands the same instance back\n * on return, preserving its messages and its still-running stream (SAGE-502).\n *\n * The accessed/created thread becomes most-recently-used (Map preserves\n * insertion order). Render-safe: it never aborts a stream. Call\n * {@link pruneThreadChats} from an effect to bound the cache.\n */\nexport function getOrCreateThreadChat<UI_MESSAGE extends UIMessage>(\n  cache: Map<string, Chat<UI_MESSAGE>>,\n  threadId: string,\n  create: () => Chat<UI_MESSAGE>,\n): Chat<UI_MESSAGE> {\n  const existing = cache.get(threadId)\n  if (existing) {\n    cache.delete(threadId)\n    cache.set(threadId, existing)\n    return existing\n  }\n\n  const created = create()\n  cache.set(threadId, created)\n  return created\n}\n\n/**\n * Evict least-recently-used chats beyond `max`, stopping each to free its\n * connection, while keeping `keepId`.\n *\n * Call this from an effect, never during render: `stop()` aborts a fetch, and\n * render may run speculatively.\n */\nexport function pruneThreadChats<UI_MESSAGE extends UIMessage>(\n  cache: Map<string, Chat<UI_MESSAGE>>,\n  keepId: string | undefined,\n  max: number = MAX_CACHED_CHATS,\n): void {\n  for (const key of [...cache.keys()]) {\n    if (cache.size <= max) break\n    if (key === keepId) continue\n    const evicted = cache.get(key)\n    cache.delete(key)\n    void evicted?.stop().catch(() => {})\n  }\n}\n","import { Chat, type UseChatOptions } from '@ai-sdk/react'\nimport type { Application, Context } from '@sanity/agent-types'\nimport type { UIMessage } from 'ai'\nimport type { RefObject } from 'react'\nimport type { OutputFormat } from '../types'\nimport { createSanityChatTransport } from './chat-transport'\nimport { buildSendBody } from './messages'\n\n/** A value, or a getter resolved at send time (avoids stale closures). */\nexport type Resolvable<T> = T | (() => T | undefined)\n\n/** `useChat` options that {@link useContentAgentChat} owns; the rest pass through. */\nexport type ManagedChatOptions =\n  | 'id'\n  | 'transport'\n  | 'resume'\n  | 'messages'\n  | 'onError'\n  | 'onFinish'\n  | 'onData'\n\n/**\n * Per-turn config and callbacks read fresh on each send/event via a ref, so a\n * Chat created once per thread still sees the current values rather than the\n * ones captured when it was created.\n */\nexport interface LatestConfig {\n  token?: string | (() => string | undefined)\n  application?: Resolvable<Application>\n  context?: Resolvable<Context>\n  format?: OutputFormat\n  clientId?: string\n  onError?: (error: Error) => void\n  onFinish?: () => void\n  onData?: (dataPart: { type: string; data: unknown }) => void\n}\n\nexport interface CreateThreadChatArgs<UI_MESSAGE extends UIMessage> {\n  organizationId: string\n  threadId: string\n  baseURL?: string\n  apiVersion?: string\n  messages?: UI_MESSAGE[]\n  chatOptions?: Omit<UseChatOptions<UI_MESSAGE>, ManagedChatOptions>\n  latest: RefObject<LatestConfig>\n}\n\n/**\n * Build the `Chat` for one thread: a Sanity transport (bearer auth, per-turn\n * body shaping, `x-request-id` idempotency) plus callbacks that read the latest\n * config. Called once per thread; the instance is then cached and reused.\n */\nexport function createThreadChat<UI_MESSAGE extends UIMessage>({\n  organizationId,\n  threadId,\n  baseURL,\n  apiVersion,\n  messages,\n  chatOptions,\n  latest,\n}: CreateThreadChatArgs<UI_MESSAGE>): Chat<UI_MESSAGE> {\n  const transport = createSanityChatTransport<UI_MESSAGE>({\n    organizationId,\n    threadId,\n    baseURL,\n    apiVersion,\n    token: () => resolve(latest.current.token),\n    prepareSendMessagesRequest: ({ messages: turnMessages, body, headers }) => {\n      const current = latest.current\n      return {\n        headers: withRequestId(headers, turnMessages),\n        body: buildSendBody({\n          messages: turnMessages,\n          body,\n          application: resolve(current.application),\n          context: resolve(current.context),\n          format: current.format,\n          clientId: current.clientId,\n        }),\n      }\n    },\n  })\n\n  return new Chat<UI_MESSAGE>({\n    ...chatOptions,\n    id: threadId,\n    transport,\n    messages,\n    onError: (error) => latest.current.onError?.(error),\n    onFinish: () => latest.current.onFinish?.(),\n    onData: (dataPart) => latest.current.onData?.(dataPart),\n  })\n}\n\nfunction resolve<T>(value: Resolvable<T> | undefined): T | undefined {\n  return typeof value === 'function' ? (value as () => T | undefined)() : value\n}\n\n/** Spread the SDK's base headers and stamp `x-request-id` from the last user\n *  turn so the optimistic SDK message and the server share one id. */\nfunction withRequestId(headers: HeadersInit | undefined, messages: UIMessage[]): HeadersInit {\n  const merged: Record<string, string> = {}\n  new Headers(headers).forEach((value, key) => {\n    merged[key] = value\n  })\n  const lastUser = [...messages].reverse().find((m) => m.role === 'user')\n  if (lastUser) merged['x-request-id'] = lastUser.id\n  return merged\n}\n","import { Chat, type UseChatHelpers, type UseChatOptions, useChat } from '@ai-sdk/react'\nimport type { Application, Context } from '@sanity/agent-types'\nimport type { UIMessage } from 'ai'\nimport { useEffect, useRef } from 'react'\nimport { getOrCreateThreadChat, pruneThreadChats } from './_internal/chat-cache'\nimport {\n  createThreadChat,\n  type LatestConfig,\n  type ManagedChatOptions,\n  type Resolvable,\n} from './_internal/create-thread-chat'\nimport type { OutputFormat } from './types'\n\n/**\n * Options for {@link useContentAgentChat}.\n *\n * @category React\n */\nexport interface UseContentAgentChatOptions<UI_MESSAGE extends UIMessage = UIMessage> {\n  /** Sanity organization ID. When undefined, no transport is built. */\n  organizationId: string | undefined\n  /** Stable thread id. When undefined, no transport is built. */\n  threadId: string | undefined\n  /**\n   * Base URL for the Sanity API.\n   * @default 'https://api.sanity.io'\n   */\n  baseURL?: string\n  /** API version segment, e.g. `'v2026-01-12'`. */\n  apiVersion?: string\n  /** Bearer token, or a getter read at request time. */\n  token?: string | (() => string | undefined)\n  /** Target application; pass a getter to read it fresh on each turn. */\n  application?: Resolvable<Application>\n  /** Per-turn request context; pass a getter to read it fresh on each turn. */\n  context?: Resolvable<Context>\n  /** Directive output format (`'directives'` for custom client rendering). */\n  format?: OutputFormat\n  /** Identifies the creating client (e.g. `'dashboard'`). */\n  clientId?: string\n  /**\n   * When true, reconnect to an in-flight stream once per thread (page reload).\n   * The SDK's `resume: true` only fires on first mount, so this hook drives\n   * resume per (Chat x threadId) pair instead.\n   */\n  autoResume?: boolean\n  /** Initial messages for a thread's first mount (e.g. server-rendered history). */\n  messages?: UI_MESSAGE[]\n  onError?: (error: Error) => void\n  onFinish?: () => void\n  onData?: (dataPart: { type: string; data: unknown }) => void\n  /**\n   * Any other `useChat`/`Chat` options to forward (e.g. `onToolCall`,\n   * `generateId`, `sendAutomaticallyWhen`). The options this hook manages\n   * (transport, id, messages, the callbacks above) take precedence. These are\n   * read once when a thread's Chat is created.\n   */\n  chatOptions?: Omit<UseChatOptions<UI_MESSAGE>, ManagedChatOptions>\n}\n\n/**\n * Options for {@link useSanityChatResume}.\n *\n * @category React\n */\nexport interface UseSanityChatResumeOptions {\n  /**\n   * When true, reconnect to an in-flight stream once per thread. The SDK's\n   * `resume: true` only fires on first mount, so this drives resume per\n   * (Chat x threadId) pair instead.\n   */\n  autoResume?: boolean\n}\n\n/**\n * Reconnect to an in-flight stream once per (Chat x threadId), when idle.\n *\n * Does not stop the stream on thread change or unmount: per the AI SDK\n * resumable-stream model, client-side aborts are disconnects that should stay\n * resumable, not cancellations. Stopping is an explicit user action (call\n * `chat.stop()` and/or a server stop endpoint), never route or unmount\n * cleanup. See https://ai-sdk.dev/docs/troubleshooting/abort-breaks-resumable-streams.\n *\n * @category React\n *\n * @example\n * ```tsx\n * const chat = useChat({ chat: myThreadChat })\n * useSanityChatResume(chat, { autoResume: threadExistsOnBackend })\n * ```\n */\nexport function useSanityChatResume<UI_MESSAGE extends UIMessage>(\n  chat: Pick<UseChatHelpers<UI_MESSAGE>, 'id' | 'status' | 'resumeStream'>,\n  options?: UseSanityChatResumeOptions,\n): void {\n  const { autoResume = false } = options ?? {}\n  const { id: threadId, status, resumeStream } = chat\n\n  const resumeEvaluatedForRef = useRef<string | undefined>(undefined)\n  useEffect(() => {\n    if (!autoResume || !threadId) return\n    if (status !== 'ready') return\n    if (resumeEvaluatedForRef.current === threadId) return\n    resumeEvaluatedForRef.current = threadId\n    void resumeStream()\n  }, [autoResume, threadId, status, resumeStream])\n}\n\n/**\n * `useChat` wired to the Sanity Content Agent: bearer auth, per-turn body\n * shaping (application/context/format/clientId), `x-request-id` idempotency,\n * and resume on reload.\n *\n * Holds one `Chat` instance per thread (the AI SDK's recommended pattern for\n * multiple conversations), so switching threads mid-stream and back keeps the\n * thread's messages and its still-running stream instead of rebuilding an empty\n * Chat. Returns the standard `useChat` helpers.\n *\n * @category React\n *\n * @example\n * ```tsx\n * const chat = useContentAgentChat({\n *   organizationId,\n *   threadId,\n *   token: () => client.config().token,\n *   application,\n *   autoResume: threadExistsOnBackend,\n * })\n * ```\n */\nexport function useContentAgentChat<UI_MESSAGE extends UIMessage = UIMessage>(\n  options: UseContentAgentChatOptions<UI_MESSAGE>,\n): UseChatHelpers<UI_MESSAGE> {\n  const {\n    organizationId,\n    threadId,\n    baseURL,\n    apiVersion,\n    token,\n    application,\n    context,\n    format,\n    clientId,\n    autoResume = false,\n    messages,\n    onError,\n    onFinish,\n    onData,\n  } = options\n\n  const latest = useRef<LatestConfig>({\n    token,\n    application,\n    context,\n    format,\n    clientId,\n    onError,\n    onFinish,\n    onData,\n  })\n  latest.current = { token, application, context, format, clientId, onError, onFinish, onData }\n\n  // One Chat per thread, retained across switches (see getOrCreateThreadChat).\n  const chatsRef = useRef(new Map<string, Chat<UI_MESSAGE>>())\n  // Stable placeholder used until a thread exists, so the hook's shape holds.\n  const fallbackRef = useRef<Chat<UI_MESSAGE> | null>(null)\n  fallbackRef.current ??= new Chat<UI_MESSAGE>({})\n\n  const threadChat =\n    organizationId && threadId\n      ? getOrCreateThreadChat(chatsRef.current, threadId, () =>\n          createThreadChat<UI_MESSAGE>({\n            organizationId,\n            threadId,\n            baseURL,\n            apiVersion,\n            messages,\n            chatOptions: options.chatOptions,\n            latest,\n          }),\n        )\n      : fallbackRef.current\n\n  // Bound the cache after commit, not in render: pruning aborts evicted streams.\n  useEffect(() => {\n    pruneThreadChats(chatsRef.current, threadId)\n  }, [threadId])\n\n  const chat = useChat<UI_MESSAGE>({ chat: threadChat })\n\n  useSanityChatResume(chat, { autoResume })\n\n  return chat\n}\n"],"names":[],"mappings":";;;;AAkBO,SAAS,sBACd,OACA,UACA,QACkB;AAClB,QAAM,WAAW,MAAM,IAAI,QAAQ;AACnC,MAAI;AACF,WAAA,MAAM,OAAO,QAAQ,GACrB,MAAM,IAAI,UAAU,QAAQ,GACrB;AAGT,QAAM,UAAU,OAAA;AAChB,SAAA,MAAM,IAAI,UAAU,OAAO,GACpB;AACT;AASO,SAAS,iBACd,OACA,QACA,MAAc,IACR;AACN,aAAW,OAAO,CAAC,GAAG,MAAM,KAAA,CAAM,GAAG;AACnC,QAAI,MAAM,QAAQ,IAAK;AACvB,QAAI,QAAQ,OAAQ;AACpB,UAAM,UAAU,MAAM,IAAI,GAAG;AAC7B,UAAM,OAAO,GAAG,GACX,SAAS,KAAA,EAAO,MAAM,MAAM;AAAA,IAAC,CAAC;AAAA,EACrC;AACF;ACFO,SAAS,iBAA+C;AAAA,EAC7D;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,GAAuD;AACrD,QAAM,YAAY,0BAAsC;AAAA,IACtD;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA,OAAO,MAAM,QAAQ,OAAO,QAAQ,KAAK;AAAA,IACzC,4BAA4B,CAAC,EAAE,UAAU,cAAc,MAAM,cAAc;AACzE,YAAM,UAAU,OAAO;AACvB,aAAO;AAAA,QACL,SAAS,cAAc,SAAS,YAAY;AAAA,QAC5C,MAAM,cAAc;AAAA,UAClB,UAAU;AAAA,UACV;AAAA,UACA,aAAa,QAAQ,QAAQ,WAAW;AAAA,UACxC,SAAS,QAAQ,QAAQ,OAAO;AAAA,UAChC,QAAQ,QAAQ;AAAA,UAChB,UAAU,QAAQ;AAAA,QAAA,CACnB;AAAA,MAAA;AAAA,IAEL;AAAA,EAAA,CACD;AAED,SAAO,IAAI,KAAiB;AAAA,IAC1B,GAAG;AAAA,IACH,IAAI;AAAA,IACJ;AAAA,IACA;AAAA,IACA,SAAS,CAAC,UAAU,OAAO,QAAQ,UAAU,KAAK;AAAA,IAClD,UAAU,MAAM,OAAO,QAAQ,WAAA;AAAA,IAC/B,QAAQ,CAAC,aAAa,OAAO,QAAQ,SAAS,QAAQ;AAAA,EAAA,CACvD;AACH;AAEA,SAAS,QAAW,OAAiD;AACnE,SAAO,OAAO,SAAU,aAAc,MAAA,IAAkC;AAC1E;AAIA,SAAS,cAAc,SAAkC,UAAoC;AAC3F,QAAM,SAAiC,CAAA;AACvC,MAAI,QAAQ,OAAO,EAAE,QAAQ,CAAC,OAAO,QAAQ;AAC3C,WAAO,GAAG,IAAI;AAAA,EAChB,CAAC;AACD,QAAM,WAAW,CAAC,GAAG,QAAQ,EAAE,QAAA,EAAU,KAAK,CAAC,MAAM,EAAE,SAAS,MAAM;AACtE,SAAI,aAAU,OAAO,cAAc,IAAI,SAAS,KACzC;AACT;ACjBO,SAAS,oBACd,MACA,SACM;AACN,QAAM,EAAE,aAAa,GAAA,IAAU,WAAW,CAAA,GACpC,EAAE,IAAI,UAAU,QAAQ,aAAA,IAAiB,MAEzC,wBAAwB,OAA2B,MAAS;AAClE,YAAU,MAAM;AACV,KAAC,cAAc,CAAC,YAChB,WAAW,WACX,sBAAsB,YAAY,aACtC,sBAAsB,UAAU,UAC3B,aAAA;AAAA,EACP,GAAG,CAAC,YAAY,UAAU,QAAQ,YAAY,CAAC;AACjD;AAyBO,SAAS,oBACd,SAC4B;AAC5B,QAAM;AAAA,IACJ;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA,aAAa;AAAA,IACb;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EAAA,IACE,SAEE,SAAS,OAAqB;AAAA,IAClC;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EAAA,CACD;AACD,SAAO,UAAU,EAAE,OAAO,aAAa,SAAS,QAAQ,UAAU,SAAS,UAAU,OAAA;AAGrF,QAAM,WAAW,OAAO,oBAAI,IAAA,CAA+B,GAErD,cAAc,OAAgC,IAAI;AACxD,cAAY,YAAY,IAAI,KAAiB,EAAE;AAE/C,QAAM,aACJ,kBAAkB,WACd;AAAA,IAAsB,SAAS;AAAA,IAAS;AAAA,IAAU,MAChD,iBAA6B;AAAA,MAC3B;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA,aAAa,QAAQ;AAAA,MACrB;AAAA,IAAA,CACD;AAAA,EAAA,IAEH,YAAY;AAGlB,YAAU,MAAM;AACd,qBAAiB,SAAS,SAAS,QAAQ;AAAA,EAC7C,GAAG,CAAC,QAAQ,CAAC;AAEb,QAAM,OAAO,QAAoB,EAAE,MAAM,YAAY;AAErD,SAAA,oBAAoB,MAAM,EAAE,WAAA,CAAY,GAEjC;AACT;"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "content-agent",
-  "version": "1.2.0",
+  "version": "1.2.1",
   "description": "Vercel AI SDK provider for Sanity Content Agent API",
   "keywords": [
     "sanity",
@@ -59,8 +59,8 @@
     "typedoc": "^0.28.17",
     "typescript": "^5.9.3",
     "vitest": "^4.0.18",
-    "@repo/package.config": "0.0.0",
-    "@repo/tsconfig": "0.0.0"
+    "@repo/tsconfig": "0.0.0",
+    "@repo/package.config": "0.0.0"
   },
   "peerDependencies": {
     "@ai-sdk/react": ">=3.0.0",