@aleph-alpha/chat-kit 3.3.0 → 5.0.0

package/dist/index.js

@@ -1,40 +1,42 @@
-import { u } from "./useMessageHandler-CMdy-lm0.js";
-import { C, d, p, u as u2, a } from "./useChatKitLabels-BM2I7-9o.js";
-import { S, c, d as d2, a as a2, u as u3, b, e, f } from "./index-C6njJ8EB.js";
-import { _, C as C2, a as a3, b as b2, c as c2, d as d3, e as e2, f as f2, g, h, i, j, k, l, m, n, o, p as p2, q, r, s, t, u as u4, v, w, x } from "./CkInput.vue_vue_type_script_setup_true_lang-BAcBFs6I.js";
+import { u } from "./useMessageHandler-N50Yj-NU.js";
+import { C, d, p, u as u2, a } from "./useChatKitLabels-zMFV7aEz.js";
+import { S, c, d as d2, a as a2, u as u3, b, e, f } from "./index-CyrVnbS9.js";
+import { _, a as a3, C as C2, b as b2, c as c2, d as d3, e as e2, f as f2, g, h, i, j, k, l, m, n, o, p as p2, q, r, s, t, u as u4, v, w, x, y, z } from "./CkHistory.vue_vue_type_script_setup_true_lang-Bp9hXdH6.js";
 export {
   C as CHAT_KIT_LABELS,
   _ as CkAgentMessage,
+  a3 as CkAgentMessageHeader,
   C2 as CkCitation,
-  a3 as CkCitationCard,
-  b2 as CkConversation,
-  c2 as CkConversationLayout,
-  d3 as CkCopyMessageActionButton,
-  e2 as CkDropZoneOverlay,
-  f2 as CkFeedbackButton,
-  g as CkFeedbackDialog,
-  h as CkFileItemsList,
-  i as CkHistory,
-  j as CkHistoryItem,
-  k as CkHistoryWrapper,
-  l as CkInput,
-  m as CkMarkdownRenderer,
-  n as CkMessageActionButton,
-  o as CkModelSelector,
-  p2 as CkPromptInput,
-  q as CkPromptInputDisclaimer,
-  r as CkPromptMenuButton,
-  s as CkPromptSendButton,
-  t as CkPromptStopButton,
-  u4 as CkSendIcon,
-  v as CkStopIcon,
-  w as CkUserMessage,
+  b2 as CkCitationCard,
+  c2 as CkConversation,
+  d3 as CkConversationLayout,
+  e2 as CkCopyMessageButton,
+  f2 as CkDropZoneOverlay,
+  g as CkFeedbackButton,
+  h as CkFeedbackDialog,
+  i as CkFileItemsList,
+  j as CkHeaderActivityButton,
+  k as CkHistory,
+  l as CkHistoryItem,
+  m as CkHistoryWrapper,
+  n as CkMarkdownRenderer,
+  o as CkMessageButton,
+  p2 as CkModelSelector,
+  q as CkPromptInput,
+  r as CkPromptInputDisclaimer,
+  s as CkPromptMenuButton,
+  t as CkPromptSendButton,
+  u4 as CkPromptStopButton,
+  v as CkRegenerateMessageButton,
+  w as CkSendIcon,
+  x as CkStopIcon,
+  y as CkUserMessage,
   S as SdkError,
   c as createDocumentUploadProvider,
   d as defaultChatKitLabels,
   d2 as defaultDocumentUploadConfig,
   a2 as defaultEndpoints,
-  x as formatRelativeTime,
+  z as formatRelativeTime,
   p as provideChatKitLabels,
   u3 as useChat,
   u2 as useChatKitLabels,

package/dist/{useChatKitLabels-BM2I7-9o.js → useChatKitLabels-zMFV7aEz.js}

@@ -28680,6 +28680,7 @@ const defaultChatKitLabels = {
   inputPlaceholder: "Type a message...",
   copyMessageAction: "Copy response",
   copyMessageActionCopied: "Copied!",
+  regenerateMessageAction: "Regenerate response",
   feedbackThumbsUp: "Good response",
   feedbackThumbsDown: "Bad response",
   feedbackGroup: "Rate this response",

package/dist/{useMessageHandler-CMdy-lm0.js → useMessageHandler-N50Yj-NU.js}

@@ -71,10 +71,8 @@ function updateNodeId(tree, oldId, newId) {
   }
   if (tree.rootId === oldId) tree.rootId = newId;
 }
-function cutBranchAfter(tree, id) {
-  const node = tree.nodes[id];
-  if (!node) return;
-  const stack = [...node.childrenIds];
+function deleteSubtrees(tree, ids) {
+  const stack = [...ids];
   while (stack.length > 0) {
     const currentId = stack.pop();
     const current = tree.nodes[currentId];
@@ -82,9 +80,33 @@ function cutBranchAfter(tree, id) {
     stack.push(...current.childrenIds);
     delete tree.nodes[currentId];
   }
+}
+function cutBranchAfter(tree, id) {
+  const node = tree.nodes[id];
+  if (!node) return;
+  deleteSubtrees(tree, node.childrenIds);
   node.childrenIds = [];
   node.activeChildIndex = 0;
 }
+function cutBranchAt(tree, id) {
+  const node = tree.nodes[id];
+  if (!node) return;
+  deleteSubtrees(tree, [id]);
+  if (node.parentId === null) {
+    if (tree.rootId === id) tree.rootId = null;
+    return;
+  }
+  const parent = tree.nodes[node.parentId];
+  if (!parent) return;
+  const idx = parent.childrenIds.indexOf(id);
+  if (idx !== -1) {
+    parent.childrenIds.splice(idx, 1);
+    if (idx < parent.activeChildIndex) parent.activeChildIndex -= 1;
+  }
+  if (parent.activeChildIndex >= parent.childrenIds.length) {
+    parent.activeChildIndex = Math.max(0, parent.childrenIds.length - 1);
+  }
+}
 function navigate(tree, targetId) {
   const node = tree.nodes[targetId];
   if (!node || node.parentId === null) return;
@@ -102,6 +124,13 @@ const useMessageHandler = () => {
     const list = messages.value;
     return list.length > 0 ? list[list.length - 1].id : null;
   });
+  function resolveParentId(parentId) {
+    if (parentId === void 0) return currentLeafId.value;
+    if (!tree.nodes[parentId]) {
+      throw new Error(`Unknown parent message id: ${parentId}`);
+    }
+    return parentId;
+  }
   function setCurrentUserMessage(text) {
     currentUserMessage.value = text;
   }
@@ -112,12 +141,13 @@ const useMessageHandler = () => {
     const id = generateId();
     const text = currentUserMessage.value.trim();
     if (!text) throw new Error("No message");
+    const parentId = resolveParentId(opts?.parentId);
     const content = [
       { type: "text", id: generateId(), text, status: "completed" }
     ];
     const node = {
       id,
-      parentId: currentLeafId.value,
+      parentId,
       childrenIds: [],
       activeChildIndex: 0,
       role: "user",
@@ -150,7 +180,7 @@ const useMessageHandler = () => {
     } else {
       const node = {
         id: message.id,
-        parentId: currentLeafId.value,
+        parentId: resolveParentId(message.parentId),
         childrenIds: [],
         activeChildIndex: 0,
         role: message.role,
@@ -191,6 +221,12 @@ const useMessageHandler = () => {
   function cutBranchAfter$1(messageId) {
     cutBranchAfter(tree, messageId);
   }
+  function cutBranchAt$1(messageId) {
+    cutBranchAt(tree, messageId);
+  }
+  function getTree() {
+    return readonly(tree);
+  }
   return {
     messages,
     currentUserMessage: readonly(currentUserMessage),
@@ -204,7 +240,9 @@ const useMessageHandler = () => {
     clearMessages,
     setTree,
     navigate: navigate$1,
-    cutBranchAfter: cutBranchAfter$1
+    cutBranchAfter: cutBranchAfter$1,
+    cutBranchAt: cutBranchAt$1,
+    getTree
   };
 };
 export {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aleph-alpha/chat-kit",
-  "version": "3.3.0",
+  "version": "5.0.0",
   "description": "Chat kit package for building chat interfaces with Vue",
   "type": "module",
   "main": "./dist/index.js",
@@ -102,9 +102,9 @@
     "vitest": "3.2.6",
     "vue": "^3.5.27",
     "vue-tsc": "^3.2.4",
-    "@aleph-alpha/tsconfig-frontend": "0.7.0",
+    "@aleph-alpha/eslint-config-frontend": "0.7.0",
     "@aleph-alpha/prettier-config-frontend": "2.1.0",
-    "@aleph-alpha/eslint-config-frontend": "0.7.0"
+    "@aleph-alpha/tsconfig-frontend": "0.7.0"
   },
   "scripts": {
     "build": "vite build && vue-tsc --project tsconfig.build.json --declaration --emitDeclarationOnly --outDir dist",

package/src/adapters/responses-api.spec.ts

@@ -142,6 +142,34 @@ describe('responsesToTree', () => {
     expect(a1.activeChildIndex).toBe(1);
   });
 
+  it('favours the latest root branch when the first turn is branched into several root responses', () => {
+    const tree = responsesToTree([
+      makeResponse({
+        id: 'r1',
+        request_input: [userInput('First attempt')],
+        output: [assistantOutput('msg-a1', 'A1')],
+      }),
+      makeResponse({
+        id: 'r2',
+        request_input: [userInput('Second attempt')],
+        output: [assistantOutput('msg-a2', 'A2')],
+      }),
+    ]);
+
+    // Both turns are root-anchored (no previous_response_id); the latest wins
+    // as the active root so getLinearPath surfaces it.
+    expect(tree.nodes['user-r1-0']!.parentId).toBeNull();
+    expect(tree.nodes['user-r2-0']!.parentId).toBeNull();
+    expect(tree.rootId).toBe('user-r2-0');
+
+    const handler = useMessageHandler();
+    handler.setTree(tree);
+    expect(handler.messages.value.map((m) => m.id)).toEqual([
+      'user-r2-0',
+      'msg-a2',
+    ]);
+  });
+
   it('skips responses with no visible content but keeps descendants attached to the parent tail', () => {
     const tree = responsesToTree([
       makeResponse({
@@ -402,15 +430,31 @@ describe('responsesToTree', () => {
         // string on the wire (transports flatten nested metadata).
         metadata: {
           attachments: JSON.stringify([
-            { fileName: 'report.pdf', fileId: 'f1', searchStoreId: 's1' },
+            {
+              fileName: 'report.pdf',
+              fileId: 'f1',
+              searchStoreId: 's1',
+              documentId: 'd1',
+            },
           ]),
         },
       }),
     ]);
 
+    // The full attachment shape (incl. `documentId`) is preserved so the
+    // turn can be faithfully re-submitted (e.g. regenerate) after a reload.
     expect(tree.nodes['user-r1-0']!.metadata).toEqual({
       responseId: 'r1',
-      attachments: [{ name: 'report.pdf', fileId: 'f1', searchStoreId: 's1' }],
+      model: 'pharia-1',
+      attachments: [
+        {
+          name: 'report.pdf',
+          fileId: 'f1',
+          fileName: 'report.pdf',
+          searchStoreId: 's1',
+          documentId: 'd1',
+        },
+      ],
     });
     // The assistant turn does not inherit attachments but still carries
     // its responseId stamp.
@@ -431,9 +475,12 @@ describe('responsesToTree', () => {
         }),
       ]);
 
-      // The malformed attachments are dropped, but `responseId` is still
-      // stamped on every node the adapter emits.
-      expect(tree.nodes['user-r1-0']!.metadata).toEqual({ responseId: 'r1' });
+      // The malformed attachments are dropped; the first user node still
+      // carries `responseId` (and `model` when present).
+      expect(tree.nodes['user-r1-0']!.metadata).toEqual({
+        responseId: 'r1',
+        model: 'pharia-1',
+      });
       expect(tree.nodes['user-r1-0']!.metadata.attachments).toBeUndefined();
       expect(warn).toHaveBeenCalled();
     } finally {

package/src/adapters/responses-api.ts

@@ -212,6 +212,7 @@ export interface ResponseAttachment {
   fileId: string;
   fileName: string;
   searchStoreId: string;
+  documentId?: string;
 }
 
 /**
@@ -219,15 +220,17 @@ export interface ResponseAttachment {
  * `metadata.attachments` on the user `TreeMessage` (chat-kit's `BaseMessage`
  * /`TreeMessage` carry `metadata` as an opaque `Record<string, unknown>`).
  *
- * The shape is intentionally minimal — a human-readable `name` plus the
- * optional ids needed to resolve a public URL for the file. Consumers
- * using a different attachment shape can either skip this adapter or
- * project this onto their own type at the read site.
+ * Carries a human-readable `name` plus every id needed to resolve a public
+ * URL for the file or to re-submit it on a later turn (e.g. regenerate).
+ * Consumers using a different attachment shape can either skip this adapter
+ * or project this onto their own type at the read site.
  */
 export interface ResponseMessageAttachment {
   name: string;
   fileId?: string;
+  fileName?: string;
   searchStoreId?: string;
+  documentId?: string;
 }
 
 /**
@@ -280,6 +283,7 @@ export interface ResponseList {
  */
 export type ResponseTreeMessageMetadata = MessageMetadata & {
   responseId: string;
+  model?: string;
   attachments?: ResponseMessageAttachment[];
 };
 
@@ -478,13 +482,18 @@ function responseToMessageNodes(resp: StoredResponse): ResponseMessageNode[] {
       const userContent = text || responseAttachments?.length;
       if (role !== 'user' || !userContent) continue;
 
-      const metadata: ResponseTreeMessageMetadata = { responseId: resp.id };
+      const metadata: ResponseTreeMessageMetadata = {
+        responseId: resp.id,
+        ...(resp.model ? { model: resp.model } : {}),
+      };
       if (responseAttachments && userNodesPushed === 0) {
         metadata.attachments = responseAttachments.map(
           (a): ResponseMessageAttachment => ({
             name: a.fileName,
             fileId: a.fileId,
+            fileName: a.fileName,
             searchStoreId: a.searchStoreId,
+            ...(a.documentId ? { documentId: a.documentId } : {}),
           }),
         );
       }
@@ -607,6 +616,12 @@ function linkResponse(state: TreeBuildState, resp: StoredResponse): void {
     parentResponseId !== null
       ? (tailMessageByResponse.get(parentResponseId) ?? null)
       : null;
+  // Whether this response opens a new root-level branch (it chains onto
+  // nothing). The first turn can be branched/regenerated, producing several
+  // root responses; since the tree has a single `rootId` with no parent node
+  // to track sibling navigation, we keep the latest such branch as the active
+  // root — mirroring the "latest sibling wins" rule used for deeper branches.
+  const isRootChain = chainParent === null;
 
   const messageNodes = responseToMessageNodes(resp);
 
@@ -624,7 +639,9 @@ function linkResponse(state: TreeBuildState, resp: StoredResponse): void {
     chainParent = node.id;
   }
 
-  if (state.rootMessageId === null && firstId !== null) {
+  // Favour the latest root branch: responses arrive in ascending `created_at`
+  // order, so a later root chain overwrites an earlier one.
+  if (firstId !== null && (state.rootMessageId === null || isRootChain)) {
     state.rootMessageId = firstId;
   }
 
@@ -639,7 +656,9 @@ function linkResponse(state: TreeBuildState, resp: StoredResponse): void {
  * pass: each response's `previous_response_id` always points to a response
  * we have already processed, so we can wire its messages onto the parent's
  * last message immediately. Sibling responses (same `previous_response_id`)
- * surface as branches; the latest one wins as the active branch.
+ * surface as branches; the latest one wins as the active branch. The same
+ * preference applies at the top level: when the first turn is branched into
+ * several root responses, the latest becomes the active `rootId`.
  *
  * Every emitted node carries `metadata.responseId` pointing back to the
  * originating `StoredResponse.id` (see {@link ResponseTreeMessage}). The

package/src/components/base/CkAgentMessage/CkAgentMessage.spec.ts

@@ -32,31 +32,26 @@ describe('CkAgentMessage', () => {
     });
   });
 
-  describe('icon', () => {
-    it('displays the icon when provided via slot', () => {
-      render(CkAgentMessage, {
-        slots: { icon: '<img src="" alt="bot-icon" />' },
-      });
-
-      expect(screen.getByAltText('bot-icon')).toBeInTheDocument();
-    });
-
-    it('renders the icon container only when the icon slot is provided', () => {
-      render(CkAgentMessage, {
-        slots: { icon: '<img src="" alt="bot-icon" />' },
+  describe('bottom separator', () => {
+    it('renders a bottom border by default', () => {
+      const { container } = render(CkAgentMessage, {
+        slots: { default: 'Some content' },
       });
 
-      expect(screen.getByTestId('ck-agent-message-icon')).toBeInTheDocument();
+      expect((container.firstElementChild as HTMLElement).className).toContain(
+        'border-b',
+      );
     });
 
-    it('does not render the icon container when no icon slot is provided', () => {
-      render(CkAgentMessage, {
+    it('omits the bottom border when bottomSeparator is false', () => {
+      const { container } = render(CkAgentMessage, {
+        props: { bottomSeparator: false },
         slots: { default: 'Some content' },
       });
 
       expect(
-        screen.queryByTestId('ck-agent-message-icon'),
-      ).not.toBeInTheDocument();
+        (container.firstElementChild as HTMLElement).className,
+      ).not.toContain('border-b');
     });
   });