[api] Fixes ChunkedBytesSupplier read timeout bug (#2716)

deepjavalibrary · Jul 15, 2023 · 7a1f5dd · 7a1f5dd
1 parent ef1147b
commit 7a1f5dd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/api/src/main/java/ai/djl/inference/streaming/ChunkedBytesSupplier.java b/api/src/main/java/ai/djl/inference/streaming/ChunkedBytesSupplier.java
@@ -50,10 +50,10 @@ public void appendContent(byte[] data, boolean lastChunk) {
      * @param lastChunk true if this is the last chunk
      */
     public void appendContent(BytesSupplier bytesSupplier, boolean lastChunk) {
-        queue.offer(bytesSupplier);
         if (lastChunk) {
             completed.set(true);
         }
+        queue.offer(bytesSupplier);
     }
 
     /**