feat(settings): add rtk compression toggle, on by default

tatoalo · web-flow · commit cb4d21686459 · 2026-07-08T09:30:59.000+01:00
One boolean in Settings &gt; Advanced, "Compress command output", default on. When off: local and worktree sessions get POSTHOG_RTK=0 on the agent environment (start and reconnect), and cloud runs send rtk_enabled: false into TaskRun.state so the sandbox launches with rtk disabled. When on, nothing is sent and the server-side default (enabled) applies.

Generated-By: PostHog Code
Task-Id: 2d6eed7f-9209-4d1f-be54-ae3e22bf2990
diff --git a/packages/api-client/src/posthog-client.test.ts b/packages/api-client/src/posthog-client.test.ts
@@ -177,6 +177,41 @@ describe("PostHogAPIClient", () => {
     );
   });
 
+  it("serializes an rtk opt-out as rtk_enabled false on run creation", async () => {
+    const fetch = vi.fn().mockResolvedValue({
+      ok: true,
+      json: async () => ({ id: "run-123", environment: "cloud" }),
+    });
+    const client = new PostHogAPIClient(
+      "http://localhost:8000",
+      async () => "token",
+      async () => "token",
+      123,
+    );
+
+    (
+      client as unknown as {
+        api: { baseUrl: string; fetcher: { fetch: typeof fetch } };
+      }
+    ).api = {
+      baseUrl: "http://localhost:8000",
+      fetcher: { fetch },
+    };
+
+    await client.createTaskRun("task-123", {
+      environment: "cloud",
+      mode: "interactive",
+      rtkEnabled: false,
+    });
+
+    const request = fetch.mock.calls[0][0] as {
+      overrides: { body: string };
+    };
+    expect(JSON.parse(request.overrides.body)).toMatchObject({
+      rtk_enabled: false,
+    });
+  });
+
   it("starts an existing cloud task run with run-scoped artifact ids", async () => {
     const fetch = vi.fn().mockResolvedValue({
       ok: true,
diff --git a/packages/api-client/src/posthog-client.ts b/packages/api-client/src/posthog-client.ts
@@ -480,6 +480,8 @@ interface CloudRunOptions {
   sandboxEnvironmentId?: string;
   prAuthorshipMode?: PrAuthorshipMode;
   autoPublish?: boolean;
+  /** Only false is sent: opts the run out of rtk command-output compression. */
+  rtkEnabled?: boolean;
   runSource?: CloudRunSource;
   signalReportId?: string;
   initialPermissionMode?: PermissionMode;
@@ -556,6 +558,9 @@ function buildCloudRunRequestBody(
   if (options?.autoPublish) {
     body.auto_publish = options.autoPublish;
   }
+  if (options?.rtkEnabled === false) {
+    body.rtk_enabled = false;
+  }
   if (options?.runSource) {
     body.run_source = options.runSource;
   }
diff --git a/packages/core/src/sessions/sessionService.ts b/packages/core/src/sessions/sessionService.ts
@@ -298,7 +298,10 @@ export interface SessionServiceDeps {
     setAdapter(taskRunId: string, adapter: Adapter): void;
     removeAdapter(taskRunId: string): void;
   };
-  readonly settings: { customInstructions?: string | null };
+  readonly settings: {
+    customInstructions?: string | null;
+    rtkEnabled?: boolean;
+  };
   usageLimit: { show: (...args: any[]) => any };
   readonly addDirectoryDialog: { open: boolean };
   taskViewedApi: { markActivity(taskId: string): void };
@@ -1022,11 +1025,12 @@ export class SessionService {
           this.d.log.warn("Failed to verify workspace", { taskId, err });
         });
 
-      const { customInstructions } = this.d.settings;
+      const { customInstructions, rtkEnabled } = this.d.settings;
       const result = await this.d.trpc.agent.reconnect.mutate({
         taskId,
         taskRunId,
         repoPath,
+        rtkEnabled,
         apiHost: auth.apiHost,
         projectId: auth.projectId,
         logUrl,
@@ -1357,6 +1361,7 @@ export class SessionService {
       permissionMode: executionMode,
       adapter,
       customInstructions: startCustomInstructions || undefined,
+      rtkEnabled: this.d.settings.rtkEnabled,
       effort: effortLevelSchema.safeParse(reasoningLevel).success
         ? (reasoningLevel as EffortLevel)
         : undefined,
@@ -2994,6 +2999,7 @@ export class SessionService {
             artifactIds.length > 0 ? artifactIds : undefined,
           prAuthorshipMode,
           autoPublish: previousState.auto_publish === true || undefined,
+          rtkEnabled: this.d.settings.rtkEnabled,
           runSource: getCloudRunSource(previousState),
           signalReportId:
             typeof previousState.signal_report_id === "string"
diff --git a/packages/core/src/task-detail/taskCreationApiClient.ts b/packages/core/src/task-detail/taskCreationApiClient.ts
@@ -15,6 +15,7 @@ export interface CreateTaskRunClientOptions {
   sandboxEnvironmentId?: string;
   prAuthorshipMode?: PrAuthorshipMode;
   autoPublish?: boolean;
+  rtkEnabled?: boolean;
   runSource?: CloudRunSource;
   signalReportId?: string;
   initialPermissionMode?: string;
diff --git a/packages/core/src/task-detail/taskCreationSaga.test.ts b/packages/core/src/task-detail/taskCreationSaga.test.ts
@@ -153,6 +153,7 @@ describe("TaskCreationSaga", () => {
       model: "gpt-5.4",
       reasoningLevel: "high",
       cloudAutoPublish: true,
+      rtkEnabled: false,
     });
 
     expect(result.success).toBe(true);
@@ -170,6 +171,7 @@ describe("TaskCreationSaga", () => {
       sandboxEnvironmentId: undefined,
       prAuthorshipMode: "user",
       autoPublish: true,
+      rtkEnabled: false,
       runSource: "manual",
       signalReportId: undefined,
       initialPermissionMode: "auto",
diff --git a/packages/core/src/task-detail/taskCreationSaga.ts b/packages/core/src/task-detail/taskCreationSaga.ts
@@ -402,6 +402,7 @@ export class TaskCreationSaga extends Saga<
             sandboxEnvironmentId: input.sandboxEnvironmentId,
             prAuthorshipMode,
             autoPublish: input.cloudAutoPublish,
+            rtkEnabled: input.rtkEnabled,
             runSource: input.cloudRunSource ?? "manual",
             signalReportId: input.signalReportId,
             homeQuickAction: input.homeQuickActionLabel,
diff --git a/packages/core/src/task-detail/taskInput.ts b/packages/core/src/task-detail/taskInput.ts
@@ -28,6 +28,7 @@ export interface PrepareTaskInputOptions {
   channelId?: string;
   customInstructions?: string;
   autoPublishCloudRuns?: boolean;
+  rtkEnabled?: boolean;
   allowNoRepo?: boolean;
 }
 
@@ -62,6 +63,7 @@ export function prepareTaskInput(
     cloudRunSource:
       options.signalReportId && isCloud ? "signal_report" : undefined,
     cloudAutoPublish: isCloud ? options.autoPublishCloudRuns : undefined,
+    rtkEnabled: options.rtkEnabled,
     signalReportId: options.signalReportId,
     additionalDirectories: isCloud ? undefined : options.additionalDirectories,
     channelContext: options.channelContext,
diff --git a/packages/shared/src/task-creation-domain.ts b/packages/shared/src/task-creation-domain.ts
@@ -42,6 +42,11 @@ export interface TaskCreationInput {
    * completion without waiting for an explicit ask (Settings → Advanced).
    */
   cloudAutoPublish?: boolean;
+  /**
+   * rtk command-output compression for the run. Only false is meaningful:
+   * it opts the run out of the server-side default (enabled).
+   */
+  rtkEnabled?: boolean;
   signalReportId?: string;
   additionalDirectories?: string[];
   /**
diff --git a/packages/ui/src/features/settings/sections/AdvancedSettings.tsx b/packages/ui/src/features/settings/sections/AdvancedSettings.tsx
@@ -27,6 +27,8 @@ export function AdvancedSettings() {
   const setAutoPublishCloudRuns = useSettingsStore(
     (s) => s.setAutoPublishCloudRuns,
   );
+  const rtkEnabled = useSettingsStore((s) => s.rtkEnabled);
+  const setRtkEnabled = useSettingsStore((s) => s.setRtkEnabled);
   const devModeClient = useServiceOptional<DevModeClient>(DEV_MODE_CLIENT);
 
   return (
@@ -41,6 +43,12 @@ export function AdvancedSettings() {
           size="1"
         />
       </SettingRow>
+      <SettingRow
+        label="Compress command output"
+        description="Route eligible shell commands through rtk so their verbose output is compressed before it reaches the model, reducing token usage. Applies to local, worktree, and cloud runs"
+      >
+        <Switch checked={rtkEnabled} onCheckedChange={setRtkEnabled} size="1" />
+      </SettingRow>
       <SettingRow
         label="Reset onboarding and tours"
         description="Re-run the onboarding tutorial and product tours on next app restart"
diff --git a/packages/ui/src/features/settings/settingsStore.ts b/packages/ui/src/features/settings/settingsStore.ts
@@ -153,10 +153,14 @@ interface SettingsStore {
   // When on, cloud runs push their work and open a draft PR on completion
   // without waiting for an explicit ask.
   autoPublishCloudRuns: boolean;
+  // When on, agent runs compress eligible command output through rtk before it
+  // reaches the model. Applies to local, worktree, and cloud runs.
+  rtkEnabled: boolean;
   setAllowBypassPermissions: (enabled: boolean) => void;
   setPreventSleepWhileRunning: (enabled: boolean) => void;
   setDebugLogsCloudRuns: (enabled: boolean) => void;
   setAutoPublishCloudRuns: (enabled: boolean) => void;
+  setRtkEnabled: (enabled: boolean) => void;
 
   // Terminal
   terminalFont: TerminalFont;
@@ -324,13 +328,15 @@ export const useSettingsStore = create<SettingsStore>()(
       preventSleepWhileRunning: false,
       debugLogsCloudRuns: false,
       autoPublishCloudRuns: true,
+      rtkEnabled: true,
       setAllowBypassPermissions: (enabled) =>
         set({ allowBypassPermissions: enabled }),
       setPreventSleepWhileRunning: (enabled) =>
         set({ preventSleepWhileRunning: enabled }),
       setDebugLogsCloudRuns: (enabled) => set({ debugLogsCloudRuns: enabled }),
       setAutoPublishCloudRuns: (enabled) =>
         set({ autoPublishCloudRuns: enabled }),
+      setRtkEnabled: (enabled) => set({ rtkEnabled: enabled }),
 
       // Terminal
       terminalFont: "berkeley-mono",
@@ -444,6 +450,7 @@ export const useSettingsStore = create<SettingsStore>()(
         preventSleepWhileRunning: state.preventSleepWhileRunning,
         debugLogsCloudRuns: state.debugLogsCloudRuns,
         autoPublishCloudRuns: state.autoPublishCloudRuns,
+        rtkEnabled: state.rtkEnabled,
 
         // Terminal
         terminalFont: state.terminalFont,
diff --git a/packages/ui/src/features/task-detail/hooks/useTaskCreation.ts b/packages/ui/src/features/task-detail/hooks/useTaskCreation.ts
@@ -326,6 +326,7 @@ export function useTaskCreation({
           channelId,
           customInstructions: settings.customInstructions,
           autoPublishCloudRuns: settings.autoPublishCloudRuns,
+          rtkEnabled: settings.rtkEnabled,
           allowNoRepo,
         });
 
diff --git a/packages/workspace-server/src/services/agent/agent.ts b/packages/workspace-server/src/services/agent/agent.ts
@@ -283,6 +283,8 @@ interface SessionConfig {
    * replayed to the client. Claude adapter only.
    */
   importedSessionId?: string;
+  /** rtk command-output compression for this session; false opts out. */
+  rtkEnabled?: boolean;
 }
 
 /** Pull the adapter's `agentCapabilities._meta.posthog.steering` from initialize. */
@@ -765,6 +767,15 @@ If a repository IS genuinely required, attach one in this priority order:
       claudeCliPath: this.getClaudeCliPath(),
     });
 
+    // The agent auto-detects rtk on PATH; an explicit opt-out from settings
+    // pins it off for sessions this process spawns. Deleting on the enabled
+    // path restores auto-detection after a re-enable without a restart.
+    if (config.rtkEnabled === false) {
+      process.env.POSTHOG_RTK = "0";
+    } else {
+      delete process.env.POSTHOG_RTK;
+    }
+
     const isPreview = taskId === "__preview__";
 
     const agent = new Agent({
@@ -1992,6 +2003,7 @@ For git operations while detached:
       jsonSchema: "jsonSchema" in params ? params.jsonSchema : undefined,
       importedSessionId:
         "importedSessionId" in params ? params.importedSessionId : undefined,
+      rtkEnabled: "rtkEnabled" in params ? params.rtkEnabled : undefined,
     };
   }
 
diff --git a/packages/workspace-server/src/services/agent/schemas.ts b/packages/workspace-server/src/services/agent/schemas.ts
@@ -75,6 +75,11 @@ export const startSessionInput = z.object({
    * history is replayed to the client. Claude adapter only.
    */
   importedSessionId: z.string().optional(),
+  /**
+   * Whether rtk command-output compression is enabled for this session.
+   * Defaults to enabled; false sets POSTHOG_RTK=0 on the agent environment.
+   */
+  rtkEnabled: z.boolean().optional(),
 });
 
 export type StartSessionInput = z.infer<typeof startSessionInput>;
@@ -206,6 +211,8 @@ export const reconnectSessionInput = z.object({
   customInstructions: z.string().max(2000).optional(),
   effort: effortLevelSchema.optional(),
   jsonSchema: z.record(z.string(), z.unknown()).nullish(),
+  /** See startSessionInput.rtkEnabled. */
+  rtkEnabled: z.boolean().optional(),
 });
 
 export type ReconnectSessionInput = z.infer<typeof reconnectSessionInput>;
