@spfunctions/cli 0.1.4 → 0.1.5

package/dist/commands/agent.d.ts

@@ -0,0 +1,18 @@
+/**
+ * sf agent — Interactive natural language terminal agent.
+ *
+ * Uses pi-agent-core's Agent class for the tool-calling loop.
+ * Uses pi-ai for unified LLM access via OpenRouter.
+ *
+ * The agent has 6 tools:
+ *   get_context   — thesis snapshot (causal tree, edges, confidence)
+ *   inject_signal — inject news/note/external signal
+ *   trigger_evaluation — trigger deep eval (heavy model)
+ *   scan_markets  — search Kalshi markets
+ *   list_theses   — list all theses
+ *   get_positions  — Kalshi portfolio positions
+ */
+export declare function agentCommand(thesisId?: string, opts?: {
+    model?: string;
+    modelKey?: string;
+}): Promise<void>;

package/dist/commands/agent.js

@@ -0,0 +1,290 @@
+"use strict";
+/**
+ * sf agent — Interactive natural language terminal agent.
+ *
+ * Uses pi-agent-core's Agent class for the tool-calling loop.
+ * Uses pi-ai for unified LLM access via OpenRouter.
+ *
+ * The agent has 6 tools:
+ *   get_context   — thesis snapshot (causal tree, edges, confidence)
+ *   inject_signal — inject news/note/external signal
+ *   trigger_evaluation — trigger deep eval (heavy model)
+ *   scan_markets  — search Kalshi markets
+ *   list_theses   — list all theses
+ *   get_positions  — Kalshi portfolio positions
+ */
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.agentCommand = agentCommand;
+const readline_1 = __importDefault(require("readline"));
+const client_js_1 = require("../client.js");
+const kalshi_js_1 = require("../kalshi.js");
+async function agentCommand(thesisId, opts) {
+    // ── Dynamic imports for ESM-only packages ──────────────────────────────────
+    const piAi = await import('@mariozechner/pi-ai');
+    const piAgent = await import('@mariozechner/pi-agent-core');
+    const { getModel, streamSimple, Type } = piAi;
+    const { Agent } = piAgent;
+    // ── Validate API keys ──────────────────────────────────────────────────────
+    const openrouterKey = opts?.modelKey || process.env.OPENROUTER_API_KEY;
+    if (!openrouterKey) {
+        console.error('Need OpenRouter API key. Set OPENROUTER_API_KEY or use --model-key.');
+        process.exit(1);
+    }
+    const sfClient = new client_js_1.SFClient();
+    // ── Resolve thesis ID ──────────────────────────────────────────────────────
+    let resolvedThesisId = thesisId;
+    if (!resolvedThesisId) {
+        const data = await sfClient.listTheses();
+        const theses = data.theses || data;
+        const active = theses.find((t) => t.status === 'active');
+        if (!active) {
+            console.error('No active thesis. Create one first: sf create "..."');
+            process.exit(1);
+        }
+        resolvedThesisId = active.id;
+    }
+    // ── Fetch initial context ──────────────────────────────────────────────────
+    const context = await sfClient.getContext(resolvedThesisId);
+    // ── Define tools ───────────────────────────────────────────────────────────
+    const thesisIdParam = Type.Object({
+        thesisId: Type.String({ description: 'Thesis ID (short or full UUID)' }),
+    });
+    const signalParams = Type.Object({
+        thesisId: Type.String({ description: 'Thesis ID' }),
+        content: Type.String({ description: 'Signal content' }),
+        type: Type.Optional(Type.String({ description: 'Signal type: news, user_note, external. Default: user_note' })),
+    });
+    const scanParams = Type.Object({
+        query: Type.Optional(Type.String({ description: 'Keyword search for Kalshi markets' })),
+        series: Type.Optional(Type.String({ description: 'Kalshi series ticker (e.g. KXWTIMAX)' })),
+        market: Type.Optional(Type.String({ description: 'Specific market ticker' })),
+    });
+    const emptyParams = Type.Object({});
+    const tools = [
+        {
+            name: 'get_context',
+            label: 'Get Context',
+            description: 'Get thesis snapshot: causal tree, edge prices, last evaluation, confidence',
+            parameters: thesisIdParam,
+            execute: async (_toolCallId, params) => {
+                const ctx = await sfClient.getContext(params.thesisId);
+                return {
+                    content: [{ type: 'text', text: JSON.stringify(ctx, null, 2) }],
+                    details: {},
+                };
+            },
+        },
+        {
+            name: 'inject_signal',
+            label: 'Inject Signal',
+            description: 'Inject a signal into the thesis (news, note, external event)',
+            parameters: signalParams,
+            execute: async (_toolCallId, params) => {
+                const result = await sfClient.injectSignal(params.thesisId, params.type || 'user_note', params.content);
+                return {
+                    content: [{ type: 'text', text: JSON.stringify(result) }],
+                    details: {},
+                };
+            },
+        },
+        {
+            name: 'trigger_evaluation',
+            label: 'Evaluate',
+            description: 'Trigger a deep evaluation cycle (heavy model, takes longer)',
+            parameters: thesisIdParam,
+            execute: async (_toolCallId, params) => {
+                const result = await sfClient.evaluate(params.thesisId);
+                return {
+                    content: [{ type: 'text', text: JSON.stringify(result) }],
+                    details: {},
+                };
+            },
+        },
+        {
+            name: 'scan_markets',
+            label: 'Scan Markets',
+            description: 'Search Kalshi prediction markets: by keywords, series ticker, or specific market ticker',
+            parameters: scanParams,
+            execute: async (_toolCallId, params) => {
+                let result;
+                if (params.market) {
+                    result = await (0, client_js_1.kalshiFetchMarket)(params.market);
+                }
+                else if (params.series) {
+                    result = await (0, client_js_1.kalshiFetchMarketsBySeries)(params.series);
+                }
+                else if (params.query) {
+                    const series = await (0, client_js_1.kalshiFetchAllSeries)();
+                    const keywords = params.query.toLowerCase().split(/\s+/);
+                    const matched = series
+                        .filter((s) => keywords.some((kw) => (s.title || '').toLowerCase().includes(kw) ||
+                        (s.ticker || '').toLowerCase().includes(kw)))
+                        .slice(0, 15);
+                    result = matched;
+                }
+                else {
+                    result = { error: 'Provide query, series, or market parameter' };
+                }
+                return {
+                    content: [{ type: 'text', text: JSON.stringify(result, null, 2) }],
+                    details: {},
+                };
+            },
+        },
+        {
+            name: 'list_theses',
+            label: 'List Theses',
+            description: 'List all theses for the current user',
+            parameters: emptyParams,
+            execute: async () => {
+                const theses = await sfClient.listTheses();
+                return {
+                    content: [{ type: 'text', text: JSON.stringify(theses, null, 2) }],
+                    details: {},
+                };
+            },
+        },
+        {
+            name: 'get_positions',
+            label: 'Get Positions',
+            description: 'Get Kalshi exchange positions with live prices and PnL (requires local Kalshi key config)',
+            parameters: emptyParams,
+            execute: async () => {
+                const positions = await (0, kalshi_js_1.getPositions)();
+                if (!positions) {
+                    return {
+                        content: [
+                            {
+                                type: 'text',
+                                text: 'Kalshi not configured. Set KALSHI_API_KEY_ID and KALSHI_PRIVATE_KEY_PATH environment variables.',
+                            },
+                        ],
+                        details: {},
+                    };
+                }
+                // Enrich with live prices (same logic as sf positions command)
+                for (const pos of positions) {
+                    const livePrice = await (0, kalshi_js_1.getMarketPrice)(pos.ticker);
+                    if (livePrice !== null) {
+                        pos.current_value = livePrice;
+                        pos.unrealized_pnl = Math.round((livePrice - pos.average_price_paid) * pos.quantity);
+                    }
+                }
+                return {
+                    content: [{ type: 'text', text: JSON.stringify(positions, null, 2) }],
+                    details: {},
+                };
+            },
+        },
+    ];
+    // ── System prompt ──────────────────────────────────────────────────────────
+    const confidencePct = typeof context.confidence === 'number'
+        ? Math.round(context.confidence * 100)
+        : context.confidence;
+    const systemPrompt = `You are a SimpleFunctions prediction market trading assistant.
+
+Current thesis: ${context.thesis || context.rawThesis || 'N/A'}
+Confidence: ${confidencePct}%
+Status: ${context.status}
+Thesis ID: ${context.thesisId || resolvedThesisId}
+
+You have six tools available. Use them when you need real-time data. Answer directly when you don't.
+Be concise. Use Chinese if the user writes in Chinese, English if they write in English.
+Do NOT make up data. Always call tools to get current state.`;
+    // ── Configure model via OpenRouter ─────────────────────────────────────────
+    const rawModelName = opts?.model || 'anthropic/claude-sonnet-4-20250514';
+    // Strip 'openrouter/' prefix if user passed it (provider is already openrouter)
+    const modelName = rawModelName.replace(/^openrouter\//, '');
+    // Try the registry first. If the model isn't registered, construct manually.
+    // OpenRouter accepts any model ID — the registry just may not have it yet.
+    let model;
+    try {
+        model = getModel('openrouter', modelName);
+    }
+    catch {
+        // Manual fallback: construct a Model object compatible with pi-ai's openai-completions API
+        // (OpenRouter uses OpenAI-compatible API)
+        model = {
+            modelId: modelName,
+            provider: 'openrouter',
+            api: 'openai-completions',
+            baseUrl: 'https://openrouter.ai/api/v1',
+            id: modelName,
+            name: modelName,
+            inputPrice: 0,
+            outputPrice: 0,
+            contextWindow: 200000,
+            supportsImages: true,
+            supportsTools: true,
+        };
+        console.log(`\x1b[33mModel '${modelName}' not in registry, using direct OpenRouter API\x1b[0m`);
+    }
+    // ── Create Agent ───────────────────────────────────────────────────────────
+    const agent = new Agent({
+        initialState: {
+            systemPrompt,
+            model,
+            tools,
+            thinkingLevel: 'off',
+        },
+        streamFn: streamSimple,
+        getApiKey: (provider) => {
+            if (provider === 'openrouter')
+                return openrouterKey;
+            return undefined;
+        },
+    });
+    // ── Subscribe to streaming events ──────────────────────────────────────────
+    agent.subscribe((event) => {
+        if (event.type === 'message_update') {
+            const e = event.assistantMessageEvent;
+            if (e.type === 'text_delta') {
+                process.stdout.write(e.delta);
+            }
+        }
+        if (event.type === 'tool_execution_start') {
+            process.stdout.write(`\n\x1b[33m⚡ ${event.toolName}\x1b[0m `);
+        }
+        if (event.type === 'tool_execution_end') {
+            if (event.isError) {
+                process.stdout.write('\x1b[31m✗\x1b[0m\n');
+            }
+            else {
+                process.stdout.write('\x1b[32m✓\x1b[0m\n');
+            }
+        }
+    });
+    // ── Terminal REPL ──────────────────────────────────────────────────────────
+    const shortId = (resolvedThesisId || '').slice(0, 8);
+    console.log(`\n\x1b[32mSimpleFunctions Agent\x1b[0m — ${shortId}`);
+    console.log(`Model: ${modelName} | Type "exit" to quit\n`);
+    const rl = readline_1.default.createInterface({
+        input: process.stdin,
+        output: process.stdout,
+    });
+    const prompt = () => {
+        rl.question('\x1b[36m> \x1b[0m', async (input) => {
+            const trimmed = input.trim();
+            if (!trimmed) {
+                prompt();
+                return;
+            }
+            if (trimmed === 'exit' || trimmed === 'quit') {
+                rl.close();
+                process.exit(0);
+            }
+            try {
+                await agent.prompt(trimmed);
+                console.log('\n');
+            }
+            catch (err) {
+                console.error(`\n\x1b[31mError: ${err.message}\x1b[0m\n`);
+            }
+            prompt();
+        });
+    };
+    prompt();
+}

package/dist/index.js

@@ -25,6 +25,7 @@ const signal_js_1 = require("./commands/signal.js");
 const evaluate_js_1 = require("./commands/evaluate.js");
 const scan_js_1 = require("./commands/scan.js");
 const positions_js_1 = require("./commands/positions.js");
+const agent_js_1 = require("./commands/agent.js");
 const utils_js_1 = require("./utils.js");
 const program = new commander_1.Command();
 program
@@ -130,6 +131,16 @@ program
         apiUrl: g.apiUrl,
     }));
 });
+// ── sf agent [thesisId] ───────────────────────────────────────────────────────
+program
+    .command('agent [thesisId]')
+    .description('Interactive agent mode — natural language interface to SimpleFunctions')
+    .option('--model <model>', 'Model via OpenRouter (default: anthropic/claude-sonnet-4-20250514)')
+    .option('--model-key <key>', 'OpenRouter API key (or set OPENROUTER_API_KEY)')
+    .action(async (thesisId, opts, cmd) => {
+    const g = cmd.optsWithGlobals();
+    await run(() => (0, agent_js_1.agentCommand)(thesisId, { model: opts.model, modelKey: opts.modelKey }));
+});
 // ── Error wrapper ─────────────────────────────────────────────────────────────
 async function run(fn) {
     try {

package/dist/kalshi.js

@@ -104,9 +104,6 @@ async function getPositions() {
         return null;
     try {
         const data = await kalshiAuthGet('/portfolio/positions');
-        // DEBUG: dump raw Kalshi response to see actual field names
-        console.log('[Kalshi DEBUG] Raw /portfolio/positions response:');
-        console.log(JSON.stringify(data, null, 2));
         // Kalshi returns { market_positions: [...] } or { positions: [...] }
         const raw = data.market_positions || data.positions || [];
         return raw.map((p) => {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@spfunctions/cli",
-  "version": "0.1.4",
+  "version": "0.1.5",
   "description": "CLI for SimpleFunctions prediction market thesis agent",
   "bin": {
     "sf": "./dist/index.js"
@@ -11,6 +11,8 @@
     "prepublishOnly": "npm run build"
   },
   "dependencies": {
+    "@mariozechner/pi-agent-core": "^0.57.1",
+    "@mariozechner/pi-ai": "^0.57.1",
     "commander": "^12.0.0",
     "kalshi-typescript": "^3.9.0"
   },