fix(livekit): bound memory background task steps

packages/common/src/base/constants.ts

@@ -24,11 +24,13 @@ export const PochiClientHeader = "x-pochi-client";
 export const PochiRequestUseCaseHeader = "x-pochi-request-use-case";
 
 /**
- * Task Memory thresholds — background extraction of session notes.
+ * Task Memory extracts once per compaction cycle. Everything after the
+ * extraction boundary survives compaction verbatim, so a second extraction in
+ * the same cycle would only shrink that tail; only failures are retried.
  */
-export const TaskMemoryInitTokenThreshold = 10_000;
-export const TaskMemoryUpdateTokenIncrement = 5_000;
-export const TaskMemoryUpdateToolCallThreshold = 3;
+export const TaskMemoryExtractionThresholdRatio = 0.8;
+export const MaxTaskMemoryExtractionAttemptsPerCycle = 2;
+export const TaskMemoryFallbackCompactThreshold = 120_000;
 
 /**
  * Timeout (ms) for any single git operation.

packages/common/src/base/index.ts

@@ -108,6 +108,8 @@ export interface BackgroundTaskState {
   tools?: readonly ToolSpecInput[];
   parentTaskId?: string;
   useCase?: ForkAgentUseCase;
+  /** Maximum number of steps this background task may run. */
+  maxSteps?: number;
   /** Step-start count inherited from the parent, excluded from the max-step guard. */
   baselineStepCount?: number;
 }

packages/common/src/base/memory.ts

@@ -1,9 +1,6 @@
 import type { AutoMemoryContext } from "./prompts/auto-memory";
 
 export interface TaskMemoryState {
-  initialized: boolean;
-  lastExtractionTokens: number;
-  lastExtractionToolCalls: number;
   /**
    * UUID of the last message incorporated into memory.md by the most recent
    * successful extraction. Compaction uses this as the boundary to know
@@ -17,6 +14,8 @@ export interface TaskMemoryState {
    * fork agent writes memory.md successfully.
    */
   pendingExtractionMessageId?: string;
+  extractionAttemptsSinceCompact: number;
+  extractedSinceCompact?: boolean;
   isExtracting: boolean;
   extractionCount: number;
   activeTaskId?: string;

packages/livekit/src/background-task/__tests__/fork-agent.test.ts

@@ -28,6 +28,7 @@ describe("createForkAgent", () => {
       parentMessages,
       parentCwd: "/repo",
       directive: "extract memory",
+      maxSteps: 3,
     }).initMessages;
 
     expect(result).toHaveLength(3);
@@ -50,6 +51,7 @@ describe("createForkAgent", () => {
       parentCwd: "/repo",
       directive: "extract memory",
       tools: ["readFile(/memory/**)", "writeToFile(/memory/**)"],
+      maxSteps: 3,
     });
 
     expect(agent).toMatchObject({
@@ -62,6 +64,7 @@ describe("createForkAgent", () => {
         "writeToFile(/memory/**)",
         "attemptCompletion",
       ],
+      maxSteps: 3,
       baselineStepCount: 0,
     });
     expect(agent).not.toHaveProperty("taskId");
@@ -97,6 +100,7 @@ describe("createForkAgent", () => {
       parentMessages,
       parentCwd: "/repo",
       directive: "extract memory",
+      maxSteps: 3,
     });
 
     expect(agent.baselineStepCount).toBe(3);

packages/livekit/src/background-task/fork-agent.ts

@@ -22,6 +22,7 @@ type ForkAgentInput<TMessage extends UIMessage> = {
   parentCwd: string | undefined;
   directive: string;
   tools?: readonly ToolSpecInput[];
+  maxSteps: number;
 };
 
 export type ForkAgent<TMessage extends UIMessage> = {
@@ -31,6 +32,7 @@ export type ForkAgent<TMessage extends UIMessage> = {
   initTitle: string | undefined;
   parentTaskId: string | undefined;
   tools: readonly ToolSpecInput[] | undefined;
+  maxSteps: number;
   baselineStepCount: number;
 };
 
@@ -94,6 +96,7 @@ export function createForkAgent<TMessage extends UIMessage>(
       parentMessages: input.parentMessages.length,
       initMessages: initMessages.length,
       tools: input.tools?.length,
+      maxSteps: input.maxSteps,
       baselineStepCount,
     },
     "Creating fork agent",
@@ -106,6 +109,7 @@ export function createForkAgent<TMessage extends UIMessage>(
     initTitle: input.initTitle,
     parentTaskId: input.parentTaskId,
     tools,
+    maxSteps: input.maxSteps,
     baselineStepCount,
   };
 }