package

Author: balogh.adam@icloud.com

examples/llm_chat.ts

@@ -26,7 +26,7 @@ async function main() {
   });
 
   console.log(`Response: ${result.chatOutput?.content}`);
-  console.log(`Payment hash: ${result.paymentHash}`);
+  console.log(`Payment hash: ${result.transactionHash}`);
 }
 
 main().catch((err) => {

package-lock.json

@@ -52,6 +52,9 @@
     "testEnvironment": "node"
   },
   "dependencies": {
+    "@x402/core": "^2.11.0",
+    "@x402/evm": "^2.11.0",
+    "@x402/fetch": "^2.11.0",
     "undici": "^6.21.0",
     "viem": "^2.21.0",
     "x402-fetch": "^1.2.0"

package.json

@@ -1,4 +1,8 @@
-import { createSigner, wrapFetchWithPayment } from "x402-fetch";
+import { wrapFetchWithPayment } from "@x402/fetch";
+import { x402Client } from "@x402/core/client";
+import { UptoEvmScheme } from "@x402/evm";
+import { registerExactEvmScheme } from "@x402/evm/exact/client";
+import { privateKeyToAccount } from "viem/accounts";
 import type { Agent } from "undici";
 import {
   ChatParams,
@@ -34,7 +38,7 @@ export interface LLMConfig {
  * `llmServerUrl` on the `Client` to override with a hardcoded URL.
  */
 export class LLM {
-  private signerPromise?: Promise<unknown>;
+  private x402ClientInstance?: x402Client;
 
   constructor(private readonly config: LLMConfig) {}
 
@@ -200,29 +204,26 @@ export class LLM {
     return payload;
   }
 
-  private async getSigner(): Promise<unknown> {
-    if (!this.signerPromise) {
-      this.signerPromise = createSigner(
-        this.config.network,
-        this.config.privateKey,
-      );
+  private getX402Client(): x402Client {
+    if (!this.x402ClientInstance) {
+      const account = privateKeyToAccount(this.config.privateKey);
+      const client = new x402Client();
+      registerExactEvmScheme(client, { signer: account });
+      // The TEE may quote the "upto" scheme — register it on EVM networks too.
+      client.register("eip155:*", new UptoEvmScheme(account));
+      this.x402ClientInstance = client;
     }
-    return this.signerPromise;
+    return this.x402ClientInstance;
   }
 
   /**
    * Build a paid fetch that injects the TEE's pinned TLS dispatcher into every
    * request (including x402 payment retries).
    */
-  private async buildPaidFetch(dispatcher: Agent): Promise<typeof fetch> {
-    const signer = await this.getSigner();
+  private buildPaidFetch(dispatcher: Agent): typeof fetch {
     const baseFetch: typeof fetch = ((input: any, init?: any) =>
       fetch(input, { ...(init ?? {}), dispatcher } as any)) as typeof fetch;
-    return wrapFetchWithPayment(
-      baseFetch,
-      signer as any,
-      this.config.maxPaymentValue,
-    ) as typeof fetch;
+    return wrapFetchWithPayment(baseFetch, this.getX402Client()) as typeof fetch;
   }
 
   /**
@@ -261,7 +262,7 @@ export class LLM {
   ): Promise<{ response: Response; tee: ActiveTEE }> {
     const tee = await this.config.connection.ensureConnected();
     const url = `${trimSlash(tee.endpoint)}${path}`;
-    const paidFetch = await this.buildPaidFetch(tee.dispatcher);
+    const paidFetch = this.buildPaidFetch(tee.dispatcher);
 
     let response: Response;
     try {

src/llm.ts

@@ -93,13 +93,20 @@ export class TEERegistry {
       return [];
     }
 
+    console.debug(
+      `[TEERegistry] getActiveTEEs(type=${teeType}) returned ${teeInfos.length} entries`,
+    );
+
     const out: TEEEndpoint[] = [];
     for (const tee of teeInfos) {
       if (
         !tee.endpoint ||
         !tee.tlsCertificate ||
         tee.tlsCertificate === "0x"
       ) {
+        console.debug(
+          `[TEERegistry] skipping TEE (endpoint=${tee.endpoint || "<empty>"}, certLen=${tee.tlsCertificate?.length ?? 0})`,
+        );
         continue;
       }
       out.push({
@@ -109,6 +116,10 @@ export class TEERegistry {
         paymentAddress: tee.paymentAddress,
       });
     }
+    console.debug(
+      `[TEERegistry] ${out.length} usable TEEs:`,
+      out.map((t) => ({ teeId: t.teeId, endpoint: t.endpoint })),
+    );
     return out;
   }
 
@@ -118,7 +129,14 @@ export class TEERegistry {
    */
   async getLLMTEE(): Promise<TEEEndpoint | null> {
     const tees = await this.getActiveTEEsByType(TEE_TYPE_LLM_PROXY);
-    if (tees.length === 0) return null;
-    return tees[Math.floor(Math.random() * tees.length)];
+    if (tees.length === 0) {
+      console.debug("[TEERegistry] getLLMTEE: no active LLM proxy TEEs found");
+      return null;
+    }
+    const picked = tees[Math.floor(Math.random() * tees.length)];
+    console.debug(
+      `[TEERegistry] getLLMTEE: picked ${picked.teeId} @ ${picked.endpoint}`,
+    );
+    return picked;
   }
 }