@spfunctions/cli 0.1.7 → 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`;
@@ -368,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
@@ -561,6 +597,30 @@ 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 = '';
@@ -633,6 +693,7 @@ Do NOT make up data. Always call tools to get current state.`;
         if (event.type === 'agent_end') {
             // Agent turn fully complete — safe to accept new input
             isProcessing = false;
+            persistSession();
             flushRender();
         }
         if (event.type === 'tool_execution_start') {
@@ -671,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;
             }
@@ -766,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();
@@ -821,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);
     }
@@ -839,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.7",
+  "version": "1.1.0",
   "description": "CLI for SimpleFunctions prediction market thesis agent",
   "bin": {
     "sf": "./dist/index.js"