cozy-search 0.23.1 → 0.23.3

package/dist/components/AssistantProvider.d.ts

@@ -1,11 +1,5 @@
 import React from 'react'
 
-export interface AssistantState {
-  message: Record<string, string>
-  status: 'idle' | 'writing' | 'pending'
-  messagesId: string[]
-}
-
 export interface TwakeKnowledgeState {
   drive: { id: string; name: string }[]
   mail: { id: string; name: string }[]
@@ -13,7 +7,6 @@ export interface TwakeKnowledgeState {
 }
 
 export interface AssistantContextValue {
-  assistantState: AssistantState
   isOpenCreateAssistant: boolean
   isOpenDeleteAssistant: boolean
   isOpenEditAssistant: boolean
@@ -24,12 +17,6 @@ export interface AssistantContextValue {
   selectedTwakeKnowledge: TwakeKnowledgeState
   setAssistantIdInAction: (id: string | null) => void
   setIsOpenDeleteAssistant: (isOpen: boolean) => void
-  setAssistantState: React.Dispatch<React.SetStateAction<AssistantState>>
-  clearAssistant: () => void
-  onAssistantExecute: (
-    params: { value: string; conversationId: string },
-    callback?: () => void
-  ) => Promise<void>
   setIsOpenCreateAssistant: (isOpen: boolean) => void
   setIsOpenEditAssistant: (isOpen: boolean) => void
   setSelectedAssistantId: (id: string) => void

package/dist/components/AssistantProvider.js

@@ -7,38 +7,16 @@ Object.defineProperty(exports, "__esModule", {
 });
 exports.useAssistant = exports.default = exports.AssistantContext = void 0;
 
-var _regenerator = _interopRequireDefault(require("@babel/runtime/regenerator"));
-
-var _asyncToGenerator2 = _interopRequireDefault(require("@babel/runtime/helpers/asyncToGenerator"));
-
-var _defineProperty2 = _interopRequireDefault(require("@babel/runtime/helpers/defineProperty"));
-
 var _slicedToArray2 = _interopRequireDefault(require("@babel/runtime/helpers/slicedToArray"));
 
-var _set = _interopRequireDefault(require("lodash/set"));
-
 var _react = _interopRequireWildcard(require("react"));
 
-var _reactRouterDom = require("react-router-dom");
-
-var _cozyClient = require("cozy-client");
-
-var _useRealtime3 = _interopRequireDefault(require("cozy-realtime/dist/useRealtime"));
-
 var _constants = require("./constants");
 
-var _helpers = require("./helpers");
-
-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; }
 
-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 AssistantContext = /*#__PURE__*/_react.default.createContext();
 /**
  * @returns {import('./AssistantProvider').AssistantContextValue}
@@ -62,153 +40,52 @@ exports.useAssistant = useAssistant;
 var AssistantProvider = function AssistantProvider(_ref) {
   var children = _ref.children;
 
-  var _useParams = (0, _reactRouterDom.useParams)(),
-      conversationId = _useParams.conversationId;
-
-  var client = (0, _cozyClient.useClient)();
-
-  var _useState = (0, _react.useState)({
-    message: {},
-    status: 'idle',
-    messagesId: []
-  }),
+  var _useState = (0, _react.useState)(false),
       _useState2 = (0, _slicedToArray2.default)(_useState, 2),
-      assistantState = _useState2[0],
-      setAssistantState = _useState2[1];
+      isOpenCreateAssistant = _useState2[0],
+      setIsOpenCreateAssistant = _useState2[1];
 
   var _useState3 = (0, _react.useState)(false),
       _useState4 = (0, _slicedToArray2.default)(_useState3, 2),
-      isOpenCreateAssistant = _useState4[0],
-      setIsOpenCreateAssistant = _useState4[1];
+      isOpenDeleteAssistant = _useState4[0],
+      setIsOpenDeleteAssistant = _useState4[1];
 
   var _useState5 = (0, _react.useState)(false),
       _useState6 = (0, _slicedToArray2.default)(_useState5, 2),
-      isOpenDeleteAssistant = _useState6[0],
-      setIsOpenDeleteAssistant = _useState6[1];
+      isOpenEditAssistant = _useState6[0],
+      setIsOpenEditAssistant = _useState6[1];
 
-  var _useState7 = (0, _react.useState)(false),
+  var _useState7 = (0, _react.useState)(null),
       _useState8 = (0, _slicedToArray2.default)(_useState7, 2),
-      isOpenEditAssistant = _useState8[0],
-      setIsOpenEditAssistant = _useState8[1];
+      assistantIdInAction = _useState8[0],
+      setAssistantIdInAction = _useState8[1];
 
-  var _useState9 = (0, _react.useState)(null),
+  var _useState9 = (0, _react.useState)(_constants.DEFAULT_ASSISTANT._id),
       _useState10 = (0, _slicedToArray2.default)(_useState9, 2),
-      assistantIdInAction = _useState10[0],
-      setAssistantIdInAction = _useState10[1];
+      selectedAssistantId = _useState10[0],
+      setSelectedAssistantId = _useState10[1];
 
-  var _useState11 = (0, _react.useState)(_constants.DEFAULT_ASSISTANT._id),
+  var _useState11 = (0, _react.useState)(false),
       _useState12 = (0, _slicedToArray2.default)(_useState11, 2),
-      selectedAssistantId = _useState12[0],
-      setSelectedAssistantId = _useState12[1];
-
-  var _useState13 = (0, _react.useState)(false),
-      _useState14 = (0, _slicedToArray2.default)(_useState13, 2),
-      isOpenSearchConversation = _useState14[0],
-      setIsOpenSearchConversation = _useState14[1];
+      isOpenSearchConversation = _useState12[0],
+      setIsOpenSearchConversation = _useState12[1];
 
-  var _useState15 = (0, _react.useState)({
+  var _useState13 = (0, _react.useState)({
     drive: [],
     mail: [],
     chat: []
   }),
+      _useState14 = (0, _slicedToArray2.default)(_useState13, 2),
+      selectedTwakeKnowledge = _useState14[0],
+      setSelectedTwakeKnowledge = _useState14[1];
+
+  var _useState15 = (0, _react.useState)(null),
       _useState16 = (0, _slicedToArray2.default)(_useState15, 2),
-      selectedTwakeKnowledge = _useState16[0],
-      setSelectedTwakeKnowledge = _useState16[1];
-
-  var _useState17 = (0, _react.useState)(null),
-      _useState18 = (0, _slicedToArray2.default)(_useState17, 2),
-      openedKnowledgePanel = _useState18[0],
-      setOpenedKnowledgePanel = _useState18[1];
-
-  (0, _useRealtime3.default)(client, (0, _defineProperty2.default)({}, _queries.CHAT_CONVERSATIONS_DOCTYPE, {
-    created: function created(res) {
-      (0, _helpers.pushMessagesIdInState)(conversationId, res, setAssistantState);
-    },
-    updated: function updated(res) {
-      (0, _helpers.pushMessagesIdInState)(conversationId, res, setAssistantState);
-    }
-  }), []);
-  (0, _useRealtime3.default)(client, (0, _defineProperty2.default)({}, _queries.CHAT_EVENTS_DOCTYPE, {
-    created: function created(res) {
-      setAssistantState(function (prevState) {
-        // to exclude realtime messages if not relevant to the actual conversation
-        if (!(0, _helpers.isMessageForThisConversation)(res, prevState.messagesId)) {
-          return prevState;
-        }
-
-        if (res.object === 'done') {
-          if (prevState.status !== 'idle') {
-            return _objectSpread(_objectSpread({}, prevState), {}, {
-              status: 'idle'
-            });
-          }
-        }
-
-        if (res.object === 'delta') {
-          var message = (0, _set.default)(prevState.message, res.position, res.content);
-          return _objectSpread(_objectSpread({}, prevState), {}, {
-            message: message,
-            status: 'writing'
-          });
-        }
-
-        return prevState;
-      });
-    }
-  }), []);
-  var clearAssistant = (0, _react.useCallback)(function () {
-    return setAssistantState({
-      message: {},
-      status: 'pending',
-      messagesId: []
-    });
-  }, []);
-  var onAssistantExecute = (0, _react.useCallback)( /*#__PURE__*/function () {
-    var _ref3 = (0, _asyncToGenerator2.default)( /*#__PURE__*/_regenerator.default.mark(function _callee(_ref2, callback) {
-      var value, conversationId;
-      return _regenerator.default.wrap(function _callee$(_context) {
-        while (1) {
-          switch (_context.prev = _context.next) {
-            case 0:
-              value = _ref2.value, conversationId = _ref2.conversationId;
-
-              if (value) {
-                _context.next = 3;
-                break;
-              }
-
-              return _context.abrupt("return");
-
-            case 3:
-              callback === null || callback === void 0 ? void 0 : callback();
-              clearAssistant();
-              _context.next = 7;
-              return client.stackClient.fetchJSON('POST', "/ai/chat/conversations/".concat(conversationId), {
-                q: value
-              });
-
-            case 7:
-              setAssistantState(function (v) {
-                return _objectSpread(_objectSpread({}, v), {}, {
-                  status: 'pending'
-                });
-              });
-
-            case 8:
-            case "end":
-              return _context.stop();
-          }
-        }
-      }, _callee);
-    }));
-
-    return function (_x, _x2) {
-      return _ref3.apply(this, arguments);
-    };
-  }(), [client, clearAssistant]);
+      openedKnowledgePanel = _useState16[0],
+      setOpenedKnowledgePanel = _useState16[1];
+
   var value = (0, _react.useMemo)(function () {
     return {
-      assistantState: assistantState,
       isOpenCreateAssistant: isOpenCreateAssistant,
       isOpenDeleteAssistant: isOpenDeleteAssistant,
       isOpenEditAssistant: isOpenEditAssistant,
@@ -219,9 +96,6 @@ var AssistantProvider = function AssistantProvider(_ref) {
       selectedTwakeKnowledge: selectedTwakeKnowledge,
       setAssistantIdInAction: setAssistantIdInAction,
       setIsOpenDeleteAssistant: setIsOpenDeleteAssistant,
-      setAssistantState: setAssistantState,
-      clearAssistant: clearAssistant,
-      onAssistantExecute: onAssistantExecute,
       setIsOpenCreateAssistant: setIsOpenCreateAssistant,
       setIsOpenEditAssistant: setIsOpenEditAssistant,
       setSelectedAssistantId: setSelectedAssistantId,
@@ -229,7 +103,7 @@ var AssistantProvider = function AssistantProvider(_ref) {
       setOpenedKnowledgePanel: setOpenedKnowledgePanel,
       setSelectedTwakeKnowledge: setSelectedTwakeKnowledge
     };
-  }, [assistantState, isOpenCreateAssistant, isOpenDeleteAssistant, isOpenEditAssistant, assistantIdInAction, selectedAssistantId, isOpenSearchConversation, openedKnowledgePanel, selectedTwakeKnowledge, clearAssistant, onAssistantExecute]);
+  }, [isOpenCreateAssistant, isOpenDeleteAssistant, isOpenEditAssistant, assistantIdInAction, selectedAssistantId, isOpenSearchConversation, openedKnowledgePanel, selectedTwakeKnowledge]);
   return /*#__PURE__*/_react.default.createElement(AssistantContext.Provider, {
     value: value
   }, children);

package/dist/components/Conversations/Sources/Sources.js

@@ -70,8 +70,18 @@ var Sources = function Sources(_ref) {
   }, []);
   (0, _react.useEffect)(function () {
     if (showSources) {
+      var _ref$current2;
+
+      var innerContainer = (_ref$current2 = ref.current) === null || _ref$current2 === void 0 ? void 0 : _ref$current2.closest('.cozyDialogContent');
+
+      if (!innerContainer) {
+        var _ref$current3;
+
+        (_ref$current3 = ref.current) === null || _ref$current3 === void 0 ? void 0 : _ref$current3.scrollIntoView(false);
+        return;
+      }
+
       var sourcesBottom = ref.current.getBoundingClientRect().bottom;
-      var innerContainer = document.getElementsByClassName('cozyDialogContent')[0];
       var innerContainerBottom = innerContainer.getBoundingClientRect().bottom;
 
       if (sourcesBottom > innerContainerBottom) {

package/dist/components/CozyAssistantRuntimeProvider.js

@@ -183,7 +183,7 @@ var CozyAssistantRuntimeProviderInner = function CozyAssistantRuntimeProviderInn
   (0, _useRealtime3.default)(client, (0, _defineProperty2.default)({}, _queries.CHAT_EVENTS_DOCTYPE, {
     created: function created(res) {
       if (cancelledMessageIdsRef.current.has(res._id)) {
-        if (res.object === 'done') {
+        if (res.object === 'done' || res.object === 'error') {
           cancelledMessageIdsRef.current.delete(res._id);
         }
 
@@ -205,10 +205,20 @@ var CozyAssistantRuntimeProviderInner = function CozyAssistantRuntimeProviderInn
           streamBridgeRef.current.onDelta(conversationId, res.content, res.position);
         }
 
+        if (res.object === 'sources') {
+          streamBridgeRef.current.onSources(conversationId, res.content);
+        }
+
         if (res.object === 'done') {
           streamBridgeRef.current.onDone(conversationId);
           currentStreamingMessageIdRef.current = null;
         }
+
+        if (res.object === 'error') {
+          log.error('LLM error:', res.message);
+          streamBridgeRef.current.onError(conversationId, new Error('LLM error'));
+          currentStreamingMessageIdRef.current = null;
+        }
       } catch (error) {
         log.error('Error handling chat real-time event:', error);
       }

package/dist/components/Messages/AssistantMessage.js

@@ -11,6 +11,8 @@ var _react = require("@assistant-ui/react");
 
 var _react2 = _interopRequireDefault(require("react"));
 
+var _Alert = _interopRequireDefault(require("cozy-ui/transpiled/react/Alert"));
+
 var _Box = _interopRequireDefault(require("cozy-ui/transpiled/react/Box"));
 
 var _Icon = _interopRequireDefault(require("cozy-ui/transpiled/react/Icon"));
@@ -25,6 +27,14 @@ var _TwakeAssistantIcon = require("../AssistantIcon/TwakeAssistantIcon");
 
 var _Sources = _interopRequireDefault(require("../Conversations/Sources/Sources"));
 
+var useIsErrorMessage = function useIsErrorMessage() {
+  return (0, _react.useMessage)(function (s) {
+    var _s$metadata, _s$metadata$custom;
+
+    return ((_s$metadata = s.metadata) === null || _s$metadata === void 0 ? void 0 : (_s$metadata$custom = _s$metadata.custom) === null || _s$metadata$custom === void 0 ? void 0 : _s$metadata$custom.isError) === true;
+  });
+};
+
 var AssistantMessage = function AssistantMessage() {
   var _useI18n = (0, _twakeI18n.useI18n)(),
       t = _useI18n.t;
@@ -34,13 +44,14 @@ var AssistantMessage = function AssistantMessage() {
 
     return ((_s$status = s.status) === null || _s$status === void 0 ? void 0 : _s$status.type) === 'requires-action';
   });
+  var isError = useIsErrorMessage();
   var messageId = (0, _react.useMessage)(function (s) {
     return s.id;
   });
   var sources = (0, _react.useMessage)(function (s) {
-    var _s$metadata, _s$metadata$custom;
+    var _s$metadata2, _s$metadata2$custom;
 
-    return (_s$metadata = s.metadata) === null || _s$metadata === void 0 ? void 0 : (_s$metadata$custom = _s$metadata.custom) === null || _s$metadata$custom === void 0 ? void 0 : _s$metadata$custom.sources;
+    return (_s$metadata2 = s.metadata) === null || _s$metadata2 === void 0 ? void 0 : (_s$metadata2$custom = _s$metadata2.custom) === null || _s$metadata2$custom === void 0 ? void 0 : _s$metadata2$custom.sources;
   });
   return /*#__PURE__*/_react2.default.createElement(_react.MessagePrimitive.Root, {
     className: "u-mt-1-half u-mr-3"
@@ -56,7 +67,9 @@ var AssistantMessage = function AssistantMessage() {
   }), /*#__PURE__*/_react2.default.createElement(_Typography.default, {
     variant: "h6",
     display: "inline"
-  }, t('assistant.message.running'))), /*#__PURE__*/_react2.default.createElement(_react.MessagePrimitive.Content, {
+  }, t('assistant.message.running'))), isError ? /*#__PURE__*/_react2.default.createElement(_Alert.default, {
+    severity: "error"
+  }, t('assistant.default_error')) : /*#__PURE__*/_react2.default.createElement(_react.MessagePrimitive.Content, {
     components: {
       Text: _MarkdownText.default
     }

package/dist/components/Search/AssistantLink.d.ts

@@ -1,4 +1,4 @@
-export default AssistantLinkWrapper;
-declare function AssistantLinkWrapper({ children }: {
+export default AssistantLink;
+declare function AssistantLink({ children }: {
     children: any;
-}): JSX.Element;
+}): any;

package/dist/components/Search/AssistantLink.js

@@ -1,30 +1,16 @@
 "use strict";
 
-var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");
-
 Object.defineProperty(exports, "__esModule", {
   value: true
 });
 exports.default = void 0;
 
-var _react = _interopRequireDefault(require("react"));
-
 var _reactRouterDom = require("react-router-dom");
 
-var _AssistantProvider = _interopRequireWildcard(require("../AssistantProvider"));
-
 var _helpers = require("../helpers");
 
-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; }
-
 var AssistantLink = function AssistantLink(_ref) {
   var children = _ref.children;
-
-  var _useAssistant = (0, _AssistantProvider.useAssistant)(),
-      onAssistantExecute = _useAssistant.onAssistantExecute;
-
   var navigate = (0, _reactRouterDom.useNavigate)();
 
   var _useLocation = (0, _reactRouterDom.useLocation)(),
@@ -33,22 +19,13 @@ var AssistantLink = function AssistantLink(_ref) {
   var openAssistant = function openAssistant() {
     if (!(0, _helpers.isAssistantEnabled)()) return;
     var conversationId = (0, _helpers.makeConversationId)();
-    onAssistantExecute({
-      value: '',
-      conversationId: conversationId
-    });
     navigate("assistant/".concat(conversationId, "?returnPath=").concat(pathname));
   };
 
-  return /*#__PURE__*/_react.default.createElement(_AssistantProvider.default, null, children({
+  return children({
     openAssistant: openAssistant
-  }));
-};
-
-var AssistantLinkWrapper = function AssistantLinkWrapper(_ref2) {
-  var children = _ref2.children;
-  return /*#__PURE__*/_react.default.createElement(_AssistantProvider.default, null, /*#__PURE__*/_react.default.createElement(AssistantLink, null, children));
+  });
 };
 
-var _default = AssistantLinkWrapper;
+var _default = AssistantLink;
 exports.default = _default;

package/dist/components/adapters/CozyRealtimeChatAdapter.js

@@ -9,6 +9,8 @@ exports.createCozyRealtimeChatAdapter = void 0;
 
 var _regenerator = _interopRequireDefault(require("@babel/runtime/regenerator"));
 
+var _defineProperty2 = _interopRequireDefault(require("@babel/runtime/helpers/defineProperty"));
+
 var _awaitAsyncGenerator2 = _interopRequireDefault(require("@babel/runtime/helpers/awaitAsyncGenerator"));
 
 var _wrapAsyncGenerator2 = _interopRequireDefault(require("@babel/runtime/helpers/wrapAsyncGenerator"));
@@ -17,6 +19,10 @@ var _cozyMinilog = _interopRequireDefault(require("cozy-minilog"));
 
 var _helpers = require("../helpers");
 
+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; }
+
 function _asyncIterator(iterable) { var method, async, sync, retry = 2; for ("undefined" != typeof Symbol && (async = Symbol.asyncIterator, sync = Symbol.iterator); retry--;) { if (async && null != (method = iterable[async])) return method.call(iterable); if (sync && null != (method = iterable[sync])) return new AsyncFromSyncIterator(method.call(iterable)); async = "@@asyncIterator", sync = "@@iterator"; } throw new TypeError("Object is not async iterable"); }
 
 function AsyncFromSyncIterator(s) { function AsyncFromSyncIteratorContinuation(r) { if (Object(r) !== r) return Promise.reject(new TypeError(r + " is not an object.")); var done = r.done; return Promise.resolve(r.value).then(function (value) { return { value: value, done: done }; }); } return AsyncFromSyncIterator = function AsyncFromSyncIterator(s) { this.s = s, this.n = s.next; }, AsyncFromSyncIterator.prototype = { s: null, n: null, next: function next() { return AsyncFromSyncIteratorContinuation(this.n.apply(this.s, arguments)); }, return: function _return(value) { var ret = this.s.return; return void 0 === ret ? Promise.resolve({ value: value, done: !0 }) : AsyncFromSyncIteratorContinuation(ret.apply(this.s, arguments)); }, throw: function _throw(value) { var thr = this.s.return; return void 0 === thr ? Promise.reject(value) : AsyncFromSyncIteratorContinuation(thr.apply(this.s, arguments)); } }, new AsyncFromSyncIterator(s); }
@@ -58,7 +64,7 @@ var createCozyRealtimeChatAdapter = function createCozyRealtimeChatAdapter(optio
       var messages = _ref.messages,
           abortSignal = _ref.abortSignal;
       return (0, _wrapAsyncGenerator2.default)( /*#__PURE__*/_regenerator.default.mark(function _callee() {
-        var client, conversationId, streamBridge, assistantId, userQuery, stream, fullText, wasAborted, _iteratorAbruptCompletion, _didIteratorError, _iteratorError, _iterator, _step, chunk, sanitizedText, finalText;
+        var client, conversationId, streamBridge, assistantId, userQuery, stream, fullText, wasAborted, _iteratorAbruptCompletion, _didIteratorError, _iteratorError, _iterator, _step, chunk, sanitizedText, finalText, sources;
 
         return _regenerator.default.wrap(function _callee$(_context) {
           while (1) {
@@ -185,13 +191,14 @@ var createCozyRealtimeChatAdapter = function createCozyRealtimeChatAdapter(optio
 
               case 48:
                 if (wasAborted) {
-                  _context.next = 53;
+                  _context.next = 54;
                   break;
                 }
 
                 finalText = (0, _helpers.sanitizeChatContent)(fullText);
-                _context.next = 52;
-                return {
+                sources = streamBridge.getSources(conversationId);
+                _context.next = 53;
+                return _objectSpread({
                   content: [{
                     type: 'text',
                     text: finalText
@@ -200,21 +207,27 @@ var createCozyRealtimeChatAdapter = function createCozyRealtimeChatAdapter(optio
                     type: 'complete',
                     reason: 'stop'
                   }
-                };
+                }, sources ? {
+                  metadata: {
+                    custom: {
+                      sources: sources
+                    }
+                  }
+                } : {});
 
-              case 52:
+              case 53:
                 streamBridge.cleanup(conversationId);
 
-              case 53:
-                _context.next = 61;
+              case 54:
+                _context.next = 62;
                 break;
 
-              case 55:
-                _context.prev = 55;
+              case 56:
+                _context.prev = 56;
                 _context.t1 = _context["catch"](6);
                 log.error('Error:', _context.t1);
                 streamBridge.cleanup(conversationId);
-                _context.next = 61;
+                _context.next = 62;
                 return {
                   content: [{
                     type: 'text',
@@ -223,15 +236,20 @@ var createCozyRealtimeChatAdapter = function createCozyRealtimeChatAdapter(optio
                   status: {
                     type: 'incomplete',
                     reason: 'error'
+                  },
+                  metadata: {
+                    custom: {
+                      isError: true
+                    }
                   }
                 };
 
-              case 61:
+              case 62:
               case "end":
                 return _context.stop();
             }
           }
-        }, _callee, null, [[6, 55], [15, 34, 38, 48], [39,, 43, 47]]);
+        }, _callee, null, [[6, 56], [15, 34, 38, 48], [39,, 43, 47]]);
       }))();
     }
   };

package/dist/components/adapters/StreamBridge.d.ts

@@ -10,6 +10,7 @@ export declare class StreamBridge {
     private cleanupCallback;
     private positionBuffers;
     private nextPositions;
+    private sourcesMap;
     /**
      * Sets a callback to be invoked when cleanup is called.
      * This allows the provider to mark the current message as cancelled.
@@ -31,6 +32,21 @@ export declare class StreamBridge {
      * Completes the stream for the conversation.
      */
     onDone(conversationId: string): void;
+    /**
+     * Called when a 'sources' event is received from WebSocket.
+     * Stores sources for the conversation to be retrieved by the adapter.
+     */
+    onSources(conversationId: string, sources: Array<{
+        id: string;
+        doctype?: string;
+    }>): void;
+    /**
+     * Returns stored sources for a conversation.
+     */
+    getSources(conversationId: string): Array<{
+        id: string;
+        doctype?: string;
+    }> | undefined;
     /**
      * Called when an error occurs.
      * Errors the stream for the conversation.

package/dist/components/adapters/StreamBridge.js

@@ -27,6 +27,7 @@ var StreamBridge = /*#__PURE__*/function () {
     (0, _defineProperty2.default)(this, "cleanupCallback", null);
     (0, _defineProperty2.default)(this, "positionBuffers", new Map());
     (0, _defineProperty2.default)(this, "nextPositions", new Map());
+    (0, _defineProperty2.default)(this, "sourcesMap", new Map());
   }
 
   (0, _createClass2.default)(StreamBridge, [{
@@ -200,6 +201,26 @@ var StreamBridge = /*#__PURE__*/function () {
         this.nextPositions.delete(conversationId);
       }
     }
+    /**
+     * Called when a 'sources' event is received from WebSocket.
+     * Stores sources for the conversation to be retrieved by the adapter.
+     */
+
+  }, {
+    key: "onSources",
+    value: function onSources(conversationId, sources) {
+      if (!this.streams.has(conversationId)) return;
+      this.sourcesMap.set(conversationId, sources);
+    }
+    /**
+     * Returns stored sources for a conversation.
+     */
+
+  }, {
+    key: "getSources",
+    value: function getSources(conversationId) {
+      return this.sourcesMap.get(conversationId);
+    }
     /**
      * Called when an error occurs.
      * Errors the stream for the conversation.
@@ -212,9 +233,11 @@ var StreamBridge = /*#__PURE__*/function () {
 
       if (stream) {
         stream.error(error);
-        this.positionBuffers.delete(conversationId);
-        this.nextPositions.delete(conversationId);
       }
+
+      this.positionBuffers.delete(conversationId);
+      this.nextPositions.delete(conversationId);
+      this.sourcesMap.delete(conversationId);
     }
     /**
      * Cleans up the stream for a conversation.
@@ -233,9 +256,11 @@ var StreamBridge = /*#__PURE__*/function () {
 
         stream.complete();
         this.streams.delete(conversationId);
-        this.positionBuffers.delete(conversationId);
-        this.nextPositions.delete(conversationId);
       }
+
+      this.positionBuffers.delete(conversationId);
+      this.nextPositions.delete(conversationId);
+      this.sourcesMap.delete(conversationId);
     }
     /**
      * Checks if there's an active stream for a conversation.

package/dist/components/adapters/StreamBridge.spec.ts

@@ -48,6 +48,18 @@ describe('StreamBridge', () => {
     await expect(iterator.next()).rejects.toThrow('Socket disconnected')
   })
 
+  it('should reject the iterator when an error occurs mid-stream', async () => {
+    const iterator = bridge.createStream('convo_1')
+
+    bridge.onDelta('convo_1', 'partial ')
+    const first = await iterator.next()
+    expect(first).toEqual({ value: 'partial ', done: false })
+
+    bridge.onError('convo_1', new Error('LLM unavailable'))
+
+    await expect(iterator.next()).rejects.toThrow('LLM unavailable')
+  })
+
   it('should call the cleanup callback and mark the stream complete on cleanup', async () => {
     const cleanupSpy = jest.fn()
     bridge.setCleanupCallback(cleanupSpy)

package/dist/components/helpers.d.ts

@@ -1,6 +1,4 @@
 export function makeConversationId(): string;
-export function pushMessagesIdInState(id: any, res: any, setState: any): void;
-export function isMessageForThisConversation(res: any, messagesId: any): any;
 export function isAssistantEnabled(): any;
 export function sanitizeChatContent(content: string): string;
 export function formatConversationDate(dateString: any, t: any, lang: any): string;

package/dist/components/helpers.js

@@ -5,42 +5,16 @@ var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefau
 Object.defineProperty(exports, "__esModule", {
   value: true
 });
-exports.sanitizeChatContent = exports.pushMessagesIdInState = exports.makeConversationId = exports.isMessageForThisConversation = exports.isAssistantEnabled = exports.getNameOfConversation = exports.getDescriptionOfConversation = exports.formatConversationDate = void 0;
-
-var _defineProperty2 = _interopRequireDefault(require("@babel/runtime/helpers/defineProperty"));
+exports.sanitizeChatContent = exports.makeConversationId = exports.isAssistantEnabled = exports.getNameOfConversation = exports.getDescriptionOfConversation = exports.formatConversationDate = void 0;
 
 var _cozyFlags = _interopRequireDefault(require("cozy-flags"));
 
-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 makeConversationId = function makeConversationId() {
   return "".concat(Date.now(), "-").concat(Math.floor(Math.random() * 90000) + 10000);
 };
 
 exports.makeConversationId = makeConversationId;
 
-var pushMessagesIdInState = function pushMessagesIdInState(id, res, setState) {
-  if (id !== res._id) return;
-  var messagesId = res.messages.map(function (message) {
-    return message.id;
-  });
-  setState(function (v) {
-    return _objectSpread(_objectSpread({}, v), {}, {
-      messagesId: messagesId
-    });
-  });
-};
-
-exports.pushMessagesIdInState = pushMessagesIdInState;
-
-var isMessageForThisConversation = function isMessageForThisConversation(res, messagesId) {
-  return messagesId.includes(res._id);
-};
-
-exports.isMessageForThisConversation = isMessageForThisConversation;
-
 var isAssistantEnabled = function isAssistantEnabled() {
   return (0, _cozyFlags.default)('cozy.assistant.enabled');
 };

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "cozy-search",
   "description": "UI components about search bar and IA assistant",
-  "version": "0.23.1",
+  "version": "0.23.3",
   "author": {
     "name": "Cozy Cloud",
     "url": "https://github.com/cozy"
@@ -94,5 +94,5 @@
     "lint": "cd .. && yarn eslint --ext js,jsx,ts packages/cozy-search"
   },
   "types": "dist/index.d.ts",
-  "gitHead": "1065f8b76289ccfb32c86e62086184de8ae4d340"
+  "gitHead": "e10a9ecf005814f4ff42fc35114172085c34abd5"
 }