Gas Optimization API for AI Agents

import requests
import time
from collections import deque

PF_BASE = "https://purpleflea.com/api"
HEADERS = {"Authorization": "Bearer YOUR_API_KEY"}

# Pending trade queue for batching
pending_trades = deque()

def get_current_gas_gwei(chain="ethereum"):
    """Fetch current base fee in gwei"""
    resp = requests.get(
        f"{PF_BASE}/v1/gas/estimate?chain={chain}",
        headers=HEADERS, timeout=5
    )
    return resp.json()["base_fee_gwei"]

def execute_trade_gas_aware(trade_params, max_gas_gwei=20):
    """Execute trade only when gas is below threshold"""
    current_gwei = get_current_gas_gwei()

    if current_gwei > max_gas_gwei:
        # Queue for off-peak execution
        pending_trades.append(trade_params)
        return {
            "status": "queued",
            "current_gas": current_gwei,
            "threshold": max_gas_gwei,
            "queue_depth": len(pending_trades)
        }

    # If we have batched trades waiting, flush them together
    if len(pending_trades) >= 5:
        return flush_batch()

    # Execute single trade with optimized gas
    return requests.post(
        f"{PF_BASE}/v1/gas/optimize",
        json=trade_params,
        headers=HEADERS
    ).json()

def flush_batch():
    """Batch all pending trades into a single transaction"""
    trades = list(pending_trades)
    pending_trades.clear()
    result = requests.post(
        f"{PF_BASE}/v1/gas/batch",
        json={"transactions": trades},
        headers=HEADERS
    ).json()
    print(f"Batched {len(trades)} trades — saved {result['savings_usd']:.2f} USD in gas")
    return result

# Main agent loop
def agent_loop():
    while True:
        trade = generate_next_trade()  # your strategy logic
        result = execute_trade_gas_aware(trade, max_gas_gwei=18)
        print(f"Trade result: {result['status']}")
        time.sleep(60)  # check every minute