@getmegabrain/cli 0.1.7 → 0.1.9

package/dist/science/agent/agent.js

@@ -28,17 +28,25 @@ export class AgentSession {
     agentContext;
     cellCounter = 0;
     tools;
-    constructor(model, tools, emit, agentContext = '') {
+    messages;
+    constructor(model, tools, emit, agentContext = '', history = []) {
         this.model = model;
         this.emit = emit;
         this.agentContext = agentContext;
         this.tools = new Map(tools.map(t => [t.definition.name, t]));
+        this.messages = [...history];
+    }
+    /** The full model-visible message history (prior turns + this run) — persist to give the
+     *  session memory across sends. */
+    getMessages() {
+        return this.messages;
     }
     async run(prompt) {
         const system = this.agentContext
             ? `${SYSTEM_PROMPT}\n\nProject context: ${this.agentContext}`
             : SYSTEM_PROMPT;
-        const messages = [{ role: 'user', content: [{ type: 'text', text: prompt }] }];
+        const messages = this.messages;
+        messages.push({ role: 'user', content: [{ type: 'text', text: prompt }] });
         const toolDefs = [...this.tools.values()].map(t => t.definition);
         try {
             for (let turn = 0; turn < MAX_TURNS; turn += 1) {

package/dist/science/kernel_driver.py

@@ -13,9 +13,14 @@ cell are visible in the next.
 import contextlib
 import io
 import json
+import os
 import sys
 import traceback
 
+# Render matplotlib to a non-interactive backend so plt.show() never blocks; we capture
+# figures ourselves after each cell (see emit_figures). Set before any user import.
+os.environ.setdefault("MPLBACKEND", "Agg")
+
 # Persistent namespace shared across all cells in this session.
 NAMESPACE: dict = {"__name__": "__mb_science__"}
 
@@ -25,6 +30,30 @@ def emit(frame: dict) -> None:
     sys.stdout.flush()
 
 
+def emit_figures(cell_id: str) -> int:
+    """Emit any open matplotlib figures as base64 PNG image frames, then close them.
+
+    This is what makes plots appear inline in the transcript (a science-notebook staple).
+    No-op if matplotlib hasn't been used in this session."""
+    pyplot = sys.modules.get("matplotlib.pyplot")
+    if pyplot is None:
+        return 0
+    import base64
+
+    count = 0
+    try:
+        for num in pyplot.get_fignums():
+            buf = io.BytesIO()
+            pyplot.figure(num).savefig(buf, format="png", bbox_inches="tight", dpi=110)
+            data = base64.b64encode(buf.getvalue()).decode("ascii")
+            emit({"type": "image", "cellId": cell_id, "mime": "image/png", "data": data})
+            count += 1
+        pyplot.close("all")
+    except Exception:  # noqa: BLE001 - figure capture is best-effort
+        pass
+    return count
+
+
 def run_cell(cell_id: str, code: str) -> None:
     emit({"type": "status", "cellId": cell_id, "state": "busy"})
     stdout, stderr = io.StringIO(), io.StringIO()
@@ -41,6 +70,11 @@ def run_cell(cell_id: str, code: str) -> None:
         emit({"type": "stream", "cellId": cell_id, "kind": "stdout", "data": out})
     if err:
         emit({"type": "stream", "cellId": cell_id, "kind": "stderr", "data": err})
+    figures = emit_figures(cell_id)
+    if figures:
+        # A note in stdout so the (text-only) model knows a figure was produced.
+        emit({"type": "stream", "cellId": cell_id, "kind": "stdout",
+              "data": f"[{figures} figure(s) displayed]\n"})
     emit({"type": "status", "cellId": cell_id, "state": "idle"})
 
 

package/dist/science/pages.js

@@ -307,6 +307,8 @@ export function sessionPage(projectId, sessionId) {
         white-space:pre-wrap; overflow-x:auto; border-top:1px solid var(--border); }
       .step .out { color:var(--muted); background:var(--bg); }
       .step .out.err { color:#c0392b; }
+      img.figure { max-width:calc(100% - 24px); border-radius:8px; margin:10px 12px; display:block;
+        background:#fff; border:1px solid var(--border); }
       /* Composer */
       .composer { flex-shrink:0; border-top:1px solid var(--border); padding:14px 20px; }
       .composer-inner { max-width:760px; margin:0 auto; display:flex; gap:10px; align-items:flex-end;

package/dist/science/server.js

@@ -224,11 +224,14 @@ export async function startScienceServer(host = '127.0.0.1') {
         if (seg[0] === 'api' &&
             seg[1] === 'sessions' &&
             seg[2] &&
-            (seg[3] === 'events' || seg[3] === 'agent' || seg[3] === 'kernel')) {
+            (seg[3] === 'events' || seg[3] === 'agent' || seg[3] === 'kernel' || seg[3] === 'transcript')) {
             const creds = readCredentials();
             if (!creds)
                 return json(res, { error: 'signed-out' }, 401);
             const sid = seg[2];
+            if (seg[3] === 'transcript' && method === 'GET') {
+                return json(res, store.getMessages(sid));
+            }
             if (seg[3] === 'events' && method === 'GET') {
                 res.writeHead(200, {
                     'content-type': 'text/event-stream',
@@ -261,8 +264,10 @@ export async function startScienceServer(host = '127.0.0.1') {
                 const project = session ? store.getProject(session.projectId) : null;
                 store.touchSession(sid, session && !session.summary ? { summary: prompt } : undefined);
                 fanout(sid, 'agent', { type: 'user', text: prompt });
-                const agent = new AgentSession(new GatewayModel({ apiKey: creds.token }), [pythonTool(kernels.get(sid))], evt => fanout(sid, 'agent', evt), project?.agentContext ?? '');
-                void agent.run(prompt);
+                const agent = new AgentSession(new GatewayModel({ apiKey: creds.token }), [pythonTool(kernels.get(sid))], evt => fanout(sid, 'agent', evt), project?.agentContext ?? '', store.getMessages(sid));
+                // Persist the full history when the turn ends, so the session remembers it and the
+                // transcript survives a refresh.
+                void agent.run(prompt).finally(() => store.setMessages(sid, agent.getMessages()));
                 return json(res, { ok: true });
             }
         }

package/dist/science/session-client.js

@@ -64,12 +64,39 @@
     if (stderr) out.classList.add('err');
     scroll();
   }
+  // Inline a rendered figure (matplotlib PNG) under the current step, or on its own if none.
+  function addFigure(mime, data) {
+    var img = el('img', 'figure');
+    img.src = 'data:' + mime + ';base64,' + data;
+    if (lastCard) lastCard.appendChild(img);
+    else {
+      var row = el('div', 'msg agent');
+      row.appendChild(img);
+      transcript.appendChild(row);
+    }
+    scroll();
+  }
 
   function setRunning(on) {
     sendBtn.disabled = on;
     sendBtn.textContent = on ? 'Running…' : 'Send';
   }
 
+  // Replay a persisted conversation (model messages) into the transcript on load.
+  function renderHistory(messages) {
+    messages.forEach(function (m) {
+      (m.content || []).forEach(function (b) {
+        if (m.role === 'user') {
+          if (b.type === 'text') addUser(b.text);
+          else if (b.type === 'tool_result') addToolResult(b.content, '');
+        } else {
+          if (b.type === 'text') addAgentText(b.text);
+          else if (b.type === 'tool_use') addToolCall(b.name, (b.input && b.input.code) || '');
+        }
+      });
+    });
+  }
+
   // ── Sidebar: this project's sessions ────────────────────────────────
   function loadSidebar() {
     if (!pid) return;
@@ -97,6 +124,7 @@
       if (m.channel === 'kernel') {
         var f = m.payload;
         if (f.type === 'status') kstate.textContent = f.state === 'busy' ? 'running' : f.state;
+        else if (f.type === 'image') addFigure(f.mime, f.data);
         return;
       }
       var p = m.payload;
@@ -150,16 +178,27 @@
       });
   });
 
-  fetch('/api/sessions/' + sid)
+  fetch('/api/sessions/' + sid + '/transcript')
     .then(function (r) {
       return r.json();
     })
-    .then(function (s) {
-      if (s && s.title) $('session-title').textContent = s.title;
-      // Prefill the composer with the seeded first task so the user can just hit Send.
-      if (s && (s.summary || s.title) && !composer.value) composer.value = s.summary || s.title;
+    .then(function (msgs) {
+      var has = Array.isArray(msgs) && msgs.length > 0;
+      if (has) renderHistory(msgs);
+      return fetch('/api/sessions/' + sid)
+        .then(function (r) {
+          return r.json();
+        })
+        .then(function (s) {
+          if (s && s.title) $('session-title').textContent = s.title;
+          // Prefill the composer with the seeded first task only for a fresh session.
+          if (!has && s && (s.summary || s.title) && !composer.value)
+            composer.value = s.summary || s.title;
+        });
+    })
+    .catch(function () {})
+    .then(function () {
+      loadSidebar();
+      connect();
     });
-
-  loadSidebar();
-  connect();
 })();

package/dist/science/store.js

@@ -28,6 +28,8 @@ function defaultSettings() {
         profile: '',
     };
 }
+const MAX_MESSAGES = 60;
+const MAX_TOOL_RESULT_CHARS = 4000;
 const STORE_FILE = path.join(MEGABRAIN_CONFIG_DIR, 'science-workspace.json');
 function id(prefix) {
     return `${prefix}_${randomBytes(8).toString('hex')}`;
@@ -48,6 +50,7 @@ export class WorkspaceStore {
                 projects: parsed.projects ?? [],
                 sessions: parsed.sessions ?? [],
                 settings: parsed.settings,
+                transcripts: parsed.transcripts ?? {},
             };
         }
         catch {
@@ -157,4 +160,19 @@ export class WorkspaceStore {
         this.save();
         return next;
     }
+    // ── Per-session conversation history (agent memory + transcript replay) ──
+    getMessages(sessionId) {
+        return this.data.transcripts?.[sessionId] ?? [];
+    }
+    setMessages(sessionId, messages) {
+        // Cap history length and truncate large tool outputs so the on-disk workspace stays small.
+        const capped = messages.slice(-MAX_MESSAGES).map(m => ({
+            role: m.role,
+            content: m.content.map(b => b.type === 'tool_result' && b.content.length > MAX_TOOL_RESULT_CHARS
+                ? { ...b, content: `${b.content.slice(0, MAX_TOOL_RESULT_CHARS)}\n…(truncated)` }
+                : b),
+        }));
+        (this.data.transcripts ??= {})[sessionId] = capped;
+        this.save();
+    }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@getmegabrain/cli",
-  "version": "0.1.7",
+  "version": "0.1.9",
   "description": "MegaBrain CLI — sign in once, then code from your terminal via OpenCode wired to the MegaBrain Gateway, or run `megabrain science` for the local research workbench.",
   "license": "MIT",
   "type": "module",