Purple Flea for NVIDIA NIM

import openai
import httpx
import json

# NVIDIA NIM is OpenAI-compatible — point the client at your NIM endpoint
nim_client = openai.OpenAI(
    base_url="https://integrate.api.nvidia.com/v1",
    api_key="YOUR_NIM_API_KEY",
)

PF_API_KEY = "pf_live_your_key_here"
PF_BASE    = "https://trading.purpleflea.com"

# Define Purple Flea tools as NIM function schemas
tools = [
    {
        "type": "function",
        "function": {
            "name": "purpleflea_trade",
            "description": "Place a perpetual futures trade on Hyperliquid via Purple Flea",
            "parameters": {
                "type": "object",
                "properties": {
                    "market":     {"type": "string", "description": "e.g. BTC-USD-PERP"},
                    "side":       {"type": "string", "enum": ["long", "short"]},
                    "size_usd":   {"type": "number", "description": "Notional in USD"},
                    "leverage":   {"type": "integer", "minimum": 1, "maximum": 50},
                    "order_type": {"type": "string", "enum": ["market", "limit"]}
                },
                "required": ["market", "side", "size_usd"]
            }
        }
    },
    {
        "type": "function",
        "function": {
            "name": "purpleflea_wallet_balance",
            "description": "Get the current balance of the agent's Purple Flea wallet",
            "parameters": {"type": "object", "properties": {}, "required": []}
        }
    },
    {
        "type": "function",
        "function": {
            "name": "purpleflea_casino_bet",
            "description": "Place a provably fair bet on crash, coin flip, or dice",
            "parameters": {
                "type": "object",
                "properties": {
                    "game":      {"type": "string", "enum": ["crash", "coinflip", "dice"]},
                    "amount":    {"type": "number", "description": "Bet amount in USDC"},
                    "cashout_at":{"type": "number", "description": "Auto cashout multiplier (crash only)"}
                },
                "required": ["game", "amount"]
            }
        }
    }
]

def execute_tool(name: str, args: dict) -> dict:
    """Dispatch NIM tool call to the appropriate Purple Flea endpoint."""
    headers = {"Authorization": f"Bearer {PF_API_KEY}", "Content-Type": "application/json"}
    if name == "purpleflea_trade":
        r = httpx.post(f"{PF_BASE}/trade", json=args, headers=headers)
    elif name == "purpleflea_wallet_balance":
        r = httpx.get("https://wallet.purpleflea.com/balance", headers=headers)
    elif name == "purpleflea_casino_bet":
        r = httpx.post("https://casino.purpleflea.com/bet", json=args, headers=headers)
    else:
        return {"error": f"unknown tool: {name}"}
    return r.json()

# Run the agent loop
messages = [{"role": "user", "content": "Check my balance, then go long 50 USD BTC at 2x."}]

response = nim_client.chat.completions.create(
    model="meta/llama-3.3-70b-instruct",   # any NIM model that supports tool use
    messages=messages,
    tools=tools,
    tool_choice="auto",
)

for tool_call in (response.choices[0].message.tool_calls or []):
    fn_name = tool_call.function.name
    fn_args = json.loads(tool_call.function.arguments)
    result  = execute_tool(fn_name, fn_args)
    print(f"{fn_name} → {result}")