@spfunctions/cli 1.4.4 → 1.5.0

package/dist/polymarket.test.js

@@ -0,0 +1,424 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const vitest_1 = require("vitest");
+const polymarket_js_1 = require("./polymarket.js");
+const mockFetch = vitest_1.vi.fn();
+vitest_1.vi.stubGlobal('fetch', mockFetch);
+(0, vitest_1.beforeEach)(() => {
+    mockFetch.mockReset();
+});
+// ============================================================================
+// PURE FUNCTIONS (no network)
+// ============================================================================
+(0, vitest_1.describe)('computeOrderbookDepth', () => {
+    (0, vitest_1.it)('computes best bid/ask and spread in cents', () => {
+        const raw = {
+            market: 'cond1', asset_id: 'tok1', timestamp: '123', hash: 'h',
+            bids: [
+                { price: '0.45', size: '200' },
+                { price: '0.44', size: '150' },
+                { price: '0.43', size: '100' },
+            ],
+            asks: [
+                { price: '0.48', size: '300' },
+                { price: '0.49', size: '250' },
+                { price: '0.50', size: '100' },
+            ],
+        };
+        const depth = (0, polymarket_js_1.computeOrderbookDepth)(raw);
+        (0, vitest_1.expect)(depth.bestBid).toBe(45);
+        (0, vitest_1.expect)(depth.bestAsk).toBe(48);
+        (0, vitest_1.expect)(depth.spread).toBe(3);
+    });
+    (0, vitest_1.it)('computes top-3 depth', () => {
+        const raw = {
+            market: 'c', asset_id: 't', timestamp: '0', hash: '',
+            bids: [
+                { price: '0.50', size: '100' },
+                { price: '0.49', size: '200' },
+                { price: '0.48', size: '300' },
+                { price: '0.47', size: '400' }, // 4th level, not counted
+            ],
+            asks: [
+                { price: '0.52', size: '50' },
+                { price: '0.53', size: '60' },
+                { price: '0.54', size: '70' },
+            ],
+        };
+        const depth = (0, polymarket_js_1.computeOrderbookDepth)(raw);
+        (0, vitest_1.expect)(depth.bidDepthTop3).toBe(600); // 100+200+300
+        (0, vitest_1.expect)(depth.askDepthTop3).toBe(180); // 50+60+70
+        (0, vitest_1.expect)(depth.totalBidDepth).toBe(1000); // includes 4th level
+        (0, vitest_1.expect)(depth.totalAskDepth).toBe(180);
+    });
+    (0, vitest_1.it)('sorts bids descending and asks ascending', () => {
+        const raw = {
+            market: 'c', asset_id: 't', timestamp: '0', hash: '',
+            bids: [
+                { price: '0.40', size: '100' },
+                { price: '0.50', size: '200' }, // best bid (highest)
+            ],
+            asks: [
+                { price: '0.60', size: '100' },
+                { price: '0.55', size: '200' }, // best ask (lowest)
+            ],
+        };
+        const depth = (0, polymarket_js_1.computeOrderbookDepth)(raw);
+        (0, vitest_1.expect)(depth.bestBid).toBe(50);
+        (0, vitest_1.expect)(depth.bestAsk).toBe(55);
+        (0, vitest_1.expect)(depth.spread).toBe(5);
+    });
+    (0, vitest_1.it)('handles empty orderbook', () => {
+        const raw = {
+            market: 'c', asset_id: 't', timestamp: '0', hash: '',
+            bids: [], asks: [],
+        };
+        const depth = (0, polymarket_js_1.computeOrderbookDepth)(raw);
+        (0, vitest_1.expect)(depth.bestBid).toBe(0);
+        (0, vitest_1.expect)(depth.bestAsk).toBe(100);
+        (0, vitest_1.expect)(depth.spread).toBe(100);
+        (0, vitest_1.expect)(depth.bidDepthTop3).toBe(0);
+        (0, vitest_1.expect)(depth.askDepthTop3).toBe(0);
+        (0, vitest_1.expect)(depth.liquidityScore).toBe('low');
+    });
+    (0, vitest_1.it)('assigns correct liquidity scores', () => {
+        // high: spread ≤ 2¢, depth ≥ 500
+        const high = {
+            market: 'c', asset_id: 't', timestamp: '0', hash: '',
+            bids: [{ price: '0.49', size: '300' }],
+            asks: [{ price: '0.51', size: '300' }],
+        };
+        (0, vitest_1.expect)((0, polymarket_js_1.computeOrderbookDepth)(high).liquidityScore).toBe('high');
+        // medium: spread ≤ 5¢, depth ≥ 100
+        const medium = {
+            market: 'c', asset_id: 't', timestamp: '0', hash: '',
+            bids: [{ price: '0.47', size: '60' }],
+            asks: [{ price: '0.52', size: '60' }],
+        };
+        (0, vitest_1.expect)((0, polymarket_js_1.computeOrderbookDepth)(medium).liquidityScore).toBe('medium');
+        // low: wide spread
+        const low = {
+            market: 'c', asset_id: 't', timestamp: '0', hash: '',
+            bids: [{ price: '0.30', size: '10' }],
+            asks: [{ price: '0.70', size: '10' }],
+        };
+        (0, vitest_1.expect)((0, polymarket_js_1.computeOrderbookDepth)(low).liquidityScore).toBe('low');
+    });
+});
+(0, vitest_1.describe)('scoreLiquidity', () => {
+    (0, vitest_1.it)('returns high for tight spread + deep book', () => {
+        (0, vitest_1.expect)((0, polymarket_js_1.scoreLiquidity)(1, 600)).toBe('high');
+        (0, vitest_1.expect)((0, polymarket_js_1.scoreLiquidity)(2, 500)).toBe('high');
+    });
+    (0, vitest_1.it)('returns medium for moderate spread + depth', () => {
+        (0, vitest_1.expect)((0, polymarket_js_1.scoreLiquidity)(3, 200)).toBe('medium');
+        (0, vitest_1.expect)((0, polymarket_js_1.scoreLiquidity)(5, 100)).toBe('medium');
+    });
+    (0, vitest_1.it)('returns low for wide spread or thin book', () => {
+        (0, vitest_1.expect)((0, polymarket_js_1.scoreLiquidity)(6, 1000)).toBe('low');
+        (0, vitest_1.expect)((0, polymarket_js_1.scoreLiquidity)(1, 50)).toBe('low');
+        (0, vitest_1.expect)((0, polymarket_js_1.scoreLiquidity)(10, 10)).toBe('low');
+    });
+});
+(0, vitest_1.describe)('parseClobTokenIds', () => {
+    (0, vitest_1.it)('parses valid JSON array', () => {
+        const result = (0, polymarket_js_1.parseClobTokenIds)('["tok_yes_123","tok_no_456"]');
+        (0, vitest_1.expect)(result).toEqual(['tok_yes_123', 'tok_no_456']);
+    });
+    (0, vitest_1.it)('returns null for invalid JSON', () => {
+        (0, vitest_1.expect)((0, polymarket_js_1.parseClobTokenIds)('not json')).toBeNull();
+    });
+    (0, vitest_1.it)('returns null for array with < 2 elements', () => {
+        (0, vitest_1.expect)((0, polymarket_js_1.parseClobTokenIds)('["only_one"]')).toBeNull();
+    });
+    (0, vitest_1.it)('returns null for empty string', () => {
+        (0, vitest_1.expect)((0, polymarket_js_1.parseClobTokenIds)('')).toBeNull();
+    });
+});
+(0, vitest_1.describe)('parseOutcomes', () => {
+    (0, vitest_1.it)('parses yes/no outcomes', () => {
+        (0, vitest_1.expect)((0, polymarket_js_1.parseOutcomes)('["Yes","No"]')).toEqual(['Yes', 'No']);
+    });
+    (0, vitest_1.it)('parses multi-outcome', () => {
+        (0, vitest_1.expect)((0, polymarket_js_1.parseOutcomes)('["Trump","Biden","Other"]')).toEqual(['Trump', 'Biden', 'Other']);
+    });
+    (0, vitest_1.it)('returns empty for invalid', () => {
+        (0, vitest_1.expect)((0, polymarket_js_1.parseOutcomes)('bad')).toEqual([]);
+    });
+});
+(0, vitest_1.describe)('parseOutcomePrices', () => {
+    (0, vitest_1.it)('parses price array', () => {
+        (0, vitest_1.expect)((0, polymarket_js_1.parseOutcomePrices)('[0.65, 0.35]')).toEqual([0.65, 0.35]);
+    });
+    (0, vitest_1.it)('returns empty for invalid', () => {
+        (0, vitest_1.expect)((0, polymarket_js_1.parseOutcomePrices)('')).toEqual([]);
+    });
+});
+(0, vitest_1.describe)('toCents', () => {
+    (0, vitest_1.it)('converts dollars to cents', () => {
+        (0, vitest_1.expect)((0, polymarket_js_1.toCents)(0.55)).toBe(55);
+        (0, vitest_1.expect)((0, polymarket_js_1.toCents)(1.0)).toBe(100);
+        (0, vitest_1.expect)((0, polymarket_js_1.toCents)(0)).toBe(0);
+        (0, vitest_1.expect)((0, polymarket_js_1.toCents)(0.123)).toBe(12); // rounds
+    });
+});
+(0, vitest_1.describe)('polymarketUrl', () => {
+    (0, vitest_1.it)('builds correct URL', () => {
+        (0, vitest_1.expect)((0, polymarket_js_1.polymarketUrl)('fed-decision')).toBe('https://polymarket.com/event/fed-decision');
+    });
+});
+// ============================================================================
+// API FUNCTIONS (mocked fetch)
+// ============================================================================
+(0, vitest_1.describe)('polymarketSearch', () => {
+    (0, vitest_1.it)('calls Gamma API with query params', async () => {
+        mockFetch.mockResolvedValue({
+            ok: true,
+            json: () => Promise.resolve({ events: [{ id: '1', title: 'Test' }] }),
+        });
+        const events = await (0, polymarket_js_1.polymarketSearch)('recession', 5);
+        (0, vitest_1.expect)(events).toHaveLength(1);
+        (0, vitest_1.expect)(events[0].title).toBe('Test');
+        const url = mockFetch.mock.calls[0][0];
+        (0, vitest_1.expect)(url).toContain('gamma-api.polymarket.com/public-search');
+        (0, vitest_1.expect)(url).toContain('q=recession');
+        (0, vitest_1.expect)(url).toContain('limit_per_type=5');
+    });
+    (0, vitest_1.it)('throws on error response', async () => {
+        mockFetch.mockResolvedValue({ ok: false, status: 500 });
+        await (0, vitest_1.expect)((0, polymarket_js_1.polymarketSearch)('test')).rejects.toThrow('500');
+    });
+});
+(0, vitest_1.describe)('polymarketListEvents', () => {
+    (0, vitest_1.it)('fetches active events with pagination', async () => {
+        mockFetch.mockResolvedValue({
+            ok: true,
+            json: () => Promise.resolve([{ id: '1' }, { id: '2' }]),
+        });
+        const events = await (0, polymarket_js_1.polymarketListEvents)({ limit: 10, offset: 20, order: 'volume_24hr' });
+        const url = mockFetch.mock.calls[0][0];
+        (0, vitest_1.expect)(url).toContain('active=true');
+        (0, vitest_1.expect)(url).toContain('limit=10');
+        (0, vitest_1.expect)(url).toContain('offset=20');
+        (0, vitest_1.expect)(url).toContain('order=volume_24hr');
+        (0, vitest_1.expect)(events).toHaveLength(2);
+    });
+});
+(0, vitest_1.describe)('polymarketGetEvent', () => {
+    (0, vitest_1.it)('fetches single event by ID', async () => {
+        mockFetch.mockResolvedValue({
+            ok: true,
+            json: () => Promise.resolve({ id: '42', title: 'Fed Decision' }),
+        });
+        const event = await (0, polymarket_js_1.polymarketGetEvent)('42');
+        (0, vitest_1.expect)(event.id).toBe('42');
+        (0, vitest_1.expect)(mockFetch.mock.calls[0][0]).toContain('/events/42');
+    });
+});
+(0, vitest_1.describe)('polymarketGetMarket', () => {
+    (0, vitest_1.it)('fetches single market', async () => {
+        mockFetch.mockResolvedValue({
+            ok: true,
+            json: () => Promise.resolve({ id: 'm1', question: 'Will X happen?' }),
+        });
+        const market = await (0, polymarket_js_1.polymarketGetMarket)('m1');
+        (0, vitest_1.expect)(market.question).toBe('Will X happen?');
+    });
+});
+(0, vitest_1.describe)('polymarketListTags', () => {
+    (0, vitest_1.it)('returns tags array', async () => {
+        mockFetch.mockResolvedValue({
+            ok: true,
+            json: () => Promise.resolve([
+                { id: '1', label: 'Politics', slug: 'politics' },
+                { id: '2', label: 'Crypto', slug: 'crypto' },
+            ]),
+        });
+        const tags = await (0, polymarket_js_1.polymarketListTags)();
+        (0, vitest_1.expect)(tags).toHaveLength(2);
+        (0, vitest_1.expect)(tags[0].label).toBe('Politics');
+    });
+});
+(0, vitest_1.describe)('polymarketGetOrderbook', () => {
+    (0, vitest_1.it)('fetches raw orderbook', async () => {
+        mockFetch.mockResolvedValue({
+            ok: true,
+            json: () => Promise.resolve({
+                market: 'cond1', asset_id: 'tok1', timestamp: '123', hash: 'h',
+                bids: [{ price: '0.45', size: '100' }],
+                asks: [{ price: '0.55', size: '100' }],
+            }),
+        });
+        const ob = await (0, polymarket_js_1.polymarketGetOrderbook)('tok1');
+        (0, vitest_1.expect)(ob).not.toBeNull();
+        (0, vitest_1.expect)(ob.bids).toHaveLength(1);
+        (0, vitest_1.expect)(mockFetch.mock.calls[0][0]).toContain('clob.polymarket.com/book?token_id=tok1');
+    });
+    (0, vitest_1.it)('returns null on error', async () => {
+        mockFetch.mockResolvedValue({ ok: false, status: 404 });
+        const ob = await (0, polymarket_js_1.polymarketGetOrderbook)('bad');
+        (0, vitest_1.expect)(ob).toBeNull();
+    });
+    (0, vitest_1.it)('returns null on API error response', async () => {
+        mockFetch.mockResolvedValue({
+            ok: true,
+            json: () => Promise.resolve({ error: 'No orderbook exists' }),
+        });
+        (0, vitest_1.expect)(await (0, polymarket_js_1.polymarketGetOrderbook)('bad')).toBeNull();
+    });
+    (0, vitest_1.it)('returns null on network error', async () => {
+        mockFetch.mockRejectedValue(new Error('network'));
+        (0, vitest_1.expect)(await (0, polymarket_js_1.polymarketGetOrderbook)('bad')).toBeNull();
+    });
+});
+(0, vitest_1.describe)('polymarketGetOrderbookWithDepth', () => {
+    (0, vitest_1.it)('fetches and computes depth in one call', async () => {
+        mockFetch.mockResolvedValue({
+            ok: true,
+            json: () => Promise.resolve({
+                market: 'c', asset_id: 't', timestamp: '0', hash: '',
+                bids: [{ price: '0.49', size: '300' }],
+                asks: [{ price: '0.51', size: '400' }],
+            }),
+        });
+        const depth = await (0, polymarket_js_1.polymarketGetOrderbookWithDepth)('tok1');
+        (0, vitest_1.expect)(depth).not.toBeNull();
+        (0, vitest_1.expect)(depth.bestBid).toBe(49);
+        (0, vitest_1.expect)(depth.bestAsk).toBe(51);
+        (0, vitest_1.expect)(depth.spread).toBe(2);
+        (0, vitest_1.expect)(depth.liquidityScore).toBe('high');
+    });
+    (0, vitest_1.it)('returns null when orderbook unavailable', async () => {
+        mockFetch.mockResolvedValue({ ok: false });
+        (0, vitest_1.expect)(await (0, polymarket_js_1.polymarketGetOrderbookWithDepth)('bad')).toBeNull();
+    });
+});
+(0, vitest_1.describe)('polymarketGetMidpoint', () => {
+    (0, vitest_1.it)('returns midpoint as number', async () => {
+        mockFetch.mockResolvedValue({
+            ok: true,
+            json: () => Promise.resolve({ mid: '0.52' }),
+        });
+        (0, vitest_1.expect)(await (0, polymarket_js_1.polymarketGetMidpoint)('tok1')).toBe(0.52);
+    });
+    (0, vitest_1.it)('returns null on error', async () => {
+        mockFetch.mockResolvedValue({ ok: false });
+        (0, vitest_1.expect)(await (0, polymarket_js_1.polymarketGetMidpoint)('bad')).toBeNull();
+    });
+});
+(0, vitest_1.describe)('polymarketGetSpread', () => {
+    (0, vitest_1.it)('returns spread as number', async () => {
+        mockFetch.mockResolvedValue({
+            ok: true,
+            json: () => Promise.resolve({ spread: '0.03' }),
+        });
+        (0, vitest_1.expect)(await (0, polymarket_js_1.polymarketGetSpread)('tok1')).toBe(0.03);
+    });
+});
+(0, vitest_1.describe)('polymarketGetMidpoints', () => {
+    (0, vitest_1.it)('posts batch of token IDs', async () => {
+        mockFetch.mockResolvedValue({
+            ok: true,
+            json: () => Promise.resolve({ tok1: 0.5, tok2: 0.7 }),
+        });
+        const result = await (0, polymarket_js_1.polymarketGetMidpoints)(['tok1', 'tok2']);
+        (0, vitest_1.expect)(result.tok1).toBe(0.5);
+        (0, vitest_1.expect)(result.tok2).toBe(0.7);
+        (0, vitest_1.expect)(mockFetch.mock.calls[0][1].method).toBe('POST');
+    });
+    (0, vitest_1.it)('returns empty for empty input', async () => {
+        const result = await (0, polymarket_js_1.polymarketGetMidpoints)([]);
+        (0, vitest_1.expect)(result).toEqual({});
+        (0, vitest_1.expect)(mockFetch).not.toHaveBeenCalled();
+    });
+});
+(0, vitest_1.describe)('polymarketGetOHLC', () => {
+    (0, vitest_1.it)('fetches candlestick data', async () => {
+        const candles = [
+            { t: 1000, o: 0.5, h: 0.55, l: 0.48, c: 0.52, v: 100 },
+        ];
+        mockFetch.mockResolvedValue({
+            ok: true,
+            json: () => Promise.resolve(candles),
+        });
+        const result = await (0, polymarket_js_1.polymarketGetOHLC)({
+            tokenId: 'tok1',
+            startTs: 1000,
+            fidelity: '1h',
+        });
+        (0, vitest_1.expect)(result).toHaveLength(1);
+        (0, vitest_1.expect)(result[0].c).toBe(0.52);
+        const url = mockFetch.mock.calls[0][0];
+        (0, vitest_1.expect)(url).toContain('clob.polymarket.com/ohlc');
+        (0, vitest_1.expect)(url).toContain('asset_id=tok1');
+        (0, vitest_1.expect)(url).toContain('fidelity=1h');
+    });
+    (0, vitest_1.it)('returns empty array on error', async () => {
+        mockFetch.mockResolvedValue({ ok: false, status: 400 });
+        (0, vitest_1.expect)(await (0, polymarket_js_1.polymarketGetOHLC)({ tokenId: 'bad', startTs: 0 })).toEqual([]);
+    });
+});
+(0, vitest_1.describe)('polymarketGetPriceHistory', () => {
+    (0, vitest_1.it)('fetches price history with interval', async () => {
+        mockFetch.mockResolvedValue({
+            ok: true,
+            json: () => Promise.resolve({
+                history: [{ t: 1000, p: 0.5 }, { t: 2000, p: 0.55 }],
+            }),
+        });
+        const result = await (0, polymarket_js_1.polymarketGetPriceHistory)({
+            tokenId: 'tok1',
+            interval: '1d',
+        });
+        (0, vitest_1.expect)(result).toHaveLength(2);
+        const url = mockFetch.mock.calls[0][0];
+        (0, vitest_1.expect)(url).toContain('prices-history');
+        (0, vitest_1.expect)(url).toContain('interval=1d');
+    });
+    (0, vitest_1.it)('fetches with absolute timestamps', async () => {
+        mockFetch.mockResolvedValue({
+            ok: true,
+            json: () => Promise.resolve({ history: [] }),
+        });
+        await (0, polymarket_js_1.polymarketGetPriceHistory)({
+            tokenId: 'tok1',
+            startTs: 1000,
+            endTs: 2000,
+            fidelity: 60,
+        });
+        const url = mockFetch.mock.calls[0][0];
+        (0, vitest_1.expect)(url).toContain('startTs=1000');
+        (0, vitest_1.expect)(url).toContain('endTs=2000');
+        (0, vitest_1.expect)(url).toContain('fidelity=60');
+        (0, vitest_1.expect)(url).not.toContain('interval=');
+    });
+});
+(0, vitest_1.describe)('polymarketGetPositions', () => {
+    (0, vitest_1.it)('fetches positions for wallet address', async () => {
+        mockFetch.mockResolvedValue({
+            ok: true,
+            json: () => Promise.resolve([
+                { asset: 'tok1', size: 100, outcome: 'Yes' },
+            ]),
+        });
+        const positions = await (0, polymarket_js_1.polymarketGetPositions)('0xabc123');
+        (0, vitest_1.expect)(positions).toHaveLength(1);
+        (0, vitest_1.expect)(positions[0].outcome).toBe('Yes');
+        (0, vitest_1.expect)(mockFetch.mock.calls[0][0]).toContain('data-api.polymarket.com/positions?user=0xabc123');
+    });
+    (0, vitest_1.it)('returns empty on error', async () => {
+        mockFetch.mockResolvedValue({ ok: false });
+        (0, vitest_1.expect)(await (0, polymarket_js_1.polymarketGetPositions)('bad')).toEqual([]);
+    });
+});
+(0, vitest_1.describe)('polymarketGetClosedPositions', () => {
+    (0, vitest_1.it)('fetches closed positions', async () => {
+        mockFetch.mockResolvedValue({
+            ok: true,
+            json: () => Promise.resolve([{ asset: 'tok1', outcome: 'No' }]),
+        });
+        const positions = await (0, polymarket_js_1.polymarketGetClosedPositions)('0xabc');
+        (0, vitest_1.expect)(positions).toHaveLength(1);
+        (0, vitest_1.expect)(mockFetch.mock.calls[0][0]).toContain('closed-positions');
+    });
+});

package/dist/telegram/agent-bridge.d.ts

@@ -0,0 +1,15 @@
+/**
+ * Agent bridge — connects Telegram to pi-agent-core
+ *
+ * Uses the SAME tools as sf agent --plain. Multi-turn tool calling
+ * is handled by pi-agent-core's Agent class (not manual OpenRouter calls).
+ */
+import { SFClient } from '../client.js';
+interface SessionState {
+    thesisId: string | null;
+    agentMessages: any[];
+    agent?: any;
+}
+export declare function getOrCreateAgent(sfClient: SFClient, session: SessionState): Promise<any>;
+export declare function runAgentMessage(client: SFClient, session: SessionState, userMessage: string): Promise<string>;
+export {};