cozy-search 0.19.0 → 0.19.2

package/dist/components/CozyAssistantRuntimeProvider.js

@@ -202,7 +202,7 @@ var CozyAssistantRuntimeProviderInner = function CozyAssistantRuntimeProviderInn
 
       try {
         if (res.object === 'delta' && res.content !== undefined) {
-          streamBridgeRef.current.onDelta(conversationId, res.content);
+          streamBridgeRef.current.onDelta(conversationId, res.content, res.position);
         }
 
         if (res.object === 'done') {

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

@@ -8,6 +8,8 @@
 export declare class StreamBridge {
     private streams;
     private cleanupCallback;
+    private positionBuffers;
+    private nextPositions;
     /**
      * Sets a callback to be invoked when cleanup is called.
      * This allows the provider to mark the current message as cancelled.
@@ -20,9 +22,10 @@ export declare class StreamBridge {
     createStream(conversationId: string): AsyncIterableIterator<string>;
     /**
      * Called when a delta event is received from WebSocket.
-     * Pushes the content chunk to the appropriate stream.
+     * When a position is provided, chunks are buffered and flushed in order.
+     * Without a position, chunks are pushed directly in arrival order.
      */
-    onDelta(conversationId: string, content: string): void;
+    onDelta(conversationId: string, content: string, position?: number): void;
     /**
      * Called when a 'done' event is received from WebSocket.
      * Completes the stream for the conversation.

package/dist/components/adapters/StreamBridge.js

@@ -25,6 +25,8 @@ var StreamBridge = /*#__PURE__*/function () {
     (0, _classCallCheck2.default)(this, StreamBridge);
     (0, _defineProperty2.default)(this, "streams", new Map());
     (0, _defineProperty2.default)(this, "cleanupCallback", null);
+    (0, _defineProperty2.default)(this, "positionBuffers", new Map());
+    (0, _defineProperty2.default)(this, "nextPositions", new Map());
   }
 
   (0, _createClass2.default)(StreamBridge, [{
@@ -128,17 +130,59 @@ var StreamBridge = /*#__PURE__*/function () {
     }
     /**
      * Called when a delta event is received from WebSocket.
-     * Pushes the content chunk to the appropriate stream.
+     * When a position is provided, chunks are buffered and flushed in order.
+     * Without a position, chunks are pushed directly in arrival order.
      */
 
   }, {
     key: "onDelta",
-    value: function onDelta(conversationId, content) {
+    value: function onDelta(conversationId, content, position) {
+      var _this$nextPositions$g;
+
       var stream = this.streams.get(conversationId);
+      if (!stream) return;
 
-      if (stream) {
+      if (position === undefined) {
+        stream.push(content);
+        return;
+      } // Fast path: chunk arrived in order, no buffering needed
+
+
+      var nextExpected = (_this$nextPositions$g = this.nextPositions.get(conversationId)) !== null && _this$nextPositions$g !== void 0 ? _this$nextPositions$g : 0;
+
+      if (position === nextExpected) {
         stream.push(content);
+        var next = nextExpected + 1; // Flush any previously buffered chunks that are now contiguous
+
+        var _buffer = this.positionBuffers.get(conversationId);
+
+        if (_buffer && _buffer.size > 0) {
+          while (_buffer.has(next)) {
+            stream.push(_buffer.get(next));
+
+            _buffer.delete(next);
+
+            next++;
+          }
+
+          if (_buffer.size === 0) {
+            this.positionBuffers.delete(conversationId);
+          }
+        }
+
+        this.nextPositions.set(conversationId, next);
+        return;
+      } // Out-of-order: buffer until the gap is filled
+
+
+      var buffer = this.positionBuffers.get(conversationId);
+
+      if (!buffer) {
+        buffer = new Map();
+        this.positionBuffers.set(conversationId, buffer);
       }
+
+      buffer.set(position, content);
     }
     /**
      * Called when a 'done' event is received from WebSocket.
@@ -152,6 +196,8 @@ var StreamBridge = /*#__PURE__*/function () {
 
       if (stream) {
         stream.complete();
+        this.positionBuffers.delete(conversationId);
+        this.nextPositions.delete(conversationId);
       }
     }
     /**
@@ -166,6 +212,8 @@ var StreamBridge = /*#__PURE__*/function () {
 
       if (stream) {
         stream.error(error);
+        this.positionBuffers.delete(conversationId);
+        this.nextPositions.delete(conversationId);
       }
     }
     /**
@@ -185,6 +233,8 @@ var StreamBridge = /*#__PURE__*/function () {
 
         stream.complete();
         this.streams.delete(conversationId);
+        this.positionBuffers.delete(conversationId);
+        this.nextPositions.delete(conversationId);
       }
     }
     /**

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

@@ -63,6 +63,54 @@ describe('StreamBridge', () => {
     expect(result.done).toBe(true)
   })
 
+  it('should reorder out-of-order deltas when position is provided', async () => {
+    const iterator = bridge.createStream('convo_1')
+
+    // Arrive out of order: position 1 before position 0
+    bridge.onDelta('convo_1', 'world!', 1)
+    bridge.onDelta('convo_1', 'Hello ', 0)
+
+    const first = await iterator.next()
+    expect(first).toEqual({ value: 'Hello ', done: false })
+
+    const second = await iterator.next()
+    expect(second).toEqual({ value: 'world!', done: false })
+  })
+
+  it('should flush buffered positions in order when gap is filled', async () => {
+    const iterator = bridge.createStream('convo_1')
+
+    // Positions 2 and 1 arrive before 0
+    bridge.onDelta('convo_1', 'c', 2)
+    bridge.onDelta('convo_1', 'b', 1)
+    // Nothing yielded yet since position 0 is missing
+
+    bridge.onDelta('convo_1', 'a', 0)
+    // Now all three should flush in order
+
+    const first = await iterator.next()
+    expect(first).toEqual({ value: 'a', done: false })
+
+    const second = await iterator.next()
+    expect(second).toEqual({ value: 'b', done: false })
+
+    const third = await iterator.next()
+    expect(third).toEqual({ value: 'c', done: false })
+  })
+
+  it('should push directly when no position is provided', async () => {
+    const iterator = bridge.createStream('convo_1')
+
+    bridge.onDelta('convo_1', 'Hello ')
+    bridge.onDelta('convo_1', 'world!')
+
+    const first = await iterator.next()
+    expect(first).toEqual({ value: 'Hello ', done: false })
+
+    const second = await iterator.next()
+    expect(second).toEqual({ value: 'world!', done: false })
+  })
+
   it('multiple unresolved next calls should reject to prevent concurrency issues', async () => {
     const iterator = bridge.createStream('convo_1')
 

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "cozy-search",
   "description": "UI components about search bar and IA assistant",
-  "version": "0.19.0",
+  "version": "0.19.2",
   "author": {
     "name": "Cozy Cloud",
     "url": "https://github.com/cozy"
@@ -29,21 +29,21 @@
     "cozy-bar": "^29.0.0",
     "cozy-client": "^60.21.0",
     "cozy-device-helper": "^4.0.3",
-    "cozy-flags": "^4.9.0",
-    "cozy-intent": "^2.31.0",
-    "cozy-logger": "^1.18.0",
+    "cozy-flags": "^4.9.1",
+    "cozy-intent": "^2.31.1",
+    "cozy-logger": "^1.18.1",
     "cozy-minilog": "^3.10.1",
     "cozy-pouch-link": "^54.0.0",
-    "cozy-realtime": "^5.9.0",
+    "cozy-realtime": "^5.9.1",
     "cozy-ui": "^135.6.0",
-    "cozy-ui-plus": "^5.3.0",
+    "cozy-ui-plus": "^5.3.1",
     "cross-fetch": "^4.0.0",
-    "jest": "28.1.3",
+    "jest": "30.3.0",
     "react": "18.2.0",
     "react-dom": "18.2.0",
     "react-router-dom": "6.14.2",
     "stylus": "^0.64.0",
-    "twake-i18n": "^0.4.0",
+    "twake-i18n": "^0.4.1",
     "typescript": "5.5.2"
   },
   "files": [
@@ -94,5 +94,5 @@
     "lint": "cd .. && yarn eslint --ext js,jsx,ts packages/cozy-search"
   },
   "types": "dist/index.d.ts",
-  "gitHead": "a2816d422234922d845a1b6c01fae4c45fe81fc8"
+  "gitHead": "661b14497f4331618699916112fbdb1d5ed2d606"
 }