@getmegabrain/cli 0.1.7 → 0.1.8

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/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

@@ -70,6 +70,21 @@
     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;
@@ -150,16 +165,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.8",
   "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",