register updates

Author: dixitaniket

src/opengradient/client/llm.py

@@ -6,6 +6,7 @@
 from typing import AsyncGenerator, Awaitable, Callable, Dict, List, Optional, TypeVar, Union
 import httpx
 import asyncio
+import os
 
 from eth_account import Account
 from eth_account.account import LocalAccount
@@ -28,6 +29,7 @@
 X402_PROCESSING_HASH_HEADER = "x-processing-hash"
 X402_PLACEHOLDER_API_KEY = "0x1234567890abcdef1234567890abcdef1234567890abcdef1234567890abcdef"
 BASE_TESTNET_NETWORK = "eip155:84532"
+BASE_TESTNET_RPC = os.getenv("BASE_TESTNET_RPC", "https://sepolia.base.org")
 
 _CHAT_ENDPOINT = "/v1/chat/completions"
 _COMPLETION_ENDPOINT = "/v1/completions"
@@ -84,7 +86,7 @@ def __init__(
             raise ValueError("A private key is required to use the LLM client. Pass a valid private_key to the constructor.")
         self._wallet_account: LocalAccount = Account.from_key(private_key)
 
-        x402_client = LLM._build_x402_client(private_key)
+        x402_client = LLM._build_x402_client(private_key, rpc_url=BASE_TESTNET_RPC)
         onchain_registry = TEERegistry(rpc_url=rpc_url, registry_address=tee_registry_address)
         self._tee: TEEConnectionInterface = RegistryTEEConnection(x402_client=x402_client, registry=onchain_registry)
 
@@ -114,13 +116,18 @@ def from_url(
         return instance
 
     @staticmethod
-    def _build_x402_client(private_key: str) -> x402Client:
+    def _build_x402_client(private_key: str, rpc_url: str = BASE_TESTNET_RPC) -> x402Client:
         """Build the x402 payment stack from a private key."""
         account = Account.from_key(private_key)
         signer = EthAccountSigner(account)
         client = x402Client()
         register_exact_evm_client(client, signer, networks=[BASE_TESTNET_NETWORK])
-        register_upto_evm_client(client, signer, networks=[BASE_TESTNET_NETWORK])
+        register_upto_evm_client(
+            client,
+            signer,
+            networks=[BASE_TESTNET_NETWORK],
+            rpc_url=rpc_url,
+        )
         return client
 
     # ── Lifecycle ───────────────────────────────────────────────────────