pmxt-core 2.24.0 → 2.25.0

package/dist/exchanges/polymarket_us/websocket.test.js

@@ -0,0 +1,162 @@
+"use strict";
+/**
+ * Unit tests for PolymarketUSWebSocket.
+ *
+ * The polymarket-us SDK is fully stubbed here so the tests run without
+ * touching the network. A fake MarketsWebSocket captures subscriptions
+ * and lets each test fire synthetic messages through a listener map.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+const websocket_1 = require("./websocket");
+const normalizer_1 = require("./normalizer");
+class FakeMarketsWebSocket {
+    listeners = new Map();
+    connectCalls = 0;
+    closeCalls = 0;
+    subscribeMarketDataCalls = [];
+    subscribeTradesCalls = [];
+    on(event, listener) {
+        const arr = this.listeners.get(event) ?? [];
+        arr.push(listener);
+        this.listeners.set(event, arr);
+        return this;
+    }
+    async connect() {
+        this.connectCalls += 1;
+    }
+    close() {
+        this.closeCalls += 1;
+    }
+    subscribeMarketData(requestId, slugs) {
+        this.subscribeMarketDataCalls.push({ requestId, slugs });
+    }
+    subscribeTrades(requestId, slugs) {
+        this.subscribeTradesCalls.push({ requestId, slugs });
+    }
+    emit(event, arg) {
+        const arr = this.listeners.get(event) ?? [];
+        for (const listener of arr)
+            listener(arg);
+    }
+}
+function makeClient(fakeSocket) {
+    return {
+        ws: {
+            markets: () => fakeSocket,
+        },
+    };
+}
+describe('PolymarketUSWebSocket', () => {
+    let fake;
+    let ws;
+    beforeEach(() => {
+        fake = new FakeMarketsWebSocket();
+        ws = new websocket_1.PolymarketUSWebSocket(makeClient(fake), new normalizer_1.PolymarketUSNormalizer());
+    });
+    describe('watchOrderBook', () => {
+        it('connects lazily, subscribes once per slug, and resolves on marketData', async () => {
+            const pending = ws.watchOrderBook('btc-100k');
+            // Let the ensureInitialized microtask run so connect resolves
+            await new Promise(r => setImmediate(r));
+            expect(fake.connectCalls).toBe(1);
+            expect(fake.subscribeMarketDataCalls).toEqual([
+                { requestId: 'book:btc-100k', slugs: ['btc-100k'] },
+            ]);
+            fake.emit('marketData', {
+                requestId: 'book:btc-100k',
+                subscriptionType: 'SUBSCRIPTION_TYPE_MARKET_DATA',
+                marketData: {
+                    marketSlug: 'btc-100k',
+                    bids: [{ px: { value: '0.55', currency: 'USD' }, qty: '10' }],
+                    offers: [{ px: { value: '0.57', currency: 'USD' }, qty: '5' }],
+                    state: 'MARKET_STATE_OPEN',
+                    transactTime: '2026-04-06T00:00:00Z',
+                },
+            });
+            const book = await pending;
+            expect(book.bids).toEqual([{ price: 0.55, size: 10 }]);
+            expect(book.asks).toEqual([{ price: 0.57, size: 5 }]);
+        });
+        it('strips :long suffix before subscribing', async () => {
+            const pending = ws.watchOrderBook('btc-100k:long');
+            await new Promise(r => setImmediate(r));
+            expect(fake.subscribeMarketDataCalls).toEqual([
+                { requestId: 'book:btc-100k', slugs: ['btc-100k'] },
+            ]);
+            fake.emit('marketData', {
+                marketData: {
+                    marketSlug: 'btc-100k',
+                    bids: [],
+                    offers: [],
+                    state: 'MARKET_STATE_OPEN',
+                },
+            });
+            await pending;
+        });
+        it('does not re-subscribe when called twice for the same slug', async () => {
+            const p1 = ws.watchOrderBook('btc-100k');
+            await new Promise(r => setImmediate(r));
+            const p2 = ws.watchOrderBook('btc-100k');
+            await new Promise(r => setImmediate(r));
+            expect(fake.subscribeMarketDataCalls).toHaveLength(1);
+            fake.emit('marketData', {
+                marketData: {
+                    marketSlug: 'btc-100k',
+                    bids: [],
+                    offers: [],
+                    state: 'MARKET_STATE_OPEN',
+                },
+            });
+            const [b1, b2] = await Promise.all([p1, p2]);
+            expect(b1.bids).toEqual([]);
+            expect(b2.bids).toEqual([]);
+        });
+    });
+    describe('watchTrades', () => {
+        it('resolves with a PMXT Trade on the next trade message', async () => {
+            const pending = ws.watchTrades('btc-100k');
+            await new Promise(r => setImmediate(r));
+            expect(fake.subscribeTradesCalls).toEqual([
+                { requestId: 'trade:btc-100k', slugs: ['btc-100k'] },
+            ]);
+            fake.emit('trade', {
+                requestId: 'trade:btc-100k',
+                subscriptionType: 'SUBSCRIPTION_TYPE_TRADE',
+                trade: {
+                    marketSlug: 'btc-100k',
+                    price: { value: '0.55', currency: 'USD' },
+                    quantity: { value: '10', currency: 'USD' },
+                    tradeTime: '2026-04-06T00:00:00Z',
+                    maker: { side: 'ORDER_SIDE_SELL', intent: 'ORDER_INTENT_SELL_LONG' },
+                    taker: { side: 'ORDER_SIDE_BUY', intent: 'ORDER_INTENT_BUY_LONG' },
+                },
+            });
+            const trades = await pending;
+            expect(trades).toHaveLength(1);
+            expect(trades[0].price).toBeCloseTo(0.55);
+            expect(trades[0].amount).toBeCloseTo(10);
+            expect(trades[0].side).toBe('buy');
+            expect(trades[0].timestamp).toBe(new Date('2026-04-06T00:00:00Z').getTime());
+        });
+    });
+    describe('close', () => {
+        it('rejects pending watchers and closes the socket', async () => {
+            const pending = ws.watchOrderBook('btc-100k');
+            await new Promise(r => setImmediate(r));
+            const pendingTrade = ws.watchTrades('btc-100k');
+            await new Promise(r => setImmediate(r));
+            await ws.close();
+            await expect(pending).rejects.toThrow('PolymarketUS WebSocket closed');
+            await expect(pendingTrade).rejects.toThrow('PolymarketUS WebSocket closed');
+            expect(fake.closeCalls).toBe(1);
+        });
+    });
+    describe('error handling', () => {
+        it('rejects pending watchers when the socket emits an error', async () => {
+            const pending = ws.watchOrderBook('btc-100k');
+            await new Promise(r => setImmediate(r));
+            fake.emit('error', new Error('boom'));
+            await expect(pending).rejects.toThrow('boom');
+        });
+    });
+});

package/dist/exchanges/probable/api.d.ts

@@ -1,6 +1,6 @@
 /**
  * Auto-generated from /home/runner/work/pmxt/pmxt/core/specs/probable/probable.yaml
- * Generated at: 2026-04-06T14:01:00.988Z
+ * Generated at: 2026-04-06T19:59:33.366Z
  * Do not edit manually -- run "npm run fetch:openapi" to regenerate.
  */
 export declare const probableApiSpec: {

package/dist/exchanges/probable/api.js

@@ -3,7 +3,7 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.probableApiSpec = void 0;
 /**
  * Auto-generated from /home/runner/work/pmxt/pmxt/core/specs/probable/probable.yaml
- * Generated at: 2026-04-06T14:01:00.988Z
+ * Generated at: 2026-04-06T19:59:33.366Z
  * Do not edit manually -- run "npm run fetch:openapi" to regenerate.
  */
 exports.probableApiSpec = {

package/dist/index.d.ts

@@ -13,6 +13,7 @@ export * from './exchanges/myriad';
 export * from './exchanges/opinion';
 export * from './exchanges/metaculus';
 export * from './exchanges/smarkets';
+export * from './exchanges/polymarket_us';
 export * from './server/app';
 export * from './server/utils/port-manager';
 export * from './server/utils/lock-file';
@@ -26,6 +27,7 @@ import { MyriadExchange } from './exchanges/myriad';
 import { OpinionExchange } from './exchanges/opinion';
 import { MetaculusExchange } from './exchanges/metaculus';
 import { SmarketsExchange } from './exchanges/smarkets';
+import { PolymarketUSExchange } from './exchanges/polymarket_us';
 declare const pmxt: {
     Polymarket: typeof PolymarketExchange;
     Limitless: typeof LimitlessExchange;
@@ -37,6 +39,7 @@ declare const pmxt: {
     Opinion: typeof OpinionExchange;
     Metaculus: typeof MetaculusExchange;
     Smarkets: typeof SmarketsExchange;
+    PolymarketUS: typeof PolymarketUSExchange;
 };
 export declare const Polymarket: typeof PolymarketExchange;
 export declare const Limitless: typeof LimitlessExchange;
@@ -48,4 +51,5 @@ export declare const Myriad: typeof MyriadExchange;
 export declare const Opinion: typeof OpinionExchange;
 export declare const Metaculus: typeof MetaculusExchange;
 export declare const Smarkets: typeof SmarketsExchange;
+export declare const PolymarketUS: typeof PolymarketUSExchange;
 export default pmxt;

package/dist/index.js

@@ -14,7 +14,7 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
     for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.Smarkets = exports.Metaculus = exports.Opinion = exports.Myriad = exports.Baozi = exports.Probable = exports.KalshiDemo = exports.Kalshi = exports.Limitless = exports.Polymarket = exports.parseOpenApiSpec = void 0;
+exports.PolymarketUS = exports.Smarkets = exports.Metaculus = exports.Opinion = exports.Myriad = exports.Baozi = exports.Probable = exports.KalshiDemo = exports.Kalshi = exports.Limitless = exports.Polymarket = exports.parseOpenApiSpec = void 0;
 __exportStar(require("./BaseExchange"), exports);
 __exportStar(require("./types"), exports);
 __exportStar(require("./utils/math"), exports);
@@ -31,6 +31,7 @@ __exportStar(require("./exchanges/myriad"), exports);
 __exportStar(require("./exchanges/opinion"), exports);
 __exportStar(require("./exchanges/metaculus"), exports);
 __exportStar(require("./exchanges/smarkets"), exports);
+__exportStar(require("./exchanges/polymarket_us"), exports);
 __exportStar(require("./server/app"), exports);
 __exportStar(require("./server/utils/port-manager"), exports);
 __exportStar(require("./server/utils/lock-file"), exports);
@@ -44,6 +45,7 @@ const myriad_1 = require("./exchanges/myriad");
 const opinion_1 = require("./exchanges/opinion");
 const metaculus_1 = require("./exchanges/metaculus");
 const smarkets_1 = require("./exchanges/smarkets");
+const polymarket_us_1 = require("./exchanges/polymarket_us");
 const pmxt = {
     Polymarket: polymarket_1.PolymarketExchange,
     Limitless: limitless_1.LimitlessExchange,
@@ -55,6 +57,7 @@ const pmxt = {
     Opinion: opinion_1.OpinionExchange,
     Metaculus: metaculus_1.MetaculusExchange,
     Smarkets: smarkets_1.SmarketsExchange,
+    PolymarketUS: polymarket_us_1.PolymarketUSExchange,
 };
 exports.Polymarket = polymarket_1.PolymarketExchange;
 exports.Limitless = limitless_1.LimitlessExchange;
@@ -66,4 +69,5 @@ exports.Myriad = myriad_1.MyriadExchange;
 exports.Opinion = opinion_1.OpinionExchange;
 exports.Metaculus = metaculus_1.MetaculusExchange;
 exports.Smarkets = smarkets_1.SmarketsExchange;
+exports.PolymarketUS = polymarket_us_1.PolymarketUSExchange;
 exports.default = pmxt;

package/dist/server/app.js

@@ -16,6 +16,7 @@ const myriad_1 = require("../exchanges/myriad");
 const opinion_1 = require("../exchanges/opinion");
 const metaculus_1 = require("../exchanges/metaculus");
 const smarkets_1 = require("../exchanges/smarkets");
+const polymarket_us_1 = require("../exchanges/polymarket_us");
 const errors_1 = require("../errors");
 // Singleton instances for local usage (when no credentials provided)
 const defaultExchanges = {
@@ -212,6 +213,11 @@ function createExchange(name, credentials) {
                 apiKey: credentials?.apiKey || process.env.SMARKETS_EMAIL,
                 privateKey: credentials?.privateKey || process.env.SMARKETS_PASSWORD,
             });
+        case "polymarket_us":
+            return new polymarket_us_1.PolymarketUSExchange({
+                apiKey: credentials?.apiKey || process.env.POLYMARKET_US_KEY_ID,
+                privateKey: credentials?.privateKey || process.env.POLYMARKET_US_SECRET_KEY,
+            });
         default:
             throw new Error(`Unknown exchange: ${name}`);
     }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pmxt-core",
-  "version": "2.24.0",
+  "version": "2.25.0",
   "description": "pmxt is a unified prediction market data API. The ccxt for prediction markets.",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -29,8 +29,8 @@
     "test": "jest -c jest.config.js",
     "server": "tsx watch src/server/index.ts",
     "server:prod": "node dist/server/index.js",
-    "generate:sdk:python": "npx @openapitools/openapi-generator-cli generate -i src/server/openapi.yaml -g python -o ../sdks/python/generated --package-name pmxt_internal --additional-properties=projectName=pmxt-internal,packageVersion=2.24.0,library=urllib3",
-    "generate:sdk:typescript": "npx @openapitools/openapi-generator-cli generate -i src/server/openapi.yaml -g typescript-fetch -o ../sdks/typescript/generated --additional-properties=npmName=pmxtjs,npmVersion=2.24.0,supportsES6=true,typescriptThreePlus=true && node ../sdks/typescript/scripts/fix-generated.js",
+    "generate:sdk:python": "npx @openapitools/openapi-generator-cli generate -i src/server/openapi.yaml -g python -o ../sdks/python/generated --package-name pmxt_internal --additional-properties=projectName=pmxt-internal,packageVersion=2.25.0,library=urllib3",
+    "generate:sdk:typescript": "npx @openapitools/openapi-generator-cli generate -i src/server/openapi.yaml -g typescript-fetch -o ../sdks/typescript/generated --additional-properties=npmName=pmxtjs,npmVersion=2.25.0,supportsES6=true,typescriptThreePlus=true && node ../sdks/typescript/scripts/fix-generated.js",
     "fetch:openapi": "node scripts/fetch-openapi-specs.js",
     "extract:jsdoc": "node ../scripts/extract-jsdoc.js",
     "generate:docs": "npm run extract:jsdoc && node ../scripts/generate-api-docs.js",
@@ -58,6 +58,7 @@
     "express": "^5.2.1",
     "isows": "^1.0.6",
     "jest": "^30.2.0",
+    "polymarket-us": "0.1.1",
     "tsx": "^4.21.0",
     "ws": "^8.18.0"
   },