@aleph-alpha/chat-kit 2.0.0 → 2.1.0

package/dist/composables/fileUpload/useFileUpload.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"useFileUpload.d.ts","sourceRoot":"","sources":["../../../src/composables/fileUpload/useFileUpload.ts"],"names":[],"mappings":"AACA,OAAO,EAkBL,KAAK,oBAAoB,EACzB,KAAK,mBAAmB,EAKzB,MAAM,SAAS,CAAC;AAEjB,cAAc,SAAS,CAAC;AAybxB,wBAAgB,aAAa,CAC3B,OAAO,EAAE,oBAAoB,GAC5B,mBAAmB,CAmwBrB"}
+{"version":3,"file":"useFileUpload.d.ts","sourceRoot":"","sources":["../../../src/composables/fileUpload/useFileUpload.ts"],"names":[],"mappings":"AACA,OAAO,EAkBL,KAAK,oBAAoB,EACzB,KAAK,mBAAmB,EAKzB,MAAM,SAAS,CAAC;AAEjB,cAAc,SAAS,CAAC;AAybxB,wBAAgB,aAAa,CAC3B,OAAO,EAAE,oBAAoB,GAC5B,mBAAmB,CA6xBrB"}

package/dist/composables/index.js

@@ -1,4 +1,4 @@
-import { c, d, u, b, e, f } from "../index-Ct5hD214.js";
+import { c, d, u, b, e, f } from "../index-DrtHeZpl.js";
 export {
   c as createDocumentUploadProvider,
   d as defaultDocumentUploadConfig,

package/dist/{index-Ct5hD214.js → index-DrtHeZpl.js}

@@ -1033,6 +1033,19 @@ function useFileUpload(options) {
     for (const key of involvedStateKeys) {
       updateEntry(key, { status: "processing", progress: 0 });
     }
+    const completeInvolvedFiles = (documents = []) => {
+      const documentIdByFileId = new Map(
+        documents.filter((doc) => doc.file_id && doc.document_id).map((doc) => [doc.file_id, doc.document_id])
+      );
+      for (const key of involvedStateKeys) {
+        updateEntry(key, { status: "success", progress: 100 });
+        const remoteFileId = files.value.get(key)?.remoteFileId;
+        const documentId = remoteFileId ? documentIdByFileId.get(remoteFileId) : void 0;
+        if (documentId) {
+          updateFileMetadata(key, { documentId });
+        }
+      }
+    };
     const runResponse = await client.post(
       endpoints.workflowRun(storeId, workflowId),
       { input },
@@ -1075,9 +1088,7 @@ function useFileUpload(options) {
       if (data.status === "completed") {
         workflowRun.value.status = "completed";
         workflowRun.value.completedAt = now();
-        for (const key of involvedStateKeys) {
-          updateEntry(key, { status: "success", progress: 100 });
-        }
+        completeInvolvedFiles(Array.isArray(data.documents) ? data.documents : []);
         emit({ type: "workflow:completed", runId });
       } else if (data.status === "failed" || data.status === "error") {
         workflowRun.value.status = "failed";

package/dist/index.js

@@ -1,6 +1,6 @@
 import { u } from "./useMessageHandler-CMdy-lm0.js";
 import { C, d, p, u as u2, a } from "./useChatKitLabels-ZutacPad.js";
-import { S, c, d as d2, a as a2, u as u3, b, e, f } from "./index-Ct5hD214.js";
+import { S, c, d as d2, a as a2, u as u3, b, e, f } from "./index-DrtHeZpl.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 } from "./CkInput.vue_vue_type_script_setup_true_lang-DPZedAQS.js";
 export {
   C as CHAT_KIT_LABELS,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aleph-alpha/chat-kit",
-  "version": "2.0.0",
+  "version": "2.1.0",
   "description": "Chat kit package for building chat interfaces with Vue",
   "type": "module",
   "main": "./dist/index.js",

package/src/composables/fileUpload/useFileUpload.spec.ts

@@ -0,0 +1,111 @@
+import { useFileUpload } from './useFileUpload';
+import type { FileEntry, HttpClient, SseEvent } from './types';
+
+/**
+ * Build a minimal `FileEntry` in the post-upload `processing` state so
+ * `runWorkflow` treats it as an involved file (it filters on `remoteFileId`).
+ */
+function makeEntry(id: string, remoteFileId: string): FileEntry {
+  return {
+    id,
+    groupId: 'group-1',
+    fileName: `${id}.pdf`,
+    fileSize: 1,
+    fileType: 'application/pdf',
+    status: 'processing',
+    progress: 0,
+    queuedAt: 'now',
+    uploadStartedAt: null,
+    uploadCompletedAt: null,
+    cancelledAt: null,
+    removedAt: null,
+    uploadDurationMs: null,
+    remoteFileId,
+    error: null,
+    attemptCount: 0,
+    lastAttemptAt: null,
+    thumbnail: null,
+    metadata: {},
+  };
+}
+
+/**
+ * Mock `HttpClient` whose `stream` captures the subscriber so the test can
+ * drive SSE events synchronously. `post` resolves the workflow run id.
+ */
+function createMockClient(captured: { cb?: (event: SseEvent) => void }) {
+  return {
+    get: vi.fn(),
+    delete: vi.fn(),
+    post: vi.fn().mockResolvedValue({ id: 'run-1' }),
+    stream: vi.fn(() => ({
+      subscribe: (cb: (event: SseEvent) => void) => {
+        captured.cb = cb;
+      },
+      close: vi.fn(),
+    })),
+  } as unknown as HttpClient;
+}
+
+function completeEvent(
+  documents: Array<{ document_id?: string; file_id?: string }>,
+): SseEvent {
+  return {
+    event: 'complete',
+    data: JSON.stringify({
+      document_ids: documents.map((d) => d.document_id),
+      documents,
+      status: 'completed',
+    }),
+  };
+}
+
+describe('useFileUpload runWorkflow', () => {
+  it('stamps documentId onto each file from the completed event, matched by file_id', async () => {
+    const captured: { cb?: (event: SseEvent) => void } = {};
+    const upload = useFileUpload({
+      client: createMockClient(captured),
+      workflowId: 'wf',
+      storeId: 'ss-1',
+    });
+
+    upload.files.value.set('key-1', makeEntry('key-1', 'file-1'));
+    upload.files.value.set('key-2', makeEntry('key-2', 'file-2'));
+
+    await upload.runWorkflow(['file-1', 'file-2']);
+    expect(captured.cb).toBeDefined();
+
+    // Reverse order on purpose: mapping must be keyed by file_id, not position.
+    captured.cb!(
+      completeEvent([
+        { document_id: 'doc-2', file_id: 'file-2' },
+        { document_id: 'doc-1', file_id: 'file-1' },
+      ]),
+    );
+
+    expect(upload.getFileState('key-1')?.metadata.documentId).toBe('doc-1');
+    expect(upload.getFileState('key-2')?.metadata.documentId).toBe('doc-2');
+    expect(upload.getFileState('key-1')?.status).toBe('success');
+    expect(upload.getFileState('key-2')?.status).toBe('success');
+  });
+
+  it('marks files successful without a documentId when the event has no documents', async () => {
+    const captured: { cb?: (event: SseEvent) => void } = {};
+    const upload = useFileUpload({
+      client: createMockClient(captured),
+      workflowId: 'wf',
+      storeId: 'ss-1',
+    });
+
+    upload.files.value.set('key-1', makeEntry('key-1', 'file-1'));
+
+    await upload.runWorkflow(['file-1']);
+    captured.cb!({
+      event: 'complete',
+      data: JSON.stringify({ status: 'completed' }),
+    });
+
+    expect(upload.getFileState('key-1')?.status).toBe('success');
+    expect(upload.getFileState('key-1')?.metadata.documentId).toBeUndefined();
+  });
+});

package/src/composables/fileUpload/useFileUpload.ts

@@ -1054,6 +1054,30 @@ export function useFileUpload(
       updateEntry(key, { status: 'processing', progress: 0 });
     }
 
+    // Mark the involved files successful and stamp the `documentId` resolved
+    // from the terminal event's `documents` array (mapping `file_id` ->
+    // `document_id`) onto each matching entry, so consumers can reference the
+    // produced document directly without a separate lookup.
+    const completeInvolvedFiles = (
+      documents: Array<{ document_id?: string; file_id?: string }> = [],
+    ) => {
+      const documentIdByFileId = new Map(
+        documents
+          .filter((doc) => doc.file_id && doc.document_id)
+          .map((doc) => [doc.file_id, doc.document_id] as const),
+      );
+      for (const key of involvedStateKeys) {
+        updateEntry(key, { status: 'success', progress: 100 });
+        const remoteFileId = files.value.get(key)?.remoteFileId;
+        const documentId = remoteFileId
+          ? documentIdByFileId.get(remoteFileId)
+          : undefined;
+        if (documentId) {
+          updateFileMetadata(key, { documentId });
+        }
+      }
+    };
+
     const runResponse = await client.post<{ id: string }>(
       endpoints.workflowRun(storeId, workflowId),
       { input },
@@ -1087,7 +1111,11 @@ export function useFileUpload(
     sseStream.subscribe((event: SseEvent) => {
       if (!workflowRun.value) return;
 
-      let data: { id?: string; status?: string };
+      let data: {
+        id?: string;
+        status?: string;
+        documents?: Array<{ document_id?: string; file_id?: string }>;
+      };
       try {
         data = JSON.parse(event.data);
       } catch {
@@ -1105,9 +1133,7 @@ export function useFileUpload(
       if (data.status === 'completed') {
         workflowRun.value.status = 'completed';
         workflowRun.value.completedAt = now();
-        for (const key of involvedStateKeys) {
-          updateEntry(key, { status: 'success', progress: 100 });
-        }
+        completeInvolvedFiles(Array.isArray(data.documents) ? data.documents : []);
         emit({ type: 'workflow:completed', runId });
       } else if (data.status === 'failed' || data.status === 'error') {
         workflowRun.value.status = 'failed';