cozy-search 2.2.29 → 2.3.0

package/dist/components/KnowledgeBase/useDefaultAssistantId.js

@@ -0,0 +1,38 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.useDefaultAssistantId = void 0;
+
+var _cozyClient = require("cozy-client");
+
+var _provisioning = require("./provisioning");
+
+var _queries = require("../queries");
+
+/**
+ * The assistant new conversations start on: the flagged default of the
+ * autoprovision flag, once its document is known to exist. Undefined while
+ * that is being checked; null when nothing is configured or when the
+ * document is missing (an id the stack cannot resolve would fail every
+ * message).
+ * @returns {string|null|undefined}
+ */
+var useDefaultAssistantId = function useDefaultAssistantId() {
+  var defaultId = (0, _provisioning.getDefaultProvisionedAssistantId)();
+
+  var _buildAssistantByIdQu = (0, _queries.buildAssistantByIdQuery)(defaultId),
+      definition = _buildAssistantByIdQu.definition,
+      options = _buildAssistantByIdQu.options;
+
+  var _ref = (0, _cozyClient.useQuery)(definition, options) || {},
+      assistant = _ref.data,
+      fetchStatus = _ref.fetchStatus;
+
+  if (!defaultId || fetchStatus === 'failed') return null;
+  if (fetchStatus !== 'loaded') return undefined;
+  return (assistant === null || assistant === void 0 ? void 0 : assistant._id) === defaultId ? defaultId : null;
+};
+
+exports.useDefaultAssistantId = useDefaultAssistantId;

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

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

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

@@ -0,0 +1,69 @@
+import { renderHook } from '@testing-library/react-hooks/dom'
+
+import { useQuery } from 'cozy-client'
+
+import { getDefaultProvisionedAssistantId } from './provisioning'
+import { useDefaultAssistantId } from './useDefaultAssistantId'
+
+jest.mock('cozy-client', () => ({
+  useQuery: jest.fn(),
+  // queries.js computes its default fetch policy at import time.
+  fetchPolicies: { olderThan: jest.fn(() => 'mock-fetch-policy') },
+  Q: jest.fn(() => ({
+    getById: jest.fn(() => ({ include: jest.fn(() => 'mock-definition') }))
+  }))
+}))
+jest.mock('./provisioning', () => ({
+  getDefaultProvisionedAssistantId: jest.fn()
+}))
+
+describe('useDefaultAssistantId', () => {
+  beforeEach(() => {
+    useQuery.mockReset().mockReturnValue({ data: null, fetchStatus: 'pending' })
+    getDefaultProvisionedAssistantId.mockReset()
+  })
+
+  it('returns null and disables the query without a configured default', () => {
+    getDefaultProvisionedAssistantId.mockReturnValue(null)
+    const { result } = renderHook(() => useDefaultAssistantId())
+    expect(result.current).toBeNull()
+    expect(useQuery).toHaveBeenCalledWith(
+      expect.anything(),
+      expect.objectContaining({ enabled: false })
+    )
+  })
+
+  it('queries the default assistant by id', () => {
+    getDefaultProvisionedAssistantId.mockReturnValue('docs')
+    renderHook(() => useDefaultAssistantId())
+    expect(useQuery).toHaveBeenCalledWith(
+      expect.anything(),
+      expect.objectContaining({
+        as: 'io.cozy.ai.chat.assistants/docs',
+        enabled: true
+      })
+    )
+  })
+
+  it('returns undefined while the document is being checked', () => {
+    getDefaultProvisionedAssistantId.mockReturnValue('docs')
+    useQuery.mockReturnValue({ data: null, fetchStatus: 'loading' })
+    const { result } = renderHook(() => useDefaultAssistantId())
+    expect(result.current).toBeUndefined()
+  })
+
+  it('returns the id once its document is loaded', () => {
+    getDefaultProvisionedAssistantId.mockReturnValue('docs')
+    useQuery.mockReturnValue({ data: { _id: 'docs' }, fetchStatus: 'loaded' })
+    const { result } = renderHook(() => useDefaultAssistantId())
+    expect(result.current).toBe('docs')
+  })
+
+  it('returns null when the document is missing', () => {
+    getDefaultProvisionedAssistantId.mockReturnValue('docs')
+    useQuery.mockReturnValue({ data: null, fetchStatus: 'failed' })
+    expect(renderHook(() => useDefaultAssistantId()).result.current).toBeNull()
+    useQuery.mockReturnValue({ data: null, fetchStatus: 'loaded' })
+    expect(renderHook(() => useDefaultAssistantId()).result.current).toBeNull()
+  })
+})

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

@@ -1,6 +1,7 @@
 export function useSelectedAssistantKnowledgeBase(): {
     dirId: any;
     folder: object | any[];
+    isRoot: boolean;
     isUnavailable: boolean;
     setKnowledgeBaseFolder: (pickedFolder: any) => Promise<void>;
     isRealAssistant: boolean;

package/dist/components/KnowledgeBase/useSelectedAssistantKnowledgeBase.js

@@ -59,7 +59,9 @@ var useSelectedAssistantKnowledgeBase = function useSelectedAssistantKnowledgeBa
       folder = _useQuery2.data,
       fetchStatus = _useQuery2.fetchStatus;
 
-  var hasEmail = (0, _knowledgeBase.hasEmailKnowledgeBase)(assistant);
+  var hasEmail = (0, _knowledgeBase.hasEmailKnowledgeBase)(assistant); // The folder is gone (or unreadable): its knowledge base cannot be used.
+
+  var isUnavailable = !!dirId && (fetchStatus === 'failed' || !!(folder !== null && folder !== void 0 && folder.trashed) || !!(folder !== null && folder !== void 0 && (_folder$path = folder.path) !== null && _folder$path !== void 0 && _folder$path.startsWith('/.cozy_trash')));
   var setKnowledgeBaseFolder = (0, _react.useCallback)( /*#__PURE__*/function () {
     var _ref = (0, _asyncToGenerator2.default)( /*#__PURE__*/_regenerator.default.mark(function _callee(pickedFolder) {
       return _regenerator.default.wrap(function _callee$(_context) {
@@ -104,10 +106,12 @@ var useSelectedAssistantKnowledgeBase = function useSelectedAssistantKnowledgeBa
       return _ref.apply(this, arguments);
     };
   }(), [client, realAssistantId, showAlert, t]);
+  var isRoot = (0, _knowledgeBase.isRootDirId)(dirId);
   return {
     dirId: dirId,
     folder: folder !== null && folder !== void 0 ? folder : null,
-    isUnavailable: !!dirId && (fetchStatus === 'failed' || !!(folder !== null && folder !== void 0 && folder.trashed) || !!(folder !== null && folder !== void 0 && (_folder$path = folder.path) !== null && _folder$path !== void 0 && _folder$path.startsWith('/.cozy_trash'))),
+    isRoot: isRoot,
+    isUnavailable: isUnavailable,
     setKnowledgeBaseFolder: setKnowledgeBaseFolder,
     isRealAssistant: !!realAssistantId,
     hasEmail: hasEmail

package/dist/components/TwakeKnowledges/TwakeKnowledgeSelector.js

@@ -40,6 +40,7 @@ var TwakeKnowledgeSelector = function TwakeKnowledgeSelector(_ref) {
   var _useSelectedAssistant = (0, _useSelectedAssistantKnowledgeBase.useSelectedAssistantKnowledgeBase)(),
       dirId = _useSelectedAssistant.dirId,
       folder = _useSelectedAssistant.folder,
+      isRoot = _useSelectedAssistant.isRoot,
       isUnavailable = _useSelectedAssistant.isUnavailable,
       setKnowledgeBaseFolder = _useSelectedAssistant.setKnowledgeBaseFolder,
       isRealAssistant = _useSelectedAssistant.isRealAssistant,
@@ -75,6 +76,7 @@ var TwakeKnowledgeSelector = function TwakeKnowledgeSelector(_ref) {
   }), hasKnowledgeBase ? /*#__PURE__*/_react.default.createElement(_KnowledgeBaseChip.default, {
     dirId: dirId,
     folder: folder,
+    isRoot: isRoot,
     isUnavailable: isUnavailable,
     isLast: !showSourceChips,
     onChangeFolder: setKnowledgeBaseFolder

package/dist/components/Views/CreateAssistantDialog.js

@@ -117,8 +117,7 @@ var CreateAssistantDialog = function CreateAssistantDialog(_ref) {
 
   var onSubmit = /*#__PURE__*/function () {
     var _ref2 = (0, _asyncToGenerator2.default)( /*#__PURE__*/_regenerator.default.mark(function _callee() {
-      var savedAssistant, _formData$knowledgeBa;
-
+      var savedAssistant;
       return _regenerator.default.wrap(function _callee$(_context) {
         while (1) {
           switch (_context.prev = _context.next) {
@@ -139,28 +138,23 @@ var CreateAssistantDialog = function CreateAssistantDialog(_ref) {
               savedAssistant = _context.sent;
 
               if (!(savedAssistant !== null && savedAssistant !== void 0 && savedAssistant._id)) {
-                _context.next = 8;
-                break;
-              }
-
-              if (!(((_formData$knowledgeBa = formData.knowledgeBase) === null || _formData$knowledgeBa === void 0 ? void 0 : _formData$knowledgeBa.length) > 0)) {
                 _context.next = 7;
                 break;
               }
 
-              _context.next = 7;
-              return (0, _knowledgeBase.saveKnowledgeBase)(client, savedAssistant._id, formData.knowledgeBase);
+              _context.next = 6;
+              return (0, _knowledgeBase.saveKnowledgeBase)(client, savedAssistant._id, formData.knowledgeBase || []);
 
-            case 7:
+            case 6:
               setSelectedAssistantId(savedAssistant._id);
 
-            case 8:
+            case 7:
               showAlert({
                 message: t('assistant_create.success'),
                 severity: 'success'
               });
 
-            case 9:
+            case 8:
             case "end":
               return _context.stop();
           }

package/dist/components/Views/DeleteAssistantDialog.js

@@ -33,6 +33,8 @@ var _locales = require("../../locales");
 
 var _AssistantProvider = require("../AssistantProvider");
 
+var _constants = require("../constants");
+
 var _queries = require("../queries");
 
 function _getRequireWildcardCache(nodeInterop) { if (typeof WeakMap !== "function") return null; var cacheBabelInterop = new WeakMap(); var cacheNodeInterop = new WeakMap(); return (_getRequireWildcardCache = function _getRequireWildcardCache(nodeInterop) { return nodeInterop ? cacheNodeInterop : cacheBabelInterop; })(nodeInterop); }
@@ -72,7 +74,9 @@ var DeleteAssistantDialog = function DeleteAssistantDialog(_ref) {
 
   var _useAssistant = (0, _AssistantProvider.useAssistant)(),
       assistantIdInAction = _useAssistant.assistantIdInAction,
-      setAssistantIdInAction = _useAssistant.setAssistantIdInAction;
+      setAssistantIdInAction = _useAssistant.setAssistantIdInAction,
+      selectedAssistantId = _useAssistant.selectedAssistantId,
+      setSelectedAssistantId = _useAssistant.setSelectedAssistantId;
 
   var _useAlert = (0, _Alert.useAlert)(),
       showAlert = _useAlert.showAlert;
@@ -120,41 +124,45 @@ var DeleteAssistantDialog = function DeleteAssistantDialog(_ref) {
               return client.destroy(assistantDoc);
 
             case 11:
+              if (selectedAssistantId === assistantDoc._id) {
+                setSelectedAssistantId(_constants.DEFAULT_ASSISTANT._id);
+              }
+
               provider = included === null || included === void 0 ? void 0 : included[0];
 
               if (!provider) {
-                _context.next = 15;
+                _context.next = 16;
                 break;
               }
 
-              _context.next = 15;
+              _context.next = 16;
               return client.destroy(provider);
 
-            case 15:
+            case 16:
               setAssistantIdInAction(null);
               onClose();
-              _context.next = 22;
+              _context.next = 23;
               break;
 
-            case 19:
-              _context.prev = 19;
+            case 20:
+              _context.prev = 20;
               _context.t0 = _context["catch"](2);
               showAlert({
                 message: t('assistant.default_error'),
                 severity: 'error'
               });
 
-            case 22:
-              _context.prev = 22;
+            case 23:
+              _context.prev = 23;
               setIsDeleting(false);
-              return _context.finish(22);
+              return _context.finish(23);
 
-            case 25:
+            case 26:
             case "end":
               return _context.stop();
           }
         }
-      }, _callee, null, [[2, 19, 22, 25]]);
+      }, _callee, null, [[2, 20, 23, 26]]);
     }));
 
     return function handleDeleteAssistant() {

package/dist/components/Views/EditAssistantDialog.js

@@ -13,10 +13,10 @@ var _defineProperty2 = _interopRequireDefault(require("@babel/runtime/helpers/de
 
 var _asyncToGenerator2 = _interopRequireDefault(require("@babel/runtime/helpers/asyncToGenerator"));
 
-var _twakeIcons = require("@linagora/twake-icons");
-
 var _react = _interopRequireWildcard(require("react"));
 
+var _twakeIcons = require("@linagora/twake-icons");
+
 var _cozyClient = require("cozy-client");
 
 var _assistant = require("cozy-client/dist/models/assistant");
@@ -45,6 +45,8 @@ var _useAssistantDialog2 = require("../CreateAssistantSteps/useAssistantDialog")
 
 var _knowledgeBase = require("../KnowledgeBase/knowledgeBase");
 
+var _queries = require("../queries");
+
 function _getRequireWildcardCache(nodeInterop) { if (typeof WeakMap !== "function") return null; var cacheBabelInterop = new WeakMap(); var cacheNodeInterop = new WeakMap(); return (_getRequireWildcardCache = function _getRequireWildcardCache(nodeInterop) { return nodeInterop ? cacheNodeInterop : cacheBabelInterop; })(nodeInterop); }
 
 function _interopRequireWildcard(obj, nodeInterop) { if (!nodeInterop && obj && obj.__esModule) { return obj; } if (obj === null || typeof obj !== "object" && typeof obj !== "function") { return { default: obj }; } var cache = _getRequireWildcardCache(nodeInterop); if (cache && cache.has(obj)) { return cache.get(obj); } var newObj = {}; var hasPropertyDescriptor = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var key in obj) { if (key !== "default" && Object.prototype.hasOwnProperty.call(obj, key)) { var desc = hasPropertyDescriptor ? Object.getOwnPropertyDescriptor(obj, key) : null; if (desc && (desc.get || desc.set)) { Object.defineProperty(newObj, key, desc); } else { newObj[key] = obj[key]; } } } newObj.default = obj; if (cache) { cache.set(obj, newObj); } return newObj; }
@@ -118,15 +120,19 @@ var EditAssistantDialog = function EditAssistantDialog(_ref) {
       var _ref2 = (0, _asyncToGenerator2.default)( /*#__PURE__*/_regenerator.default.mark(function _callee() {
         var _assistant$relationsh, _assistant$relationsh2, _assistant$relationsh3, _assistant$relationsh4, _provider$data, _provider$auth, _provider$data2, _provider$auth2, _provider$auth3, _provider$data3, _provider$auth4;
 
-        var response, assistant, provider, providerId, model, selectProviderDefault;
+        var _buildAssistantByIdQu, definition, options, response, assistant, provider, providerId, model, selectProviderDefault;
+
         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)('io.cozy.ai.chat.assistants').getById(assistantIdInAction).include(['provider']));
+                _buildAssistantByIdQu = (0, _queries.buildAssistantByIdQuery)(assistantIdInAction), definition = _buildAssistantByIdQu.definition, options = _buildAssistantByIdQu.options;
+                _context.next = 3;
+                return client.query(definition(), {
+                  as: options.as
+                });
 
-              case 2:
+              case 3:
                 response = _context.sent;
                 assistant = response.data;
                 provider = response.included[0];
@@ -152,7 +158,7 @@ var EditAssistantDialog = function EditAssistantDialog(_ref) {
                   name: selectProviderDefault.id === 'custom' ? (provider === null || provider === void 0 ? void 0 : (_provider$auth4 = provider.auth) === null || _provider$auth4 === void 0 ? void 0 : _provider$auth4.accountName) || model : selectProviderDefault.name
                 }));
 
-              case 10:
+              case 11:
               case "end":
                 return _context.stop();
             }

package/dist/components/queries.d.ts

@@ -1,6 +1,8 @@
 export const CHAT_CONVERSATIONS_DOCTYPE: "io.cozy.ai.chat.conversations";
 export const CHAT_EVENTS_DOCTYPE: "io.cozy.ai.chat.events";
 export const FILES_DOCTYPE: "io.cozy.files";
+export const ASSISTANTS_DOCTYPE: "io.cozy.ai.chat.assistants";
+export const ACCOUNTS_DOCTYPE: "io.cozy.accounts";
 export const EMAIL_DOCTYPE: "com.linagora.email";
 export function buildFilesByIds(ids: any, enabled: any): {
     definition: import("cozy-client").QueryDefinition;
@@ -62,3 +64,9 @@ export function buildChatConversationsQuery(): {
         fetchPolicy: Function;
     };
 };
+export function buildAllAssistantsQuery(): {
+    definition: () => import("cozy-client").QueryDefinition;
+    options: {
+        as: string;
+    };
+};

package/dist/components/queries.js

@@ -3,7 +3,7 @@
 Object.defineProperty(exports, "__esModule", {
   value: true
 });
-exports.buildMyselfQuery = exports.buildFilesByIds = exports.buildFileByIdQuery = exports.buildChatConversationsQuery = exports.buildChatConversationQueryById = exports.buildAssistantsQuery = exports.buildAssistantByIdQuery = exports.FILES_DOCTYPE = exports.EMAIL_DOCTYPE = exports.CHAT_EVENTS_DOCTYPE = exports.CHAT_CONVERSATIONS_DOCTYPE = void 0;
+exports.buildMyselfQuery = exports.buildFilesByIds = exports.buildFileByIdQuery = exports.buildChatConversationsQuery = exports.buildChatConversationQueryById = exports.buildAssistantsQuery = exports.buildAssistantByIdQuery = exports.buildAllAssistantsQuery = exports.FILES_DOCTYPE = exports.EMAIL_DOCTYPE = exports.CHAT_EVENTS_DOCTYPE = exports.CHAT_CONVERSATIONS_DOCTYPE = exports.ASSISTANTS_DOCTYPE = exports.ACCOUNTS_DOCTYPE = void 0;
 
 var _cozyClient = require("cozy-client");
 
@@ -16,6 +16,10 @@ var CHAT_EVENTS_DOCTYPE = 'io.cozy.ai.chat.events';
 exports.CHAT_EVENTS_DOCTYPE = CHAT_EVENTS_DOCTYPE;
 var FILES_DOCTYPE = 'io.cozy.files';
 exports.FILES_DOCTYPE = FILES_DOCTYPE;
+var ASSISTANTS_DOCTYPE = 'io.cozy.ai.chat.assistants';
+exports.ASSISTANTS_DOCTYPE = ASSISTANTS_DOCTYPE;
+var ACCOUNTS_DOCTYPE = 'io.cozy.accounts';
+exports.ACCOUNTS_DOCTYPE = ACCOUNTS_DOCTYPE;
 var EMAIL_DOCTYPE = 'com.linagora.email';
 exports.EMAIL_DOCTYPE = EMAIL_DOCTYPE;
 
@@ -130,5 +134,20 @@ var buildChatConversationsQuery = function buildChatConversationsQuery() {
     }
   };
 };
+/** Every assistant of the instance, paged by 1000. */
 
-exports.buildChatConversationsQuery = buildChatConversationsQuery;
+
+exports.buildChatConversationsQuery = buildChatConversationsQuery;
+
+var buildAllAssistantsQuery = function buildAllAssistantsQuery() {
+  return {
+    definition: function definition() {
+      return (0, _cozyClient.Q)(ASSISTANTS_DOCTYPE).limitBy(1000);
+    },
+    options: {
+      as: "".concat(ASSISTANTS_DOCTYPE, "/all")
+    }
+  };
+};
+
+exports.buildAllAssistantsQuery = buildAllAssistantsQuery;

package/dist/index.d.ts

@@ -9,3 +9,5 @@ export { default as CozyAssistantRuntimeProvider } from './components/CozyAssist
 export { default as CozyComposer } from './components/Conversations/ConversationComposer';
 export { StreamBridge, createCozyRealtimeChatAdapter } from './components/adapters';
 export { default as AssistantView } from './components/Views/AssistantView';
+export { assistantIdFromName, AUTOPROVISION_FLAG, ensureProvisionedAssistants, getDefaultProvisionedAssistantId } from './components/KnowledgeBase/provisioning';
+export { setupRagIndexing } from './components/KnowledgeBase/ragIndexing';

package/dist/index.js

@@ -5,6 +5,12 @@ var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefau
 Object.defineProperty(exports, "__esModule", {
   value: true
 });
+Object.defineProperty(exports, "AUTOPROVISION_FLAG", {
+  enumerable: true,
+  get: function get() {
+    return _provisioning.AUTOPROVISION_FLAG;
+  }
+});
 Object.defineProperty(exports, "AiText", {
   enumerable: true,
   get: function get() {
@@ -71,12 +77,36 @@ Object.defineProperty(exports, "StreamBridge", {
     return _adapters.StreamBridge;
   }
 });
+Object.defineProperty(exports, "assistantIdFromName", {
+  enumerable: true,
+  get: function get() {
+    return _provisioning.assistantIdFromName;
+  }
+});
 Object.defineProperty(exports, "createCozyRealtimeChatAdapter", {
   enumerable: true,
   get: function get() {
     return _adapters.createCozyRealtimeChatAdapter;
   }
 });
+Object.defineProperty(exports, "ensureProvisionedAssistants", {
+  enumerable: true,
+  get: function get() {
+    return _provisioning.ensureProvisionedAssistants;
+  }
+});
+Object.defineProperty(exports, "getDefaultProvisionedAssistantId", {
+  enumerable: true,
+  get: function get() {
+    return _provisioning.getDefaultProvisionedAssistantId;
+  }
+});
+Object.defineProperty(exports, "setupRagIndexing", {
+  enumerable: true,
+  get: function get() {
+    return _ragIndexing.setupRagIndexing;
+  }
+});
 
 var _AssistantLink = _interopRequireDefault(require("./components/Search/AssistantLink"));
 
@@ -98,4 +128,8 @@ var _ConversationComposer = _interopRequireDefault(require("./components/Convers
 
 var _adapters = require("./components/adapters");
 
-var _AssistantView = _interopRequireDefault(require("./components/Views/AssistantView"));
+var _AssistantView = _interopRequireDefault(require("./components/Views/AssistantView"));
+
+var _provisioning = require("./components/KnowledgeBase/provisioning");
+
+var _ragIndexing = require("./components/KnowledgeBase/ragIndexing");

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "cozy-search",
   "description": "UI components about search bar and IA assistant",
-  "version": "2.2.29",
+  "version": "2.3.0",
   "author": {
     "name": "Cozy Cloud",
     "url": "https://github.com/cozy"
@@ -37,7 +37,7 @@
     "cozy-pouch-link": "^54.0.0",
     "cozy-realtime": "^5.9.4",
     "cozy-ui": "^138.4.0",
-    "cozy-ui-plus": "^13.0.1",
+    "cozy-ui-plus": "^13.0.2",
     "cross-fetch": "^4.0.0",
     "jest": "30.3.0",
     "react": "18.2.0",
@@ -96,5 +96,5 @@
     "lint": "cd .. && yarn eslint --ext js,jsx,ts packages/cozy-search"
   },
   "types": "dist/index.d.ts",
-  "gitHead": "1e26037833a2fc579309c5f87ef613b03f6f2515"
+  "gitHead": "2635589989da94e3286a64a32fca81d9f7d323e9"
 }