@aleph-alpha/chat-kit 3.1.1 → 3.1.2

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

@@ -1 +1 @@
-{"version":3,"file":"useFileUploadWrapper.d.ts","sourceRoot":"","sources":["../../../src/composables/fileUpload/useFileUploadWrapper.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAOV,oBAAoB,EACpB,sBAAsB,EAIvB,MAAM,SAAS,CAAC;AAiJjB,wBAAgB,4BAA4B,CAC1C,MAAM,EAAE,oBAAoB,GAC3B,sBAAsB,CAqPxB"}
+{"version":3,"file":"useFileUploadWrapper.d.ts","sourceRoot":"","sources":["../../../src/composables/fileUpload/useFileUploadWrapper.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAOV,oBAAoB,EACpB,sBAAsB,EAIvB,MAAM,SAAS,CAAC;AAiJjB,wBAAgB,4BAA4B,CAC1C,MAAM,EAAE,oBAAoB,GAC3B,sBAAsB,CAqQxB"}

package/dist/composables/index.js

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

package/dist/{index-CnuiV5vd.js → index-BgQf4bMm.js}

@@ -1354,23 +1354,28 @@ function createDocumentUploadProvider(config) {
     );
     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;
+      throw err;
+    }
+  }
   async function resolveSearchStore() {
     if (resolvedStoreId) return resolvedStoreId;
     if (resolvePromise) return resolvePromise;
-    resolvePromise = (async () => {
-      try {
-        const stores = await httpClient.get("search-stores");
-        if (stores.length > 0 && stores[0]) {
-          resolvedStoreId = stores[0].id;
-          return resolvedStoreId;
-        }
-        resolvedStoreId = await createSearchStoreWithWorkflow();
-        return resolvedStoreId;
-      } catch (err) {
-        resolvePromise = null;
-        throw err;
-      }
-    })();
+    resolvePromise = fetchOrCreateStore();
     return resolvePromise;
   }
   let instance = null;

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-CLsWWwgG.js";
-import { S, c, d as d2, a as a2, u as u3, b, e, f } from "./index-CnuiV5vd.js";
+import { S, c, d as d2, a as a2, u as u3, b, e, f } from "./index-BgQf4bMm.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-DKzc21Yp.js";
 export {
   C as CHAT_KIT_LABELS,

package/package.json

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

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

@@ -59,7 +59,16 @@ describe('useFileUploadWrapper handleFilesSelected', () => {
       const method = init?.method ?? 'GET';
 
       if (method === 'GET' && url.includes('search-stores')) {
-        return createMockFetchResponse([{ id: 'store-1' }]);
+        return createMockFetchResponse({
+          items: [
+            {
+              id: 'store-1',
+              name: 'test-store',
+              models: [{ embedding_model: 'test-model' }],
+            },
+          ],
+          total: 1,
+        });
       }
       if (method === 'POST' && url.includes('/files')) {
         return createMockFetchResponse([{ id: 'remote-1' }]);

package/src/composables/fileUpload/useFileUploadWrapper.ts

@@ -198,27 +198,43 @@ export function createDocumentUploadProvider(
     return newStoreId;
   }
 
-  async function resolveSearchStore(): Promise<string> {
-    if (resolvedStoreId) return resolvedStoreId;
-    if (resolvePromise) return resolvePromise;
+  interface SearchStoreListResponse {
+    items: Array<{
+      id: string;
+      name: string;
+      models: Array<{ embedding_model: string }>;
+    }>;
+    total: number;
+  }
 
-    resolvePromise = (async () => {
-      try {
-        const stores = await httpClient.get<{ id: string }[]>('search-stores');
+  async function fetchOrCreateStore(): Promise<string> {
+    try {
+      const response =
+        await httpClient.get<SearchStoreListResponse>('search-stores');
 
-        if (stores.length > 0 && stores[0]) {
-          resolvedStoreId = stores[0].id;
-          return resolvedStoreId;
-        }
+      const expectedModel = searchStoreConfig.models[0]?.embedding_model;
+      const match = response.items.find((store) =>
+        store.models.some((m) => m.embedding_model === expectedModel),
+      );
 
-        resolvedStoreId = await createSearchStoreWithWorkflow();
+      if (match) {
+        resolvedStoreId = match.id;
         return resolvedStoreId;
-      } catch (err) {
-        resolvePromise = null;
-        throw err;
       }
-    })();
 
+      resolvedStoreId = await createSearchStoreWithWorkflow();
+      return resolvedStoreId;
+    } catch (err) {
+      resolvePromise = null;
+      throw err;
+    }
+  }
+
+  async function resolveSearchStore(): Promise<string> {
+    if (resolvedStoreId) return resolvedStoreId;
+    if (resolvePromise) return resolvePromise;
+
+    resolvePromise = fetchOrCreateStore();
     return resolvePromise;
   }