atp-protocol 1.0.0__py3-none-any.whl → 1.0.1__py3-none-any.whl

atp/__init__.py

@@ -1,11 +1,8 @@
-"""ATP Protocol package."""
-
-from atp.middleware import (
-    ATPSettlementMiddleware,
-    create_settlement_middleware,
-)
+from atp.middleware import ATPSettlementMiddleware, create_settlement_middleware
+from atp.settlement_client import SettlementServiceClient
 
 __all__ = [
     "ATPSettlementMiddleware",
     "create_settlement_middleware",
-]
+    "SettlementServiceClient",
+]

atp/config.py

@@ -73,5 +73,5 @@ ATP_SOLANA_DEBUG = _bool_env("ATP_SOLANA_DEBUG", default=False)
 
 # Settlement Service URL
 ATP_SETTLEMENT_URL = os.getenv(
-    "ATP_SETTLEMENT_URL", "http://localhost:8001"
+    "ATP_SETTLEMENT_URL", "https://facilitator.swarms.world"
 )

atp/middleware.py

@@ -67,8 +67,6 @@ class ATPSettlementMiddleware(BaseHTTPMiddleware):
         recipient_pubkey: Optional[str] = None,
         skip_preflight: bool = False,
         commitment: str = "confirmed",
-        usage_response_key: str = "usage",
-        include_usage_in_response: bool = True,
         require_wallet: bool = True,
         settlement_service_url: Optional[str] = None,
     ):
@@ -92,8 +90,6 @@ class ATPSettlementMiddleware(BaseHTTPMiddleware):
                 This wallet receives the main payment (after processing fee). Required.
             skip_preflight: Whether to skip preflight simulation for Solana transactions.
             commitment: Solana commitment level (processed|confirmed|finalized).
-            usage_response_key: Key in response JSON where usage data is located (default: "usage").
-            include_usage_in_response: Whether to add usage/cost info to the response.
             require_wallet: Whether to require wallet private key (if False, skips settlement when missing).
             settlement_service_url: Base URL of the settlement service. If not provided, uses
                 ATP_SETTLEMENT_URL environment variable (default: http://localhost:8001).
@@ -119,8 +115,6 @@ class ATPSettlementMiddleware(BaseHTTPMiddleware):
             )
         self.skip_preflight = skip_preflight
         self.commitment = commitment
-        self.usage_response_key = usage_response_key
-        self.include_usage_in_response = include_usage_in_response
         self.require_wallet = require_wallet
         # Always use settlement service - initialize client with config value or provided URL
         service_url = (
@@ -146,10 +140,9 @@ class ATPSettlementMiddleware(BaseHTTPMiddleware):
         """
         Extract usage information from response body.
 
-        Tries multiple strategies:
-        1. Look for usage data at the configured usage_response_key
-        2. Check if the entire response contains usage-like keys
-        3. Try nested structures (usage.usage, meta.usage, etc.)
+        Automatically detects usage data using multiple strategies:
+        1. Check if the entire response contains usage-like keys
+        2. Try nested structures with common usage key names (usage, token_usage, etc.)
 
         The usage data is then sent to the settlement service for parsing,
         so we just need to extract the raw usage object.
@@ -160,12 +153,7 @@ class ATPSettlementMiddleware(BaseHTTPMiddleware):
                 return None
             data = json.loads(body_str)
 
-            # Strategy 1: Try the configured usage key first
-            usage = data.get(self.usage_response_key)
-            if usage and isinstance(usage, dict):
-                return usage
-
-            # Strategy 2: Check if the entire response is usage-like
+            # Strategy 1: Check if the entire response is usage-like
             if isinstance(data, dict):
                 # Check for common usage keys at top level
                 usage_keys = [
@@ -182,7 +170,7 @@ class ATPSettlementMiddleware(BaseHTTPMiddleware):
                 if any(key in data for key in usage_keys):
                     return data
 
-            # Strategy 3: Try nested structures
+            # Strategy 2: Try nested structures
             # Check for usage nested in common locations
             for nested_key in [
                 "usage",
@@ -284,21 +272,20 @@ class ATPSettlementMiddleware(BaseHTTPMiddleware):
                 detail=f"Settlement failed: {str(e)}",
             )
 
-        # Modify response to include usage/payment info if requested
-        if self.include_usage_in_response:
-            try:
-                response_data = json.loads(
-                    response_body.decode("utf-8")
-                )
-                response_data["atp_settlement"] = payment_result
-                response_data["atp_usage"] = usage
-                response_body = json.dumps(response_data).encode(
-                    "utf-8"
-                )
-            except Exception as e:
-                logger.warning(
-                    f"Failed to add settlement info to response: {e}"
-                )
+        # Always include settlement information in the response
+        try:
+            response_data = json.loads(
+                response_body.decode("utf-8")
+            )
+            response_data["atp_settlement"] = payment_result
+            response_data["atp_usage"] = usage
+            response_body = json.dumps(response_data).encode(
+                "utf-8"
+            )
+        except Exception as e:
+            logger.warning(
+                f"Failed to add settlement info to response: {e}"
+            )
 
         return Response(
             content=response_body,

{atp_protocol-1.0.0.dist-info → atp_protocol-1.0.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: atp-protocol
-Version: 1.0.0
+Version: 1.0.1
 Summary: ATP Protocol - Agentic Trade Protocol. The easiest way to enable agent-to-agent payments powered by Solana.
 License: MIT
 Keywords: atp,atp-protocol,solana,blockchain,cryptocurrency,payment-gated,agent-to-agent,agent payments,agent execution,payment settlement,solana payments,usdc,sol,fastapi,agent api,payment gateway,settlement service,agent marketplace,decentralized payments,web3,crypto payments,agent infrastructure,payment middleware,automated settlement
@@ -17,6 +17,7 @@ Classifier: Programming Language :: Python :: 3.13
 Classifier: Topic :: Internet :: WWW/HTTP :: HTTP Servers
 Classifier: Topic :: Office/Business :: Financial
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Requires-Dist: fastapi
 Requires-Dist: httpx
 Requires-Dist: loguru
 Requires-Dist: pydantic
@@ -24,6 +25,8 @@ Requires-Dist: python-dotenv
 Requires-Dist: setuptools
 Requires-Dist: solana
 Requires-Dist: solders
+Requires-Dist: starlette
+Requires-Dist: swarms
 Project-URL: Documentation, https://github.com/The-Swarm-Corporation/ATP-Protocol
 Project-URL: Homepage, https://github.com/The-Swarm-Corporation/ATP-Protocol
 Project-URL: Repository, https://github.com/The-Swarm-Corporation/ATP-Protocol
@@ -303,8 +306,6 @@ app.add_middleware(
     recipient_pubkey="YourPublicKeyHere",  # Required: endpoint host receives payment
     skip_preflight=False,  # Skip Solana transaction preflight simulation
     commitment="confirmed",  # Solana commitment level
-    usage_response_key="usage",  # Key in response where usage data is located
-    include_usage_in_response=True,  # Add usage/cost info to response
     require_wallet=True,  # Require wallet key (if False, skips settlement when missing)
 )
 ```
@@ -319,8 +320,6 @@ app.add_middleware(
 - **`recipient_pubkey`** (required): Solana public key of the recipient wallet (endpoint host receives main payment)
 - **`skip_preflight`** (default: `False`): Whether to skip preflight simulation for Solana transactions
 - **`commitment`** (default: `"confirmed"`): Solana commitment level (`processed`|`confirmed`|`finalized`)
-- **`usage_response_key`** (default: `"usage"`): Key in response JSON where usage data is located
-- **`include_usage_in_response`** (default: `True`): Whether to add usage/cost info to the response
 - **`require_wallet`** (default: `True`): Whether to require wallet private key (if False, skips settlement when missing)
 
 ### Usage Example
@@ -409,10 +408,9 @@ The middleware automatically detects and parses usage data from multiple API for
 - **Generic**: `input_tokens`, `output_tokens`, `total_tokens`
 - **Nested**: `usage.prompt_tokens`, `meta.usage`, `statistics`, etc.
 
-The middleware searches for usage data in this order:
-1. The configured `usage_response_key` (default: `"usage"`)
-2. Top-level keys in the response
-3. Nested structures (`usage`, `meta.usage`, `statistics`, etc.)
+The middleware automatically detects usage data by:
+1. Checking if the entire response contains usage-like keys at the top level
+2. Searching nested structures with common usage key names (`usage`, `token_usage`, `meta`, `statistics`, etc.)
 
 ### Payment Flow
 

atp_protocol-1.0.1.dist-info/RECORD

@@ -0,0 +1,9 @@
+atp/__init__.py,sha256=fpKQQCi9HQEj4t2Rs5ng5qmp0KwGV6bZIl1tnS-3HQ8,251
+atp/config.py,sha256=lYERbrySRTuyS_jeaAnLHYuBEQNLqaKLQTy_nEFP7lY,2041
+atp/middleware.py,sha256=GlRBXEba_hlGMPi69LAGUJSQsKfmwSnAENKZQ9QcXhU,13185
+atp/schemas.py,sha256=iASVBPpAhrO-LDXs2UC9ABtfV1oDYsu4kZcz3dVeJNA,6220
+atp/settlement_client.py,sha256=jfvhxDqp2kDISWG7tjX5s88goAPxp-lMXvuJtpMyOys,6742
+atp_protocol-1.0.1.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+atp_protocol-1.0.1.dist-info/METADATA,sha256=BdbWCKlFe31u-00ijy9aKpoIcSIpEwHvjJVS5n3GumA,18192
+atp_protocol-1.0.1.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+atp_protocol-1.0.1.dist-info/RECORD,,