Add task branching feature (#15)

1 files changed, 21 insertions, 1 deletions

diff --git a/makima/frontend/src/routes/mesh.tsx b/makima/frontend/src/routes/mesh.tsx
index 142cc54..453bdff 100644
--- a/makima/frontend/src/routes/mesh.tsx
+++ b/makima/frontend/src/routes/mesh.tsx
@@ -9,7 +9,7 @@ import { ContractCompleteQuestion } from "../components/mesh/ContractCompleteQue
 import { useTasks } from "../hooks/useTasks";
 import { useTaskSubscription, type TaskUpdateEvent, type TaskOutputEvent } from "../hooks/useTaskSubscription";
 import type { TaskWithSubtasks, MeshChatContext, ContractSummary, ContractWithRelations, DaemonDirectory, TaskSummary } from "../lib/api";
-import { startTask as startTaskApi, stopTask as stopTaskApi, getTaskOutput, listContracts, getContract, getDaemonDirectories, continueTask as continueTaskApi, resumeSupervisor } from "../lib/api";
+import { startTask as startTaskApi, stopTask as stopTaskApi, getTaskOutput, listContracts, getContract, getDaemonDirectories, continueTask as continueTaskApi, resumeSupervisor, branchTask } from "../lib/api";
 import { DirectoryInput } from "../components/mesh/DirectoryInput";
 import { useAuth } from "../contexts/AuthContext";
 import { useSupervisorQuestions } from "../contexts/SupervisorQuestionsContext";
@@ -461,6 +461,25 @@ export default function MeshPage() {
     [editTask, taskDetail]
   );
 
+  const handleBranch = useCallback(
+    async (taskId: string, message: string, name?: string) => {
+      try {
+        const result = await branchTask(taskId, {
+          message,
+          name,
+          includeConversation: true,
+        });
+        console.log(`[Mesh] Task branched, new task ID: ${result.task.id}`);
+        // Navigate to the new branched task
+        navigate(`/mesh/${result.task.id}`);
+      } catch (e) {
+        console.error("Failed to branch task:", e);
+        throw e; // Re-throw so the modal can display the error
+      }
+    },
+    [navigate]
+  );
+
   // Open contract selection modal
   const handleCreate = useCallback(async () => {
     if (creating || contractsLoading) return;
@@ -742,6 +761,7 @@ export default function MeshPage() {
                       onToggleSubtaskOutput={handleToggleSubtaskOutput}
                       viewingSubtaskId={viewingSubtaskId}
                       onViewContract={(contractId) => navigate(`/contracts/${contractId}`)}
+                      onBranch={handleBranch}
                       contractTasks={taskDetail.isSupervisor ? contractTasks : undefined}
                     />
                   </div>