Tutorials

README.md

@@ -269,6 +269,16 @@ result = client.llm.chat(
 )
 ```
 
+### Payment Network Selection
+
+By default, payments settle on the OG EVM network. To pay via Base Sepolia instead, set the `payment_network` parameter at client initialization. The Base Sepolia payment network uses the OG token at address `0x240b09731D96979f50B2C649C9CE10FcF9C7987F`.
+```python
+client = og.Client(
+    private_key=os.environ.get("OG_PRIVATE_KEY"),
+    payment_network=og.PaymentNetwork.BASE_SEPOLIA,
+)
+```
+
 ## Examples
 
 Additional code examples are available in the [examples](./examples) directory.

docs/opengradient/client/client.md

@@ -21,7 +21,7 @@ blockchain private key and optional Model Hub credentials.
 #### Constructor
 
 ```python
-def __init__(private_key: str, email: Optional[str] = None, password: Optional[str] = None, rpc_url: str = 'https://ogevmdevnet.opengradient.ai', api_url: str = 'https://sdk-devnet.opengradient.ai', contract_address: str = '0x8383C9bD7462F12Eb996DD02F78234C0421A6FaE', og_llm_server_url: Optional[str] = 'https://llmogevm.opengradient.ai', og_llm_streaming_server_url: Optional[str] = 'https://llmogevm.opengradient.ai')
+def __init__(private_key: str, email: Optional[str] = None, password: Optional[str] = None, rpc_url: str = 'https://ogevmdevnet.opengradient.ai', api_url: str = 'https://sdk-devnet.opengradient.ai', contract_address: str = '0x8383C9bD7462F12Eb996DD02F78234C0421A6FaE', og_llm_server_url: Optional[str] = 'https://llmogevm.opengradient.ai', og_llm_streaming_server_url: Optional[str] = 'https://llmogevm.opengradient.ai', payment_network: `PaymentNetwork` = PaymentNetwork.OG_EVM)
 ```
 
 **Arguments**
@@ -34,6 +34,7 @@ def __init__(private_key: str, email: Optional[str] = None, password: Optio
 * **`contract_address`**: Inference contract address.
 * **`og_llm_server_url`**: OpenGradient LLM server URL.
 * **`og_llm_streaming_server_url`**: OpenGradient LLM streaming server URL.
+* **`payment_network`**: Payment network for x402 transactions. Defaults to OG_EVM.
 
 #### Variables
 

docs/opengradient/client/index.md

@@ -66,7 +66,7 @@ blockchain private key and optional Model Hub credentials.
 #### Constructor
 
 ```python
-def __init__(private_key: str, email: Optional[str] = None, password: Optional[str] = None, rpc_url: str = 'https://ogevmdevnet.opengradient.ai', api_url: str = 'https://sdk-devnet.opengradient.ai', contract_address: str = '0x8383C9bD7462F12Eb996DD02F78234C0421A6FaE', og_llm_server_url: Optional[str] = 'https://llmogevm.opengradient.ai', og_llm_streaming_server_url: Optional[str] = 'https://llmogevm.opengradient.ai')
+def __init__(private_key: str, email: Optional[str] = None, password: Optional[str] = None, rpc_url: str = 'https://ogevmdevnet.opengradient.ai', api_url: str = 'https://sdk-devnet.opengradient.ai', contract_address: str = '0x8383C9bD7462F12Eb996DD02F78234C0421A6FaE', og_llm_server_url: Optional[str] = 'https://llmogevm.opengradient.ai', og_llm_streaming_server_url: Optional[str] = 'https://llmogevm.opengradient.ai', payment_network: `PaymentNetwork` = PaymentNetwork.OG_EVM)
 ```
 
 **Arguments**
@@ -79,6 +79,7 @@ def __init__(private_key: str, email: Optional[str] = None, password: Optio
 * **`contract_address`**: Inference contract address.
 * **`og_llm_server_url`**: OpenGradient LLM server URL.
 * **`og_llm_streaming_server_url`**: OpenGradient LLM streaming server URL.
+* **`payment_network`**: Payment network for x402 transactions. Defaults to OG_EVM.
 
 #### Variables
 

docs/opengradient/client/llm.md

@@ -21,7 +21,7 @@ Supports both streaming and non-streaming responses.
 #### Constructor
 
 ```python
-def __init__(wallet_account: `LocalAccount`, og_llm_server_url: str, og_llm_streaming_server_url: str)
+def __init__(wallet_account: `LocalAccount`, og_llm_server_url: str, og_llm_streaming_server_url: str, network_filter: str = 'og-evm')
 ```
 
 #### Methods

docs/opengradient/index.md

@@ -126,7 +126,7 @@ blockchain private key and optional Model Hub credentials.
 #### Constructor
 
 ```python
-def __init__(private_key: str, email: Optional[str] = None, password: Optional[str] = None, rpc_url: str = 'https://ogevmdevnet.opengradient.ai', api_url: str = 'https://sdk-devnet.opengradient.ai', contract_address: str = '0x8383C9bD7462F12Eb996DD02F78234C0421A6FaE', og_llm_server_url: Optional[str] = 'https://llmogevm.opengradient.ai', og_llm_streaming_server_url: Optional[str] = 'https://llmogevm.opengradient.ai')
+def __init__(private_key: str, email: Optional[str] = None, password: Optional[str] = None, rpc_url: str = 'https://ogevmdevnet.opengradient.ai', api_url: str = 'https://sdk-devnet.opengradient.ai', contract_address: str = '0x8383C9bD7462F12Eb996DD02F78234C0421A6FaE', og_llm_server_url: Optional[str] = 'https://llmogevm.opengradient.ai', og_llm_streaming_server_url: Optional[str] = 'https://llmogevm.opengradient.ai', payment_network: `PaymentNetwork` = PaymentNetwork.OG_EVM)
 ```
 
 **Arguments**
@@ -139,6 +139,7 @@ def __init__(private_key: str, email: Optional[str] = None, password: Optio
 * **`contract_address`**: Inference contract address.
 * **`og_llm_server_url`**: OpenGradient LLM server URL.
 * **`og_llm_streaming_server_url`**: OpenGradient LLM streaming server URL.
+* **`payment_network`**: Payment network for x402 transactions. Defaults to OG_EVM.
 
 #### Variables
 
@@ -156,6 +157,23 @@ Enum for the different inference modes available for inference (VANILLA, ZKML, T
 * static `VANILLA`
 * static `ZKML`
 
+### `PaymentNetwork`
+
+Payment network for x402 payment protocol transactions.
+
+Controls which blockchain network is used for paying for LLM inference.
+
+**Attributes**
+
+* **`OG_EVM`**: Pay on the OpenGradient EVM network (default).
+* **`BASE_SEPOLIA`**: Pay on the Base Sepolia testnet using the OG token
+        at address ``0x240b09731D96979f50B2C649C9CE10FcF9C7987F``.
+
+#### Variables
+
+* static `BASE_SEPOLIA`
+* static `OG_EVM`
+
 ### `TEE_LLM`
 
 Enum for LLM models available for TEE (Trusted Execution Environment) execution.

docs/opengradient/types.md

@@ -220,6 +220,23 @@ def __init__(name: str, values: List[Tuple[int, int]])
 * static `name` : str
 * static `values` : List[Tuple[int, int]]
 
+### `PaymentNetwork`
+
+Payment network for x402 payment protocol transactions.
+
+Controls which blockchain network is used for paying for LLM inference.
+
+**Attributes**
+
+* **`OG_EVM`**: Pay on the OpenGradient EVM network (default).
+* **`BASE_SEPOLIA`**: Pay on the Base Sepolia testnet using the OG token
+        at address ``0x240b09731D96979f50B2C649C9CE10FcF9C7987F``.
+
+#### Variables
+
+* static `BASE_SEPOLIA`
+* static `OG_EVM`
+
 ### `SchedulerParams`
 
 #### Constructor

examples/run_x402_llm.py

@@ -15,6 +15,8 @@
 
 client = og.Client(
     private_key=os.environ.get("OG_PRIVATE_KEY"),
+    # To pay via Base Sepolia instead of OG EVM:
+    # payment_network=og.PaymentNetwork.BASE_SEPOLIA,
 )
 
 messages = [

src/opengradient/__init__.py

@@ -86,6 +86,7 @@
     InferenceResult,
     ModelOutput,
     ModelRepository,
+    PaymentNetwork,
     SchedulerParams,
     TextGenerationOutput,
     TextGenerationStream,
@@ -132,6 +133,7 @@ def init(
     "init",
     "TEE_LLM",
     "InferenceMode",
+    "PaymentNetwork",
     "HistoricalInputQuery",
     "SchedulerParams",
     "CandleType",

src/opengradient/cli.py

@@ -324,9 +324,7 @@ def infer(ctx, model_cid: str, inference_mode: str, input_data, input_file: Path
                 model_input = json.load(file)
 
         click.echo(f'Running {inference_mode} inference for model "{model_cid}"')
-        inference_result = client.alpha.infer(
-            model_cid=model_cid, inference_mode=InferenceModes[inference_mode], model_input=model_input
-        )
+        inference_result = client.alpha.infer(model_cid=model_cid, inference_mode=InferenceModes[inference_mode], model_input=model_input)
 
         click.echo()  # Add a newline for better spacing
         click.secho("✅ Transaction successful", fg="green", bold=True)

src/opengradient/client/client.py

@@ -11,6 +11,7 @@
     DEFAULT_OPENGRADIENT_LLM_STREAMING_SERVER_URL,
     DEFAULT_RPC_URL,
 )
+from ..types import PaymentNetwork
 from .alpha import Alpha
 from .llm import LLM
 from .model_hub import ModelHub
@@ -50,6 +51,7 @@ def __init__(
         contract_address: str = DEFAULT_INFERENCE_CONTRACT_ADDRESS,
         og_llm_server_url: Optional[str] = DEFAULT_OPENGRADIENT_LLM_SERVER_URL,
         og_llm_streaming_server_url: Optional[str] = DEFAULT_OPENGRADIENT_LLM_STREAMING_SERVER_URL,
+        payment_network: PaymentNetwork = PaymentNetwork.OG_EVM,
     ):
         """
         Initialize the OpenGradient client.
@@ -63,6 +65,7 @@ def __init__(
             contract_address: Inference contract address.
             og_llm_server_url: OpenGradient LLM server URL.
             og_llm_streaming_server_url: OpenGradient LLM streaming server URL.
+            payment_network: Payment network for x402 transactions. Defaults to OG_EVM.
         """
         blockchain = Web3(Web3.HTTPProvider(rpc_url))
         wallet_account = blockchain.eth.account.from_key(private_key)
@@ -78,6 +81,7 @@ def __init__(
             wallet_account=wallet_account,
             og_llm_server_url=og_llm_server_url,
             og_llm_streaming_server_url=og_llm_streaming_server_url,
+            network_filter=payment_network.value,
         )
 
         self.alpha = Alpha(
@@ -86,4 +90,3 @@ def __init__(
             inference_hub_contract_address=contract_address,
             api_url=api_url,
         )
-

src/opengradient/client/llm.py

@@ -53,16 +53,23 @@ class LLM:
         result = client.llm.completion(model=TEE_LLM.CLAUDE_3_5_HAIKU, prompt="Hello")
     """
 
-    def __init__(self, wallet_account: LocalAccount, og_llm_server_url: str, og_llm_streaming_server_url: str):
+    def __init__(
+        self,
+        wallet_account: LocalAccount,
+        og_llm_server_url: str,
+        og_llm_streaming_server_url: str,
+        network_filter: str = DEFAULT_NETWORK_FILTER,
+    ):
         self._wallet_account = wallet_account
         self._og_llm_server_url = og_llm_server_url
         self._og_llm_streaming_server_url = og_llm_streaming_server_url
+        self._network_filter = network_filter
 
-    def _og_payment_selector(self, accepts, network_filter=DEFAULT_NETWORK_FILTER, scheme_filter=None, max_value=None):
+    def _og_payment_selector(self, accepts, network_filter=None, scheme_filter=None, max_value=None):
         """Custom payment selector for OpenGradient network."""
         return x402Client.default_payment_requirements_selector(
             accepts,
-            network_filter=network_filter,
+            network_filter=network_filter or self._network_filter,
             scheme_filter=scheme_filter,
             max_value=max_value,
         )
@@ -418,7 +425,7 @@ async def _tee_llm_chat_stream_async(
             limits=LIMITS,
             http2=False,
             follow_redirects=False,
-            auth=X402Auth(account=self._wallet_account, network_filter=DEFAULT_NETWORK_FILTER),  # type: ignore
+            auth=X402Auth(account=self._wallet_account, network_filter=self._network_filter),  # type: ignore
             verify=True,
         ) as client:
             headers = {

src/opengradient/defaults.py

@@ -9,3 +9,4 @@
 DEFAULT_OPENGRADIENT_LLM_SERVER_URL = "https://llmogevm.opengradient.ai"
 DEFAULT_OPENGRADIENT_LLM_STREAMING_SERVER_URL = "https://llmogevm.opengradient.ai"
 DEFAULT_NETWORK_FILTER = "og-evm"
+DEFAULT_BASE_SEPOLIA_NETWORK_FILTER = "base-sepolia"

src/opengradient/types.py

@@ -10,6 +10,28 @@
 import numpy as np
 
 
+class PaymentNetwork(str, Enum):
+    """
+    Payment network for x402 payment protocol transactions.
+
+    Controls which blockchain network is used for paying for LLM inference.
+
+    Attributes:
+        OG_EVM: Pay on the OpenGradient EVM network (default).
+        BASE_SEPOLIA: Pay on the Base Sepolia testnet using the OG token
+            at address ``0x240b09731D96979f50B2C649C9CE10FcF9C7987F``.
+
+    Examples:
+        >>> client = og.Client(
+        ...     private_key="0x...",
+        ...     payment_network=og.PaymentNetwork.BASE_SEPOLIA,
+        ... )
+    """
+
+    OG_EVM = "og-evm"
+    BASE_SEPOLIA = "base-sepolia"
+
+
 class x402SettlementMode(str, Enum):
     """
     Settlement modes for x402 payment protocol transactions.