lightnode-sdk 0.7.6 → 0.7.8

package/README.md

@@ -226,6 +226,15 @@ forever and **blocks deregistration** - and no official tool clears it. The
 JobRegistry's `claimTimeout` is permissionless, so the operator can self-clear
 it. `unstickAndDeregister()` is the one-call rescue.
 
+The second thing it gets right is **gas-correct writes**. The worker daemon
+under-sets the gas limit on its WorkerRegistry writes, so `addSupportedModel` and
+`deregisterWorker` run out of gas and revert on-chain - the daemon reports a
+failure (or, for deregister, some indexers still flip the worker to
+"deregistered" while the stake never moves). Every write here estimates the gas
+first and sends with a margin, so the transaction lands. `addModel()` is the
+gas-correct version of the model-add the daemon botches; `deregister()` is the
+gas-correct exit.
+
 ```ts
 import { WorkerOperator } from "lightnode-sdk";
 import { createPublicClient, createWalletClient, http } from "viem";
@@ -275,8 +284,9 @@ Full method reference (`jobIds` are the worker's IDs from
 | `withdraw()` | yes | pull the earned balance into the worker wallet |
 | `topUpStake(lcai)` | yes | add stake |
 | `withdrawStake(lcai)` | yes | remove stake above the floor |
+| `addModel(tagOrId)` | yes | add a supported model to a registered worker on-chain, gas-correct (no-op if already served) |
 | `reinstate()` | yes | reactivate a suspended worker |
-| `deregister()` | yes | exit and release stake (reverts if in-flight jobs remain) |
+| `deregister()` | yes | exit and release stake, gas-correct (reverts if in-flight jobs remain) |
 | `unstickAndDeregister(jobIds)` | yes | clear stuck + release + withdraw + deregister, in one call |
 
 > **Mainnet slashing.** `claimTimeout` / `clearStuck` / `unstickAndDeregister`
@@ -443,6 +453,8 @@ controls. Mainnet `clearstuck` and `deregister` realize a slash, so they require
 
 ```bash
                   npx lightnode worker status 0x...        # registration, stake, claimable, live config
+                  npx lightnode worker models 0x...        # models served, reconciled vs chain (servingNow truth)
+PRIVATE_KEY=0x... npx lightnode worker preflight           # one real test inference, print verdict + timings
 PRIVATE_KEY=0x... npx lightnode worker can-deregister      # what blocks the exit, before spending gas
 PRIVATE_KEY=0x... npx lightnode worker settle              # release completed jobs past their window + withdraw
 PRIVATE_KEY=0x... npx lightnode worker withdraw            # pull the earned balance into the worker wallet

package/dist/index.d.ts

@@ -133,7 +133,7 @@ export declare class LightNode {
  * (especially in registry-proxy environments like StackBlitz where lockfiles
  * may pin an older minor than the local install command suggests).
  */
-export declare const SDK_VERSION = "0.7.6";
+export declare const SDK_VERSION = "0.7.8";
 export { NETWORKS, WORKER_REGISTRY, REGISTRY_TOPICS, aggregateModelStats, aggregateWorkerStats, networkAnalytics, modelStatsCsv, workerStatsCsv, workerJobsCsv, fromWei, resolveJobTransactions, fetchWorkerModels, computeModelId as modelId, estimateJobFee, JOB_REGISTRY_CONSUMER_ABI, consumerGatewayUrl, consumerGatewayHost, GatewayClient, GatewayHttpError, prepareSession, submitPrompt, decryptResponse, generateEcdhKeyPair, crypto, runInference, runInferenceWithKey, runInferenceStream, Conversation, chat, runInferenceBatch, Agent, parseAgentOutput, workerPreflight, workerWatch, Bridge, BRIDGE_ROUTE, HYPERLANE_ROUTER_ABI, ERC20_ABI, addressToBytes32, quoteBridgeFee, bridgeableBalance, bridgeAllowance, approveBridge, bridgeTransfer, DAO, DAO_ADDRESSES, ProposalState, PROPOSAL_STATE_LABEL, VoteSupport, GOVERNOR_ABI, VOTES_ABI, OnchainModelRegistry, AIVM_MODEL_REGISTRY_ABI, BENCHMARK_REGISTRY_ABI, ModelStatus, MODEL_STATUS_LABEL, StalledWorkerError, OnChainRevertError, RelayTokenTimeoutError, GatewayAuthError, isStalledWorker, WorkerOperator, WORKER_REGISTRY_ABI, JOB_REGISTRY_OPERATOR_ABI, AI_CONFIG_ABI, JOB_STATE, decodeWorkerError, WorkerOpError, isWorkerOpError, };
 export type { BearerSource, GatewayClientOptions, SelectSessionResult, PrepareSessionResult, UploadBlobResult, SessionTokenResult } from "./gateway.js";
 export type { SessionPreparation, RunInferenceArgs, RunInferenceResult, RunInferenceWithKeyArgs, RunInferenceStreamResult } from "./inference.js";

package/dist/index.js

@@ -212,7 +212,7 @@ export class LightNode {
  * (especially in registry-proxy environments like StackBlitz where lockfiles
  * may pin an older minor than the local install command suggests).
  */
-export const SDK_VERSION = "0.7.6";
+export const SDK_VERSION = "0.7.8";
 export { NETWORKS, WORKER_REGISTRY, REGISTRY_TOPICS, aggregateModelStats, aggregateWorkerStats, networkAnalytics, modelStatsCsv, workerStatsCsv, workerJobsCsv, fromWei, 
 // v0.7.3 per-job transaction-hash resolver (lifts the upstream
 // subgraph's "block-only" Job entity to a deep-linkable Job + tx pair).

package/dist/inference.js

@@ -75,35 +75,57 @@ export const JOB_REGISTRY_CONSUMER_ABI = [
  */
 export async function prepareSession(gateway, modelTag) {
     const id = modelId(modelTag);
-    const selected = await gateway.selectSession(id);
-    const sessionKey = await generateSessionKey();
-    // Workers' pubkeys arrive as base64; disputer's as hex - decodePublicKey
-    // accepts either.
-    const workerPub = await importPublicKey(decodePublicKey(selected.workerEncryptionKey));
-    const encWorker = await encryptSessionKey(sessionKey, workerPub);
-    const encDisputer = selected.disputerEncryptionKey
-        ? await encryptSessionKey(sessionKey, await importPublicKey(decodePublicKey(selected.disputerEncryptionKey)))
-        : new Uint8Array(0);
-    // The gateway expects the wrapped keys as BASE64; the same bytes are passed
-    // as HEX to the on-chain createSession. Sending hex to the gateway makes the
-    // dispatcher reject the prepare with an opaque error.
-    const prepared = await gateway.prepareSession({
-        modelId: id,
-        encWorkerKey: bytesToBase64(encWorker),
-        encDisputerKey: bytesToBase64(encDisputer),
-    });
-    return {
-        sessionKey,
-        nonce: prepared.nonce,
-        createSessionArgs: {
-            paramsHash: id,
-            worker: prepared.worker,
-            encWorkerKey: bytesToHex(encWorker),
-            ephemeralPubKey: bytesToHex(encDisputer),
-            initState: prepared.signature,
-            expiry: BigInt(prepared.expiry),
-        },
-    };
+    // The gateway returns 409 selection_mismatch when a NEWER selectSession()
+    // for the same wallet supersedes ours between the select and the prepare.
+    // The error message is literally "re-run POST /api/sessions/select", so we
+    // do exactly that: rebuild from a fresh selection. The cap stops a busy
+    // pool of callers from looping forever - 4 attempts at 250ms / 750ms /
+    // 1500ms covers every churn pattern we have seen.
+    const MAX_ATTEMPTS = 4;
+    const BACKOFFS_MS = [0, 250, 750, 1500];
+    let lastErr = null;
+    for (let attempt = 0; attempt < MAX_ATTEMPTS; attempt++) {
+        if (attempt > 0)
+            await new Promise((r) => setTimeout(r, BACKOFFS_MS[attempt]));
+        const selected = await gateway.selectSession(id);
+        const sessionKey = await generateSessionKey();
+        // Workers' pubkeys arrive as base64; disputer's as hex - decodePublicKey
+        // accepts either.
+        const workerPub = await importPublicKey(decodePublicKey(selected.workerEncryptionKey));
+        const encWorker = await encryptSessionKey(sessionKey, workerPub);
+        const encDisputer = selected.disputerEncryptionKey
+            ? await encryptSessionKey(sessionKey, await importPublicKey(decodePublicKey(selected.disputerEncryptionKey)))
+            : new Uint8Array(0);
+        try {
+            const prepared = await gateway.prepareSession({
+                modelId: id,
+                encWorkerKey: bytesToBase64(encWorker),
+                encDisputerKey: bytesToBase64(encDisputer),
+            });
+            return {
+                sessionKey,
+                nonce: prepared.nonce,
+                createSessionArgs: {
+                    paramsHash: id,
+                    worker: prepared.worker,
+                    encWorkerKey: bytesToHex(encWorker),
+                    ephemeralPubKey: bytesToHex(encDisputer),
+                    initState: prepared.signature,
+                    expiry: BigInt(prepared.expiry),
+                },
+            };
+        }
+        catch (e) {
+            lastErr = e;
+            const msg = e instanceof Error ? e.message : String(e);
+            if (!/selection_mismatch|selection was superseded|409/.test(msg))
+                throw e;
+            // else loop: a newer select stole this session, try again from select.
+        }
+    }
+    throw lastErr instanceof Error
+        ? lastErr
+        : new Error("prepareSession: gateway selection_mismatch did not clear");
 }
 /**
  * Encrypt a UTF-8 prompt with the session key, upload as a blob, and return

package/dist/worker-operator.d.ts

@@ -210,9 +210,10 @@ export declare class WorkerOperator {
     readonly network: NetworkConfig;
     private readonly pub;
     private readonly wallet?;
-    private readonly addr;
+    private readonly maybeAddr;
     private cfgCache?;
     constructor(network: NetworkId | NetworkConfig, opts: WorkerOperatorOpts);
+    private get addr();
     private requireWallet;
     private get jobReg();
     private get workerReg();
@@ -282,6 +283,17 @@ export declare class WorkerOperator {
     withdraw(): Promise<`0x${string}`>;
     /** Deregister - releases stake to the wallet. Reverts (ActiveJobsExist) if any in-flight job remains. */
     deregister(): Promise<`0x${string}`>;
+    /**
+     * Add a supported model to THIS (already-registered) worker on-chain. Accepts a
+     * model tag (e.g. "gemma4:e2b") or a raw bytes32 modelId.
+     *
+     * This is the gas-correct version of the step the worker daemon's one-shot
+     * register botches: the daemon sends addSupportedModel with an under-set gas
+     * limit, so it OutOfGas-reverts (and its rollback deregister can too). viem
+     * estimates the gas here, so it lands. Use it to finish a worker that staked but
+     * failed to add its model. No-op (returns null) if already serving the model.
+     */
+    addModel(modelTagOrId: string): Promise<`0x${string}` | null>;
     /**
      * The flagship rescue: clear stuck jobs then release any settled completed jobs +
      * withdraw earnings then deregister. The one flow no official tool provides.

package/dist/worker-operator.js

@@ -287,9 +287,13 @@ export class WorkerOperator {
         const acct = opts.walletClient?.account;
         const fromWallet = typeof acct === "string" ? acct : acct?.address;
         const a = (opts.workerAddress ?? fromWallet);
-        if (!a)
-            throw new Error("WorkerOperator: provide workerAddress or a walletClient with an account");
-        this.addr = a.toLowerCase();
+        this.maybeAddr = a ? a.toLowerCase() : undefined;
+    }
+    get addr() {
+        if (!this.maybeAddr) {
+            throw new Error("WorkerOperator: this method needs the worker address. Pass `workerAddress` (or a `walletClient` with an account) when constructing WorkerOperator.");
+        }
+        return this.maybeAddr;
     }
     requireWallet(op) {
         if (!this.wallet)
@@ -525,6 +529,25 @@ export class WorkerOperator {
     async deregister() {
         return this.send("deregister", this.workerReg, WORKER_REGISTRY_ABI_PARSED, "deregisterWorker", []);
     }
+    /**
+     * Add a supported model to THIS (already-registered) worker on-chain. Accepts a
+     * model tag (e.g. "gemma4:e2b") or a raw bytes32 modelId.
+     *
+     * This is the gas-correct version of the step the worker daemon's one-shot
+     * register botches: the daemon sends addSupportedModel with an under-set gas
+     * limit, so it OutOfGas-reverts (and its rollback deregister can too). viem
+     * estimates the gas here, so it lands. Use it to finish a worker that staked but
+     * failed to add its model. No-op (returns null) if already serving the model.
+     */
+    async addModel(modelTagOrId) {
+        const id = modelTagOrId.startsWith("0x") && modelTagOrId.length === 66
+            ? modelTagOrId.toLowerCase()
+            : (await import("./inference.js")).modelId(modelTagOrId);
+        const already = (await this.read(this.workerReg, WORKER_REGISTRY_ABI_PARSED, "isEligible", [this.addr, id]));
+        if (already)
+            return null;
+        return this.send("addModel", this.workerReg, WORKER_REGISTRY_ABI_PARSED, "addSupportedModel", [id]);
+    }
     /**
      * The flagship rescue: clear stuck jobs then release any settled completed jobs +
      * withdraw earnings then deregister. The one flow no official tool provides.

package/package.json

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