Replace reportActivity with recordActivity using taskRunId directly

Author: charlesvien

apps/code/src/main/services/agent/schemas.ts

@@ -183,9 +183,9 @@ export const subscribeSessionInput = z.object({
   taskRunId: z.string(),
 });
 
-// Report activity input — keeps the idle timeout debounce alive for the given task
-export const reportActivityInput = z.object({
-  taskId: z.string().nullable(),
+// Record activity input — keeps the idle timeout debounce alive for the given session
+export const recordActivityInput = z.object({
+  taskRunId: z.string(),
 });
 
 // Agent events

apps/code/src/main/services/agent/service.ts

@@ -397,16 +397,7 @@ export class AgentService extends TypedEventEmitter<AgentServiceEvents> {
     return false;
   }
 
-  public reportActivity(taskId: string | null): void {
-    if (!taskId) return;
-    for (const session of this.sessions.values()) {
-      if (session.taskId === taskId) {
-        this.recordActivity(session.taskRunId);
-      }
-    }
-  }
-
-  private recordActivity(taskRunId: string): void {
+  public recordActivity(taskRunId: string): void {
     const existing = this.idleTimeoutHandles.get(taskRunId);
     if (existing) clearTimeout(existing);
 

apps/code/src/main/trpc/routers/agent.ts

@@ -14,7 +14,7 @@ import {
   promptInput,
   promptOutput,
   reconnectSessionInput,
-  reportActivityInput,
+  recordActivityInput,
   respondToPermissionInput,
   sessionResponseSchema,
   setConfigOptionInput,
@@ -184,9 +184,9 @@ export const agentRouter = router({
     log.info("All sessions reset successfully");
   }),
 
-  reportActivity: publicProcedure
-    .input(reportActivityInput)
-    .mutation(({ input }) => getService().reportActivity(input.taskId)),
+  recordActivity: publicProcedure
+    .input(recordActivityInput)
+    .mutation(({ input }) => getService().recordActivity(input.taskRunId)),
 
   onSessionIdleKilled: publicProcedure.subscription(async function* (opts) {
     const service = getService();

apps/code/src/renderer/features/task-detail/components/TaskLogsPanel.tsx

@@ -133,15 +133,17 @@ export function TaskLogsPanel({ taskId, task }: TaskLogsPanelProps) {
   }, [taskId, requestFocus]);
 
   useEffect(() => {
-    trpcClient.agent.reportActivity.mutate({ taskId }).catch(() => {});
+    const taskRunId = session?.taskRunId;
+    if (!taskRunId) return;
+    trpcClient.agent.recordActivity.mutate({ taskRunId }).catch(() => {});
     const heartbeat = setInterval(
       () => {
-        trpcClient.agent.reportActivity.mutate({ taskId }).catch(() => {});
+        trpcClient.agent.recordActivity.mutate({ taskRunId }).catch(() => {});
       },
       5 * 60 * 1000,
     );
     return () => clearInterval(heartbeat);
-  }, [taskId]);
+  }, [session?.taskRunId]);
 
   // Keep cloud session title aligned with latest task metadata.
   useEffect(() => {