lightnode-sdk 0.8.9 → 0.8.10

package/dist/inference.js

@@ -293,6 +293,8 @@ async function runOneAttempt(args, attempt) {
         setTimeout(() => reject(new Error("relay WebSocket open timeout")), 20000);
     });
     const chunks = [];
+    let streamDone = false;
+    let streamDoneAt = null;
     const handleMessage = async (rawData) => {
         const raw = typeof rawData === "string"
             ? rawData
@@ -308,20 +310,26 @@ async function runOneAttempt(args, attempt) {
         catch {
             return;
         }
-        if (!frame?.payload)
-            return;
-        if (frame.type === "chunk") {
-            try {
-                const piece = await decryptResponse(prepared.sessionKey, frame.payload);
-                chunks.push(piece);
-                if (onChunk)
-                    onChunk(piece, chunks.join(""));
-            }
-            catch {
-                /* control frame */
+        // "complete" marks end-of-stream (it may or may not carry a payload).
+        // Record it so the JobCompleted wait can stop promptly instead of polling
+        // the full grace window after the answer is already in hand.
+        if (frame.type === "complete") {
+            streamDone = true;
+            streamDoneAt = Date.now();
+            if (chunks.length === 0 && frame.payload) {
+                try {
+                    const piece = await decryptResponse(prepared.sessionKey, frame.payload);
+                    chunks.push(piece);
+                    if (onChunk)
+                        onChunk(piece, chunks.join(""));
+                }
+                catch {
+                    /* ignore */
+                }
             }
+            return;
         }
-        else if (frame.type === "complete" && chunks.length === 0) {
+        if (frame.type === "chunk" && frame.payload) {
             try {
                 const piece = await decryptResponse(prepared.sessionKey, frame.payload);
                 chunks.push(piece);
@@ -329,7 +337,7 @@ async function runOneAttempt(args, attempt) {
                     onChunk(piece, chunks.join(""));
             }
             catch {
-                /* ignore */
+                /* control frame */
             }
         }
     };
@@ -376,7 +384,8 @@ async function runOneAttempt(args, attempt) {
     //     answer with txs.jobCompleted=null (the answer is still session-key
     //     authentic; the on-chain proof can be polled for separately by callers).
     const deadline = Date.now() + jobCompletedTimeoutMs;
-    const POST_CHUNKS_GRACE_MS = 45000;
+    const POST_CHUNKS_GRACE_MS = 45000; // fallback if the relay never sends a 'complete' frame
+    const POST_DONE_GRACE_MS = 8000; // once the answer is fully in, the worker commits JobCompleted within ~seconds
     const waitStart = Date.now();
     let firstChunkAt = chunks.length > 0 ? waitStart : null;
     const jobIdTopic = (`0x${jobId.toString(16).padStart(64, "0")}`);
@@ -385,9 +394,14 @@ async function runOneAttempt(args, attempt) {
         const now = Date.now();
         if (now >= deadline)
             break;
+        // Answer fully received (relay sent 'complete'): wait only briefly for the
+        // on-chain proof, then return. The answer is already session-key authentic;
+        // callers get txs.jobCompleted=null and can poll the proof later if needed.
+        if (streamDone && now - (streamDoneAt ?? now) >= POST_DONE_GRACE_MS)
+            break;
         if (firstChunkAt != null && now - firstChunkAt >= POST_CHUNKS_GRACE_MS)
             break;
-        await new Promise((res) => setTimeout(res, 3000));
+        await new Promise((res) => setTimeout(res, 1500));
         if (firstChunkAt == null && chunks.length > 0)
             firstChunkAt = Date.now();
         const logs = await publicClient.getLogs({
@@ -414,8 +428,9 @@ async function runOneAttempt(args, attempt) {
             feeLcai: fee,
         });
     }
-    // 7. grace period for the last relay frame, then close
-    await new Promise((res) => setTimeout(res, 4000));
+    // 7. grace period for the last relay frame, then close. If the stream already
+    // signaled 'complete', no more frames are coming - skip the wait.
+    await new Promise((res) => setTimeout(res, streamDone ? 300 : 4000));
     try {
         ws.close();
     }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "lightnode-sdk",
-  "version": "0.8.9",
+  "version": "0.8.10",
   "description": "Read-only TypeScript client for LightChain AI: workers, jobs, models, on-chain registration, and per-model network analytics. Independent, community-built (not an official LightChain package).",
   "type": "module",
   "main": "./dist/index.js",