@spfunctions/cli 0.1.6 → 1.1.0

package/dist/commands/agent.d.ts

@@ -15,4 +15,5 @@
 export declare function agentCommand(thesisId?: string, opts?: {
     model?: string;
     modelKey?: string;
+    newSession?: boolean;
 }): Promise<void>;

package/dist/commands/agent.js

@@ -13,10 +13,43 @@
  * Slash commands (bypass LLM):
  *   /help   /tree   /edges   /pos   /eval   /model   /clear   /exit
  */
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.agentCommand = agentCommand;
+const fs_1 = __importDefault(require("fs"));
+const path_1 = __importDefault(require("path"));
+const os_1 = __importDefault(require("os"));
 const client_js_1 = require("../client.js");
 const kalshi_js_1 = require("../kalshi.js");
+// ─── Session persistence ─────────────────────────────────────────────────────
+function getSessionDir() {
+    return path_1.default.join(os_1.default.homedir(), '.sf', 'sessions');
+}
+function getSessionPath(thesisId) {
+    return path_1.default.join(getSessionDir(), `${thesisId}.json`);
+}
+function loadSession(thesisId) {
+    const p = getSessionPath(thesisId);
+    try {
+        if (fs_1.default.existsSync(p)) {
+            return JSON.parse(fs_1.default.readFileSync(p, 'utf-8'));
+        }
+    }
+    catch { /* corrupt file, ignore */ }
+    return null;
+}
+function saveSession(thesisId, model, messages) {
+    const dir = getSessionDir();
+    fs_1.default.mkdirSync(dir, { recursive: true });
+    fs_1.default.writeFileSync(getSessionPath(thesisId), JSON.stringify({
+        thesisId,
+        model,
+        updatedAt: new Date().toISOString(),
+        messages,
+    }, null, 2));
+}
 // ─── ANSI 24-bit color helpers (no chalk dependency) ─────────────────────────
 const rgb = (r, g, b) => (s) => `\x1b[38;2;${r};${g};${b}m${s}\x1b[39m`;
 const bgRgb = (r, g, b) => (s) => `\x1b[48;2;${r};${g};${b}m${s}\x1b[49m`;
@@ -197,16 +230,19 @@ function renderEdges(context, piTui) {
     const positions = context._positions || [];
     const lines = [];
     for (const e of edges) {
-        const name = (e.marketTitle || e.ticker || '').slice(0, 18).padEnd(18);
-        const market = typeof e.marketPrice === 'number' ? `${e.marketPrice}\u00A2` : '?';
-        const thesis = typeof e.thesisImpliedPrice === 'number' ? `${e.thesisImpliedPrice}\u00A2` : '?';
-        const edge = typeof e.edgeSize === 'number' ? (e.edgeSize > 0 ? `+${e.edgeSize}` : `${e.edgeSize}`) : '?';
-        const spread = typeof e.spread === 'number' ? `${e.spread}\u00A2` : '?';
-        const liq = e.liquidityScore || 'low';
+        // Context API field names: market, marketId, thesisPrice, edge, orderbook.spread, orderbook.liquidityScore
+        const name = (e.market || e.marketId || '').slice(0, 18).padEnd(18);
+        const marketStr = typeof e.marketPrice === 'number' ? `${e.marketPrice}\u00A2` : '?';
+        const thesisStr = typeof e.thesisPrice === 'number' ? `${e.thesisPrice}\u00A2` : '?';
+        const edgeVal = typeof e.edge === 'number' ? (e.edge > 0 ? `+${e.edge}` : `${e.edge}`) : '?';
+        const ob = e.orderbook || {};
+        const spreadStr = typeof ob.spread === 'number' ? `${ob.spread}\u00A2` : '?';
+        const liq = ob.liquidityScore || 'low';
         const liqBars = liq === 'high' ? '\u25A0\u25A0\u25A0' : liq === 'medium' ? '\u25A0\u25A0 ' : '\u25A0  ';
         const liqColor = liq === 'high' ? C.emerald : liq === 'medium' ? C.amber : C.red;
-        // Check if we have a position on this edge
-        const pos = positions.find((p) => p.ticker === e.ticker);
+        // Check if we have a position on this edge (match by marketId prefix in ticker)
+        const pos = positions.find((p) => p.ticker === e.marketId ||
+            (e.marketId && p.ticker?.includes(e.marketId)));
         let posStr = C.zinc600('\u2014');
         if (pos) {
             const side = pos.side?.toUpperCase() || 'YES';
@@ -215,7 +251,7 @@ function renderEdges(context, piTui) {
                 : '';
             posStr = C.emerald(`${side} (${pos.quantity}@${pos.average_price_paid}\u00A2 ${pnl})`);
         }
-        lines.push(`  ${C.zinc200(name)} ${C.zinc400(market)} \u2192 ${C.zinc400(thesis)}  edge ${edge.includes('+') ? C.emerald(edge) : C.red(edge)}  spread ${C.zinc600(spread)}  ${liqColor(liqBars)} ${liqColor(liq.padEnd(4))}  ${posStr}`);
+        lines.push(`  ${C.zinc200(name)} ${C.zinc400(marketStr)} \u2192 ${C.zinc400(thesisStr)}  edge ${edgeVal.includes('+') ? C.emerald(edgeVal) : C.red(edgeVal)}  spread ${C.zinc600(spreadStr)}  ${liqColor(liqBars)} ${liqColor(liq.padEnd(4))}  ${posStr}`);
     }
     return lines.join('\n');
 }
@@ -283,7 +319,7 @@ async function agentCommand(thesisId, opts) {
     // ── Fetch initial context ──────────────────────────────────────────────────
     let latestContext = await sfClient.getContext(resolvedThesisId);
     // ── Model setup ────────────────────────────────────────────────────────────
-    const rawModelName = opts?.model || 'anthropic/claude-sonnet-4-20250514';
+    const rawModelName = opts?.model || 'anthropic/claude-sonnet-4.6';
     let currentModelName = rawModelName.replace(/^openrouter\//, '');
     function resolveModel(name) {
         try {
@@ -365,9 +401,12 @@ async function agentCommand(thesisId, opts) {
         { name: 'edges', description: 'Display edge/spread table' },
         { name: 'pos', description: 'Display Kalshi positions' },
         { name: 'eval', description: 'Trigger deep evaluation' },
+        { name: 'switch', description: 'Switch thesis (e.g. /switch f582bf76)' },
+        { name: 'compact', description: 'Compress conversation history' },
+        { name: 'new', description: 'Start fresh session' },
         { name: 'model', description: 'Switch model (e.g. /model anthropic/claude-sonnet-4)' },
-        { name: 'clear', description: 'Clear chat' },
-        { name: 'exit', description: 'Exit agent' },
+        { name: 'clear', description: 'Clear screen (keeps history)' },
+        { name: 'exit', description: 'Exit agent (auto-saves)' },
     ], process.cwd());
     editor.setAutocompleteProvider(autocompleteProvider);
     // Assemble TUI tree
@@ -558,12 +597,53 @@ Do NOT make up data. Always call tools to get current state.`;
             return undefined;
         },
     });
+    // ── Session restore ────────────────────────────────────────────────────────
+    let sessionRestored = false;
+    if (!opts?.newSession) {
+        const saved = loadSession(resolvedThesisId);
+        if (saved?.messages?.length > 0) {
+            try {
+                agent.replaceMessages(saved.messages);
+                // Always update system prompt with fresh context
+                agent.setSystemPrompt(systemPrompt);
+                sessionRestored = true;
+            }
+            catch { /* corrupt session, start fresh */ }
+        }
+    }
+    // Helper to persist session after each turn
+    function persistSession() {
+        try {
+            const msgs = agent.state.messages;
+            if (msgs.length > 0) {
+                saveSession(resolvedThesisId, currentModelName, msgs);
+            }
+        }
+        catch { /* best-effort save */ }
+    }
     // ── Subscribe to agent events → update TUI ────────────────────────────────
     let currentAssistantMd = null;
     let currentAssistantText = '';
     let currentLoader = null;
     const toolStartTimes = new Map();
     const toolLines = new Map();
+    // Throttle renders during streaming to prevent flicker (max ~15fps)
+    let renderTimer = null;
+    function throttledRender() {
+        if (renderTimer)
+            return;
+        renderTimer = setTimeout(() => {
+            renderTimer = null;
+            tui.requestRender();
+        }, 66);
+    }
+    function flushRender() {
+        if (renderTimer) {
+            clearTimeout(renderTimer);
+            renderTimer = null;
+        }
+        tui.requestRender();
+    }
     agent.subscribe((event) => {
         if (event.type === 'message_start') {
             // Show loader while waiting for first text
@@ -590,30 +670,31 @@ Do NOT make up data. Always call tools to get current state.`;
                 if (currentAssistantMd) {
                     currentAssistantMd.setText(currentAssistantText);
                 }
-                tui.requestRender();
+                // Throttled render to prevent flicker during fast token streaming
+                throttledRender();
             }
         }
-        if (event.type === 'message_complete') {
+        if (event.type === 'message_end') {
             // Clean up loader if still present (no text was generated)
             if (currentLoader) {
                 currentLoader.stop();
                 chatContainer.removeChild(currentLoader);
                 currentLoader = null;
             }
-            // Add spacer after message
+            // Final render of the complete message
+            if (currentAssistantMd && currentAssistantText) {
+                currentAssistantMd.setText(currentAssistantText);
+            }
             addSpacer();
             currentAssistantMd = null;
             currentAssistantText = '';
+            flushRender();
+        }
+        if (event.type === 'agent_end') {
+            // Agent turn fully complete — safe to accept new input
             isProcessing = false;
-            tui.requestRender();
-            // Update token/cost tracking from event if available
-            if (event.usage) {
-                totalTokens += (event.usage.inputTokens || 0) + (event.usage.outputTokens || 0);
-                totalCost += event.usage.totalCost || 0;
-                footerBar.tokens = totalTokens;
-                footerBar.cost = totalCost;
-                footerBar.update();
-            }
+            persistSession();
+            flushRender();
         }
         if (event.type === 'tool_execution_start') {
             const toolLine = new MutableLine(C.zinc600(`  \u26A1 ${event.toolName}...`));
@@ -651,14 +732,17 @@ Do NOT make up data. Always call tools to get current state.`;
             case '/help': {
                 addSpacer();
                 addSystemText(C.zinc200(bold('Commands')) + '\n' +
-                    C.emerald('/help    ') + C.zinc400('  Show this help') + '\n' +
-                    C.emerald('/tree    ') + C.zinc400('  Display causal tree') + '\n' +
-                    C.emerald('/edges   ') + C.zinc400('  Display edge/spread table') + '\n' +
-                    C.emerald('/pos     ') + C.zinc400('  Display Kalshi positions') + '\n' +
-                    C.emerald('/eval    ') + C.zinc400('  Trigger deep evaluation') + '\n' +
-                    C.emerald('/model   ') + C.zinc400('  Switch model (e.g. /model anthropic/claude-sonnet-4)') + '\n' +
-                    C.emerald('/clear   ') + C.zinc400('  Clear chat') + '\n' +
-                    C.emerald('/exit    ') + C.zinc400('  Exit agent'));
+                    C.emerald('/help      ') + C.zinc400('Show this help') + '\n' +
+                    C.emerald('/tree      ') + C.zinc400('Display causal tree') + '\n' +
+                    C.emerald('/edges     ') + C.zinc400('Display edge/spread table') + '\n' +
+                    C.emerald('/pos       ') + C.zinc400('Display Kalshi positions') + '\n' +
+                    C.emerald('/eval      ') + C.zinc400('Trigger deep evaluation') + '\n' +
+                    C.emerald('/switch <id>') + C.zinc400(' Switch thesis') + '\n' +
+                    C.emerald('/compact   ') + C.zinc400('Compress conversation history') + '\n' +
+                    C.emerald('/new       ') + C.zinc400('Start fresh session') + '\n' +
+                    C.emerald('/model <m> ') + C.zinc400('Switch model') + '\n' +
+                    C.emerald('/clear     ') + C.zinc400('Clear screen (keeps history)') + '\n' +
+                    C.emerald('/exit      ') + C.zinc400('Exit (auto-saves)'));
                 addSpacer();
                 return true;
             }
@@ -746,6 +830,98 @@ Do NOT make up data. Always call tools to get current state.`;
                 tui.requestRender();
                 return true;
             }
+            case '/switch': {
+                const newId = parts[1]?.trim();
+                if (!newId) {
+                    addSystemText(C.zinc400('Usage: /switch <thesisId>'));
+                    return true;
+                }
+                addSpacer();
+                try {
+                    // Save current session
+                    persistSession();
+                    // Load new thesis context
+                    const newContext = await sfClient.getContext(newId);
+                    resolvedThesisId = newContext.thesisId || newId;
+                    latestContext = newContext;
+                    // Build new system prompt
+                    const newConf = typeof newContext.confidence === 'number'
+                        ? Math.round(newContext.confidence * 100) : 0;
+                    const newSysPrompt = `You are a SimpleFunctions prediction market trading assistant.\n\nCurrent thesis: ${newContext.thesis || newContext.rawThesis || 'N/A'}\nConfidence: ${newConf}%\nStatus: ${newContext.status}\nThesis ID: ${resolvedThesisId}\n\nYou have six tools available. Use them when you need real-time data. Answer directly when you don't.\nBe concise. Use Chinese if the user writes in Chinese, English if they write in English.\nDo NOT make up data. Always call tools to get current state.`;
+                    // Load saved session or start fresh
+                    const saved = loadSession(resolvedThesisId);
+                    if (saved?.messages?.length > 0) {
+                        agent.replaceMessages(saved.messages);
+                        agent.setSystemPrompt(newSysPrompt);
+                        addSystemText(C.emerald(`Switched to ${resolvedThesisId.slice(0, 8)}`) + C.zinc400(` (resumed ${saved.messages.length} messages)`));
+                    }
+                    else {
+                        agent.clearMessages();
+                        agent.setSystemPrompt(newSysPrompt);
+                        addSystemText(C.emerald(`Switched to ${resolvedThesisId.slice(0, 8)}`) + C.zinc400(' (new session)'));
+                    }
+                    // Update header
+                    headerBar.update(C.emerald(bold('SF Agent')) + C.zinc600(` \u2014 ${resolvedThesisId.slice(0, 8)}`), newConf > 0 ? C.zinc200(`${newConf}%`) : '', undefined);
+                    chatContainer.clear();
+                    const thText = (newContext.thesis || newContext.rawThesis || '').slice(0, 120);
+                    addSystemText(C.zinc600('\u2500'.repeat(50)) + '\n' +
+                        C.zinc200(bold(thText)) + '\n' +
+                        C.zinc600(`${newContext.status || 'active'}  ${newConf > 0 ? newConf + '%' : ''}  ${(newContext.edges || []).length} edges`) + '\n' +
+                        C.zinc600('\u2500'.repeat(50)));
+                }
+                catch (err) {
+                    addSystemText(C.red(`Switch failed: ${err.message}`));
+                }
+                addSpacer();
+                tui.requestRender();
+                return true;
+            }
+            case '/compact': {
+                addSpacer();
+                const msgs = agent.state.messages;
+                if (msgs.length <= 10) {
+                    addSystemText(C.zinc400('Conversation too short to compact'));
+                    addSpacer();
+                    return true;
+                }
+                // Keep recent 6 messages (3 turns) + create summary of the rest
+                const recentCount = 6;
+                const toCompress = msgs.slice(0, -recentCount);
+                const toKeep = msgs.slice(-recentCount);
+                // Extract text for summary (no LLM, just bullet points)
+                const bulletPoints = [];
+                for (const m of toCompress) {
+                    const content = typeof m.content === 'string' ? m.content : '';
+                    if (m.role === 'user' && content) {
+                        bulletPoints.push(`- User: ${content.slice(0, 100)}`);
+                    }
+                    else if (m.role === 'assistant' && content) {
+                        bulletPoints.push(`- Assistant: ${content.slice(0, 150)}`);
+                    }
+                }
+                const summary = bulletPoints.slice(-20).join('\n');
+                // Replace messages: summary + recent
+                const compactedMessages = [
+                    { role: 'assistant', content: `[Conversation summary - ${toCompress.length} messages compressed]\n${summary}` },
+                    ...toKeep,
+                ];
+                agent.replaceMessages(compactedMessages);
+                persistSession();
+                addSystemText(C.emerald(`Compacted: ${toCompress.length} messages \u2192 summary + ${toKeep.length} recent`));
+                addSpacer();
+                return true;
+            }
+            case '/new': {
+                addSpacer();
+                persistSession(); // save current before clearing
+                agent.clearMessages();
+                agent.setSystemPrompt(systemPrompt);
+                chatContainer.clear();
+                addSystemText(C.emerald('Session cleared') + C.zinc400(' \u2014 fresh start'));
+                addSpacer();
+                tui.requestRender();
+                return true;
+            }
             case '/clear': {
                 chatContainer.clear();
                 tui.requestRender();
@@ -801,6 +977,7 @@ Do NOT make up data. Always call tools to get current state.`;
     function cleanup() {
         if (currentLoader)
             currentLoader.stop();
+        persistSession();
         tui.stop();
         process.exit(0);
     }
@@ -819,9 +996,12 @@ Do NOT make up data. Always call tools to get current state.`;
     // ── Show initial welcome ───────────────────────────────────────────────────
     const thesisText = latestContext.thesis || latestContext.rawThesis || 'N/A';
     const truncatedThesis = thesisText.length > 120 ? thesisText.slice(0, 120) + '...' : thesisText;
+    const sessionStatus = sessionRestored
+        ? C.zinc600(`  resumed (${agent.state.messages.length} messages)`)
+        : C.zinc600('  new session');
     addSystemText(C.zinc600('\u2500'.repeat(50)) + '\n' +
         C.zinc200(bold(truncatedThesis)) + '\n' +
-        C.zinc600(`${latestContext.status || 'active'}  ${confidencePct > 0 ? confidencePct + '%' : ''}  ${(latestContext.edges || []).length} edges`) + '\n' +
+        C.zinc600(`${latestContext.status || 'active'}  ${confidencePct > 0 ? confidencePct + '%' : ''}  ${(latestContext.edges || []).length} edges`) + sessionStatus + '\n' +
         C.zinc600('\u2500'.repeat(50)));
     addSpacer();
     // ── Start TUI ──────────────────────────────────────────────────────────────

package/dist/index.js

@@ -135,11 +135,12 @@ program
 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 <model>', 'Model via OpenRouter (default: anthropic/claude-sonnet-4.6)')
     .option('--model-key <key>', 'OpenRouter API key (or set OPENROUTER_API_KEY)')
+    .option('--new', 'Start a fresh session (default: continue last session)')
     .action(async (thesisId, opts, cmd) => {
     const g = cmd.optsWithGlobals();
-    await run(() => (0, agent_js_1.agentCommand)(thesisId, { model: opts.model, modelKey: opts.modelKey }));
+    await run(() => (0, agent_js_1.agentCommand)(thesisId, { model: opts.model, modelKey: opts.modelKey, newSession: opts.new }));
 });
 // ── Error wrapper ─────────────────────────────────────────────────────────────
 async function run(fn) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@spfunctions/cli",
-  "version": "0.1.6",
+  "version": "1.1.0",
   "description": "CLI for SimpleFunctions prediction market thesis agent",
   "bin": {
     "sf": "./dist/index.js"