import asyncio
import json
import requests
import websockets

API_KEY = "ps_live_..."
BASE = "https://api.polysimulator.com/v1"

# Mint WS token
ws_token = requests.post(
    f"{BASE}/keys/ws-token",
    headers={"X-API-Key": API_KEY},
).json()["token"]

async def stream_prices():
    async with websockets.connect(
        f"wss://api.polysimulator.com/v1/ws/prices?token={ws_token}"
    ) as ws:
        # Subscribe to markets
        await ws.send(json.dumps({
            "action": "subscribe",
            "markets": ["0xabc123...", "0xdef456..."],
        }))

        async for raw in ws:
            msg = json.loads(raw)
            if msg["type"] == "price":
                mid = msg["market_id"][:16]
                buy = msg["buy"]
                print(f"{mid}... → Yes: {buy}")

asyncio.run(stream_prices())

import json
import requests
import websocket

API_KEY = "ps_live_..."
BASE = "https://api.polysimulator.com/v1"
WS_URL = "wss://api.polysimulator.com/v1"

token = requests.post(
    f"{BASE}/keys/ws-token",
    headers={"X-API-Key": API_KEY},
).json()["token"]

def on_message(ws, message):
    data = json.loads(message)
    if data["type"] == "price":
        print(f"{data['market_id'][:16]}... → Yes: {data['buy']}")
    elif data["type"] == "connected":
        ws.send(json.dumps({
            "action": "subscribe",
            "markets": ["0xabc123...", "0xdef456..."],
        }))

ws = websocket.WebSocketApp(
    f"{WS_URL}/ws/prices?token={token}",
    on_message=on_message,
)
ws.run_forever()

const API_KEY = "ps_live_...";
const BASE = "https://api.polysimulator.com/v1";

// Mint WS token
const tokenResp = await fetch(`${BASE}/keys/ws-token`, {
  method: "POST",
  headers: { "X-API-Key": API_KEY },
});
const { token } = await tokenResp.json();

const ws = new WebSocket(
  `wss://api.polysimulator.com/v1/ws/prices?token=${token}`
);

ws.onopen = () => {
  ws.send(JSON.stringify({
    action: "subscribe",
    markets: ["0xabc123...", "0xdef456..."],
  }));
};

ws.onmessage = (event) => {
  const msg = JSON.parse(event.data);
  if (msg.type === "price") {
    console.log(`${msg.market_id}: Yes=${msg.buy}`);
  }
};