revert to old name

balogh.adam@icloud.com · balogh.adam@icloud.com · commit 78c86d712d18 · 2026-03-27T11:51:12.000-04:00
diff --git a/README.md b/README.md
@@ -91,7 +91,7 @@ hub = og.ModelHub(email="you@example.com", password="...")
 Before making LLM requests, your wallet must approve OPG token spending via the [Permit2](https://github.com/Uniswap/permit2) protocol. This only sends an on-chain transaction when the current allowance drops below the threshold:
 
 ```python
-llm.ensure_opg_allowance(min_allowance=5)
+llm.ensure_opg_approval(min_allowance=5)
 ```
 
 See [Payment Settlement](#payment-settlement) for details on settlement modes.
diff --git a/docs/CLAUDE_SDK_USERS.md b/docs/CLAUDE_SDK_USERS.md
@@ -22,7 +22,7 @@ import os
 llm = og.LLM(private_key=os.environ["OG_PRIVATE_KEY"])
 
 # Ensure sufficient OPG allowance (only sends tx when below threshold)
-llm.ensure_opg_allowance(min_allowance=0.1)
+llm.ensure_opg_approval(min_allowance=0.1)
 
 # LLM Chat (TEE-verified with x402 payments, async)
 result = await llm.chat(
diff --git a/docs/opengradient/client/index.md b/docs/opengradient/client/index.md
@@ -24,7 +24,7 @@ import opengradient as og
 
 # LLM inference (Base Sepolia OPG tokens)
 llm = og.LLM(private_key="0x...")
-llm.ensure_opg_allowance(min_allowance=5)
+llm.ensure_opg_approval(min_allowance=5)
 result = await llm.chat(model=og.TEE_LLM.CLAUDE_HAIKU_4_5, messages=[...])
 
 # On-chain model inference (OpenGradient testnet gas tokens)
diff --git a/docs/opengradient/client/llm.md b/docs/opengradient/client/llm.md
@@ -21,7 +21,7 @@ Supports both streaming and non-streaming responses.
 All request methods (``chat``, ``completion``) are async.
 
 Before making LLM requests, ensure your wallet has approved sufficient
-OPG tokens for Permit2 spending by calling ``ensure_opg_allowance``.
+OPG tokens for Permit2 spending by calling ``ensure_opg_approval``.
 
 #### Constructor
 
@@ -188,10 +188,10 @@ TextGenerationOutput: Generated text results including:
 
 ---
 
-#### `ensure_opg_allowance()`
+#### `ensure_opg_approval()`
 
 ```python
-def ensure_opg_allowance(
+def ensure_opg_approval(
     self,
     min_allowance: float,
     approve_amount: Optional[float] = None
@@ -206,7 +206,7 @@ multiple service restarts without re-approving.
 
 Best for backend servers that call this on startup::
 
-    llm.ensure_opg_allowance(min_allowance=5.0, approve_amount=100.0)
+    llm.ensure_opg_approval(min_allowance=5.0, approve_amount=100.0)
 
 **Arguments**
 
diff --git a/docs/opengradient/index.md b/docs/opengradient/index.md
@@ -35,7 +35,7 @@ import opengradient as og
 llm = og.LLM(private_key="0x...")
 
 # One-time OPG token approval (idempotent -- skips if allowance is sufficient)
-llm.ensure_opg_allowance(min_allowance=5)
+llm.ensure_opg_approval(min_allowance=5)
 
 # Chat with an LLM (TEE-verified)
 response = asyncio.run(llm.chat(
diff --git a/examples/langchain_react_agent.py b/examples/langchain_react_agent.py
@@ -19,7 +19,7 @@
 
 # Ensure sufficient OPG allowance for Permit2 spending
 llm_client = og.LLM(private_key=private_key)
-llm_client.ensure_opg_allowance(min_allowance=5)
+llm_client.ensure_opg_approval(min_allowance=5)
 
 # Create the OpenGradient LangChain adapter
 llm = og.agents.langchain_adapter(
diff --git a/examples/llm_chat.py b/examples/llm_chat.py
@@ -10,7 +10,7 @@
 
 async def main():
     llm = og.LLM(private_key=os.environ.get("OG_PRIVATE_KEY"))
-    llm.ensure_opg_allowance(min_allowance=0.1)
+    llm.ensure_opg_approval(min_allowance=0.1)
 
     messages = [
         {"role": "user", "content": "What is the capital of France?"},
diff --git a/examples/llm_chat_streaming.py b/examples/llm_chat_streaming.py
@@ -6,7 +6,7 @@
 
 async def main():
     llm = og.LLM(private_key=os.environ.get("OG_PRIVATE_KEY"))
-    llm.ensure_opg_allowance(min_allowance=0.1)
+    llm.ensure_opg_approval(min_allowance=0.1)
 
     messages = [
         {"role": "user", "content": "What is Python?"},
diff --git a/examples/llm_tool_calling.py b/examples/llm_tool_calling.py
@@ -51,7 +51,7 @@ async def main():
     ]
 
     # Ensure sufficient OPG allowance for Permit2 spending
-    llm.ensure_opg_allowance(min_allowance=0.1)
+    llm.ensure_opg_approval(min_allowance=0.1)
 
     print("Testing Gemini tool calls...")
     print(f"Model: {og.TEE_LLM.GEMINI_2_5_FLASH_LITE}")
diff --git a/integrationtest/llm/test_llm_chat.py b/integrationtest/llm/test_llm_chat.py
@@ -99,7 +99,7 @@ def llm_client():
     print("Account funded with ETH and OPG")
 
     llm = og.LLM(private_key=account.key.hex())
-    llm.ensure_opg_allowance(min_allowance=OPG_FUND_AMOUNT, approve_amount=OPG_FUND_AMOUNT)
+    llm.ensure_opg_approval(min_allowance=OPG_FUND_AMOUNT, approve_amount=OPG_FUND_AMOUNT)
     print("Permit2 approval complete")
 
     # Wait for the approval to propagate on-chain
diff --git a/src/opengradient/__init__.py b/src/opengradient/__init__.py
@@ -26,7 +26,7 @@
 llm = og.LLM(private_key="0x...")
 
 # One-time OPG token approval (idempotent -- skips if allowance is sufficient)
-llm.ensure_opg_allowance(min_allowance=5)
+llm.ensure_opg_approval(min_allowance=5)
 
 # Chat with an LLM (TEE-verified)
 response = asyncio.run(llm.chat(
diff --git a/src/opengradient/client/__init__.py b/src/opengradient/client/__init__.py
@@ -17,7 +17,7 @@
 
 # LLM inference (Base Sepolia OPG tokens)
 llm = og.LLM(private_key="0x...")
-llm.ensure_opg_allowance(min_allowance=5)
+llm.ensure_opg_approval(min_allowance=5)
 result = await llm.chat(model=og.TEE_LLM.CLAUDE_HAIKU_4_5, messages=[...])
 
 # On-chain model inference (OpenGradient testnet gas tokens)
diff --git a/src/opengradient/client/llm.py b/src/opengradient/client/llm.py
@@ -15,7 +15,7 @@
 from x402.mechanisms.evm.upto.register import register_upto_evm_client
 
 from ..types import TEE_LLM, StreamChoice, StreamChunk, StreamDelta, TextGenerationOutput, x402SettlementMode
-from .opg_token import Permit2ApprovalResult, ensure_opg_allowance
+from .opg_token import Permit2ApprovalResult, ensure_opg_approval
 from .tee_connection import RegistryTEEConnection, StaticTEEConnection, TEEConnectionInterface
 from .tee_registry import TEERegistry
 
@@ -58,7 +58,7 @@ class LLM:
     All request methods (``chat``, ``completion``) are async.
 
     Before making LLM requests, ensure your wallet has approved sufficient
-    OPG tokens for Permit2 spending by calling ``ensure_opg_allowance``.
+    OPG tokens for Permit2 spending by calling ``ensure_opg_approval``.
 
     Usage:
         # Via on-chain registry (default)
@@ -68,7 +68,7 @@ class LLM:
         llm = og.LLM.from_url(private_key="0x...", llm_server_url="https://1.2.3.4")
 
         # Ensure sufficient OPG allowance (only sends tx when below threshold)
-        llm.ensure_opg_allowance(min_allowance=5)
+        llm.ensure_opg_approval(min_allowance=5)
 
         result = await llm.chat(model=TEE_LLM.CLAUDE_HAIKU_4_5, messages=[...])
         result = await llm.completion(model=TEE_LLM.CLAUDE_HAIKU_4_5, prompt="Hello")
@@ -182,7 +182,7 @@ async def _call_with_tee_retry(
 
     # ── Public API ──────────────────────────────────────────────────────
 
-    def ensure_opg_allowance(
+    def ensure_opg_approval(
         self,
         min_allowance: float,
         approve_amount: Optional[float] = None,
@@ -196,7 +196,7 @@ def ensure_opg_allowance(
 
         Best for backend servers that call this on startup::
 
-            llm.ensure_opg_allowance(min_allowance=5.0, approve_amount=100.0)
+            llm.ensure_opg_approval(min_allowance=5.0, approve_amount=100.0)
 
         Args:
             min_allowance: The minimum acceptable allowance in OPG. Must be
@@ -217,7 +217,7 @@ def ensure_opg_allowance(
         """
         if min_allowance < 0.1:
             raise ValueError("min_allowance must be at least 0.1.")
-        return ensure_opg_allowance(self._wallet_account, min_allowance, approve_amount)
+        return ensure_opg_approval(self._wallet_account, min_allowance, approve_amount)
 
     async def completion(
         self,
diff --git a/src/opengradient/client/opg_token.py b/src/opengradient/client/opg_token.py
@@ -144,7 +144,7 @@ def _get_web3_and_contract():
     return w3, token, spender
 
 
-def ensure_opg_allowance(
+def ensure_opg_approval(
     wallet_account: LocalAccount,
     min_allowance: float,
     approve_amount: Optional[float] = None,
@@ -160,7 +160,7 @@ def ensure_opg_allowance(
 
         # On startup — only sends a tx when allowance < 5 OPG,
         # then approves 100 OPG so subsequent restarts are free.
-        result = ensure_opg_allowance(wallet, min_allowance=5.0, approve_amount=100.0)
+        result = ensure_opg_approval(wallet, min_allowance=5.0, approve_amount=100.0)
 
     Args:
         wallet_account: The wallet account to check and approve from.
diff --git a/stresstest/llm.py b/stresstest/llm.py
@@ -13,7 +13,7 @@
 
 async def main(private_key: str):
     llm = og.LLM(private_key=private_key)
-    llm.ensure_opg_allowance(min_allowance=0.1)
+    llm.ensure_opg_approval(min_allowance=0.1)
 
     async def run_prompt(prompt: str):
         messages = [{"role": "user", "content": prompt}]
diff --git a/tests/opg_token_test.py b/tests/opg_token_test.py
@@ -4,7 +4,7 @@
 import pytest
 
 from opengradient.client.opg_token import (
-    ensure_opg_allowance,
+    ensure_opg_approval,
 )
 
 OWNER_ADDRESS = "0x1234567890abcdef1234567890ABCDEF12345678"
@@ -70,7 +70,7 @@ def _setup_approval_mocks(mock_web3, mock_wallet, contract):
     return approve_fn, tx_hash
 
 
-# ── ensure_opg_allowance tests ──────────────────────────────────────
+# ── ensure_opg_approval tests ──────────────────────────────────────
 
 
 class TestEnsureOpgAllowanceSkips:
@@ -81,7 +81,7 @@ def test_above_minimum_skips(self, mock_wallet, mock_web3):
         min_base = int(5.0 * 10**18)
         _setup_allowance(mock_web3, min_base * 2)
 
-        result = ensure_opg_allowance(mock_wallet, min_allowance=5.0)
+        result = ensure_opg_approval(mock_wallet, min_allowance=5.0)
 
         assert result.tx_hash is None
         assert result.allowance_before == min_base * 2
@@ -91,7 +91,7 @@ def test_exact_minimum_skips(self, mock_wallet, mock_web3):
         min_base = int(5.0 * 10**18)
         _setup_allowance(mock_web3, min_base)
 
-        result = ensure_opg_allowance(mock_wallet, min_allowance=5.0)
+        result = ensure_opg_approval(mock_wallet, min_allowance=5.0)
 
         assert result.tx_hash is None
 
@@ -107,7 +107,7 @@ def test_approves_default_with_greater_amount(self, mock_wallet, mock_web3):
         approve_base = int(10.0 * 10**18)  # 2x of 5.0
         contract.functions.allowance.return_value.call.side_effect = [0, 0, approve_base]
 
-        result = ensure_opg_allowance(mock_wallet, min_allowance=5.0)
+        result = ensure_opg_approval(mock_wallet, min_allowance=5.0)
 
         assert result.tx_hash == "0xabc123"
         # Verify approve was called with 10x amount
@@ -122,7 +122,7 @@ def test_approves_custom_amount(self, mock_wallet, mock_web3):
         approve_base = int(100.0 * 10**18)
         contract.functions.allowance.return_value.call.side_effect = [0, 0, approve_base]
 
-        result = ensure_opg_allowance(mock_wallet, min_allowance=5.0, approve_amount=100.0)
+        result = ensure_opg_approval(mock_wallet, min_allowance=5.0, approve_amount=100.0)
 
         assert result.tx_hash == "0xabc123"
         args = contract.functions.approve.call_args[0]
@@ -135,7 +135,7 @@ def test_no_tx_on_restart_when_above_min(self, mock_wallet, mock_web3):
         remaining = int(60.0 * 10**18)
         _setup_allowance(mock_web3, remaining)
 
-        result = ensure_opg_allowance(mock_wallet, min_allowance=5.0, approve_amount=100.0)
+        result = ensure_opg_approval(mock_wallet, min_allowance=5.0, approve_amount=100.0)
 
         assert result.tx_hash is None
         assert result.allowance_before == remaining
@@ -153,7 +153,7 @@ def test_approve_amount_capped_to_balance(self, mock_wallet, mock_web3):
         # allowance calls: 1st for the check, 2nd in _send_approve_tx before, 3rd in post-tx poll
         contract.functions.allowance.return_value.call.side_effect = [0, 0, balance]
 
-        result = ensure_opg_allowance(mock_wallet, min_allowance=0.1)
+        result = ensure_opg_approval(mock_wallet, min_allowance=0.1)
 
         # Default approve_amount would be 0.2, but balance is only 0.1 — capped
         args = contract.functions.approve.call_args[0]
@@ -165,7 +165,7 @@ def test_zero_balance_raises(self, mock_wallet, mock_web3):
         _setup_allowance(mock_web3, 0, balance=0)
 
         with pytest.raises(ValueError, match="has no OPG tokens"):
-            ensure_opg_allowance(mock_wallet, min_allowance=0.1)
+            ensure_opg_approval(mock_wallet, min_allowance=0.1)
 
     def test_no_cap_when_balance_sufficient(self, mock_wallet, mock_web3):
         """When balance >= approve_amount, no capping occurs."""
@@ -176,7 +176,7 @@ def test_no_cap_when_balance_sufficient(self, mock_wallet, mock_web3):
 
         contract.functions.allowance.return_value.call.side_effect = [0, 0, approve_base]
 
-        result = ensure_opg_allowance(mock_wallet, min_allowance=0.1)
+        result = ensure_opg_approval(mock_wallet, min_allowance=0.1)
 
         args = contract.functions.approve.call_args[0]
         assert args[1] == approve_base
@@ -190,7 +190,7 @@ def test_approve_amount_less_than_min_raises(self, mock_wallet, mock_web3):
         _setup_allowance(mock_web3, 0)
 
         with pytest.raises(ValueError, match="approve_amount.*must be >= min_allowance"):
-            ensure_opg_allowance(mock_wallet, min_allowance=10.0, approve_amount=5.0)
+            ensure_opg_approval(mock_wallet, min_allowance=10.0, approve_amount=5.0)
 
 
 class TestAmountConversion:
@@ -201,7 +201,7 @@ def test_fractional_amount(self, mock_wallet, mock_web3):
         expected_base = int(0.5 * 10**18)
         _setup_allowance(mock_web3, expected_base)
 
-        result = ensure_opg_allowance(mock_wallet, min_allowance=0.5)
+        result = ensure_opg_approval(mock_wallet, min_allowance=0.5)
 
         assert result.allowance_before == expected_base
         assert result.tx_hash is None
@@ -211,7 +211,7 @@ def test_large_amount(self, mock_wallet, mock_web3):
         expected_base = int(1000.0 * 10**18)
         _setup_allowance(mock_web3, expected_base)
 
-        result = ensure_opg_allowance(mock_wallet, min_allowance=1000.0)
+        result = ensure_opg_approval(mock_wallet, min_allowance=1000.0)
 
         assert result.allowance_before == expected_base
         assert result.tx_hash is None
diff --git a/tutorials/01-verifiable-ai-agent.md b/tutorials/01-verifiable-ai-agent.md
@@ -36,7 +36,7 @@ export OG_PRIVATE_KEY="0x..."
 ## Step 1: Initialize and Create the LangChain Adapter
 
 Before making any LLM calls, you need to approve OPG token spending for the x402
-payment protocol. The `ensure_opg_allowance` method checks your wallet's current
+payment protocol. The `ensure_opg_approval` method checks your wallet's current
 Permit2 allowance and only sends an on-chain transaction if the allowance drops
 below the threshold -- so it's safe to call every time.
 
@@ -48,7 +48,7 @@ private_key = os.environ["OG_PRIVATE_KEY"]
 
 # Ensure sufficient OPG allowance for x402 payments (only sends tx when below threshold).
 llm_client = og.LLM(private_key=private_key)
-llm_client.ensure_opg_allowance(min_allowance=5)
+llm_client.ensure_opg_approval(min_allowance=5)
 
 # Create the LangChain chat model backed by OpenGradient TEE.
 # The adapter creates its own internal LLM client. The approval above applies
@@ -289,7 +289,7 @@ private_key = os.environ["OG_PRIVATE_KEY"]
 
 # Ensure sufficient OPG allowance for x402 payments (only sends tx when below threshold).
 llm_client = og.LLM(private_key=private_key)
-llm_client.ensure_opg_allowance(min_allowance=5)
+llm_client.ensure_opg_approval(min_allowance=5)
 
 # Alpha client for on-chain model inference.
 alpha = og.Alpha(private_key=private_key)
diff --git a/tutorials/02-streaming-multi-provider.md b/tutorials/02-streaming-multi-provider.md
@@ -36,7 +36,7 @@ export OG_PRIVATE_KEY="0x..."
 
 Start with the simplest possible call -- send a message and get a response. Before
 making any LLM calls, ensure sufficient OPG token allowance for the x402 payment
-protocol using `ensure_opg_allowance`. This only sends a transaction when the
+protocol using `ensure_opg_approval`. This only sends a transaction when the
 current allowance drops below the threshold.
 
 ```python
@@ -53,7 +53,7 @@ if not private_key:
 llm = og.LLM(private_key=private_key)
 
 # Ensure sufficient OPG allowance for x402 payments (only sends tx when below threshold).
-llm.ensure_opg_allowance(min_allowance=5)
+llm.ensure_opg_approval(min_allowance=5)
 
 async def main():
     result = await llm.chat(
@@ -274,7 +274,7 @@ if not private_key:
 llm = og.LLM(private_key=private_key)
 
 # Ensure sufficient OPG allowance for x402 payments (only sends tx when below threshold).
-llm.ensure_opg_allowance(min_allowance=5)
+llm.ensure_opg_approval(min_allowance=5)
 
 PROMPT = "Explain what a Trusted Execution Environment is in two sentences."
 
diff --git a/tutorials/03-verified-tool-calling.md b/tutorials/03-verified-tool-calling.md

Original file line number	Diff line number	Diff line change
`@@ -51,7 +51,7 @@ async def main():`
`51`	`51`	`]`
`52`	`52`
`53`	`53`	`# Ensure sufficient OPG allowance for Permit2 spending`
`54`		`- llm.ensure_opg_allowance(min_allowance=0.1)`
	`54`	`+ llm.ensure_opg_approval(min_allowance=0.1)`
`55`	`55`
`56`	`56`	`print("Testing Gemini tool calls...")`
`57`	`57`	`print(f"Model: {og.TEE_LLM.GEMINI_2_5_FLASH_LITE}")`