cozy-search 2.2.29 → 2.3.0

package/README.md

@@ -34,15 +34,59 @@ import 'cozy-search/dist/stylesheet.css'
   "description": "Required by the cozy Assistant",
   "type": "io.cozy.ai.chat.events",
   "verbs": ["GET"]
+},
+"chatAssistants": {
+  "description": "Required by the cozy Assistant",
+  "type": "io.cozy.ai.chat.assistants",
+  "verbs": ["GET", "POST", "PUT", "DELETE"]
+},
+"accounts": {
+  "description": "Required by the cozy Assistant (provider accounts)",
+  "type": "io.cozy.accounts",
+  "verbs": ["GET", "POST", "PUT", "DELETE"]
 }
 ```
 
+The assistants and their provider accounts are created, edited and
+deleted from the library's dialogs, so the host app declares the full set
+of verbs (cozy-home and cozy-drive use `ALL`).
+
 2. Add realtime queries for chat conversations in your tree :
 
 ```jsx
 <RealTimeQueries doctype="io.cozy.ai.chat.conversations" />
 ```
 
+### RAG indexing setup
+
+Call `setupRagIndexing(client)` once per session at startup (before
+`ensureProvisionedAssistants` if your app provisions assistants from the
+`rag.assistants.autoprovision` flag). It makes sure the instance's two
+`rag-index` triggers exist (one on `io.cozy.files`, one on
+`io.cozy.ai.chat.assistants`) and gives the root folder to any assistant
+that has no knowledge base folder yet — the stack's `rag-index` worker
+reads the assistants to know what to index, cozy-search only has to keep
+the triggers and the assistants' `knowledgeBase` in shape.
+
+It needs the following permissions: `io.cozy.triggers` and `io.cozy.jobs`
+(to create and launch the triggers), `io.cozy.ai.chat.assistants` and
+`io.cozy.files` (to read and migrate the assistants). It is idempotent and
+never throws: on a stack whose `rag-index` worker is still reserved, the
+403 is logged and the app keeps working. `ensureProvisionedAssistants`
+additionally creates one `io.cozy.accounts` document per provisioned
+assistant, and the `dirName` folder in `io.cozy.files`.
+
+An entry of `rag.assistants.autoprovision` can carry `"default": true`
+(the first flagged entry wins if there are several). Every NEW
+conversation then starts on that assistant once its document exists —
+from the session after it was provisioned onward. Existing conversations
+are unaffected: they keep their own assistant, or stay unscoped if they
+had none.
+
+```json
+{ "name": "Mes documents", "dirName": "Documents", "default": true }
+```
+
 ### On desktop
 
 You can add the search bar like this :

package/dist/components/Assistant/AssistantSelection.js

@@ -165,7 +165,7 @@ var AssistantSelection = function AssistantSelection(_ref) {
     assistant: _constants.DEFAULT_ASSISTANT,
     onClose: handleClose,
     onSelect: function onSelect() {
-      return setSelectedAssistantId(_constants.DEFAULT_ASSISTANT.id);
+      return setSelectedAssistantId(_constants.DEFAULT_ASSISTANT._id);
     },
     selectedAssistant: selectedAssistant,
     disableActions: true

package/dist/components/CozyAssistantRuntimeProvider.js

@@ -43,12 +43,14 @@ var _twakeI18n = require("twake-i18n");
 
 var _AssistantProvider = require("./AssistantProvider");
 
+var _conversationAssistant = require("./KnowledgeBase/conversationAssistant");
+
+var _useDefaultAssistantId = require("./KnowledgeBase/useDefaultAssistantId");
+
 var _CozyRealtimeChatAdapter = require("./adapters/CozyRealtimeChatAdapter");
 
 var _StreamBridge = require("./adapters/StreamBridge");
 
-var _constants = require("./constants");
-
 var _helpers = require("./helpers");
 
 var _queries = require("./queries");
@@ -80,7 +82,7 @@ var convertMessagesToThreadMessages = function convertMessagesToThreadMessages(m
 };
 
 var ConversationLoader = function ConversationLoader(_ref) {
-  var _conversation$relatio4, _conversation$relatio5, _conversation$relatio6;
+  var _conversation$relatio, _conversation$relatio2, _conversation$relatio3;
 
   var children = _ref.children,
       conversationId = _ref.conversationId;
@@ -95,16 +97,25 @@ var ConversationLoader = function ConversationLoader(_ref) {
   var queryResult = (0, _cozyClient.useQuery)(conversationQuery.definition, conversationQuery.options);
   var conversation = queryResult.data;
   var isLoading = (0, _cozyClient.isQueryLoading)(queryResult);
+  var boundId = conversation === null || conversation === void 0 ? void 0 : (_conversation$relatio = conversation.relationships) === null || _conversation$relatio === void 0 ? void 0 : (_conversation$relatio2 = _conversation$relatio.assistant) === null || _conversation$relatio2 === void 0 ? void 0 : (_conversation$relatio3 = _conversation$relatio2.data) === null || _conversation$relatio3 === void 0 ? void 0 : _conversation$relatio3._id;
+  var defaultId = (0, _useDefaultAssistantId.useDefaultAssistantId)();
   var initialMessages = (0, _react2.useMemo)(function () {
     return convertMessagesToThreadMessages(conversation === null || conversation === void 0 ? void 0 : conversation.messages, t);
-  }, [conversation === null || conversation === void 0 ? void 0 : conversation.messages, t]);
+  }, [conversation === null || conversation === void 0 ? void 0 : conversation.messages, t]); // Undefined until the conversation, and the default a new one starts
+  // on, are known: the runtime waits, or a message could go to the
+  // sentinel assistant before the configured default is selected.
+
+  var resolvedAssistantId = (0, _conversationAssistant.resolveConversationAssistantId)({
+    boundId: boundId,
+    conversation: conversation,
+    isLoading: isLoading,
+    defaultId: defaultId
+  });
   (0, _react2.useEffect)(function () {
-    var _conversation$relatio, _conversation$relatio2, _conversation$relatio3;
-
-    setSelectedAssistantId((conversation === null || conversation === void 0 ? void 0 : (_conversation$relatio = conversation.relationships) === null || _conversation$relatio === void 0 ? void 0 : (_conversation$relatio2 = _conversation$relatio.assistant) === null || _conversation$relatio2 === void 0 ? void 0 : (_conversation$relatio3 = _conversation$relatio2.data) === null || _conversation$relatio3 === void 0 ? void 0 : _conversation$relatio3._id) || _constants.DEFAULT_ASSISTANT._id);
-  }, [conversation === null || conversation === void 0 ? void 0 : (_conversation$relatio4 = conversation.relationships) === null || _conversation$relatio4 === void 0 ? void 0 : (_conversation$relatio5 = _conversation$relatio4.assistant) === null || _conversation$relatio5 === void 0 ? void 0 : (_conversation$relatio6 = _conversation$relatio5.data) === null || _conversation$relatio6 === void 0 ? void 0 : _conversation$relatio6._id, setSelectedAssistantId]);
+    if (resolvedAssistantId) setSelectedAssistantId(resolvedAssistantId);
+  }, [resolvedAssistantId, setSelectedAssistantId]);
 
-  if (isLoading) {
+  if (resolvedAssistantId === undefined) {
     return /*#__PURE__*/_react2.default.createElement("div", {
       className: "u-flex u-flex-items-center u-flex-justify-center u-h-100 u-w-100"
     }, /*#__PURE__*/_react2.default.createElement(_Spinner.default, {

package/dist/components/KnowledgeBase/KnowledgeBaseChip.d.ts

@@ -7,9 +7,10 @@ export default KnowledgeBaseChip;
  * open the folder in Drive (new tab) or change the knowledge base through
  * the Drive folder picker (persisted on the assistant immediately).
  */
-declare function KnowledgeBaseChip({ dirId, folder, isUnavailable, isLast, onChangeFolder }: {
+declare function KnowledgeBaseChip({ dirId, folder, isRoot, isUnavailable, isLast, onChangeFolder }: {
     dirId: any;
     folder: any;
+    isRoot: any;
     isUnavailable: any;
     isLast: any;
     onChangeFolder: any;

package/dist/components/KnowledgeBase/KnowledgeBaseChip.js

@@ -48,6 +48,7 @@ var KnowledgeBaseChip = function KnowledgeBaseChip(_ref) {
 
   var dirId = _ref.dirId,
       folder = _ref.folder,
+      isRoot = _ref.isRoot,
       isUnavailable = _ref.isUnavailable,
       isLast = _ref.isLast,
       onChangeFolder = _ref.onChangeFolder;
@@ -83,7 +84,7 @@ var KnowledgeBaseChip = function KnowledgeBaseChip(_ref) {
     subDomainType: client === null || client === void 0 ? void 0 : client.getInstanceOptions().subdomain,
     hash: "/folder/".concat(dirId)
   });
-  var label = isUnavailable ? t('assistant.knowledge_base.unavailable') : (_folder$name = folder === null || folder === void 0 ? void 0 : folder.name) !== null && _folder$name !== void 0 ? _folder$name : '…';
+  var label = isUnavailable ? t('assistant.knowledge_base.unavailable') : isRoot ? t('assistant.twake_knowledges.drive') : (_folder$name = folder === null || folder === void 0 ? void 0 : folder.name) !== null && _folder$name !== void 0 ? _folder$name : '…';
   return /*#__PURE__*/_react.default.createElement(_react.default.Fragment, null, /*#__PURE__*/_react.default.createElement("div", {
     ref: chipRef,
     className: (0, _classnames.default)({

package/dist/components/KnowledgeBase/KnowledgeBaseSection.js

@@ -102,7 +102,7 @@ var KnowledgeBaseSection = function KnowledgeBaseSection(_ref) {
       src: _tdrive.default,
       width: 16
     }),
-    label: isUnavailable ? t('assistant.knowledge_base.unavailable') : (_folder$name = folder === null || folder === void 0 ? void 0 : folder.name) !== null && _folder$name !== void 0 ? _folder$name : '…',
+    label: isUnavailable ? t('assistant.knowledge_base.unavailable') : (0, _knowledgeBase.isRootDirId)(dirId) ? t('assistant.twake_knowledges.drive') : (_folder$name = folder === null || folder === void 0 ? void 0 : folder.name) !== null && _folder$name !== void 0 ? _folder$name : '…',
     deleteIcon: /*#__PURE__*/_react.default.createElement(_twakeIcons.Icon, {
       icon: _twakeIcons.Cross,
       size: 10 // inline width/height beats MUI v4's fixed deleteIcon class size

package/dist/components/KnowledgeBase/conversationAssistant.d.ts

@@ -0,0 +1,6 @@
+export function resolveConversationAssistantId({ boundId, conversation, isLoading, defaultId }: {
+    boundId: string | undefined;
+    conversation: object | undefined;
+    isLoading: boolean;
+    defaultId: string | null | undefined;
+}): string | undefined;

package/dist/components/KnowledgeBase/conversationAssistant.js

@@ -0,0 +1,42 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.resolveConversationAssistantId = void 0;
+
+var _constants = require("../constants");
+
+/**
+ * Which assistant a loaded conversation selects. Undefined means "leave the
+ * selection alone" (the conversation, or the default a new one starts on,
+ * is still loading).
+ * @param {object} params
+ * @param {string|undefined} params.boundId - The conversation's bound
+ * assistant id (`conversation?.relationships?.assistant?.data?._id`).
+ * @param {object|undefined} params.conversation - The conversation document,
+ * or undefined when it does not exist (yet).
+ * @param {boolean} params.isLoading - Whether the conversation query is
+ * still loading.
+ * @param {string|null|undefined} params.defaultId - The configured default
+ * assistant id, only when its document is known to exist; undefined while
+ * that is being checked.
+ * @returns {string|undefined}
+ */
+var resolveConversationAssistantId = function resolveConversationAssistantId(_ref) {
+  var boundId = _ref.boundId,
+      conversation = _ref.conversation,
+      isLoading = _ref.isLoading,
+      defaultId = _ref.defaultId;
+  if (boundId) return boundId;
+  if (isLoading) return undefined;
+
+  if (!conversation) {
+    if (defaultId === undefined) return undefined;
+    return defaultId || _constants.DEFAULT_ASSISTANT._id;
+  }
+
+  return _constants.DEFAULT_ASSISTANT._id;
+};
+
+exports.resolveConversationAssistantId = resolveConversationAssistantId;

package/dist/components/KnowledgeBase/conversationAssistant.spec.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/components/KnowledgeBase/conversationAssistant.spec.js

@@ -0,0 +1,80 @@
+import { resolveConversationAssistantId } from './conversationAssistant'
+
+describe('resolveConversationAssistantId', () => {
+  it('selects the bound assistant, regardless of anything else', () => {
+    expect(
+      resolveConversationAssistantId({
+        boundId: 'bound',
+        conversation: { _id: 'c1' },
+        isLoading: false,
+        defaultId: 'docs'
+      })
+    ).toBe('bound')
+  })
+
+  it('leaves the selection alone while the conversation query is loading', () => {
+    expect(
+      resolveConversationAssistantId({
+        boundId: undefined,
+        conversation: undefined,
+        isLoading: true,
+        defaultId: 'docs'
+      })
+    ).toBeUndefined()
+  })
+
+  it('leaves the selection alone while the default of a new conversation is being checked', () => {
+    expect(
+      resolveConversationAssistantId({
+        boundId: undefined,
+        conversation: undefined,
+        isLoading: false,
+        defaultId: undefined
+      })
+    ).toBeUndefined()
+  })
+
+  it('does not wait for the default on an existing conversation', () => {
+    expect(
+      resolveConversationAssistantId({
+        boundId: undefined,
+        conversation: { _id: 'c1' },
+        isLoading: false,
+        defaultId: undefined
+      })
+    ).toBe('default_ai_assistant')
+  })
+
+  it('selects the default for a new conversation when one is configured', () => {
+    expect(
+      resolveConversationAssistantId({
+        boundId: undefined,
+        conversation: undefined,
+        isLoading: false,
+        defaultId: 'docs'
+      })
+    ).toBe('docs')
+  })
+
+  it('selects the sentinel for a new conversation without a configured default', () => {
+    expect(
+      resolveConversationAssistantId({
+        boundId: undefined,
+        conversation: undefined,
+        isLoading: false,
+        defaultId: null
+      })
+    ).toBe('default_ai_assistant')
+  })
+
+  it('selects the sentinel for an existing conversation without an assistant, even with a default configured', () => {
+    expect(
+      resolveConversationAssistantId({
+        boundId: undefined,
+        conversation: { _id: 'c1' },
+        isLoading: false,
+        defaultId: 'docs'
+      })
+    ).toBe('default_ai_assistant')
+  })
+})

package/dist/components/KnowledgeBase/knowledgeBase.d.ts

@@ -1,3 +1,11 @@
+/** The root folder of the instance: the assistant covers the whole Drive. */
+export const ROOT_DIR_ID: "io.cozy.files.root-dir";
+export function isRootDirId(dirId: any): boolean;
+export function makeRootKnowledgeBaseEntry(): {
+    doctype: string;
+    dirId: string;
+};
+export function withRootFolderIfMissing(knowledgeBase?: any[]): any[];
 export function makeKnowledgeBaseEntry(pickedFolder: any): {
     doctype: string;
     dirId: any;

package/dist/components/KnowledgeBase/knowledgeBase.js

@@ -5,7 +5,7 @@ var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefau
 Object.defineProperty(exports, "__esModule", {
   value: true
 });
-exports.withoutKnowledgeBaseDoctype = exports.withKnowledgeBaseEntry = exports.saveKnowledgeBase = exports.makeKnowledgeBaseEntry = exports.makeEmailKnowledgeBaseEntry = exports.hasEmailKnowledgeBase = exports.getKnowledgeBaseDirId = void 0;
+exports.withoutKnowledgeBaseDoctype = exports.withRootFolderIfMissing = exports.withKnowledgeBaseEntry = exports.saveKnowledgeBase = exports.makeRootKnowledgeBaseEntry = exports.makeKnowledgeBaseEntry = exports.makeEmailKnowledgeBaseEntry = exports.isRootDirId = exports.hasEmailKnowledgeBase = exports.getKnowledgeBaseDirId = exports.ROOT_DIR_ID = void 0;
 
 var _regenerator = _interopRequireDefault(require("@babel/runtime/regenerator"));
 
@@ -15,15 +15,54 @@ var _asyncToGenerator2 = _interopRequireDefault(require("@babel/runtime/helpers/
 
 var _toConsumableArray2 = _interopRequireDefault(require("@babel/runtime/helpers/toConsumableArray"));
 
-var _cozyClient = require("cozy-client");
-
 var _queries = require("../queries");
 
 function ownKeys(object, enumerableOnly) { var keys = Object.keys(object); if (Object.getOwnPropertySymbols) { var symbols = Object.getOwnPropertySymbols(object); enumerableOnly && (symbols = symbols.filter(function (sym) { return Object.getOwnPropertyDescriptor(object, sym).enumerable; })), keys.push.apply(keys, symbols); } return keys; }
 
 function _objectSpread(target) { for (var i = 1; i < arguments.length; i++) { var source = null != arguments[i] ? arguments[i] : {}; i % 2 ? ownKeys(Object(source), !0).forEach(function (key) { (0, _defineProperty2.default)(target, key, source[key]); }) : Object.getOwnPropertyDescriptors ? Object.defineProperties(target, Object.getOwnPropertyDescriptors(source)) : ownKeys(Object(source)).forEach(function (key) { Object.defineProperty(target, key, Object.getOwnPropertyDescriptor(source, key)); }); } return target; }
 
-var ASSISTANTS_DOCTYPE = 'io.cozy.ai.chat.assistants';
+/** The root folder of the instance: the assistant covers the whole Drive. */
+var ROOT_DIR_ID = 'io.cozy.files.root-dir';
+exports.ROOT_DIR_ID = ROOT_DIR_ID;
+
+var isRootDirId = function isRootDirId(dirId) {
+  return dirId === ROOT_DIR_ID;
+};
+
+exports.isRootDirId = isRootDirId;
+
+var makeRootKnowledgeBaseEntry = function makeRootKnowledgeBaseEntry() {
+  return {
+    doctype: _queries.FILES_DOCTYPE,
+    dirId: ROOT_DIR_ID
+  };
+};
+
+exports.makeRootKnowledgeBaseEntry = makeRootKnowledgeBaseEntry;
+
+var isFolderEntry = function isFolderEntry(entry) {
+  return entry.doctype === _queries.FILES_DOCTYPE && !!entry.dirId;
+};
+
+var hasFolderEntry = function hasFolderEntry(knowledgeBase) {
+  return (knowledgeBase || []).some(isFolderEntry);
+};
+/**
+ * An assistant always has a knowledge base folder: without one, it covers
+ * the whole Drive. Adds the root entry when no folder entry exists,
+ * replacing a files entry without dirId (getKnowledgeBaseDirId would find
+ * that one first); returns the very same array otherwise.
+ */
+
+
+var withRootFolderIfMissing = function withRootFolderIfMissing() {
+  var knowledgeBase = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : [];
+  return hasFolderEntry(knowledgeBase) ? knowledgeBase : [].concat((0, _toConsumableArray2.default)(knowledgeBase.filter(function (entry) {
+    return entry.doctype !== _queries.FILES_DOCTYPE;
+  })), [makeRootKnowledgeBaseEntry()]);
+};
+
+exports.withRootFolderIfMissing = withRootFolderIfMissing;
 
 var makeKnowledgeBaseEntry = function makeKnowledgeBaseEntry(pickedFolder) {
   return {
@@ -90,6 +129,10 @@ var getKnowledgeBaseDirId = function getKnowledgeBaseDirId(assistant) {
  * applied to the freshly fetched doc's knowledgeBase. Computing the update
  * from the fresh doc rather than a value the caller may have cached avoids
  * dropping concurrent changes (stale-read race).
+ *
+ * The saved knowledge base always has a folder entry (the root when none
+ * was chosen): the stack's rag-index worker reads the assistants to know
+ * what to index, there is nothing else to do here.
  */
 
 
@@ -97,25 +140,28 @@ exports.getKnowledgeBaseDirId = getKnowledgeBaseDirId;
 
 var saveKnowledgeBase = /*#__PURE__*/function () {
   var _ref = (0, _asyncToGenerator2.default)( /*#__PURE__*/_regenerator.default.mark(function _callee(client, assistantId, knowledgeBaseOrUpdater) {
-    var _yield$client$query, assistant, knowledgeBase;
+    var _buildAssistantByIdQu, definition, options, _yield$client$query, assistant, knowledgeBase;
 
     return _regenerator.default.wrap(function _callee$(_context) {
       while (1) {
         switch (_context.prev = _context.next) {
           case 0:
-            _context.next = 2;
-            return client.query((0, _cozyClient.Q)(ASSISTANTS_DOCTYPE).getById(assistantId));
+            _buildAssistantByIdQu = (0, _queries.buildAssistantByIdQuery)(assistantId), definition = _buildAssistantByIdQu.definition, options = _buildAssistantByIdQu.options;
+            _context.next = 3;
+            return client.query(definition(), {
+              as: options.as
+            });
 
-          case 2:
+          case 3:
             _yield$client$query = _context.sent;
             assistant = _yield$client$query.data;
             knowledgeBase = typeof knowledgeBaseOrUpdater === 'function' ? knowledgeBaseOrUpdater(assistant === null || assistant === void 0 ? void 0 : assistant.knowledgeBase) : knowledgeBaseOrUpdater;
-            _context.next = 7;
+            _context.next = 8;
             return client.save(_objectSpread(_objectSpread({}, assistant), {}, {
-              knowledgeBase: knowledgeBase
+              knowledgeBase: withRootFolderIfMissing(knowledgeBase)
             }));
 
-          case 7:
+          case 8:
           case "end":
             return _context.stop();
         }

package/dist/components/KnowledgeBase/knowledgeBase.spec.js

@@ -5,6 +5,9 @@ import {
   withKnowledgeBaseEntry,
   withoutKnowledgeBaseDoctype,
   getKnowledgeBaseDirId,
+  ROOT_DIR_ID,
+  isRootDirId,
+  withRootFolderIfMissing,
   saveKnowledgeBase
 } from './knowledgeBase'
 
@@ -125,6 +128,43 @@ describe('getKnowledgeBaseDirId', () => {
   })
 })
 
+describe('withRootFolderIfMissing', () => {
+  it('appends the root entry when there is no files entry', () => {
+    expect(withRootFolderIfMissing([{ doctype: 'io.cozy.email' }])).toEqual([
+      { doctype: 'io.cozy.email' },
+      { doctype: 'io.cozy.files', dirId: ROOT_DIR_ID }
+    ])
+    expect(withRootFolderIfMissing()).toEqual([
+      { doctype: 'io.cozy.files', dirId: ROOT_DIR_ID }
+    ])
+  })
+
+  it('returns the same array when a files entry exists', () => {
+    const kb = [{ doctype: 'io.cozy.files', dirId: 'folder-1' }]
+    expect(withRootFolderIfMissing(kb)).toBe(kb)
+  })
+
+  it('replaces a files entry without dirId, keeping the other entries', () => {
+    expect(
+      withRootFolderIfMissing([
+        { doctype: 'io.cozy.email' },
+        { doctype: 'io.cozy.files' }
+      ])
+    ).toEqual([
+      { doctype: 'io.cozy.email' },
+      { doctype: 'io.cozy.files', dirId: ROOT_DIR_ID }
+    ])
+  })
+})
+
+describe('isRootDirId', () => {
+  it('recognizes the root folder id only', () => {
+    expect(isRootDirId(ROOT_DIR_ID)).toBe(true)
+    expect(isRootDirId('folder-1')).toBe(false)
+    expect(isRootDirId(null)).toBe(false)
+  })
+})
+
 describe('saveKnowledgeBase', () => {
   it('refetches the assistant and saves it with the new knowledgeBase', async () => {
     const assistantDoc = {
@@ -174,4 +214,56 @@ describe('saveKnowledgeBase', () => {
       ]
     })
   })
+
+  it('saves the root entry when the knowledge base has no folder', async () => {
+    const assistantDoc = {
+      _id: 'assistant-1',
+      _type: 'io.cozy.ai.chat.assistants'
+    }
+    const client = {
+      query: jest.fn().mockResolvedValue({ data: assistantDoc }),
+      save: jest.fn().mockResolvedValue({ data: assistantDoc })
+    }
+
+    await saveKnowledgeBase(client, 'assistant-1', [
+      { doctype: 'io.cozy.email' }
+    ])
+
+    expect(client.save).toHaveBeenCalledWith({
+      ...assistantDoc,
+      knowledgeBase: [
+        { doctype: 'io.cozy.email' },
+        { doctype: 'io.cozy.files', dirId: ROOT_DIR_ID }
+      ]
+    })
+  })
+
+  it('applies the root invariant to the updater result too', async () => {
+    const assistantDoc = {
+      _id: 'assistant-1',
+      _type: 'io.cozy.ai.chat.assistants',
+      knowledgeBase: [{ doctype: 'io.cozy.files', dirId: 'folder-1' }]
+    }
+    const client = {
+      query: jest.fn().mockResolvedValue({ data: assistantDoc }),
+      save: jest.fn().mockResolvedValue({ data: assistantDoc })
+    }
+
+    await saveKnowledgeBase(client, 'assistant-1', () => [])
+
+    expect(client.save).toHaveBeenCalledWith({
+      ...assistantDoc,
+      knowledgeBase: [{ doctype: 'io.cozy.files', dirId: ROOT_DIR_ID }]
+    })
+  })
+
+  it('propagates a save failure', async () => {
+    const client = {
+      query: jest.fn().mockResolvedValue({ data: { _id: 'assistant-1' } }),
+      save: jest.fn().mockRejectedValue(new Error('boom'))
+    }
+    await expect(saveKnowledgeBase(client, 'assistant-1', [])).rejects.toThrow(
+      'boom'
+    )
+  })
 })

package/dist/components/KnowledgeBase/provisioning.d.ts

@@ -0,0 +1,24 @@
+export const AUTOPROVISION_FLAG: "rag.assistants.autoprovision";
+export function assistantIdFromName(name: any): string;
+export function getDefaultProvisionedAssistantId(): string | null;
+export function isMagicFolderId(dirId: any): boolean;
+export function resolveProvisionedFolder(client: any, { dirId, dirName }: {
+    dirId: any;
+    dirName: any;
+}): Promise<any>;
+export function createOpenragAccount(client: any): Promise<any>;
+export function ensureProvisionedAssistants(client: import("cozy-client").CozyClient, configs: Array<{
+    name: string;
+    dirName?: string;
+    dirId?: string;
+    prompt?: string;
+    icon?: string | null;
+    default?: boolean;
+}>): Promise<{
+    created: string[];
+    ensured: string[];
+    skipped: {
+        id: string;
+        reason: string;
+    }[];
+}>;