@btst/stack 4.3.0 → 4.5.0

package/src/plugins/ai-chat/client/components/chat-interface.tsx

@@ -1,5 +1,7 @@
 "use client";
 
+import { persistedHistory } from "../../history";
+
 import { useChat } from "@ai-sdk/react";
 import {
 	useEffect,
@@ -46,6 +48,11 @@ import {
 import { usePageAIContext } from "../context/page-ai-context";
 import { useChatAnalytics } from "../context/chat-analytics";
 import { navigateAiChatCrossOrigin } from "../navigation";
+import {
+	captureContextCheckpoint,
+	contextRequest,
+	type ContextCheckpoint,
+} from "../context-checkpoint";
 
 interface ChatInterfaceProps {
 	initialMessages?: UIMessage[];
@@ -89,30 +96,28 @@ function reconcilePersistedMessageIds(
 	return messages.map((message) => ({
 		...message,
 		id: persistedIds.get(message) ?? message.id,
+		...(persistedMessages.some(
+			(saved) => saved.id === message.id && saved.interrupted,
+		)
+			? { metadata: { ...(message.metadata as object), interrupted: true } }
+			: {}),
 	}));
 }
 
 function persistedMessagesToUiMessages(
 	messages: readonly SerializedMessage[],
 ): UIMessage[] {
-	return messages
-		.filter((message) => message.role !== "data")
-		.map((message) => {
-			let parts: UIMessage["parts"];
-			try {
-				const parsed = JSON.parse(message.content);
-				parts = Array.isArray(parsed)
-					? parsed
-					: [{ type: "text" as const, text: message.content }];
-			} catch {
-				parts = [{ type: "text" as const, text: message.content }];
-			}
-			return {
-				id: message.id,
-				role: message.role as "user" | "assistant" | "system",
-				parts,
-			};
-		});
+	return persistedHistory(messages).map((message) => {
+		// Browser tool callbacks do not resume when saved history is restored.
+		const abandoned = message.parts.some(
+			(part) =>
+				"toolCallId" in part &&
+				(part.state === "input-streaming" || part.state === "input-available"),
+		);
+		return abandoned
+			? { ...message, metadata: { interrupted: true } }
+			: message;
+	});
 }
 
 function ChatActionCheck({
@@ -314,6 +319,9 @@ export function ChatInterface({
 		Array<{ generation: number; identityPartitionKey: string }>
 	>([]);
 	const latestMessages = useRef<UIMessage[]>([]);
+	const contextCheckpoint = useRef<ContextCheckpoint | undefined>(undefined);
+	const submittedMessages = useRef<UIMessage[]>([]);
+	const [isCompacting, setIsCompacting] = useState(false);
 
 	const conversationsListQueryKey = useMemo(() => {
 		// In public mode, we don't need conversation queries
@@ -465,6 +473,7 @@ export function ChatInterface({
 					pendingStreamRequests.current.push(request);
 					activeStreamPartitionKey.current = request.identityPartitionKey;
 				}
+				setIsCompacting(false);
 				const currentPageContext = pageAIContextRef.current;
 
 				// Build page context fields to include in every request
@@ -483,20 +492,26 @@ export function ChatInterface({
 					if (newUserMessage) {
 						messagesToSend.push(newUserMessage);
 					}
+					submittedMessages.current = messagesToSend;
 					// Clear the ref after use
 					editMessagesRef.current = null;
 					return {
 						body: {
-							messages: messagesToSend,
+							...(isPublicMode
+								? contextRequest(messagesToSend, contextCheckpoint.current)
+								: { messages: messagesToSend }),
 							conversationId: conversationIdRef.current,
 							...pageContextBody,
 						},
 					};
 				}
-				// Normal case - use the messages as-is
+				submittedMessages.current = hookMessages;
+				// Retain the full UI history; compact only the transport payload.
 				return {
 					body: {
-						messages: hookMessages,
+						...(isPublicMode
+							? contextRequest(hookMessages, contextCheckpoint.current)
+							: { messages: hookMessages }),
 						conversationId: conversationIdRef.current,
 						...pageContextBody,
 					},
@@ -550,6 +565,33 @@ export function ChatInterface({
 	} = useChat({
 		id: `${chatInstanceId}:${isPublicMode ? "public" : identitySessionVersion}`,
 		transport,
+		onData: ({ type, data }) => {
+			if (!mounted.current || !data || typeof data !== "object") return;
+			if (
+				!isPublicMode &&
+				(identitySessionVersion !== identitySessionGeneration.current ||
+					activeStreamPartitionKey.current !==
+						latestIdentityPartitionKey.current)
+			)
+				return;
+			if (type === "data-context-status" && "compacting" in data) {
+				setIsCompacting(data.compacting === true);
+			}
+			if (
+				isPublicMode &&
+				type === "data-context-checkpoint" &&
+				"summary" in data &&
+				"throughMessageId" in data &&
+				typeof data.summary === "string" &&
+				typeof data.throughMessageId === "string"
+			) {
+				contextCheckpoint.current = captureContextCheckpoint(
+					submittedMessages.current,
+					data.summary,
+					data.throughMessageId,
+				);
+			}
+		},
 		// Automatically resubmit after all client-side tool results are provided
 		sendAutomaticallyWhen: (options) =>
 			mounted.current &&
@@ -630,6 +672,7 @@ export function ChatInterface({
 			}
 		},
 		onError: (err) => {
+			setIsCompacting(false);
 			if (!mounted.current) return;
 			console.error("useChat onError:", err);
 			if (
@@ -653,6 +696,7 @@ export function ChatInterface({
 			// the provider or response stream failed.
 		},
 		onFinish: async (completion) => {
+			setIsCompacting(false);
 			if (
 				!mounted.current ||
 				(!isPublicMode &&
@@ -684,6 +728,25 @@ export function ChatInterface({
 					).length,
 				});
 			}
+			if (
+				completion?.isAbort ||
+				completion?.isError ||
+				completion?.isDisconnect
+			) {
+				setMessages((current) =>
+					current.map((message) =>
+						message.id === completion.message.id
+							? {
+									...message,
+									metadata: {
+										...(message.metadata as object),
+										interrupted: true,
+									},
+								}
+							: message,
+					),
+				);
+			}
 			// In public mode, skip all persistence-related operations
 			if (isPublicMode) {
 				activeStreamPartitionKey.current = undefined;
@@ -1328,7 +1391,20 @@ export function ChatInterface({
 									/>
 								))
 							)}
+							{isLoading && isCompacting && (
+								<div
+									role="status"
+									className="text-muted-foreground text-sm py-4"
+								>
+									{tr(
+										"CHAT_COMPACTING",
+										"aiChat.chat.compacting",
+										"Summarizing earlier conversation…",
+									)}
+								</div>
+							)}
 							{isLoading &&
+								!isCompacting &&
 								messages[messages.length - 1]?.role !== "assistant" && (
 									<div className="flex items-center gap-2 text-muted-foreground text-sm py-4">
 										<div className="animate-pulse">

package/src/plugins/ai-chat/client/components/chat-message.tsx

@@ -132,6 +132,12 @@ export function ChatMessage({
 	isRetrying = false,
 }: ChatMessageProps) {
 	const isUser = message.role === "user";
+	const interrupted = Boolean(
+		message.metadata &&
+			typeof message.metadata === "object" &&
+			"interrupted" in message.metadata &&
+			message.metadata.interrupted,
+	);
 
 	const {
 		localization: customLocalization,
@@ -479,10 +485,22 @@ export function ChatMessage({
 									{toolParts.map((part: any) => {
 										const toolName = getToolName(part);
 										const toolCallId = part.toolCallId;
-										const state = part.state as ToolCallState;
+										const unfinished =
+											interrupted &&
+											(part.state === "input-streaming" ||
+												part.state === "input-available");
+										const state = (
+											unfinished ? "output-error" : part.state
+										) as ToolCallState;
 										const input = part.input;
 										const output = part.output;
-										const errorText = part.errorText;
+										const errorText = unfinished
+											? tr(
+													"CHAT_INTERRUPTED",
+													"aiChat.interrupted",
+													"Response interrupted",
+												)
+											: part.errorText;
 										const isLoading =
 											state === "input-streaming" ||
 											state === "input-available";
@@ -513,6 +531,18 @@ export function ChatMessage({
 								</div>
 							)}
 							{/* Text content */}
+							{interrupted && !isStreaming && (
+								<p
+									className="text-xs text-muted-foreground"
+									data-testid="chat-interrupted"
+								>
+									{tr(
+										"CHAT_INTERRUPTED",
+										"aiChat.interrupted",
+										"Response interrupted",
+									)}
+								</p>
+							)}
 							{displayContent ? (
 								<MarkdownContent
 									markdown={displayContent}

package/src/plugins/ai-chat/client/context-checkpoint.ts

@@ -0,0 +1,47 @@
+import type { UIMessage } from "ai";
+
+/** Browser-only snapshot; never replaces or mutates the visible transcript. */
+export interface ContextCheckpoint {
+	summary: string;
+	throughMessageId: string;
+	prefix: string;
+}
+
+export function captureContextCheckpoint(
+	messages: UIMessage[],
+	summary: string,
+	throughMessageId: string,
+): ContextCheckpoint | undefined {
+	const index = messages.findIndex(
+		(message) => message.id === throughMessageId,
+	);
+	if (index < 0) return undefined;
+	return {
+		summary,
+		throughMessageId,
+		prefix: JSON.stringify(messages.slice(0, index + 1)),
+	};
+}
+
+export function contextRequest(
+	messages: UIMessage[],
+	checkpoint?: ContextCheckpoint,
+) {
+	if (checkpoint) {
+		const index = messages.findIndex(
+			(message) => message.id === checkpoint.throughMessageId,
+		);
+		if (
+			index >= 0 &&
+			JSON.stringify(messages.slice(0, index + 1)) === checkpoint.prefix
+		) {
+			return {
+				messages: messages.slice(index + 1),
+				contextSummary: checkpoint.summary,
+			};
+		}
+	}
+	// Editing or retrying a summarized turn invalidates the snapshot. Resend
+	// the original transcript so the server builds a fresh summary if needed.
+	return { messages };
+}

package/src/plugins/ai-chat/client/localization/index.ts

@@ -12,6 +12,8 @@ export interface AiChatLocalization {
 	CHAT_SEND_BUTTON: string;
 	CHAT_EMPTY_STATE: string;
 	CHAT_LOADING: string;
+	CHAT_COMPACTING?: string;
+	CHAT_INTERRUPTED?: string;
 	CHAT_ERROR: string;
 	CHAT_RETRY: string;
 

package/src/plugins/ai-chat/db.ts

@@ -21,6 +21,10 @@ export const aiChatSchema = createDbPlugin("aiChat", {
 				type: "string",
 				required: true,
 			},
+			contextCheckpoint: {
+				type: "string",
+				required: false,
+			},
 			createdAt: {
 				type: "date",
 				defaultValue: () => new Date(),
@@ -56,6 +60,10 @@ export const aiChatSchema = createDbPlugin("aiChat", {
 				type: "string",
 				required: true,
 			},
+			interrupted: {
+				type: "boolean",
+				required: false,
+			},
 			createdAt: {
 				type: "date",
 				defaultValue: () => new Date(),

package/src/plugins/ai-chat/history.ts

@@ -0,0 +1,68 @@
+import type { UIMessage } from "ai";
+import type { Message } from "./types";
+
+function isMessagePart(value: unknown): boolean {
+	if (!value || typeof value !== "object" || !("type" in value)) return false;
+	const part = value as Record<string, unknown>;
+	if (typeof part.type !== "string") return false;
+	switch (part.type) {
+		case "text":
+		case "reasoning":
+			return typeof part.text === "string";
+		case "file":
+			return typeof part.url === "string" && typeof part.mediaType === "string";
+		case "source-url":
+			return typeof part.sourceId === "string" && typeof part.url === "string";
+		case "source-document":
+			return (
+				typeof part.sourceId === "string" &&
+				typeof part.mediaType === "string" &&
+				typeof part.title === "string"
+			);
+		case "step-start":
+			return true;
+		default:
+			if (part.type.startsWith("data-")) return "data" in part;
+			return (
+				(part.type.startsWith("tool-") ||
+					(part.type === "dynamic-tool" &&
+						typeof part.toolName === "string")) &&
+				typeof part.toolCallId === "string" &&
+				[
+					"input-streaming",
+					"input-available",
+					"output-available",
+					"output-error",
+				].includes(String(part.state))
+			);
+	}
+}
+
+export function persistedHistory(
+	messages: readonly Pick<Message, "id" | "role" | "content" | "interrupted">[],
+): UIMessage[] {
+	return messages
+		.filter((message) => message.role !== "data")
+		.map((message) => {
+			let parts: UIMessage["parts"];
+			try {
+				const parsed = JSON.parse(message.content);
+				// The old completion writer saved [] for empty assistant replies.
+				// Empty arrays in legacy user text must remain literal text.
+				parts =
+					Array.isArray(parsed) &&
+					(parsed.length > 0 || message.role === "assistant") &&
+					parsed.every(isMessagePart)
+						? parsed
+						: [{ type: "text", text: message.content }];
+			} catch {
+				parts = [{ type: "text", text: message.content }];
+			}
+			return {
+				id: message.id,
+				role: message.role as UIMessage["role"],
+				parts,
+				...(message.interrupted ? { metadata: { interrupted: true } } : {}),
+			};
+		});
+}

package/src/plugins/ai-chat/schemas.ts

@@ -36,6 +36,8 @@ export const chatRequestSchema = z.object({
 		]),
 	),
 	conversationId: z.string().optional(),
+	/** Untrusted continuation summary for opt-in public context compaction. */
+	contextSummary: z.string().max(32_000).optional(),
 	model: z.string().optional(),
 	/**
 	 * Description of the current page context, injected into the AI system prompt.

package/src/plugins/ai-chat/types.ts

@@ -15,6 +15,8 @@ export type Message = {
 	conversationId: string;
 	role: "system" | "user" | "assistant" | "data";
 	content: string;
+	/** True when generation stopped before a successful completion. */
+	interrupted?: boolean | null;
 	createdAt: Date;
 };
 

package/dist/shared/{stack.lsbdKRj7.d.mts → stack.B8PG21zX.d.mts}

@@ -95,8 +95,8 @@ declare const cmsBackendPlugin: (config: CMSBackendConfig) => _btst_stack_plugin
             readonly id: string;
             readonly slug: string;
             readonly contentTypeId: string;
-            readonly data: string;
             readonly authorId?: string | undefined;
+            readonly data: string;
         }[];
         readonly total: number;
     }>;
@@ -128,8 +128,8 @@ declare const cmsBackendPlugin: (config: CMSBackendConfig) => _btst_stack_plugin
         readonly id: string;
         readonly slug: string;
         readonly contentTypeId: string;
-        readonly data: string;
         readonly authorId?: string | undefined;
+        readonly data: string;
     }>;
     readonly createContentItem: better_call.StrictEndpoint<"/content/:typeSlug", {
         params: z.ZodObject<{
@@ -165,8 +165,8 @@ declare const cmsBackendPlugin: (config: CMSBackendConfig) => _btst_stack_plugin
         readonly id: string;
         readonly slug: string;
         readonly contentTypeId: string;
-        readonly data: string;
         readonly authorId?: string | undefined;
+        readonly data: string;
     }>;
     readonly updateContentItem: better_call.StrictEndpoint<"/content/:typeSlug/:id", {
         params: z.ZodObject<{
@@ -203,8 +203,8 @@ declare const cmsBackendPlugin: (config: CMSBackendConfig) => _btst_stack_plugin
         readonly id: string;
         readonly slug: string;
         readonly contentTypeId: string;
-        readonly data: string;
         readonly authorId?: string | undefined;
+        readonly data: string;
     }>;
     readonly deleteContentItem: better_call.StrictEndpoint<"/content/:typeSlug/:id", {
         params: z.ZodObject<{
@@ -248,8 +248,8 @@ declare const cmsBackendPlugin: (config: CMSBackendConfig) => _btst_stack_plugin
                 readonly id: string;
                 readonly slug: string;
                 readonly contentTypeId: string;
-                readonly data: string;
                 readonly authorId?: string | undefined;
+                readonly data: string;
             }[];
         };
         readonly contentType?: {
@@ -271,8 +271,8 @@ declare const cmsBackendPlugin: (config: CMSBackendConfig) => _btst_stack_plugin
         readonly id: string;
         readonly slug: string;
         readonly contentTypeId: string;
-        readonly data: string;
         readonly authorId?: string | undefined;
+        readonly data: string;
     }>;
     readonly listContentByRelation: better_call.StrictEndpoint<"/content/:typeSlug/by-relation", {
         params: z.ZodObject<{
@@ -314,8 +314,8 @@ declare const cmsBackendPlugin: (config: CMSBackendConfig) => _btst_stack_plugin
             readonly id: string;
             readonly slug: string;
             readonly contentTypeId: string;
-            readonly data: string;
             readonly authorId?: string | undefined;
+            readonly data: string;
         }[];
         readonly total: number;
         readonly limit: number;
@@ -383,8 +383,8 @@ declare const cmsBackendPlugin: (config: CMSBackendConfig) => _btst_stack_plugin
             readonly id: string;
             readonly slug: string;
             readonly contentTypeId: string;
-            readonly data: string;
             readonly authorId?: string | undefined;
+            readonly data: string;
         }[];
         readonly total: number;
         readonly limit: number;
@@ -459,8 +459,8 @@ declare const cmsBackendPlugin: (config: CMSBackendConfig) => _btst_stack_plugin
             id: string;
             slug: string;
             contentTypeId: string;
-            data: string;
             authorId?: string | undefined;
+            data: string;
         }[];
         total: number;
     }, false>;
@@ -493,8 +493,8 @@ declare const cmsBackendPlugin: (config: CMSBackendConfig) => _btst_stack_plugin
         id: string;
         slug: string;
         contentTypeId: string;
-        data: string;
         authorId?: string | undefined;
+        data: string;
     }, false>;
     readonly createContentItem: _btst_stack_plugins_api.Operation<z.ZodObject<{
         typeSlug: z.ZodString;
@@ -522,8 +522,8 @@ declare const cmsBackendPlugin: (config: CMSBackendConfig) => _btst_stack_plugin
         id: string;
         slug: string;
         contentTypeId: string;
-        data: string;
         authorId?: string | undefined;
+        data: string;
     }, false>;
     readonly updateContentItem: _btst_stack_plugins_api.Operation<z.ZodObject<{
         typeSlug: z.ZodString;
@@ -554,8 +554,8 @@ declare const cmsBackendPlugin: (config: CMSBackendConfig) => _btst_stack_plugin
         id: string;
         slug: string;
         contentTypeId: string;
-        data: string;
         authorId?: string | undefined;
+        data: string;
     }, false>;
     readonly deleteContentItem: _btst_stack_plugins_api.Operation<z.ZodObject<{
         typeSlug: z.ZodString;
@@ -598,8 +598,8 @@ declare const cmsBackendPlugin: (config: CMSBackendConfig) => _btst_stack_plugin
                 id: string;
                 slug: string;
                 contentTypeId: string;
-                data: string;
                 authorId?: string | undefined;
+                data: string;
             }[];
         };
         contentType?: {
@@ -619,8 +619,8 @@ declare const cmsBackendPlugin: (config: CMSBackendConfig) => _btst_stack_plugin
         id: string;
         slug: string;
         contentTypeId: string;
-        data: string;
         authorId?: string | undefined;
+        data: string;
     }, false>;
     readonly listContentByRelation: _btst_stack_plugins_api.Operation<z.ZodObject<{
         typeSlug: z.ZodString;
@@ -657,8 +657,8 @@ declare const cmsBackendPlugin: (config: CMSBackendConfig) => _btst_stack_plugin
             id: string;
             slug: string;
             contentTypeId: string;
-            data: string;
             authorId?: string | undefined;
+            data: string;
         }[];
         total: number;
         limit: number;
@@ -715,8 +715,8 @@ declare const cmsBackendPlugin: (config: CMSBackendConfig) => _btst_stack_plugin
             id: string;
             slug: string;
             contentTypeId: string;
-            data: string;
             authorId?: string | undefined;
+            data: string;
         }[];
         total: number;
         limit: number;

package/dist/shared/{stack.CGmh-Lpe.d.ts → stack.BuxUQqL8.d.ts}

@@ -127,11 +127,11 @@ declare const formBuilderBackendPlugin: (config?: FormBuilderBackendConfig) => _
             readonly status: Form["status"];
             readonly createdAt: string;
             readonly updatedAt: string;
-            readonly id: string;
-            readonly name: string;
             readonly schema: string;
+            readonly id: string;
             readonly description?: string | undefined;
             readonly slug: string;
+            readonly name: string;
             readonly successMessage?: string | undefined;
             readonly redirectUrl?: string | undefined;
             readonly createdBy?: string | undefined;
@@ -151,11 +151,11 @@ declare const formBuilderBackendPlugin: (config?: FormBuilderBackendConfig) => _
         readonly status: Form["status"];
         readonly createdAt: string;
         readonly updatedAt: string;
-        readonly id: string;
-        readonly name: string;
         readonly schema: string;
+        readonly id: string;
         readonly description?: string | undefined;
         readonly slug: string;
+        readonly name: string;
         readonly successMessage?: string | undefined;
         readonly redirectUrl?: string | undefined;
         readonly createdBy?: string | undefined;
@@ -171,11 +171,11 @@ declare const formBuilderBackendPlugin: (config?: FormBuilderBackendConfig) => _
         readonly status: Form["status"];
         readonly createdAt: string;
         readonly updatedAt: string;
-        readonly id: string;
-        readonly name: string;
         readonly schema: string;
+        readonly id: string;
         readonly description?: string | undefined;
         readonly slug: string;
+        readonly name: string;
         readonly successMessage?: string | undefined;
         readonly redirectUrl?: string | undefined;
         readonly createdBy?: string | undefined;
@@ -191,11 +191,11 @@ declare const formBuilderBackendPlugin: (config?: FormBuilderBackendConfig) => _
         readonly status: Form["status"];
         readonly createdAt: string;
         readonly updatedAt: string;
-        readonly id: string;
-        readonly name: string;
         readonly schema: string;
+        readonly id: string;
         readonly description?: string | undefined;
         readonly slug: string;
+        readonly name: string;
         readonly successMessage?: string | undefined;
         readonly redirectUrl?: string | undefined;
         readonly createdBy?: string | undefined;
@@ -225,11 +225,11 @@ declare const formBuilderBackendPlugin: (config?: FormBuilderBackendConfig) => _
         readonly status: Form["status"];
         readonly createdAt: string;
         readonly updatedAt: string;
-        readonly id: string;
-        readonly name: string;
         readonly schema: string;
+        readonly id: string;
         readonly description?: string | undefined;
         readonly slug: string;
+        readonly name: string;
         readonly successMessage?: string | undefined;
         readonly redirectUrl?: string | undefined;
         readonly createdBy?: string | undefined;
@@ -259,11 +259,11 @@ declare const formBuilderBackendPlugin: (config?: FormBuilderBackendConfig) => _
         readonly status: Form["status"];
         readonly createdAt: string;
         readonly updatedAt: string;
-        readonly id: string;
-        readonly name: string;
         readonly schema: string;
+        readonly id: string;
         readonly description?: string | undefined;
         readonly slug: string;
+        readonly name: string;
         readonly successMessage?: string | undefined;
         readonly redirectUrl?: string | undefined;
         readonly createdBy?: string | undefined;