@aleph-alpha/chat-kit 3.2.0 → 4.0.0

package/dist/{index-DAmWGYNH.js → index-CyrVnbS9.js}

@@ -1,6 +1,6 @@
 import { g as generateId } from "./generateId-BHf0W4Rq.js";
 import { ref, shallowRef, computed, readonly, toRaw, toValue, watch } from "vue";
-import { u as useMessageHandler } from "./useMessageHandler-CMdy-lm0.js";
+import { u as useMessageHandler } from "./useMessageHandler-N50Yj-NU.js";
 import "./katex.min-md74IFlY.js";
 function secondsToDate(seconds) {
   return new Date(seconds * 1e3);
@@ -765,13 +765,10 @@ function useFileUpload(options) {
   }
   async function upload(params) {
     const fileList = Array.isArray(params.files) ? params.files : [params.files];
-    const activeFileCount = [...files.value.values()].filter(
-      (e) => e.status !== "failed" && e.status !== "cancelled" && e.status !== "removed"
-    ).length;
-    if (activeFileCount + fileList.length > constraints.maxFilesPerUpload) {
+    if (fileList.length > constraints.maxFilesPerUpload) {
       throw new SdkError(
         "MAX_FILES_EXCEEDED",
-        `Cannot upload ${fileList.length} file(s) — ${activeFileCount} already active, maximum is ${constraints.maxFilesPerUpload}.`
+        `Cannot upload ${fileList.length} file(s) — maximum per selection is ${constraints.maxFilesPerUpload}.`
       );
     }
     const mergedMeta = {
@@ -1346,68 +1343,80 @@ function createDocumentUploadProvider(config) {
     ...config.headers ?? {}
   };
   const httpClient = buildHttpClient(baseUrl, defaultHeaders);
-  const endpoints = defaultEndpoints;
-  let resolvedStoreId = null;
-  let resolvePromise = null;
-  const resolvedWorkflowIdByStore = /* @__PURE__ */ new Map();
-  async function createWorkflowForStore(storeId) {
-    const workflowRes = await httpClient.post(
-      endpoints.createWorkflow(storeId),
+  const storeIdByConversation = /* @__PURE__ */ new Map();
+  const resolvePromises = /* @__PURE__ */ new Map();
+  function searchStoreNameForConversation(conversationId) {
+    return conversationId;
+  }
+  function findConfiguredWorkflow(items) {
+    const configuredWorkflowName = typeof workflowConfig.name === "string" ? workflowConfig.name : void 0;
+    return items?.find(
+      (workflow) => workflow.id === workflowId || workflow.name === workflowId || configuredWorkflowName != null && workflow.name === configuredWorkflowName
+    );
+  }
+  async function resolveWorkflowId(storeId) {
+    const response = await httpClient.get(
+      `search-stores/${storeId}/workflows`
+    );
+    const existing = findConfiguredWorkflow(response.items);
+    if (existing) {
+      return existing.id;
+    }
+    const created = await httpClient.post(
+      `search-stores/${storeId}/workflows`,
       workflowConfig
     );
-    return workflowRes.id;
+    return created.id;
   }
-  async function createSearchStoreWithWorkflow() {
-    const name = `store-${(/* @__PURE__ */ new Date()).toISOString().replace(/[:.]/g, "-")}`;
+  async function createSearchStore(conversationId) {
     const storeRes = await httpClient.post("search-stores", {
-      name,
+      name: searchStoreNameForConversation(conversationId),
       ...searchStoreConfig
     });
-    const newStoreId = storeRes.id;
-    const workflowApiId = await createWorkflowForStore(newStoreId);
-    resolvedWorkflowIdByStore.set(newStoreId, workflowApiId);
-    return newStoreId;
+    await httpClient.post(
+      `search-stores/${storeRes.id}/workflows`,
+      workflowConfig
+    );
+    return storeRes.id;
   }
-  async function fetchOrCreateWorkflow(storeId) {
-    const cached = resolvedWorkflowIdByStore.get(storeId);
-    if (cached) return cached;
-    const response = await httpClient.get(
-      endpoints.workflows(storeId)
+  function findStoreForConversation(items, conversationId) {
+    const expectedName = searchStoreNameForConversation(conversationId);
+    const expectedModel = searchStoreConfig.models[0]?.embedding_model;
+    return items.find(
+      (store) => store.name === expectedName && (!expectedModel || store.models.some((m) => m.embedding_model === expectedModel))
     );
-    if (response.items.length > 0) {
-      const match = response.items.find((item) => item.name === workflowId);
-      const selected = match ?? response.items[0];
-      resolvedWorkflowIdByStore.set(storeId, selected.id);
-      return selected.id;
+  }
+  async function fetchOrCreateStore(conversationId) {
+    const response = await httpClient.get("search-stores");
+    const existing = findStoreForConversation(response.items, conversationId);
+    if (existing) {
+      await resolveWorkflowId(existing.id);
+      storeIdByConversation.set(conversationId, existing.id);
+      return existing.id;
     }
-    const workflowApiId = await createWorkflowForStore(storeId);
-    resolvedWorkflowIdByStore.set(storeId, workflowApiId);
-    return workflowApiId;
+    const newStoreId = await createSearchStore(conversationId);
+    storeIdByConversation.set(conversationId, newStoreId);
+    return newStoreId;
   }
-  async function fetchOrCreateStore() {
-    try {
-      const response = await httpClient.get("search-stores");
-      const expectedModel = searchStoreConfig.models[0]?.embedding_model;
-      const match = response.items.find(
-        (store) => store.models.some((m) => m.embedding_model === expectedModel)
-      );
-      if (match) {
-        resolvedStoreId = match.id;
-        return resolvedStoreId;
-      }
-      resolvedStoreId = await createSearchStoreWithWorkflow();
-      return resolvedStoreId;
-    } catch (err) {
-      resolvePromise = null;
+  async function resolveSearchStore(conversationId) {
+    if (!conversationId) {
+      throw new Error("conversationId is required to resolve a search store.");
+    }
+    const cached = storeIdByConversation.get(conversationId);
+    if (cached) return cached;
+    const pending = resolvePromises.get(conversationId);
+    if (pending) return pending;
+    const promise = fetchOrCreateStore(conversationId).catch((err) => {
+      storeIdByConversation.delete(conversationId);
       throw err;
+    });
+    resolvePromises.set(conversationId, promise);
+    try {
+      return await promise;
+    } finally {
+      resolvePromises.delete(conversationId);
     }
   }
-  async function resolveSearchStore() {
-    if (resolvedStoreId) return resolvedStoreId;
-    if (resolvePromise) return resolvePromise;
-    resolvePromise = fetchOrCreateStore();
-    return resolvePromise;
-  }
   let instance = null;
   function purgeInterruptedFiles() {
     if (!instance) return;
@@ -1441,33 +1450,40 @@ function createDocumentUploadProvider(config) {
     const fileInput = ref(null);
     const upload = getUploadInstance();
     const resolvedMaxFiles = upload.getSupportedFileTypes().maxFilesPerUpload;
-    const storeId = ref(resolvedStoreId ?? "");
+    const storeId = ref(
+      storeIdByConversation.get(toValue(conversationIdRef)) ?? ""
+    );
     const currentScope = shallowRef(
       upload.scopeBy("conversationId", toValue(conversationIdRef))
     );
     watch(
       () => toValue(conversationIdRef),
-      (newId, oldId) => {
-        if (oldId && oldId !== newId) {
-          const oldScope = upload.scopeBy("conversationId", oldId);
-          for (const entry of oldScope.files.value) {
-            upload.updateFileMetadata(entry.id, { conversationId: newId });
-          }
-        }
+      async (newId) => {
         currentScope.value = upload.scopeBy("conversationId", newId);
-      }
+        if (!newId) {
+          storeId.value = "";
+          return;
+        }
+        try {
+          storeId.value = await resolveSearchStore(newId);
+        } catch {
+          storeId.value = "";
+        }
+      },
+      { immediate: true }
     );
     function scopedFiles() {
       return currentScope.value?.files.value ?? [];
     }
     async function uploadFiles(files) {
       if (!files.length) return;
-      const userStoreId = await resolveSearchStore();
-      const workflowApiId = await fetchOrCreateWorkflow(userStoreId);
+      const conversationId = toValue(conversationIdRef);
+      if (!conversationId) return;
+      const userStoreId = await resolveSearchStore(conversationId);
+      const workflowApiId = await resolveWorkflowId(userStoreId);
       upload.setStoreId(userStoreId);
       upload.setWorkflowId(workflowApiId);
       storeId.value = userStoreId;
-      const conversationId = toValue(conversationIdRef);
       const result = await upload.upload({
         id: `${conversationId}::${Date.now()}`,
         files,
@@ -1521,12 +1537,11 @@ function createDocumentUploadProvider(config) {
     function handleFilesSelected(event, options) {
       const input = event.target;
       let fileList = Array.from(input.files ?? []);
-      const availableSlots = resolvedMaxFiles - unassignedFiles.value.length;
-      if (fileList.length > availableSlots) {
-        const kept = Math.max(0, availableSlots);
+      if (fileList.length > resolvedMaxFiles) {
+        const kept = Math.max(0, resolvedMaxFiles);
         options?.onFilesExceeded?.({
           selected: fileList.length,
-          allowed: availableSlots,
+          allowed: resolvedMaxFiles,
           kept
         });
         fileList = fileList.slice(0, kept);

package/dist/index.d.ts

@@ -5,8 +5,8 @@ export { useChat, useConversationSessions, useConversations, } from './composabl
 export type { ChatActions, ChatState, ConversationSession, CreateConversationInput, MessageHandler, MutationOptions, UseChatReturn, UseConversationSessionsReturn, UseConversationsOptions, UseConversationsReturn, } from './composables';
 export { SdkError, createDocumentUploadProvider, defaultDocumentUploadConfig, defaultEndpoints, useDocumentUploadWithDefaultConfig, } from './composables/fileUpload';
 export type { DocumentUploadConfig, DocumentUploadProvider, FileEntry, FileStatus, } from './composables/fileUpload';
-export { CkAgentMessage, CkConversation, CkConversationLayout, CkCopyMessageActionButton, CkDropZoneOverlay, CkFeedbackButton, CkFeedbackDialog, CkFileItemsList, CkHistory, CkHistoryItem, CkHistoryWrapper, CkInput, CkMarkdownRenderer, CkMessageActionButton, CkModelSelector, CkPromptSendButton, CkPromptMenuButton, CkPromptStopButton, CkPromptInput, CkPromptInputDisclaimer, CkSendIcon, CkStopIcon, CkUserMessage, } from './components';
-export type { CkConversationLayoutProps, CkConversationProps, CkDropZoneOverlayProps, CkDropZoneOverlayEmits, CkDropZoneRejection, CkDropZoneRejectionReason, CkFeedbackButtonProps, CkFeedbackDialogEmits, CkFeedbackDialogProps, CkFeedbackDialogSubmitPayload, CkFeedbackReason, CkFeedbackSentiment, CkFeedbackValue, CkFileEntry, CkFileItemsListEmits, CkFileItemsListProps, CkHistoryEmits, CkHistoryItemEmits, CkHistoryItemProps, CkHistoryProps, CkHistoryWrapperProps, CkMarkdownElementClasses, CkMarkdownRendererProps, CkModelOption, CkModelSelectorProps, CkModelSelectorEmits, CkPromptMenuButtonProps, CkUserMessageProps, } from './components';
+export { CkAgentMessage, CkCitation, CkCitationCard, CkConversation, CkConversationLayout, CkCopyMessageButton, CkDropZoneOverlay, CkFeedbackButton, CkFeedbackDialog, CkFileItemsList, CkHistory, CkHistoryItem, CkHistoryWrapper, CkInput, CkMarkdownRenderer, CkMessageButton, CkModelSelector, CkPromptSendButton, CkPromptMenuButton, CkPromptStopButton, CkPromptInput, CkPromptInputDisclaimer, CkRegenerateMessageButton, CkSendIcon, CkStopIcon, CkUserMessage, } from './components';
+export type { CkCitationProps, CkCitationEmits, CkCitationSource, CkCitationCardProps, CkCitationCardEmits, CkConversationLayoutProps, CkConversationProps, CkDropZoneOverlayProps, CkDropZoneOverlayEmits, CkDropZoneRejection, CkDropZoneRejectionReason, CkFeedbackButtonProps, CkFeedbackDialogEmits, CkFeedbackDialogProps, CkFeedbackDialogSubmitPayload, CkFeedbackReason, CkFeedbackSentiment, CkFeedbackValue, CkFileEntry, CkFileItemsListEmits, CkFileItemsListProps, CkHistoryEmits, CkHistoryItemEmits, CkHistoryItemProps, CkHistoryProps, CkHistoryWrapperProps, CkMarkdownElementClasses, CkMarkdownRendererProps, CkMarkdownRendererEmits, CkModelOption, CkModelSelectorProps, CkModelSelectorEmits, CkPromptMenuButtonProps, CkUserMessageProps, } from './components';
 export { formatRelativeTime } from './helpers';
 export type { BaseMessage, Conversation, ConversationTree, MessageContent, MessageContentStatus, MessageMetadata, MessageReasoningContent, MessageRole, MessageStatus, MessageTextContent, MessageToolContent, TreeMessage, } from './types';
 //# 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":"AACA,OAAO,EACL,KAAK,UAAU,EACf,KAAK,WAAW,EAChB,KAAK,YAAY,EACjB,KAAK,mBAAmB,EACxB,KAAK,UAAU,EACf,KAAK,eAAe,EACpB,KAAK,mBAAmB,EACxB,KAAK,WAAW,GACjB,MAAM,YAAY,CAAC;AASpB,OAAO,EACL,iBAAiB,EACjB,iBAAiB,EACjB,oBAAoB,EACpB,gBAAgB,EAChB,oBAAoB,EACpB,eAAe,GAChB,MAAM,YAAY,CAAC;AACpB,YAAY,EAAE,aAAa,EAAE,MAAM,YAAY,CAAC;AAGhD,OAAO,EACL,OAAO,EACP,uBAAuB,EACvB,gBAAgB,GACjB,MAAM,eAAe,CAAC;AACvB,YAAY,EACV,WAAW,EACX,SAAS,EACT,mBAAmB,EACnB,uBAAuB,EACvB,cAAc,EACd,eAAe,EACf,aAAa,EACb,6BAA6B,EAC7B,uBAAuB,EACvB,sBAAsB,GACvB,MAAM,eAAe,CAAC;AAEvB,OAAO,EACL,QAAQ,EACR,4BAA4B,EAC5B,2BAA2B,EAC3B,gBAAgB,EAChB,kCAAkC,GACnC,MAAM,0BAA0B,CAAC;AAClC,YAAY,EACV,oBAAoB,EACpB,sBAAsB,EACtB,SAAS,EACT,UAAU,GACX,MAAM,0BAA0B,CAAC;AAGlC,OAAO,EACL,cAAc,EACd,cAAc,EACd,oBAAoB,EACpB,yBAAyB,EACzB,iBAAiB,EACjB,gBAAgB,EAChB,gBAAgB,EAChB,eAAe,EACf,SAAS,EACT,aAAa,EACb,gBAAgB,EAChB,OAAO,EACP,kBAAkB,EAClB,qBAAqB,EACrB,eAAe,EACf,kBAAkB,EAClB,kBAAkB,EAClB,kBAAkB,EAClB,aAAa,EACb,uBAAuB,EACvB,UAAU,EACV,UAAU,EACV,aAAa,GACd,MAAM,cAAc,CAAC;AACtB,YAAY,EACV,yBAAyB,EACzB,mBAAmB,EACnB,sBAAsB,EACtB,sBAAsB,EACtB,mBAAmB,EACnB,yBAAyB,EACzB,qBAAqB,EACrB,qBAAqB,EACrB,qBAAqB,EACrB,6BAA6B,EAC7B,gBAAgB,EAChB,mBAAmB,EACnB,eAAe,EACf,WAAW,EACX,oBAAoB,EACpB,oBAAoB,EACpB,cAAc,EACd,kBAAkB,EAClB,kBAAkB,EAClB,cAAc,EACd,qBAAqB,EACrB,wBAAwB,EACxB,uBAAuB,EACvB,aAAa,EACb,oBAAoB,EACpB,oBAAoB,EACpB,uBAAuB,EACvB,kBAAkB,GACnB,MAAM,cAAc,CAAC;AAGtB,OAAO,EAAE,kBAAkB,EAAE,MAAM,WAAW,CAAC;AAG/C,YAAY,EACV,WAAW,EACX,YAAY,EACZ,gBAAgB,EAChB,cAAc,EACd,oBAAoB,EACpB,eAAe,EACf,uBAAuB,EACvB,WAAW,EACX,aAAa,EACb,kBAAkB,EAClB,kBAAkB,EAClB,WAAW,GACZ,MAAM,SAAS,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AACA,OAAO,EACL,KAAK,UAAU,EACf,KAAK,WAAW,EAChB,KAAK,YAAY,EACjB,KAAK,mBAAmB,EACxB,KAAK,UAAU,EACf,KAAK,eAAe,EACpB,KAAK,mBAAmB,EACxB,KAAK,WAAW,GACjB,MAAM,YAAY,CAAC;AASpB,OAAO,EACL,iBAAiB,EACjB,iBAAiB,EACjB,oBAAoB,EACpB,gBAAgB,EAChB,oBAAoB,EACpB,eAAe,GAChB,MAAM,YAAY,CAAC;AACpB,YAAY,EAAE,aAAa,EAAE,MAAM,YAAY,CAAC;AAGhD,OAAO,EACL,OAAO,EACP,uBAAuB,EACvB,gBAAgB,GACjB,MAAM,eAAe,CAAC;AACvB,YAAY,EACV,WAAW,EACX,SAAS,EACT,mBAAmB,EACnB,uBAAuB,EACvB,cAAc,EACd,eAAe,EACf,aAAa,EACb,6BAA6B,EAC7B,uBAAuB,EACvB,sBAAsB,GACvB,MAAM,eAAe,CAAC;AAEvB,OAAO,EACL,QAAQ,EACR,4BAA4B,EAC5B,2BAA2B,EAC3B,gBAAgB,EAChB,kCAAkC,GACnC,MAAM,0BAA0B,CAAC;AAClC,YAAY,EACV,oBAAoB,EACpB,sBAAsB,EACtB,SAAS,EACT,UAAU,GACX,MAAM,0BAA0B,CAAC;AAGlC,OAAO,EACL,cAAc,EACd,UAAU,EACV,cAAc,EACd,cAAc,EACd,oBAAoB,EACpB,mBAAmB,EACnB,iBAAiB,EACjB,gBAAgB,EAChB,gBAAgB,EAChB,eAAe,EACf,SAAS,EACT,aAAa,EACb,gBAAgB,EAChB,OAAO,EACP,kBAAkB,EAClB,eAAe,EACf,eAAe,EACf,kBAAkB,EAClB,kBAAkB,EAClB,kBAAkB,EAClB,aAAa,EACb,uBAAuB,EACvB,yBAAyB,EACzB,UAAU,EACV,UAAU,EACV,aAAa,GACd,MAAM,cAAc,CAAC;AACtB,YAAY,EACV,eAAe,EACf,eAAe,EACf,gBAAgB,EAChB,mBAAmB,EACnB,mBAAmB,EACnB,yBAAyB,EACzB,mBAAmB,EACnB,sBAAsB,EACtB,sBAAsB,EACtB,mBAAmB,EACnB,yBAAyB,EACzB,qBAAqB,EACrB,qBAAqB,EACrB,qBAAqB,EACrB,6BAA6B,EAC7B,gBAAgB,EAChB,mBAAmB,EACnB,eAAe,EACf,WAAW,EACX,oBAAoB,EACpB,oBAAoB,EACpB,cAAc,EACd,kBAAkB,EAClB,kBAAkB,EAClB,cAAc,EACd,qBAAqB,EACrB,wBAAwB,EACxB,uBAAuB,EACvB,uBAAuB,EACvB,aAAa,EACb,oBAAoB,EACpB,oBAAoB,EACpB,uBAAuB,EACvB,kBAAkB,GACnB,MAAM,cAAc,CAAC;AAGtB,OAAO,EAAE,kBAAkB,EAAE,MAAM,WAAW,CAAC;AAG/C,YAAY,EACV,WAAW,EACX,YAAY,EACZ,gBAAgB,EAChB,cAAc,EACd,oBAAoB,EACpB,eAAe,EACf,uBAAuB,EACvB,WAAW,EACX,aAAa,EACb,kBAAkB,EAClB,kBAAkB,EAClB,WAAW,GACZ,MAAM,SAAS,CAAC"}

package/dist/index.js

@@ -1,38 +1,41 @@
-import { u } from "./useMessageHandler-CMdy-lm0.js";
-import { C, d, p, u as u2, a } from "./useChatKitLabels-D5gNv8se.js";
-import { S, c, d as d2, a as a2, u as u3, b, e, f } from "./index-DAmWGYNH.js";
-import { _, a as a3, b as b2, C as C2, 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 } from "./CkInput.vue_vue_type_script_setup_true_lang-B-4xnH2c.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 { _, 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, y } from "./CkInput.vue_vue_type_script_setup_true_lang-DJN-nZ2-.js";
 export {
   C as CHAT_KIT_LABELS,
   _ as CkAgentMessage,
-  a3 as CkConversation,
-  b2 as CkConversationLayout,
-  C2 as CkCopyMessageActionButton,
-  c2 as CkDropZoneOverlay,
-  d3 as CkFeedbackButton,
-  e2 as CkFeedbackDialog,
-  f2 as CkFileItemsList,
-  g as CkHistory,
-  h as CkHistoryItem,
-  i as CkHistoryWrapper,
-  j as CkInput,
-  k as CkMarkdownRenderer,
-  l as CkMessageActionButton,
-  m as CkModelSelector,
-  n as CkPromptInput,
-  o as CkPromptInputDisclaimer,
-  p2 as CkPromptMenuButton,
-  q as CkPromptSendButton,
-  r as CkPromptStopButton,
-  s as CkSendIcon,
-  t as CkStopIcon,
-  u4 as CkUserMessage,
+  C2 as CkCitation,
+  a3 as CkCitationCard,
+  b2 as CkConversation,
+  c2 as CkConversationLayout,
+  d3 as CkCopyMessageButton,
+  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 CkMessageButton,
+  o as CkModelSelector,
+  p2 as CkPromptInput,
+  q as CkPromptInputDisclaimer,
+  r as CkPromptMenuButton,
+  s as CkPromptSendButton,
+  t as CkPromptStopButton,
+  u4 as CkRegenerateMessageButton,
+  v as CkSendIcon,
+  w as CkStopIcon,
+  x as CkUserMessage,
   S as SdkError,
   c as createDocumentUploadProvider,
   d as defaultChatKitLabels,
   d2 as defaultDocumentUploadConfig,
   a2 as defaultEndpoints,
-  v as formatRelativeTime,
+  y as formatRelativeTime,
   p as provideChatKitLabels,
   u3 as useChat,
   u2 as useChatKitLabels,

package/dist/{useChatKitLabels-D5gNv8se.js → useChatKitLabels-zMFV7aEz.js}

@@ -28622,8 +28622,8 @@ const SANITIZE_CONFIG = {
 };
 function buildProcessor(plugins) {
   let proc = unified().use(remarkParse).use(remarkNormalizeMath).use(remarkGfm).use(remarkMath);
-  for (const plugin of plugins) {
-    proc = proc.use(plugin);
+  if (plugins.length > 0) {
+    proc = proc.use(plugins);
   }
   return proc.use(remarkRehype).use(rehypeKatex).use(rehypeStringify);
 }
@@ -28650,12 +28650,37 @@ function useMarkdownParser(source, plugins) {
   });
   return { html: html2 };
 }
+const CITATION_RE = /\[(\d+)\]/g;
+const remarkCitations = (options = {}) => {
+  const { indices } = options;
+  return (tree) => {
+    findAndReplace(tree, [
+      CITATION_RE,
+      (_match, digits) => {
+        const index2 = Number(digits);
+        if (index2 < 1) return false;
+        if (indices && !indices.has(index2)) return false;
+        return {
+          type: "citationReference",
+          data: {
+            hName: "span",
+            // Use the literal attribute name so the placeholder contract matches
+            // the renderer's `getAttribute('data-citation-index')` lookup exactly.
+            hProperties: { "data-citation-index": index2 },
+            hChildren: [{ type: "text", value: String(index2) }]
+          }
+        };
+      }
+    ]);
+  };
+};
 const defaultChatKitLabels = {
   agentAvatarAlt: "Agent avatar",
   agentLoadingAvatarAlt: "Agent loading",
   inputPlaceholder: "Type a message...",
   copyMessageAction: "Copy response",
   copyMessageActionCopied: "Copied!",
+  regenerateMessageAction: "Regenerate response",
   feedbackThumbsUp: "Good response",
   feedbackThumbsDown: "Bad response",
   feedbackGroup: "Rate this response",
@@ -28674,6 +28699,8 @@ const defaultChatKitLabels = {
   modelSelectorListAriaLabel: "Models",
   modelSelectorPlaceholder: "Select model",
   markdownTableLabel: "Table",
+  citationOpenSource: "Open source",
+  citationSourceLabel: "Source",
   historyItemUntitled: "Untitled conversation",
   historyItemActionAriaLabel: "Conversation options",
   historyItemEditAction: "Edit",
@@ -28696,9 +28723,10 @@ function useChatKitLabels() {
 export {
   CHAT_KIT_LABELS as C,
   useMarkdownParser as a,
+  remarkNormalizeMath as b,
   defaultChatKitLabels as d,
   normalizeMathDelimiters as n,
   provideChatKitLabels as p,
-  remarkNormalizeMath as r,
+  remarkCitations as r,
   useChatKitLabels as u
 };

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.2.0",
+  "version": "4.0.0",
   "description": "Chat kit package for building chat interfaces with Vue",
   "type": "module",
   "main": "./dist/index.js",
@@ -68,6 +68,7 @@
     "@aleph-alpha/ui-library": "^2.8.3",
     "dompurify": "^3.3.2",
     "katex": "^0.17.0",
+    "mdast-util-find-and-replace": "^3.0.2",
     "rehype-katex": "^7.0.1",
     "rehype-stringify": "^10.0.1",
     "remark-gfm": "^4.0.1",
@@ -84,6 +85,7 @@
     "@testing-library/jest-dom": "6.9.1",
     "@testing-library/vue": "8.1.0",
     "@types/dompurify": "^3.2.0",
+    "@types/mdast": "^4.0.4",
     "@types/node": "^25.1.0",
     "@unocss/preset-wind4": "^66.6.2",
     "@unocss/reset": "^66.6.2",
@@ -100,9 +102,9 @@
     "vitest": "3.2.6",
     "vue": "^3.5.27",
     "vue-tsc": "^3.2.4",
+    "@aleph-alpha/eslint-config-frontend": "0.7.0",
     "@aleph-alpha/prettier-config-frontend": "2.1.0",
-    "@aleph-alpha/tsconfig-frontend": "0.7.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 {