feat: /prune

Author: spoons-and-mirrors

packages/opencode/src/cli/cmd/tui/component/prompt/autocomplete.tsx

@@ -219,6 +219,11 @@ export function Autocomplete(props: {
           description: "compact the session",
           onSelect: () => command.trigger("session.compact"),
         },
+        {
+          display: "/prune",
+          description: "prune all tool call outputs",
+          onSelect: () => command.trigger("session.prune"),
+        },
         {
           display: "/unshare",
           disabled: !s.share,

packages/opencode/src/cli/cmd/tui/event.ts

@@ -13,6 +13,7 @@ export const TuiEvent = {
           "session.share",
           "session.interrupt",
           "session.compact",
+          "session.prune",
           "session.page.up",
           "session.page.down",
           "session.half.page.up",

packages/opencode/src/cli/cmd/tui/routes/session/index.tsx

@@ -217,6 +217,23 @@ export function Session() {
         dialog.clear()
       },
     },
+    {
+      title: "Prune session",
+      value: "session.prune",
+      keybind: "session_prune",
+      category: "Session",
+      onSelect: (dialog) => {
+        sdk.client.session
+          .prune({
+            path: {
+              id: route.sessionID,
+            },
+          })
+          .then(() => toast.show({ message: "Session pruned successfully!", variant: "success" }))
+          .catch(() => toast.show({ message: "Failed to prune session", variant: "error" }))
+        dialog.clear()
+      },
+    },
     ...(sync.data.config.share !== "disabled"
       ? [
           {

packages/opencode/src/config/config.ts

@@ -396,6 +396,7 @@ export namespace Config {
       session_unshare: z.string().optional().default("none").describe("Unshare current session"),
       session_interrupt: z.string().optional().default("escape").describe("Interrupt current session"),
       session_compact: z.string().optional().default("<leader>c").describe("Compact the session"),
+      session_prune: z.string().optional().default("none").describe("Prune all tool call outputs from the session"),
       messages_page_up: z.string().optional().default("pageup").describe("Scroll messages up by one page"),
       messages_page_down: z.string().optional().default("pagedown").describe("Scroll messages down by one page"),
       messages_half_page_up: z.string().optional().default("ctrl+alt+u").describe("Scroll messages up by half page"),

packages/opencode/src/server/server.ts

@@ -724,6 +724,35 @@ export namespace Server {
           return c.json(true)
         },
       )
+      .post(
+        "/session/:id/prune",
+        describeRoute({
+          description: "Prune all tool call outputs from the session",
+          operationId: "session.prune",
+          responses: {
+            200: {
+              description: "Pruned session",
+              content: {
+                "application/json": {
+                  schema: resolver(z.boolean()),
+                },
+              },
+            },
+            ...errors(400, 404),
+          },
+        }),
+        validator(
+          "param",
+          z.object({
+            id: z.string().meta({ description: "Session ID" }),
+          }),
+        ),
+        async (c) => {
+          const id = c.req.valid("param").id
+          await SessionCompaction.prune({ sessionID: id, force: true })
+          return c.json(true)
+        },
+      )
       .get(
         "/session/:id/message",
         describeRoute({
@@ -1607,6 +1636,7 @@ export namespace Server {
               session_share: "session.share",
               session_interrupt: "session.interrupt",
               session_compact: "session.compact",
+              session_prune: "session.prune",
               messages_page_up: "session.page.up",
               messages_page_down: "session.page.down",
               messages_half_page_up: "session.half.page.up",

packages/opencode/src/session/compaction.ts

@@ -47,9 +47,9 @@ export namespace SessionCompaction {
   // goes backwards through parts until there are 40_000 tokens worth of tool
   // calls. then erases output of previous tool calls. idea is to throw away old
   // tool calls that are no longer relevant.
-  export async function prune(input: { sessionID: string }) {
+  export async function prune(input: { sessionID: string; force?: boolean }) {
     if (Flag.OPENCODE_DISABLE_PRUNE) return
-    log.info("pruning")
+    log.info("pruning", { force: input.force })
     const msgs = await Session.messages({ sessionID: input.sessionID })
     let total = 0
     let pruned = 0
@@ -68,15 +68,15 @@ export namespace SessionCompaction {
             if (part.state.time.compacted) break loop
             const estimate = Token.estimate(part.state.output)
             total += estimate
-            if (total > PRUNE_PROTECT) {
+            if (input.force || total > PRUNE_PROTECT) {
               pruned += estimate
               toPrune.push(part)
             }
           }
       }
     }
     log.info("found", { pruned, total })
-    if (pruned > PRUNE_MINIMUM) {
+    if (input.force || pruned > PRUNE_MINIMUM) {
       for (const part of toPrune) {
         if (part.state.status === "completed") {
           part.state.time.compacted = Date.now()