feat: update task run status during local session lifecycle

Author: tatoalo

apps/twig/src/renderer/features/sessions/service/service.test.ts

@@ -61,6 +61,7 @@ const mockAuthStore = vi.hoisted(() => ({
       client: {
         createTaskRun: vi.fn(),
         appendTaskRunLog: vi.fn(),
+        updateTaskRun: vi.fn().mockResolvedValue(undefined),
       },
     })),
   },
@@ -237,6 +238,7 @@ describe("SessionService", () => {
         client: {
           createTaskRun: createTaskRunMock,
           appendTaskRunLog: vi.fn(),
+          updateTaskRun: vi.fn().mockResolvedValue(undefined),
         },
       });
 
@@ -373,6 +375,7 @@ describe("SessionService", () => {
         client: {
           createTaskRun: createTaskRunMock,
           appendTaskRunLog: vi.fn(),
+          updateTaskRun: vi.fn().mockResolvedValue(undefined),
         },
       });
       mockTrpcAgent.start.mutate.mockResolvedValue({

apps/twig/src/renderer/features/sessions/service/service.ts

@@ -390,6 +390,11 @@ export class SessionService {
       throw new Error("Failed to create task run. Please try again.");
     }
 
+    // Signal that the local run is now in progress
+    auth.client
+      .updateTaskRun(taskId, taskRun.id, { status: "in_progress" })
+      .catch((err) => log.warn("Failed to update task run status", { err }));
+
     const result = await trpcVanilla.agent.start.mutate({
       taskId,
       taskRunId: taskRun.id,
@@ -896,6 +901,16 @@ export class SessionService {
       promptStartedAt: Date.now(),
     });
 
+    // Ensure task run status reflects active work
+    const auth = useAuthStore.getState();
+    if (auth.client) {
+      auth.client
+        .updateTaskRun(session.taskId, session.taskRunId, {
+          status: "in_progress",
+        })
+        .catch((err) => log.warn("Failed to update task run status", { err }));
+    }
+
     try {
       const result = await trpcVanilla.agent.prompt.mutate({
         sessionId: session.taskRunId,