@assistant-ui/react-ai-sdk 1.3.13 → 1.3.14

package/dist/index.d.ts

@@ -1,7 +1,11 @@
 /// <reference types="@assistant-ui/core/store" />
 /// <reference types="@assistant-ui/core/react" />
-export * from "./ui/index.js";
+export { useAISDKRuntime } from "./ui/use-chat/useAISDKRuntime.js";
+export { useChatRuntime } from "./ui/use-chat/useChatRuntime.js";
+export type { UseChatRuntimeOptions } from "./ui/use-chat/useChatRuntime.js";
+export { AssistantChatTransport } from "./ui/use-chat/AssistantChatTransport.js";
 export { frontendTools } from "./frontendTools.js";
+export { injectQuoteContext } from "./injectQuoteContext.js";
 export type { ThreadTokenUsage, TokenUsageExtractableMessage } from "./usage.js";
 export { getThreadMessageTokenUsage, useThreadTokenUsage } from "./usage.js";
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAGA,8BAAqB;AACrB,OAAO,EAAE,aAAa,EAAE,2BAAwB;AAChD,YAAY,EAAE,gBAAgB,EAAE,4BAA4B,EAAE,mBAAgB;AAC9E,OAAO,EAAE,0BAA0B,EAAE,mBAAmB,EAAE,mBAAgB"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,eAAe,EAAE,yCAAsC;AAChE,OAAO,EAAE,cAAc,EAAE,wCAAqC;AAC9D,YAAY,EAAE,qBAAqB,EAAE,wCAAqC;AAC1E,OAAO,EAAE,sBAAsB,EAAE,gDAA6C;AAC9E,OAAO,EAAE,aAAa,EAAE,2BAAwB;AAChD,OAAO,EAAE,kBAAkB,EAAE,gCAA6B;AAC1D,YAAY,EAAE,gBAAgB,EAAE,4BAA4B,EAAE,mBAAgB;AAC9E,OAAO,EAAE,0BAA0B,EAAE,mBAAmB,EAAE,mBAAgB"}

package/dist/index.js

@@ -1,6 +1,9 @@
 /// <reference types="@assistant-ui/core/store" />
 /// <reference types="@assistant-ui/core/react" />
-export * from "./ui/index.js";
+export { useAISDKRuntime } from "./ui/use-chat/useAISDKRuntime.js";
+export { useChatRuntime } from "./ui/use-chat/useChatRuntime.js";
+export { AssistantChatTransport } from "./ui/use-chat/AssistantChatTransport.js";
 export { frontendTools } from "./frontendTools.js";
+export { injectQuoteContext } from "./injectQuoteContext.js";
 export { getThreadMessageTokenUsage, useThreadTokenUsage } from "./usage.js";
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,kDAAkD;AAClD,kDAAkD;AAElD,8BAAqB;AACrB,OAAO,EAAE,aAAa,EAAE,2BAAwB;AAEhD,OAAO,EAAE,0BAA0B,EAAE,mBAAmB,EAAE,mBAAgB"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,kDAAkD;AAClD,kDAAkD;AAElD,OAAO,EAAE,eAAe,EAAE,yCAAsC;AAChE,OAAO,EAAE,cAAc,EAAE,wCAAqC;AAE9D,OAAO,EAAE,sBAAsB,EAAE,gDAA6C;AAC9E,OAAO,EAAE,aAAa,EAAE,2BAAwB;AAChD,OAAO,EAAE,kBAAkB,EAAE,gCAA6B;AAE1D,OAAO,EAAE,0BAA0B,EAAE,mBAAmB,EAAE,mBAAgB"}

package/dist/injectQuoteContext.d.ts

@@ -0,0 +1,24 @@
+import type { UIMessage } from "ai";
+/**
+ * Injects quote context into messages as markdown blockquotes.
+ *
+ * Use this in your route handler before `convertToModelMessages` so the LLM
+ * sees the quoted text that the user is referring to.
+ *
+ * @example
+ * ```ts
+ * import { convertToModelMessages, streamText } from "ai";
+ * import { injectQuoteContext } from "@assistant-ui/react-ai-sdk";
+ *
+ * export async function POST(req: Request) {
+ *   const { messages } = await req.json();
+ *   const result = streamText({
+ *     model: myModel,
+ *     messages: await convertToModelMessages(injectQuoteContext(messages)),
+ *   });
+ *   return result.toUIMessageStreamResponse();
+ * }
+ * ```
+ */
+export declare function injectQuoteContext(messages: UIMessage[]): UIMessage[];
+//# sourceMappingURL=injectQuoteContext.d.ts.map

package/dist/injectQuoteContext.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"injectQuoteContext.d.ts","sourceRoot":"","sources":["../src/injectQuoteContext.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,IAAI,CAAC;AAepC;;;;;;;;;;;;;;;;;;;;GAoBG;AACH,wBAAgB,kBAAkB,CAAC,QAAQ,EAAE,SAAS,EAAE,GAAG,SAAS,EAAE,CAyBrE"}

package/dist/injectQuoteContext.js

@@ -0,0 +1,58 @@
+const getQuoteText = (metadata) => {
+    if (!metadata || typeof metadata !== "object")
+        return undefined;
+    const custom = metadata.custom;
+    if (!custom || typeof custom !== "object")
+        return undefined;
+    const quote = custom.quote;
+    if (!quote || typeof quote !== "object")
+        return undefined;
+    const text = quote.text;
+    return typeof text === "string" ? text : undefined;
+};
+/**
+ * Injects quote context into messages as markdown blockquotes.
+ *
+ * Use this in your route handler before `convertToModelMessages` so the LLM
+ * sees the quoted text that the user is referring to.
+ *
+ * @example
+ * ```ts
+ * import { convertToModelMessages, streamText } from "ai";
+ * import { injectQuoteContext } from "@assistant-ui/react-ai-sdk";
+ *
+ * export async function POST(req: Request) {
+ *   const { messages } = await req.json();
+ *   const result = streamText({
+ *     model: myModel,
+ *     messages: await convertToModelMessages(injectQuoteContext(messages)),
+ *   });
+ *   return result.toUIMessageStreamResponse();
+ * }
+ * ```
+ */
+export function injectQuoteContext(messages) {
+    return messages.map((msg) => {
+        if (msg.role !== "user")
+            return msg;
+        const text = getQuoteText(msg.metadata);
+        if (!text)
+            return msg;
+        const blockquote = text
+            .split(/\r?\n/)
+            .map((line) => `> ${line}`)
+            .join("\n");
+        const alreadyInjected = msg.parts[0]?.type === "text" &&
+            msg.parts[0].text === `${blockquote}\n\n`;
+        if (alreadyInjected)
+            return msg;
+        return {
+            ...msg,
+            parts: [
+                { type: "text", text: `${blockquote}\n\n` },
+                ...(msg.parts ?? []),
+            ],
+        };
+    });
+}
+//# sourceMappingURL=injectQuoteContext.js.map

package/dist/injectQuoteContext.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"injectQuoteContext.js","sourceRoot":"","sources":["../src/injectQuoteContext.ts"],"names":[],"mappings":"AAEA,MAAM,YAAY,GAAG,CAAC,QAAiB,EAAsB,EAAE;IAC7D,IAAI,CAAC,QAAQ,IAAI,OAAO,QAAQ,KAAK,QAAQ;QAAE,OAAO,SAAS,CAAC;IAEhE,MAAM,MAAM,GAAI,QAAoC,CAAC,MAAM,CAAC;IAC5D,IAAI,CAAC,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ;QAAE,OAAO,SAAS,CAAC;IAE5D,MAAM,KAAK,GAAI,MAAkC,CAAC,KAAK,CAAC;IACxD,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,SAAS,CAAC;IAE1D,MAAM,IAAI,GAAI,KAAiC,CAAC,IAAI,CAAC;IACrD,OAAO,OAAO,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC;AACrD,CAAC,CAAC;AAEF;;;;;;;;;;;;;;;;;;;;GAoBG;AACH,MAAM,UAAU,kBAAkB,CAAC,QAAqB;IACtD,OAAO,QAAQ,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;QAC1B,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM;YAAE,OAAO,GAAG,CAAC;QAEpC,MAAM,IAAI,GAAG,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACxC,IAAI,CAAC,IAAI;YAAE,OAAO,GAAG,CAAC;QAEtB,MAAM,UAAU,GAAG,IAAI;aACpB,KAAK,CAAC,OAAO,CAAC;aACd,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC;aAC1B,IAAI,CAAC,IAAI,CAAC,CAAC;QAEd,MAAM,eAAe,GACnB,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,IAAI,KAAK,MAAM;YAC7B,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,GAAG,UAAU,MAAM,CAAC;QAC5C,IAAI,eAAe;YAAE,OAAO,GAAG,CAAC;QAEhC,OAAO;YACL,GAAG,GAAG;YACN,KAAK,EAAE;gBACL,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,GAAG,UAAU,MAAM,EAAE;gBACpD,GAAG,CAAC,GAAG,CAAC,KAAK,IAAI,EAAE,CAAC;aACrB;SACF,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC"}

package/dist/ui/use-chat/useChatRuntime.js

@@ -1,6 +1,6 @@
 "use client";
 import { useChat } from "@ai-sdk/react";
-import { useCloudThreadListAdapter as unstable_useCloudThreadListAdapter, useRemoteThreadListRuntime as unstable_useRemoteThreadListRuntime, } from "@assistant-ui/core/react";
+import { useCloudThreadListAdapter, useRemoteThreadListRuntime, } from "@assistant-ui/core/react";
 import { useAuiState } from "@assistant-ui/store";
 import { useAISDKRuntime, } from "./useAISDKRuntime.js";
 import { AssistantChatTransport } from "./AssistantChatTransport.js";
@@ -39,8 +39,8 @@ const useChatThreadRuntime = (options) => {
     return runtime;
 };
 export const useChatRuntime = ({ cloud, ...options } = {}) => {
-    const cloudAdapter = unstable_useCloudThreadListAdapter({ cloud });
-    return unstable_useRemoteThreadListRuntime({
+    const cloudAdapter = useCloudThreadListAdapter({ cloud });
+    return useRemoteThreadListRuntime({
         runtimeHook: function RuntimeHook() {
             return useChatThreadRuntime(options);
         },

package/dist/ui/use-chat/useChatRuntime.js.map

@@ -1 +1 @@
-{"version":3,"file":"useChatRuntime.js","sourceRoot":"","sources":["../../../src/ui/use-chat/useChatRuntime.ts"],"names":[],"mappings":"AAAA,YAAY,CAAC;AAEb,OAAO,EAAE,OAAO,EAAkB,MAAM,eAAe,CAAC;AAGxD,OAAO,EACL,yBAAyB,IAAI,kCAAkC,EAC/D,0BAA0B,IAAI,mCAAmC,GAClE,MAAM,0BAA0B,CAAC;AAClC,OAAO,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAClD,OAAO,EACL,eAAe,GAGhB,6BAA0B;AAE3B,OAAO,EAAE,sBAAsB,EAAE,oCAAiC;AAClE,OAAO,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,OAAO,CAAC;AASnD,MAAM,uBAAuB,GAAG,CAC9B,SAAoC,EACT,EAAE;IAC7B,MAAM,YAAY,GAAG,MAAM,CAA4B,SAAS,CAAC,CAAC;IAClE,SAAS,CAAC,GAAG,EAAE;QACb,YAAY,CAAC,OAAO,GAAG,SAAS,CAAC;IACnC,CAAC,CAAC,CAAC;IACH,MAAM,gBAAgB,GAAG,OAAO,CAC9B,GAAG,EAAE,CACH,IAAI,KAAK,CAAC,YAAY,CAAC,OAAO,EAAE;QAC9B,GAAG,CAAC,CAAC,EAAE,IAAI;YACT,MAAM,GAAG,GACP,YAAY,CAAC,OAAO,CAAC,IAAuC,CAAC,CAAC;YAChE,OAAO,OAAO,GAAG,KAAK,UAAU;gBAC9B,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC;gBAChC,CAAC,CAAC,GAAG,CAAC;QACV,CAAC;KACF,CAAC,EACJ,EAAE,CACH,CAAC;IACF,OAAO,gBAAgB,CAAC;AAC1B,CAAC,CAAC;AAEF,MAAM,oBAAoB,GAAG,CAC3B,OAA2C,EACzB,EAAE;IACpB,MAAM,EACJ,QAAQ,EACR,SAAS,EAAE,gBAAgB,EAC3B,eAAe,EACf,GAAG,WAAW,EACf,GAAG,OAAO,IAAI,EAAE,CAAC;IAElB,MAAM,SAAS,GAAG,uBAAuB,CACvC,gBAAgB,IAAI,IAAI,sBAAsB,EAAE,CACjD,CAAC;IAEF,MAAM,EAAE,GAAG,WAAW,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,cAAc,CAAC,EAAE,CAAC,CAAC;IACnD,MAAM,IAAI,GAAG,OAAO,CAAC;QACnB,GAAG,WAAW;QACd,EAAE;QACF,SAAS;KACV,CAAC,CAAC;IAEH,MAAM,OAAO,GAAG,eAAe,CAAC,IAAI,EAAE;QACpC,QAAQ;QACR,GAAG,CAAC,eAAe,IAAI,EAAE,eAAe,EAAE,CAAC;KAC5C,CAAC,CAAC;IAEH,IAAI,SAAS,YAAY,sBAAsB,EAAE,CAAC;QAChD,SAAS,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;IAChC,CAAC;IAED,OAAO,OAAO,CAAC;AACjB,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,cAAc,GAAG,CAA2C,EACvE,KAAK,EACL,GAAG,OAAO,KAC2B,EAAE,EAAoB,EAAE;IAC7D,MAAM,YAAY,GAAG,kCAAkC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;IACnE,OAAO,mCAAmC,CAAC;QACzC,WAAW,EAAE,SAAS,WAAW;YAC/B,OAAO,oBAAoB,CAAC,OAAO,CAAC,CAAC;QACvC,CAAC;QACD,OAAO,EAAE,YAAY;QACrB,YAAY,EAAE,IAAI;KACnB,CAAC,CAAC;AACL,CAAC,CAAC"}
+{"version":3,"file":"useChatRuntime.js","sourceRoot":"","sources":["../../../src/ui/use-chat/useChatRuntime.ts"],"names":[],"mappings":"AAAA,YAAY,CAAC;AAEb,OAAO,EAAE,OAAO,EAAkB,MAAM,eAAe,CAAC;AAGxD,OAAO,EACL,yBAAyB,EACzB,0BAA0B,GAC3B,MAAM,0BAA0B,CAAC;AAClC,OAAO,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAClD,OAAO,EACL,eAAe,GAGhB,6BAA0B;AAE3B,OAAO,EAAE,sBAAsB,EAAE,oCAAiC;AAClE,OAAO,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,OAAO,CAAC;AASnD,MAAM,uBAAuB,GAAG,CAC9B,SAAoC,EACT,EAAE;IAC7B,MAAM,YAAY,GAAG,MAAM,CAA4B,SAAS,CAAC,CAAC;IAClE,SAAS,CAAC,GAAG,EAAE;QACb,YAAY,CAAC,OAAO,GAAG,SAAS,CAAC;IACnC,CAAC,CAAC,CAAC;IACH,MAAM,gBAAgB,GAAG,OAAO,CAC9B,GAAG,EAAE,CACH,IAAI,KAAK,CAAC,YAAY,CAAC,OAAO,EAAE;QAC9B,GAAG,CAAC,CAAC,EAAE,IAAI;YACT,MAAM,GAAG,GACP,YAAY,CAAC,OAAO,CAAC,IAAuC,CAAC,CAAC;YAChE,OAAO,OAAO,GAAG,KAAK,UAAU;gBAC9B,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC;gBAChC,CAAC,CAAC,GAAG,CAAC;QACV,CAAC;KACF,CAAC,EACJ,EAAE,CACH,CAAC;IACF,OAAO,gBAAgB,CAAC;AAC1B,CAAC,CAAC;AAEF,MAAM,oBAAoB,GAAG,CAC3B,OAA2C,EACzB,EAAE;IACpB,MAAM,EACJ,QAAQ,EACR,SAAS,EAAE,gBAAgB,EAC3B,eAAe,EACf,GAAG,WAAW,EACf,GAAG,OAAO,IAAI,EAAE,CAAC;IAElB,MAAM,SAAS,GAAG,uBAAuB,CACvC,gBAAgB,IAAI,IAAI,sBAAsB,EAAE,CACjD,CAAC;IAEF,MAAM,EAAE,GAAG,WAAW,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,cAAc,CAAC,EAAE,CAAC,CAAC;IACnD,MAAM,IAAI,GAAG,OAAO,CAAC;QACnB,GAAG,WAAW;QACd,EAAE;QACF,SAAS;KACV,CAAC,CAAC;IAEH,MAAM,OAAO,GAAG,eAAe,CAAC,IAAI,EAAE;QACpC,QAAQ;QACR,GAAG,CAAC,eAAe,IAAI,EAAE,eAAe,EAAE,CAAC;KAC5C,CAAC,CAAC;IAEH,IAAI,SAAS,YAAY,sBAAsB,EAAE,CAAC;QAChD,SAAS,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;IAChC,CAAC;IAED,OAAO,OAAO,CAAC;AACjB,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,cAAc,GAAG,CAA2C,EACvE,KAAK,EACL,GAAG,OAAO,KAC2B,EAAE,EAAoB,EAAE;IAC7D,MAAM,YAAY,GAAG,yBAAyB,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;IAC1D,OAAO,0BAA0B,CAAC;QAChC,WAAW,EAAE,SAAS,WAAW;YAC/B,OAAO,oBAAoB,CAAC,OAAO,CAAC,CAAC;QACvC,CAAC;QACD,OAAO,EAAE,YAAY;QACrB,YAAY,EAAE,IAAI;KACnB,CAAC,CAAC;AACL,CAAC,CAAC"}

package/dist/ui/utils/convertMessage.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"convertMessage.d.ts","sourceRoot":"","sources":["../../../src/ui/utils/convertMessage.ts"],"names":[],"mappings":"AAAA,OAAO,EAA6B,KAAK,SAAS,EAAE,MAAM,IAAI,CAAC;AAC/D,OAAO,EAEL,KAAK,2BAA2B,EACjC,MAAM,0BAA0B,CAAC;AAYlC,MAAM,MAAM,6BAA6B,GACvC,2BAA2B,CAAC,QAAQ,GAAG;IACrC,qBAAqB,CAAC,EAAE,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;CAC5D,CAAC;AAmNJ,eAAO,MAAM,qBAAqB;;;;;;;;wHA7MO,qDACtB;uHAIjB,qDACM;;;CAiQP,CAAC"}
+{"version":3,"file":"convertMessage.d.ts","sourceRoot":"","sources":["../../../src/ui/utils/convertMessage.ts"],"names":[],"mappings":"AAAA,OAAO,EAA6B,KAAK,SAAS,EAAE,MAAM,IAAI,CAAC;AAC/D,OAAO,EAEL,KAAK,2BAA2B,EACjC,MAAM,0BAA0B,CAAC;AAYlC,MAAM,MAAM,6BAA6B,GACvC,2BAA2B,CAAC,QAAQ,GAAG;IACrC,qBAAqB,CAAC,EAAE,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;CAC5D,CAAC;AAmNJ,eAAO,MAAM,qBAAqB;;;;;;;;wHA7MU,qDACvB;uHAInB,qDACQ;;;CAiQT,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@assistant-ui/react-ai-sdk",
-  "version": "1.3.13",
+  "version": "1.3.14",
   "description": "Vercel AI SDK adapter for assistant-ui",
   "keywords": [
     "ai-sdk",
@@ -32,21 +32,18 @@
   "sideEffects": false,
   "dependencies": {
     "@ai-sdk/react": "^3.0.118",
-    "ai": "^6.0.116"
+    "@assistant-ui/core": "^0.1.6",
+    "@assistant-ui/store": "^0.2.3",
+    "ai": "^6.0.116",
+    "assistant-cloud": "*"
   },
   "peerDependencies": {
-    "@assistant-ui/core": "^0.1.3",
-    "@assistant-ui/store": "^0.2.2",
     "@types/react": "*",
-    "assistant-cloud": "*",
     "react": "^18 || ^19"
   },
   "peerDependenciesMeta": {
     "@types/react": {
       "optional": true
-    },
-    "assistant-cloud": {
-      "optional": true
     }
   },
   "devDependencies": {
@@ -56,11 +53,9 @@
     "@types/react-dom": "^19.2.3",
     "jsdom": "^28.1.0",
     "react": "^19.2.4",
-    "vitest": "^4.0.18",
-    "@assistant-ui/core": "0.1.3",
-    "@assistant-ui/store": "0.2.2",
-    "@assistant-ui/x-buildutils": "0.0.2",
-    "assistant-stream": "0.3.4"
+    "vitest": "^4.1.0",
+    "@assistant-ui/x-buildutils": "0.0.3",
+    "assistant-stream": "0.3.6"
   },
   "publishConfig": {
     "access": "public",

package/src/index.ts

@@ -1,7 +1,11 @@
 /// <reference types="@assistant-ui/core/store" />
 /// <reference types="@assistant-ui/core/react" />
 
-export * from "./ui";
+export { useAISDKRuntime } from "./ui/use-chat/useAISDKRuntime";
+export { useChatRuntime } from "./ui/use-chat/useChatRuntime";
+export type { UseChatRuntimeOptions } from "./ui/use-chat/useChatRuntime";
+export { AssistantChatTransport } from "./ui/use-chat/AssistantChatTransport";
 export { frontendTools } from "./frontendTools";
+export { injectQuoteContext } from "./injectQuoteContext";
 export type { ThreadTokenUsage, TokenUsageExtractableMessage } from "./usage";
 export { getThreadMessageTokenUsage, useThreadTokenUsage } from "./usage";

package/src/injectQuoteContext.ts

@@ -0,0 +1,62 @@
+import type { UIMessage } from "ai";
+
+const getQuoteText = (metadata: unknown): string | undefined => {
+  if (!metadata || typeof metadata !== "object") return undefined;
+
+  const custom = (metadata as Record<string, unknown>).custom;
+  if (!custom || typeof custom !== "object") return undefined;
+
+  const quote = (custom as Record<string, unknown>).quote;
+  if (!quote || typeof quote !== "object") return undefined;
+
+  const text = (quote as Record<string, unknown>).text;
+  return typeof text === "string" ? text : undefined;
+};
+
+/**
+ * Injects quote context into messages as markdown blockquotes.
+ *
+ * Use this in your route handler before `convertToModelMessages` so the LLM
+ * sees the quoted text that the user is referring to.
+ *
+ * @example
+ * ```ts
+ * import { convertToModelMessages, streamText } from "ai";
+ * import { injectQuoteContext } from "@assistant-ui/react-ai-sdk";
+ *
+ * export async function POST(req: Request) {
+ *   const { messages } = await req.json();
+ *   const result = streamText({
+ *     model: myModel,
+ *     messages: await convertToModelMessages(injectQuoteContext(messages)),
+ *   });
+ *   return result.toUIMessageStreamResponse();
+ * }
+ * ```
+ */
+export function injectQuoteContext(messages: UIMessage[]): UIMessage[] {
+  return messages.map((msg) => {
+    if (msg.role !== "user") return msg;
+
+    const text = getQuoteText(msg.metadata);
+    if (!text) return msg;
+
+    const blockquote = text
+      .split(/\r?\n/)
+      .map((line) => `> ${line}`)
+      .join("\n");
+
+    const alreadyInjected =
+      msg.parts[0]?.type === "text" &&
+      msg.parts[0].text === `${blockquote}\n\n`;
+    if (alreadyInjected) return msg;
+
+    return {
+      ...msg,
+      parts: [
+        { type: "text" as const, text: `${blockquote}\n\n` },
+        ...(msg.parts ?? []),
+      ],
+    };
+  });
+}

package/src/ui/use-chat/useChatRuntime.ts

@@ -4,8 +4,8 @@ import { useChat, type UIMessage } from "@ai-sdk/react";
 import type { AssistantCloud } from "assistant-cloud";
 import type { AssistantRuntime } from "@assistant-ui/core";
 import {
-  useCloudThreadListAdapter as unstable_useCloudThreadListAdapter,
-  useRemoteThreadListRuntime as unstable_useRemoteThreadListRuntime,
+  useCloudThreadListAdapter,
+  useRemoteThreadListRuntime,
 } from "@assistant-ui/core/react";
 import { useAuiState } from "@assistant-ui/store";
 import {
@@ -84,8 +84,8 @@ export const useChatRuntime = <UI_MESSAGE extends UIMessage = UIMessage>({
   cloud,
   ...options
 }: UseChatRuntimeOptions<UI_MESSAGE> = {}): AssistantRuntime => {
-  const cloudAdapter = unstable_useCloudThreadListAdapter({ cloud });
-  return unstable_useRemoteThreadListRuntime({
+  const cloudAdapter = useCloudThreadListAdapter({ cloud });
+  return useRemoteThreadListRuntime({
     runtimeHook: function RuntimeHook() {
       return useChatThreadRuntime(options);
     },

package/dist/ui/index.d.ts

@@ -1,5 +0,0 @@
-export { useAISDKRuntime } from "./use-chat/useAISDKRuntime.js";
-export { useChatRuntime } from "./use-chat/useChatRuntime.js";
-export type { UseChatRuntimeOptions } from "./use-chat/useChatRuntime.js";
-export { AssistantChatTransport } from "./use-chat/AssistantChatTransport.js";
-//# sourceMappingURL=index.d.ts.map

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

@@ -1 +0,0 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/ui/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,sCAAmC;AAC7D,OAAO,EAAE,cAAc,EAAE,qCAAkC;AAC3D,YAAY,EAAE,qBAAqB,EAAE,qCAAkC;AACvE,OAAO,EAAE,sBAAsB,EAAE,6CAA0C"}

package/dist/ui/index.js

@@ -1,4 +0,0 @@
-export { useAISDKRuntime } from "./use-chat/useAISDKRuntime.js";
-export { useChatRuntime } from "./use-chat/useChatRuntime.js";
-export { AssistantChatTransport } from "./use-chat/AssistantChatTransport.js";
-//# sourceMappingURL=index.js.map

package/dist/ui/index.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/ui/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,sCAAmC;AAC7D,OAAO,EAAE,cAAc,EAAE,qCAAkC;AAE3D,OAAO,EAAE,sBAAsB,EAAE,6CAA0C"}

package/src/ui/index.ts

@@ -1,4 +0,0 @@
-export { useAISDKRuntime } from "./use-chat/useAISDKRuntime";
-export { useChatRuntime } from "./use-chat/useChatRuntime";
-export type { UseChatRuntimeOptions } from "./use-chat/useChatRuntime";
-export { AssistantChatTransport } from "./use-chat/AssistantChatTransport";