lightnode-sdk 0.9.0 → 0.9.2

package/dist/add.js

@@ -1085,17 +1085,17 @@ export default function ChatWeb3() {
         setBusyStage("");
         patchLastAssistant({ text: totalSoFar });
       };
+      const onStage = (s: string) => setBusyStage(s);
+      const sendOpts = { onChunk, onStage };
 
       const chat = await ensureSession();
-      setBusyStage("Approve the per-turn transaction in your wallet...");
-      const result = await chat.send(prompt, { onChunk }).catch(async () => {
+      const result = await chat.send(prompt, sendOpts).catch(async () => {
         // Session expired or the worker stopped serving - reopen once and retry.
         sessionRef.current = null;
         patchLastAssistant({ text: "" });
         setBusyStage("Re-opening session...");
         const fresh = await ensureSession();
-        setBusyStage("Approve the per-turn transaction in your wallet...");
-        return fresh.send(prompt, { onChunk });
+        return fresh.send(prompt, sendOpts);
       });
 
       patchLastAssistant({
@@ -1181,7 +1181,7 @@ export default function ChatWeb3() {
                     )
                   ) : (
                     <div className="animate-pulse-dot pt-1 text-sm text-muted-foreground">
-                      {busyStage || "Writing on chain..."}
+                      {busyStage || "Thinking..."}
                     </div>
                   )}
                   {t.submitTx && (

package/dist/inference.d.ts

@@ -235,6 +235,8 @@ export interface OpenSessionArgs {
 export declare function openSession(args: OpenSessionArgs): Promise<OpenSession>;
 export interface RunJobOpts {
     onChunk?: (chunk: string, totalSoFar: string) => void;
+    /** Human-readable progress, e.g. "Uploading prompt to chain..." then "Thinking...". */
+    onStage?: (stage: string) => void;
     jobCompletedTimeoutMs?: number;
     WebSocket?: WebSocketCtor;
     relayUrl?: string;

package/dist/inference.js

@@ -277,19 +277,28 @@ export async function openSession(args) {
  */
 export async function runJobOnSession(session, prompt, opts = {}, attempt = 1) {
     const { gateway, wallet, publicClient, network, sessionId, sessionKey, worker, fee, createTx } = session;
-    const { onChunk, jobCompletedTimeoutMs = 120000 } = opts;
+    const { onChunk, onStage, jobCompletedTimeoutMs = 120000 } = opts;
     const WS = pickWebSocket(opts.WebSocket);
     const relayUrl = opts.relayUrl ?? `wss://relay.${network.id}.lightchain.ai/ws`;
     // Shim so the job body below can keep referencing prepared.* unchanged.
     const prepared = { sessionKey, createSessionArgs: { worker } };
     // 3. relay token + WebSocket
+    // Poll the gateway for the relay token with a fast backoff (it's usually ready
+    // within ~1s of createSession). Catch it sooner than a fixed 1s interval, and
+    // cap the total at a 20s deadline instead of 30 fixed iterations.
     let relayToken;
-    for (let i = 0; i < 30 && !relayToken; i++) {
+    const tokenDeadline = Date.now() + 20000;
+    let tokenDelay = 250;
+    while (!relayToken) {
         const r = await gateway.getSessionToken(Number(sessionId));
-        if ("token" in r && r.token)
+        if ("token" in r && r.token) {
             relayToken = r.token;
-        else
-            await new Promise((res) => setTimeout(res, 1000));
+            break;
+        }
+        if (Date.now() >= tokenDeadline)
+            break;
+        await new Promise((res) => setTimeout(res, tokenDelay));
+        tokenDelay = Math.min(tokenDelay * 2, 2000);
     }
     if (!relayToken)
         throw new RelayTokenTimeoutError();
@@ -373,6 +382,7 @@ export async function runJobOnSession(session, prompt, opts = {}, attempt = 1) {
         ws.addEventListener("message", (ev) => handleMessage(ev.data));
     }
     // 4. encrypt + upload prompt
+    onStage?.("Uploading prompt to chain...");
     const promptHash = await submitPrompt(gateway, prepared.sessionKey, prompt);
     // 5. submitJob on-chain
     const submitTx = await wallet.writeContract({
@@ -397,6 +407,7 @@ export async function runJobOnSession(session, prompt, opts = {}, attempt = 1) {
     if (!jobLog)
         throw new Error("JobSubmitted log missing in submitJob receipt");
     const jobId = topicAsUint(jobLog.topics[1]);
+    onStage?.("Thinking...");
     // 6. wait for JobCompleted
     // The actual *result* is the WS-delivered, session-key-decrypted ciphertext.
     // JobCompleted is an explorer pointer (the worker's commit-result tx).

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "lightnode-sdk",
-  "version": "0.9.0",
+  "version": "0.9.2",
   "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",