Prevent auto-generated title from overwriting manual rename

Author: charlesvien

apps/code/src/renderer/features/sessions/hooks/useChatTitleGenerator.ts

@@ -65,8 +65,12 @@ export function useChatTitleGenerator(taskId: string): void {
 
     const run = async () => {
       try {
-        const cachedTasks = queryClient.getQueryData<Task[]>(["tasks", "list"]);
-        const cachedTask = cachedTasks?.find((t) => t.id === taskId);
+        const allTaskQueries = queryClient.getQueriesData<Task[]>({
+          queryKey: ["tasks", "list"],
+        });
+        const cachedTask = allTaskQueries
+          .flatMap(([, tasks]) => tasks ?? [])
+          .find((t) => t.id === taskId);
         if (cachedTask?.title_manually_set) {
           log.debug("Skipping auto-title, user renamed task", { taskId });
           return;

apps/code/src/renderer/sagas/task/task-creation.ts

@@ -39,6 +39,17 @@ async function generateTaskTitle(
   if (!result?.title) return;
   const { title } = result;
 
+  const allTaskQueries = queryClient.getQueriesData<Task[]>({
+    queryKey: ["tasks", "list"],
+  });
+  const cachedTask = allTaskQueries
+    .flatMap(([, tasks]) => tasks ?? [])
+    .find((t) => t.id === taskId);
+  if (cachedTask?.title_manually_set) {
+    log.debug("Skipping auto-title, user renamed task", { taskId });
+    return;
+  }
+
   try {
     await posthogClient.updateTask(taskId, { title });