feat: bump agent to use new research mode (#119)

Author: k11kirky

package.json

@@ -63,7 +63,7 @@
     "@ai-sdk/openai": "^2.0.52",
     "@dnd-kit/react": "^0.1.21",
     "@phosphor-icons/react": "^2.1.10",
-    "@posthog/agent": "1.16.4",
+    "@posthog/agent": "1.17.0",
     "@radix-ui/react-icons": "^1.3.2",
     "@radix-ui/themes": "^3.2.1",
     "@recallai/desktop-sdk": "^1.3.2",

src/main/services/agent.ts

@@ -307,34 +307,4 @@ export function registerAgentIpc(
     },
   );
 
-  ipcMain.handle(
-    "agent-extract-questions",
-    async (
-      _event: IpcMainInvokeEvent,
-      {
-        taskId,
-        repoPath,
-        apiKey,
-        apiHost,
-        projectId,
-      }: {
-        taskId: string;
-        repoPath: string;
-        apiKey: string;
-        apiHost: string;
-        projectId: number;
-      },
-    ): Promise<Array<{ id: string; question: string; options: string[] }>> => {
-      const agent = new Agent({
-        workingDirectory: repoPath,
-        posthogApiKey: apiKey,
-        posthogApiUrl: apiHost,
-        posthogProjectId: projectId,
-        debug: true,
-      });
-
-      const questions = await agent.extractQuestionsFromResearch(taskId, false);
-      return questions;
-    },
-  );
 }

src/main/services/fs.ts

@@ -307,41 +307,50 @@ export function registerFsIpc(): void {
       }>,
     ): Promise<void> => {
       try {
-        const questionsPath = path.join(
+        const researchPath = path.join(
           repoPath,
           ".posthog",
           taskId,
-          "questions.json",
+          "research.json",
         );
 
-        // Read existing questions.json
-        let questionsData: {
-          answered: boolean;
-          answers: Array<{
+        // Read existing research.json
+        let researchData: {
+          actionabilityScore: number;
+          context: string;
+          keyFiles: string[];
+          blockers?: string[];
+          questions?: Array<{
+            id: string;
+            question: string;
+            options: string[];
+          }>;
+          answered?: boolean;
+          answers?: Array<{
             questionId: string;
             selectedOption: string;
             customInput?: string;
           }>;
         };
         try {
-          const content = await fsPromises.readFile(questionsPath, "utf-8");
-          questionsData = JSON.parse(content);
+          const content = await fsPromises.readFile(researchPath, "utf-8");
+          researchData = JSON.parse(content);
         } catch {
-          throw new Error(`questions.json not found for task ${taskId}`);
+          throw new Error(`research.json not found for task ${taskId}`);
         }
 
         // Update with answers
-        questionsData.answered = true;
-        questionsData.answers = answers;
+        researchData.answered = true;
+        researchData.answers = answers;
 
         // Write back to file
         await fsPromises.writeFile(
-          questionsPath,
-          JSON.stringify(questionsData, null, 2),
+          researchPath,
+          JSON.stringify(researchData, null, 2),
           "utf-8",
         );
 
-        console.log(`Saved answers to questions.json for task ${taskId}`);
+        console.log(`Saved answers to research.json for task ${taskId}`);
 
         // Commit the answers (local mode - no push)
         try {

src/renderer/features/tasks/components/TaskDetail.tsx

@@ -176,23 +176,23 @@ export function TaskDetail({ task: initialTask }: TaskDetailProps) {
       addQuestionAnswer(task.id, answer);
     }
 
-    // Save answers to questions.json
+    // Save answers to research.json
     if (repoPath) {
       try {
         await window.electronAPI?.saveQuestionAnswers(
           repoPath,
           task.id,
           answers,
         );
-        console.log("Answers saved to questions.json");
+        console.log("Answers saved to research.json");
 
         // Set phase to planning and trigger next run
         setPlanModePhase(task.id, "planning");
 
         // Trigger the next phase (planning) by running the task again
         runTask(task.id, task);
       } catch (error) {
-        console.error("Failed to save answers to questions.json:", error);
+        console.error("Failed to save answers to research.json:", error);
       }
     }
   };

src/renderer/features/tasks/stores/taskExecutionStore.ts

@@ -641,11 +641,17 @@ export const useTaskExecutionStore = create<TaskExecutionStore>()(
 
         if (taskState.clarifyingQuestions.length > 0) return;
 
-        const artifactEvent = taskState.logs.find(isArtifactEvent);
+        // Look specifically for research_questions artifact
+        const artifactEvent = taskState.logs.find(
+          (log): log is AgentEvent & ArtifactEvent =>
+            isArtifactEvent(log) && 
+            (log as ArtifactEvent).kind === "research_questions"
+        );
+        
         if (!artifactEvent) return;
 
         const event = artifactEvent as ArtifactEvent;
-        if (event.kind === "research_questions" && event.content) {
+        if (event.content) {
           const questions = toClarifyingQuestions(event.content);
           store.setClarifyingQuestions(taskId, questions);
           store.setPlanModePhase(taskId, "questions");