alpha private key

Author: balogh.adam@icloud.com

examples/run_x402_llm.py

@@ -26,7 +26,5 @@
     {"role": "user", "content": "What makes it good for beginners?"},
 ]
 
-result = client.llm.chat(
-    model=og.TEE_LLM.GPT_4_1_2025_04_14, messages=messages, x402_settlement_mode=og.x402SettlementMode.SETTLE_METADATA
-)
+result = client.llm.chat(model=og.TEE_LLM.GPT_4_1_2025_04_14, messages=messages, x402_settlement_mode=og.x402SettlementMode.SETTLE_METADATA)
 print(f"Response: {result.chat_output['content']}")

src/opengradient/__init__.py

@@ -99,6 +99,7 @@
 
 def init(
     private_key: str,
+    alpha_private_key: Optional[str] = None,
     email: Optional[str] = None,
     password: Optional[str] = None,
     **kwargs,
@@ -109,7 +110,9 @@ def init(
     and stores it as the global client for convenience.
 
     Args:
-        private_key: Private key for OpenGradient transactions.
+        private_key: Private key for LLM inference (Base Sepolia, x402 payments).
+        alpha_private_key: Private key for Alpha Testnet features (on-chain inference).
+            When omitted, falls back to ``private_key`` for backward compatibility.
         email: Email for Model Hub authentication. Optional.
         password: Password for Model Hub authentication. Optional.
         **kwargs: Additional arguments forwarded to `Client`.
@@ -123,7 +126,13 @@ def init(
         response = client.llm.chat(model=og.TEE_LLM.GPT_4O, messages=[...])
     """
     global global_client
-    global_client = Client(private_key=private_key, email=email, password=password, **kwargs)
+    global_client = Client(
+        private_key=private_key,
+        alpha_private_key=alpha_private_key,
+        email=email,
+        password=password,
+        **kwargs,
+    )
     return global_client
 
 

src/opengradient/cli.py

@@ -96,6 +96,15 @@ def initialize_config(ctx):
     else:
         ctx.obj["private_key"] = click.prompt("Enter your OpenGradient private key", type=str)
 
+    # Optional separate private key for Alpha Testnet
+    alpha_pk = click.prompt(
+        "Enter a separate Alpha Testnet private key (optional, press Enter to reuse the main key)",
+        type=str,
+        default="",
+        show_default=False,
+    )
+    ctx.obj["alpha_private_key"] = alpha_pk if alpha_pk else None
+
     # Make email and password optional
     email = click.prompt(
         "Enter your OpenGradient Hub email address (optional, press Enter to skip)", type=str, default="", show_default=False
@@ -132,6 +141,7 @@ def cli(ctx):
         try:
             ctx.obj["client"] = Client(
                 private_key=ctx.obj["private_key"],
+                alpha_private_key=ctx.obj.get("alpha_private_key"),
                 rpc_url=DEFAULT_RPC_URL,
                 api_url=DEFAULT_API_URL,
                 contract_address=DEFAULT_INFERENCE_CONTRACT_ADDRESS,
@@ -173,7 +183,7 @@ def show(ctx):
     click.echo("Current config:")
     for key, value in ctx.obj.items():
         if key != "client":  # Don't display the client object
-            if (key == "password" or key == "private_key") and value is not None:
+            if key in ("password", "private_key", "alpha_private_key") and value is not None:
                 click.echo(f"{key}: {'*' * len(value)}")  # Mask the password
             elif value is None:
                 click.echo(f"{key}: Not set")

src/opengradient/client/__init__.py

@@ -17,6 +17,9 @@
 
 client = og.init(private_key="0x...")
 
+# Use separate keys for LLM (Base Sepolia) and Alpha Testnet
+client = og.init(private_key="0xLLM_KEY...", alpha_private_key="0xALPHA_KEY...")
+
 # LLM chat (TEE-verified, streamed)
 for chunk in client.llm.chat(
     model=og.TEE_LLM.CLAUDE_3_5_HAIKU,

src/opengradient/client/client.py

@@ -47,6 +47,7 @@ class Client:
     def __init__(
         self,
         private_key: str,
+        alpha_private_key: Optional[str] = None,
         email: Optional[str] = None,
         password: Optional[str] = None,
         twins_api_key: Optional[str] = None,
@@ -61,11 +62,13 @@ def __init__(
         Initialize the OpenGradient client.
 
         Args:
-            private_key: Private key for OpenGradient transactions.
+            private_key: Private key for LLM inference (Base Sepolia, x402 payments).
+            alpha_private_key: Private key for Alpha Testnet features (on-chain inference).
+                When omitted, falls back to ``private_key`` for backward compatibility.
             email: Email for Model Hub authentication. Optional.
             password: Password for Model Hub authentication. Optional.
             twins_api_key: API key for digital twins chat (twin.fun). Optional.
-            rpc_url: RPC URL for the blockchain network.
+            rpc_url: RPC URL for the Alpha Testnet blockchain network.
             api_url: API URL for the OpenGradient API.
             contract_address: Inference contract address.
             og_llm_server_url: OpenGradient LLM server URL.
@@ -74,6 +77,12 @@ def __init__(
         blockchain = Web3(Web3.HTTPProvider(rpc_url))
         wallet_account = blockchain.eth.account.from_key(private_key)
 
+        # Use a separate account for Alpha Testnet when provided
+        if alpha_private_key is not None:
+            alpha_wallet_account = blockchain.eth.account.from_key(alpha_private_key)
+        else:
+            alpha_wallet_account = wallet_account
+
         hub_user = None
         if email is not None:
             hub_user = ModelHub._login_to_hub(email, password)
@@ -90,7 +99,7 @@ def __init__(
 
         self.alpha = Alpha(
             blockchain=blockchain,
-            wallet_account=wallet_account,
+            wallet_account=alpha_wallet_account,
             inference_hub_contract_address=contract_address,
             api_url=api_url,
         )

src/opengradient/client/llm.py

@@ -264,9 +264,7 @@ async def make_request_v2():
                 try:
                     # Non-streaming with x402
                     endpoint = "/v1/chat/completions"
-                    response = await client.post(
-                        self._og_llm_server_url + endpoint, json=payload, headers=headers, timeout=60
-                    )
+                    response = await client.post(self._og_llm_server_url + endpoint, json=payload, headers=headers, timeout=60)
 
                     content = await response.aread()
                     result = json.loads(content.decode())
@@ -464,4 +462,3 @@ async def _parse_sse_response(response) -> AsyncGenerator[StreamChunk, None]:
             ) as response:
                 async for parsed_chunk in _parse_sse_response(response):
                     yield parsed_chunk
-

src/opengradient/defaults.py

@@ -8,4 +8,3 @@
 DEFAULT_BLOCKCHAIN_EXPLORER = "https://explorer.opengradient.ai/tx/"
 DEFAULT_OPENGRADIENT_LLM_SERVER_URL = "https://llm.opengradient.ai"
 DEFAULT_OPENGRADIENT_LLM_STREAMING_SERVER_URL = "https://llm.opengradient.ai"
-