@posthog/agent 1.19.0 → 1.21.0

package/dist/src/workflow/steps/plan.js

@@ -1,6 +1,7 @@
 import { query } from '@anthropic-ai/claude-agent-sdk';
 import { PLANNING_SYSTEM_PROMPT } from '../../agents/planning.js';
 import { finalizeStepGitActions } from '../utils.js';
+import { TodoManager } from '../../todo-manager.js';
 
 const planStep = async ({ step, context }) => {
     const { task, cwd, isCloudMode, options, logger, fileManager, gitManager, promptBuilder, adapter, mcpServers, emitEvent, } = context;
@@ -72,17 +73,23 @@ const planStep = async ({ step, context }) => {
         prompt: fullPrompt,
         options: { ...baseOptions, ...(options.queryOverrides || {}) },
     });
+    const todoManager = new TodoManager(fileManager, stepLogger);
     let planContent = '';
     for await (const message of response) {
         emitEvent(adapter.createRawSDKEvent(message));
-        const transformed = adapter.transform(message);
-        if (transformed) {
-            emitEvent(transformed);
+        const transformedEvents = adapter.transform(message);
+        for (const event of transformedEvents) {
+            emitEvent(event);
         }
+        const todoList = await todoManager.checkAndPersistFromMessage(message, task.id);
+        if (todoList) {
+            emitEvent(adapter.createArtifactEvent('todos', todoList));
+        }
+        // Extract text content for plan
         if (message.type === 'assistant' && message.message?.content) {
-            for (const c of message.message.content) {
-                if (c.type === 'text' && c.text) {
-                    planContent += `${c.text}\n`;
+            for (const block of message.message.content) {
+                if (block.type === 'text' && block.text) {
+                    planContent += `${block.text}\n`;
                 }
             }
         }

package/dist/src/workflow/steps/plan.js.map

@@ -1 +1 @@
-{"version":3,"file":"plan.js","sources":["../../../../src/workflow/steps/plan.ts"],"sourcesContent":["import { query } from '@anthropic-ai/claude-agent-sdk';\nimport { PLANNING_SYSTEM_PROMPT } from '../../agents/planning.js';\nimport type { WorkflowStepRunner } from '../types.js';\nimport { finalizeStepGitActions } from '../utils.js';\n\nexport const planStep: WorkflowStepRunner = async ({ step, context }) => {\n    const {\n        task,\n        cwd,\n        isCloudMode,\n        options,\n        logger,\n        fileManager,\n        gitManager,\n        promptBuilder,\n        adapter,\n        mcpServers,\n        emitEvent,\n    } = context;\n\n    const stepLogger = logger.child('PlanStep');\n\n    const existingPlan = await fileManager.readPlan(task.id);\n    if (existingPlan) {\n        stepLogger.info('Plan already exists, skipping step', { taskId: task.id });\n        return { status: 'skipped' };\n    }\n\n    const researchData = await fileManager.readResearch(task.id);\n    if (researchData?.questions && !researchData.answered) {\n        stepLogger.info('Waiting for answered research questions', { taskId: task.id });\n        emitEvent(adapter.createStatusEvent('phase_complete', { phase: 'research_questions' }));\n        return { status: 'skipped', halt: true };\n    }\n\n    stepLogger.info('Starting planning phase', { taskId: task.id });\n    emitEvent(adapter.createStatusEvent('phase_start', { phase: 'planning' }));\n    let researchContext = '';\n    if (researchData) {\n        researchContext += `## Research Context\\n\\n${researchData.context}\\n\\n`;\n        if (researchData.keyFiles.length > 0) {\n            researchContext += `**Key Files:**\\n${researchData.keyFiles.map(f => `- ${f}`).join('\\n')}\\n\\n`;\n        }\n        if (researchData.blockers && researchData.blockers.length > 0) {\n            researchContext += `**Considerations:**\\n${researchData.blockers.map(b => `- ${b}`).join('\\n')}\\n\\n`;\n        }\n\n        // Add answered questions if they exist\n        if (researchData.questions && researchData.answers && researchData.answered) {\n            researchContext += `## Implementation Decisions\\n\\n`;\n            for (const question of researchData.questions) {\n                const answer = researchData.answers.find(\n                    (a) => a.questionId === question.id\n                );\n\n                researchContext += `### ${question.question}\\n\\n`;\n                if (answer) {\n                    researchContext += `**Selected:** ${answer.selectedOption}\\n`;\n                    if (answer.customInput) {\n                        researchContext += `**Details:** ${answer.customInput}\\n`;\n                    }\n                } else {\n                    researchContext += `**Selected:** Not answered\\n`;\n                }\n                researchContext += `\\n`;\n            }\n        }\n    }\n\n    const planningPrompt = await promptBuilder.buildPlanningPrompt(task, cwd);\n    const fullPrompt = `${PLANNING_SYSTEM_PROMPT}\\n\\n${planningPrompt}\\n\\n${researchContext}`;\n\n    const baseOptions: Record<string, any> = {\n        model: step.model,\n        cwd,\n        permissionMode: 'plan',\n        settingSources: ['local'],\n        mcpServers,\n        // Allow research tools: read-only operations, web search, MCP resources, and ExitPlanMode\n        allowedTools: [\n            'Read',\n            'Glob',\n            'Grep',\n            'WebFetch',\n            'WebSearch',\n            'ListMcpResources',\n            'ReadMcpResource',\n            'ExitPlanMode',\n            'TodoWrite',\n            'BashOutput',\n        ],\n    };\n\n    const response = query({\n        prompt: fullPrompt,\n        options: { ...baseOptions, ...(options.queryOverrides || {}) },\n    });\n\n    let planContent = '';\n    for await (const message of response) {\n        emitEvent(adapter.createRawSDKEvent(message));\n        const transformed = adapter.transform(message);\n        if (transformed) {\n            emitEvent(transformed);\n        }\n        if (message.type === 'assistant' && message.message?.content) {\n            for (const c of message.message.content) {\n                if (c.type === 'text' && c.text) {\n                    planContent += `${c.text}\\n`;\n                }\n            }\n        }\n    }\n\n    if (planContent.trim()) {\n        await fileManager.writePlan(task.id, planContent.trim());\n        stepLogger.info('Plan completed', { taskId: task.id });\n    }\n\n    await gitManager.addAllPostHogFiles();\n    await finalizeStepGitActions(context, step, {\n        commitMessage: `Planning phase for ${task.title}`,\n    });\n\n    if (!isCloudMode) {\n        emitEvent(adapter.createStatusEvent('phase_complete', { phase: 'planning' }));\n        return { status: 'completed', halt: true };\n    }\n\n    emitEvent(adapter.createStatusEvent('phase_complete', { phase: 'planning' }));\n    return { status: 'completed' };\n};\n"],"names":[],"mappings":";;;;AAKO,MAAM,QAAQ,GAAuB,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,KAAI;IACpE,MAAM,EACF,IAAI,EACJ,GAAG,EACH,WAAW,EACX,OAAO,EACP,MAAM,EACN,WAAW,EACX,UAAU,EACV,aAAa,EACb,OAAO,EACP,UAAU,EACV,SAAS,GACZ,GAAG,OAAO;IAEX,MAAM,UAAU,GAAG,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC;IAE3C,MAAM,YAAY,GAAG,MAAM,WAAW,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;IACxD,IAAI,YAAY,EAAE;AACd,QAAA,UAAU,CAAC,IAAI,CAAC,oCAAoC,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC;AAC1E,QAAA,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE;IAChC;IAEA,MAAM,YAAY,GAAG,MAAM,WAAW,CAAC,YAAY,CAAC,IAAI,CAAC,EAAE,CAAC;IAC5D,IAAI,YAAY,EAAE,SAAS,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE;AACnD,QAAA,UAAU,CAAC,IAAI,CAAC,yCAAyC,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC;AAC/E,QAAA,SAAS,CAAC,OAAO,CAAC,iBAAiB,CAAC,gBAAgB,EAAE,EAAE,KAAK,EAAE,oBAAoB,EAAE,CAAC,CAAC;QACvF,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE;IAC5C;AAEA,IAAA,UAAU,CAAC,IAAI,CAAC,yBAAyB,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC;AAC/D,IAAA,SAAS,CAAC,OAAO,CAAC,iBAAiB,CAAC,aAAa,EAAE,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC;IAC1E,IAAI,eAAe,GAAG,EAAE;IACxB,IAAI,YAAY,EAAE;AACd,QAAA,eAAe,IAAI,CAAA,uBAAA,EAA0B,YAAY,CAAC,OAAO,MAAM;QACvE,IAAI,YAAY,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE;YAClC,eAAe,IAAI,mBAAmB,YAAY,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,IAAI,CAAA,EAAA,EAAK,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA,IAAA,CAAM;QACnG;AACA,QAAA,IAAI,YAAY,CAAC,QAAQ,IAAI,YAAY,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE;YAC3D,eAAe,IAAI,wBAAwB,YAAY,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,IAAI,CAAA,EAAA,EAAK,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA,IAAA,CAAM;QACxG;;AAGA,QAAA,IAAI,YAAY,CAAC,SAAS,IAAI,YAAY,CAAC,OAAO,IAAI,YAAY,CAAC,QAAQ,EAAE;YACzE,eAAe,IAAI,iCAAiC;AACpD,YAAA,KAAK,MAAM,QAAQ,IAAI,YAAY,CAAC,SAAS,EAAE;gBAC3C,MAAM,MAAM,GAAG,YAAY,CAAC,OAAO,CAAC,IAAI,CACpC,CAAC,CAAC,KAAK,CAAC,CAAC,UAAU,KAAK,QAAQ,CAAC,EAAE,CACtC;AAED,gBAAA,eAAe,IAAI,CAAA,IAAA,EAAO,QAAQ,CAAC,QAAQ,MAAM;gBACjD,IAAI,MAAM,EAAE;AACR,oBAAA,eAAe,IAAI,CAAA,cAAA,EAAiB,MAAM,CAAC,cAAc,IAAI;AAC7D,oBAAA,IAAI,MAAM,CAAC,WAAW,EAAE;AACpB,wBAAA,eAAe,IAAI,CAAA,aAAA,EAAgB,MAAM,CAAC,WAAW,IAAI;oBAC7D;gBACJ;qBAAO;oBACH,eAAe,IAAI,8BAA8B;gBACrD;gBACA,eAAe,IAAI,IAAI;YAC3B;QACJ;IACJ;IAEA,MAAM,cAAc,GAAG,MAAM,aAAa,CAAC,mBAAmB,CAAC,IAAI,EAAE,GAAG,CAAC;IACzE,MAAM,UAAU,GAAG,CAAA,EAAG,sBAAsB,OAAO,cAAc,CAAA,IAAA,EAAO,eAAe,CAAA,CAAE;AAEzF,IAAA,MAAM,WAAW,GAAwB;QACrC,KAAK,EAAE,IAAI,CAAC,KAAK;QACjB,GAAG;AACH,QAAA,cAAc,EAAE,MAAM;QACtB,cAAc,EAAE,CAAC,OAAO,CAAC;QACzB,UAAU;;AAEV,QAAA,YAAY,EAAE;YACV,MAAM;YACN,MAAM;YACN,MAAM;YACN,UAAU;YACV,WAAW;YACX,kBAAkB;YAClB,iBAAiB;YACjB,cAAc;YACd,WAAW;YACX,YAAY;AACf,SAAA;KACJ;IAED,MAAM,QAAQ,GAAG,KAAK,CAAC;AACnB,QAAA,MAAM,EAAE,UAAU;AAClB,QAAA,OAAO,EAAE,EAAE,GAAG,WAAW,EAAE,IAAI,OAAO,CAAC,cAAc,IAAI,EAAE,CAAC,EAAE;AACjE,KAAA,CAAC;IAEF,IAAI,WAAW,GAAG,EAAE;AACpB,IAAA,WAAW,MAAM,OAAO,IAAI,QAAQ,EAAE;QAClC,SAAS,CAAC,OAAO,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;QAC7C,MAAM,WAAW,GAAG,OAAO,CAAC,SAAS,CAAC,OAAO,CAAC;QAC9C,IAAI,WAAW,EAAE;YACb,SAAS,CAAC,WAAW,CAAC;QAC1B;AACA,QAAA,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,IAAI,OAAO,CAAC,OAAO,EAAE,OAAO,EAAE;YAC1D,KAAK,MAAM,CAAC,IAAI,OAAO,CAAC,OAAO,CAAC,OAAO,EAAE;gBACrC,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM,IAAI,CAAC,CAAC,IAAI,EAAE;AAC7B,oBAAA,WAAW,IAAI,CAAA,EAAG,CAAC,CAAC,IAAI,IAAI;gBAChC;YACJ;QACJ;IACJ;AAEA,IAAA,IAAI,WAAW,CAAC,IAAI,EAAE,EAAE;AACpB,QAAA,MAAM,WAAW,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,EAAE,WAAW,CAAC,IAAI,EAAE,CAAC;AACxD,QAAA,UAAU,CAAC,IAAI,CAAC,gBAAgB,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC;IAC1D;AAEA,IAAA,MAAM,UAAU,CAAC,kBAAkB,EAAE;AACrC,IAAA,MAAM,sBAAsB,CAAC,OAAO,EAAE,IAAI,EAAE;AACxC,QAAA,aAAa,EAAE,CAAA,mBAAA,EAAsB,IAAI,CAAC,KAAK,CAAA,CAAE;AACpD,KAAA,CAAC;IAEF,IAAI,CAAC,WAAW,EAAE;AACd,QAAA,SAAS,CAAC,OAAO,CAAC,iBAAiB,CAAC,gBAAgB,EAAE,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC;QAC7E,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,IAAI,EAAE,IAAI,EAAE;IAC9C;AAEA,IAAA,SAAS,CAAC,OAAO,CAAC,iBAAiB,CAAC,gBAAgB,EAAE,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC;AAC7E,IAAA,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE;AAClC;;;;"}
+{"version":3,"file":"plan.js","sources":["../../../../src/workflow/steps/plan.ts"],"sourcesContent":["import { query } from '@anthropic-ai/claude-agent-sdk';\nimport { PLANNING_SYSTEM_PROMPT } from '../../agents/planning.js';\nimport type { WorkflowStepRunner } from '../types.js';\nimport { finalizeStepGitActions } from '../utils.js';\nimport { TodoManager } from '../../todo-manager.js';\n\nexport const planStep: WorkflowStepRunner = async ({ step, context }) => {\n    const {\n        task,\n        cwd,\n        isCloudMode,\n        options,\n        logger,\n        fileManager,\n        gitManager,\n        promptBuilder,\n        adapter,\n        mcpServers,\n        emitEvent,\n    } = context;\n\n    const stepLogger = logger.child('PlanStep');\n\n    const existingPlan = await fileManager.readPlan(task.id);\n    if (existingPlan) {\n        stepLogger.info('Plan already exists, skipping step', { taskId: task.id });\n        return { status: 'skipped' };\n    }\n\n    const researchData = await fileManager.readResearch(task.id);\n    if (researchData?.questions && !researchData.answered) {\n        stepLogger.info('Waiting for answered research questions', { taskId: task.id });\n        emitEvent(adapter.createStatusEvent('phase_complete', { phase: 'research_questions' }));\n        return { status: 'skipped', halt: true };\n    }\n\n    stepLogger.info('Starting planning phase', { taskId: task.id });\n    emitEvent(adapter.createStatusEvent('phase_start', { phase: 'planning' }));\n    let researchContext = '';\n    if (researchData) {\n        researchContext += `## Research Context\\n\\n${researchData.context}\\n\\n`;\n        if (researchData.keyFiles.length > 0) {\n            researchContext += `**Key Files:**\\n${researchData.keyFiles.map(f => `- ${f}`).join('\\n')}\\n\\n`;\n        }\n        if (researchData.blockers && researchData.blockers.length > 0) {\n            researchContext += `**Considerations:**\\n${researchData.blockers.map(b => `- ${b}`).join('\\n')}\\n\\n`;\n        }\n\n        // Add answered questions if they exist\n        if (researchData.questions && researchData.answers && researchData.answered) {\n            researchContext += `## Implementation Decisions\\n\\n`;\n            for (const question of researchData.questions) {\n                const answer = researchData.answers.find(\n                    (a) => a.questionId === question.id\n                );\n\n                researchContext += `### ${question.question}\\n\\n`;\n                if (answer) {\n                    researchContext += `**Selected:** ${answer.selectedOption}\\n`;\n                    if (answer.customInput) {\n                        researchContext += `**Details:** ${answer.customInput}\\n`;\n                    }\n                } else {\n                    researchContext += `**Selected:** Not answered\\n`;\n                }\n                researchContext += `\\n`;\n            }\n        }\n    }\n\n    const planningPrompt = await promptBuilder.buildPlanningPrompt(task, cwd);\n    const fullPrompt = `${PLANNING_SYSTEM_PROMPT}\\n\\n${planningPrompt}\\n\\n${researchContext}`;\n\n    const baseOptions: Record<string, any> = {\n        model: step.model,\n        cwd,\n        permissionMode: 'plan',\n        settingSources: ['local'],\n        mcpServers,\n        // Allow research tools: read-only operations, web search, MCP resources, and ExitPlanMode\n        allowedTools: [\n            'Read',\n            'Glob',\n            'Grep',\n            'WebFetch',\n            'WebSearch',\n            'ListMcpResources',\n            'ReadMcpResource',\n            'ExitPlanMode',\n            'TodoWrite',\n            'BashOutput',\n        ],\n    };\n\n    const response = query({\n        prompt: fullPrompt,\n        options: { ...baseOptions, ...(options.queryOverrides || {}) },\n    });\n\n    const todoManager = new TodoManager(fileManager, stepLogger);\n\n    let planContent = '';\n    for await (const message of response) {\n        emitEvent(adapter.createRawSDKEvent(message));\n        const transformedEvents = adapter.transform(message);\n        for (const event of transformedEvents) {\n            emitEvent(event);\n        }\n\n        const todoList = await todoManager.checkAndPersistFromMessage(message, task.id);\n        if (todoList) {\n            emitEvent(adapter.createArtifactEvent('todos', todoList));\n        }\n\n        // Extract text content for plan\n        if (message.type === 'assistant' && message.message?.content) {\n            for (const block of message.message.content) {\n                if (block.type === 'text' && block.text) {\n                    planContent += `${block.text}\\n`;\n                }\n            }\n        }\n    }\n\n    if (planContent.trim()) {\n        await fileManager.writePlan(task.id, planContent.trim());\n        stepLogger.info('Plan completed', { taskId: task.id });\n    }\n\n    await gitManager.addAllPostHogFiles();\n    await finalizeStepGitActions(context, step, {\n        commitMessage: `Planning phase for ${task.title}`,\n    });\n\n    if (!isCloudMode) {\n        emitEvent(adapter.createStatusEvent('phase_complete', { phase: 'planning' }));\n        return { status: 'completed', halt: true };\n    }\n\n    emitEvent(adapter.createStatusEvent('phase_complete', { phase: 'planning' }));\n    return { status: 'completed' };\n};\n"],"names":[],"mappings":";;;;;AAMO,MAAM,QAAQ,GAAuB,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,KAAI;IACpE,MAAM,EACF,IAAI,EACJ,GAAG,EACH,WAAW,EACX,OAAO,EACP,MAAM,EACN,WAAW,EACX,UAAU,EACV,aAAa,EACb,OAAO,EACP,UAAU,EACV,SAAS,GACZ,GAAG,OAAO;IAEX,MAAM,UAAU,GAAG,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC;IAE3C,MAAM,YAAY,GAAG,MAAM,WAAW,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;IACxD,IAAI,YAAY,EAAE;AACd,QAAA,UAAU,CAAC,IAAI,CAAC,oCAAoC,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC;AAC1E,QAAA,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE;IAChC;IAEA,MAAM,YAAY,GAAG,MAAM,WAAW,CAAC,YAAY,CAAC,IAAI,CAAC,EAAE,CAAC;IAC5D,IAAI,YAAY,EAAE,SAAS,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE;AACnD,QAAA,UAAU,CAAC,IAAI,CAAC,yCAAyC,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC;AAC/E,QAAA,SAAS,CAAC,OAAO,CAAC,iBAAiB,CAAC,gBAAgB,EAAE,EAAE,KAAK,EAAE,oBAAoB,EAAE,CAAC,CAAC;QACvF,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE;IAC5C;AAEA,IAAA,UAAU,CAAC,IAAI,CAAC,yBAAyB,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC;AAC/D,IAAA,SAAS,CAAC,OAAO,CAAC,iBAAiB,CAAC,aAAa,EAAE,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC;IAC1E,IAAI,eAAe,GAAG,EAAE;IACxB,IAAI,YAAY,EAAE;AACd,QAAA,eAAe,IAAI,CAAA,uBAAA,EAA0B,YAAY,CAAC,OAAO,MAAM;QACvE,IAAI,YAAY,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE;YAClC,eAAe,IAAI,mBAAmB,YAAY,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,IAAI,CAAA,EAAA,EAAK,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA,IAAA,CAAM;QACnG;AACA,QAAA,IAAI,YAAY,CAAC,QAAQ,IAAI,YAAY,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE;YAC3D,eAAe,IAAI,wBAAwB,YAAY,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,IAAI,CAAA,EAAA,EAAK,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA,IAAA,CAAM;QACxG;;AAGA,QAAA,IAAI,YAAY,CAAC,SAAS,IAAI,YAAY,CAAC,OAAO,IAAI,YAAY,CAAC,QAAQ,EAAE;YACzE,eAAe,IAAI,iCAAiC;AACpD,YAAA,KAAK,MAAM,QAAQ,IAAI,YAAY,CAAC,SAAS,EAAE;gBAC3C,MAAM,MAAM,GAAG,YAAY,CAAC,OAAO,CAAC,IAAI,CACpC,CAAC,CAAC,KAAK,CAAC,CAAC,UAAU,KAAK,QAAQ,CAAC,EAAE,CACtC;AAED,gBAAA,eAAe,IAAI,CAAA,IAAA,EAAO,QAAQ,CAAC,QAAQ,MAAM;gBACjD,IAAI,MAAM,EAAE;AACR,oBAAA,eAAe,IAAI,CAAA,cAAA,EAAiB,MAAM,CAAC,cAAc,IAAI;AAC7D,oBAAA,IAAI,MAAM,CAAC,WAAW,EAAE;AACpB,wBAAA,eAAe,IAAI,CAAA,aAAA,EAAgB,MAAM,CAAC,WAAW,IAAI;oBAC7D;gBACJ;qBAAO;oBACH,eAAe,IAAI,8BAA8B;gBACrD;gBACA,eAAe,IAAI,IAAI;YAC3B;QACJ;IACJ;IAEA,MAAM,cAAc,GAAG,MAAM,aAAa,CAAC,mBAAmB,CAAC,IAAI,EAAE,GAAG,CAAC;IACzE,MAAM,UAAU,GAAG,CAAA,EAAG,sBAAsB,OAAO,cAAc,CAAA,IAAA,EAAO,eAAe,CAAA,CAAE;AAEzF,IAAA,MAAM,WAAW,GAAwB;QACrC,KAAK,EAAE,IAAI,CAAC,KAAK;QACjB,GAAG;AACH,QAAA,cAAc,EAAE,MAAM;QACtB,cAAc,EAAE,CAAC,OAAO,CAAC;QACzB,UAAU;;AAEV,QAAA,YAAY,EAAE;YACV,MAAM;YACN,MAAM;YACN,MAAM;YACN,UAAU;YACV,WAAW;YACX,kBAAkB;YAClB,iBAAiB;YACjB,cAAc;YACd,WAAW;YACX,YAAY;AACf,SAAA;KACJ;IAED,MAAM,QAAQ,GAAG,KAAK,CAAC;AACnB,QAAA,MAAM,EAAE,UAAU;AAClB,QAAA,OAAO,EAAE,EAAE,GAAG,WAAW,EAAE,IAAI,OAAO,CAAC,cAAc,IAAI,EAAE,CAAC,EAAE;AACjE,KAAA,CAAC;IAEF,MAAM,WAAW,GAAG,IAAI,WAAW,CAAC,WAAW,EAAE,UAAU,CAAC;IAE5D,IAAI,WAAW,GAAG,EAAE;AACpB,IAAA,WAAW,MAAM,OAAO,IAAI,QAAQ,EAAE;QAClC,SAAS,CAAC,OAAO,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;QAC7C,MAAM,iBAAiB,GAAG,OAAO,CAAC,SAAS,CAAC,OAAO,CAAC;AACpD,QAAA,KAAK,MAAM,KAAK,IAAI,iBAAiB,EAAE;YACnC,SAAS,CAAC,KAAK,CAAC;QACpB;AAEA,QAAA,MAAM,QAAQ,GAAG,MAAM,WAAW,CAAC,0BAA0B,CAAC,OAAO,EAAE,IAAI,CAAC,EAAE,CAAC;QAC/E,IAAI,QAAQ,EAAE;YACV,SAAS,CAAC,OAAO,CAAC,mBAAmB,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;QAC7D;;AAGA,QAAA,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,IAAI,OAAO,CAAC,OAAO,EAAE,OAAO,EAAE;YAC1D,KAAK,MAAM,KAAK,IAAI,OAAO,CAAC,OAAO,CAAC,OAAO,EAAE;gBACzC,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,IAAI,KAAK,CAAC,IAAI,EAAE;AACrC,oBAAA,WAAW,IAAI,CAAA,EAAG,KAAK,CAAC,IAAI,IAAI;gBACpC;YACJ;QACJ;IACJ;AAEA,IAAA,IAAI,WAAW,CAAC,IAAI,EAAE,EAAE;AACpB,QAAA,MAAM,WAAW,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,EAAE,WAAW,CAAC,IAAI,EAAE,CAAC;AACxD,QAAA,UAAU,CAAC,IAAI,CAAC,gBAAgB,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC;IAC1D;AAEA,IAAA,MAAM,UAAU,CAAC,kBAAkB,EAAE;AACrC,IAAA,MAAM,sBAAsB,CAAC,OAAO,EAAE,IAAI,EAAE;AACxC,QAAA,aAAa,EAAE,CAAA,mBAAA,EAAsB,IAAI,CAAC,KAAK,CAAA,CAAE;AACpD,KAAA,CAAC;IAEF,IAAI,CAAC,WAAW,EAAE;AACd,QAAA,SAAS,CAAC,OAAO,CAAC,iBAAiB,CAAC,gBAAgB,EAAE,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC;QAC7E,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,IAAI,EAAE,IAAI,EAAE;IAC9C;AAEA,IAAA,SAAS,CAAC,OAAO,CAAC,iBAAiB,CAAC,gBAAgB,EAAE,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC;AAC7E,IAAA,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE;AAClC;;;;"}

package/dist/src/workflow/steps/research.js

@@ -58,9 +58,9 @@ const researchStep = async ({ step, context }) => {
     let jsonContent = '';
     for await (const message of response) {
         emitEvent(adapter.createRawSDKEvent(message));
-        const transformed = adapter.transform(message);
-        if (transformed) {
-            emitEvent(transformed);
+        const transformedEvents = adapter.transform(message);
+        for (const event of transformedEvents) {
+            emitEvent(event);
         }
         if (message.type === 'assistant' && message.message?.content) {
             for (const c of message.message.content) {

package/dist/src/workflow/steps/research.js.map

@@ -1 +1 @@
-{"version":3,"file":"research.js","sources":["../../../../src/workflow/steps/research.ts"],"sourcesContent":["import { query } from '@anthropic-ai/claude-agent-sdk';\nimport { RESEARCH_SYSTEM_PROMPT } from '../../agents/research.js';\nimport type { WorkflowStepRunner } from '../types.js';\nimport type { ResearchEvaluation } from '../../types.js';\nimport { finalizeStepGitActions } from '../utils.js';\n\nexport const researchStep: WorkflowStepRunner = async ({ step, context }) => {\n    const {\n        task,\n        cwd,\n        isCloudMode,\n        options,\n        logger,\n        fileManager,\n        gitManager,\n        promptBuilder,\n        adapter,\n        mcpServers,\n        emitEvent,\n    } = context;\n\n    const stepLogger = logger.child('ResearchStep');\n\n    const existingResearch = await fileManager.readResearch(task.id);\n    if (existingResearch) {\n        stepLogger.info('Research already exists', { taskId: task.id, hasQuestions: !!existingResearch.questions, answered: existingResearch.answered });\n        \n        // If there are unanswered questions, re-emit them so UI can prompt user\n        if (existingResearch.questions && !existingResearch.answered) {\n            stepLogger.info('Re-emitting unanswered research questions', { \n                taskId: task.id,\n                questionCount: existingResearch.questions.length \n            });\n            \n            emitEvent({\n                type: 'artifact',\n                ts: Date.now(),\n                kind: 'research_questions',\n                content: existingResearch.questions,\n            });\n            \n            // In local mode, halt to allow user to answer\n            if (!isCloudMode) {\n                emitEvent(adapter.createStatusEvent('phase_complete', { phase: 'research' }));\n                return { status: 'skipped', halt: true };\n            }\n        }\n        \n        return { status: 'skipped' };\n    }\n\n    stepLogger.info('Starting research phase', { taskId: task.id });\n    emitEvent(adapter.createStatusEvent('phase_start', { phase: 'research' }));\n\n    const researchPrompt = await promptBuilder.buildResearchPrompt(task, cwd);\n    const fullPrompt = `${RESEARCH_SYSTEM_PROMPT}\\n\\n${researchPrompt}`;\n\n    const baseOptions: Record<string, any> = {\n        model: step.model,\n        cwd,\n        permissionMode: 'plan',\n        settingSources: ['local'],\n        mcpServers,\n        // Allow research tools: read-only operations, web search, and MCP resources\n        allowedTools: [\n            'Read',\n            'Glob',\n            'Grep',\n            'WebFetch',\n            'WebSearch',\n            'ListMcpResources',\n            'ReadMcpResource',\n            'TodoWrite',\n            'BashOutput',\n        ],\n    };\n\n    const response = query({\n        prompt: fullPrompt,\n        options: { ...baseOptions, ...(options.queryOverrides || {}) },\n    });\n\n    let jsonContent = '';\n    for await (const message of response) {\n        emitEvent(adapter.createRawSDKEvent(message));\n        const transformed = adapter.transform(message);\n        if (transformed) {\n            emitEvent(transformed);\n        }\n        if (message.type === 'assistant' && message.message?.content) {\n            for (const c of message.message.content) {\n                if (c.type === 'text' && c.text) {\n                    jsonContent += c.text;\n                }\n            }\n        }\n    }\n\n    if (!jsonContent.trim()) {\n        stepLogger.error('No JSON output from research agent', { taskId: task.id });\n        emitEvent({\n            type: 'error',\n            ts: Date.now(),\n            message: 'Research agent returned no output',\n        });\n        return { status: 'completed', halt: true };\n    }\n\n    // Parse JSON response\n    let evaluation: ResearchEvaluation;\n    try {\n        // Extract JSON from potential markdown code blocks or other wrapping\n        const jsonMatch = jsonContent.match(/\\{[\\s\\S]*\\}/);\n        if (!jsonMatch) {\n            throw new Error('No JSON object found in response');\n        }\n        evaluation = JSON.parse(jsonMatch[0]);\n        stepLogger.info('Parsed research evaluation', {\n            taskId: task.id,\n            score: evaluation.actionabilityScore,\n            hasQuestions: !!evaluation.questions,\n        });\n    } catch (error) {\n        stepLogger.error('Failed to parse research JSON', {\n            taskId: task.id,\n            error: error instanceof Error ? error.message : String(error),\n            content: jsonContent.substring(0, 500),\n        });\n        emitEvent({\n            type: 'error',\n            ts: Date.now(),\n            message: `Failed to parse research JSON: ${\n                error instanceof Error ? error.message : String(error)\n            }`,\n        });\n        return { status: 'completed', halt: true };\n    }\n\n    // Add answered/answers fields to evaluation\n    if (evaluation.questions && evaluation.questions.length > 0) {\n        evaluation.answered = false;\n        evaluation.answers = undefined;\n    }\n\n    // Always write research.json\n    await fileManager.writeResearch(task.id, evaluation);\n    stepLogger.info('Research evaluation written', {\n        taskId: task.id,\n        score: evaluation.actionabilityScore,\n        hasQuestions: !!evaluation.questions,\n    });\n\n    emitEvent({\n        type: 'artifact',\n        ts: Date.now(),\n        kind: 'research_evaluation',\n        content: evaluation,\n    });\n\n    await gitManager.addAllPostHogFiles();\n    await finalizeStepGitActions(context, step, {\n        commitMessage: `Research phase for ${task.title}`,\n    });\n\n    // Log whether questions need answering\n    if (evaluation.actionabilityScore < 0.7 && evaluation.questions && evaluation.questions.length > 0) {\n        stepLogger.info('Actionability score below threshold, questions needed', {\n            taskId: task.id,\n            score: evaluation.actionabilityScore,\n            questionCount: evaluation.questions.length,\n        });\n        \n        emitEvent({\n            type: 'artifact',\n            ts: Date.now(),\n            kind: 'research_questions',\n            content: evaluation.questions,\n        });\n    } else {\n        stepLogger.info('Actionability score acceptable, proceeding to planning', {\n            taskId: task.id,\n            score: evaluation.actionabilityScore,\n        });\n    }\n\n    // In local mode, always halt after research for user review\n    if (!isCloudMode) {\n        emitEvent(adapter.createStatusEvent('phase_complete', { phase: 'research' }));\n        return { status: 'completed', halt: true };\n    }\n\n    // In cloud mode, check if questions need answering\n    const researchData = await fileManager.readResearch(task.id);\n    if (researchData?.questions && !researchData.answered) {\n        // Questions need answering - halt for user input in cloud mode too\n        emitEvent(adapter.createStatusEvent('phase_complete', { phase: 'research' }));\n        return { status: 'completed', halt: true };\n    }\n\n    // No questions or questions already answered - proceed to planning\n    emitEvent(adapter.createStatusEvent('phase_complete', { phase: 'research' }));\n    return { status: 'completed' };\n};\n"],"names":[],"mappings":";;;;AAMO,MAAM,YAAY,GAAuB,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,KAAI;IACxE,MAAM,EACF,IAAI,EACJ,GAAG,EACH,WAAW,EACX,OAAO,EACP,MAAM,EACN,WAAW,EACX,UAAU,EACV,aAAa,EACb,OAAO,EACP,UAAU,EACV,SAAS,GACZ,GAAG,OAAO;IAEX,MAAM,UAAU,GAAG,MAAM,CAAC,KAAK,CAAC,cAAc,CAAC;IAE/C,MAAM,gBAAgB,GAAG,MAAM,WAAW,CAAC,YAAY,CAAC,IAAI,CAAC,EAAE,CAAC;IAChE,IAAI,gBAAgB,EAAE;QAClB,UAAU,CAAC,IAAI,CAAC,yBAAyB,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,YAAY,EAAE,CAAC,CAAC,gBAAgB,CAAC,SAAS,EAAE,QAAQ,EAAE,gBAAgB,CAAC,QAAQ,EAAE,CAAC;;QAGhJ,IAAI,gBAAgB,CAAC,SAAS,IAAI,CAAC,gBAAgB,CAAC,QAAQ,EAAE;AAC1D,YAAA,UAAU,CAAC,IAAI,CAAC,2CAA2C,EAAE;gBACzD,MAAM,EAAE,IAAI,CAAC,EAAE;AACf,gBAAA,aAAa,EAAE,gBAAgB,CAAC,SAAS,CAAC;AAC7C,aAAA,CAAC;AAEF,YAAA,SAAS,CAAC;AACN,gBAAA,IAAI,EAAE,UAAU;AAChB,gBAAA,EAAE,EAAE,IAAI,CAAC,GAAG,EAAE;AACd,gBAAA,IAAI,EAAE,oBAAoB;gBAC1B,OAAO,EAAE,gBAAgB,CAAC,SAAS;AACtC,aAAA,CAAC;;YAGF,IAAI,CAAC,WAAW,EAAE;AACd,gBAAA,SAAS,CAAC,OAAO,CAAC,iBAAiB,CAAC,gBAAgB,EAAE,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC;gBAC7E,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE;YAC5C;QACJ;AAEA,QAAA,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE;IAChC;AAEA,IAAA,UAAU,CAAC,IAAI,CAAC,yBAAyB,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC;AAC/D,IAAA,SAAS,CAAC,OAAO,CAAC,iBAAiB,CAAC,aAAa,EAAE,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC;IAE1E,MAAM,cAAc,GAAG,MAAM,aAAa,CAAC,mBAAmB,CAAC,IAAI,EAAE,GAAG,CAAC;AACzE,IAAA,MAAM,UAAU,GAAG,CAAA,EAAG,sBAAsB,CAAA,IAAA,EAAO,cAAc,EAAE;AAEnE,IAAA,MAAM,WAAW,GAAwB;QACrC,KAAK,EAAE,IAAI,CAAC,KAAK;QACjB,GAAG;AACH,QAAA,cAAc,EAAE,MAAM;QACtB,cAAc,EAAE,CAAC,OAAO,CAAC;QACzB,UAAU;;AAEV,QAAA,YAAY,EAAE;YACV,MAAM;YACN,MAAM;YACN,MAAM;YACN,UAAU;YACV,WAAW;YACX,kBAAkB;YAClB,iBAAiB;YACjB,WAAW;YACX,YAAY;AACf,SAAA;KACJ;IAED,MAAM,QAAQ,GAAG,KAAK,CAAC;AACnB,QAAA,MAAM,EAAE,UAAU;AAClB,QAAA,OAAO,EAAE,EAAE,GAAG,WAAW,EAAE,IAAI,OAAO,CAAC,cAAc,IAAI,EAAE,CAAC,EAAE;AACjE,KAAA,CAAC;IAEF,IAAI,WAAW,GAAG,EAAE;AACpB,IAAA,WAAW,MAAM,OAAO,IAAI,QAAQ,EAAE;QAClC,SAAS,CAAC,OAAO,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;QAC7C,MAAM,WAAW,GAAG,OAAO,CAAC,SAAS,CAAC,OAAO,CAAC;QAC9C,IAAI,WAAW,EAAE;YACb,SAAS,CAAC,WAAW,CAAC;QAC1B;AACA,QAAA,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,IAAI,OAAO,CAAC,OAAO,EAAE,OAAO,EAAE;YAC1D,KAAK,MAAM,CAAC,IAAI,OAAO,CAAC,OAAO,CAAC,OAAO,EAAE;gBACrC,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM,IAAI,CAAC,CAAC,IAAI,EAAE;AAC7B,oBAAA,WAAW,IAAI,CAAC,CAAC,IAAI;gBACzB;YACJ;QACJ;IACJ;AAEA,IAAA,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,EAAE;AACrB,QAAA,UAAU,CAAC,KAAK,CAAC,oCAAoC,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC;AAC3E,QAAA,SAAS,CAAC;AACN,YAAA,IAAI,EAAE,OAAO;AACb,YAAA,EAAE,EAAE,IAAI,CAAC,GAAG,EAAE;AACd,YAAA,OAAO,EAAE,mCAAmC;AAC/C,SAAA,CAAC;QACF,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,IAAI,EAAE,IAAI,EAAE;IAC9C;;AAGA,IAAA,IAAI,UAA8B;AAClC,IAAA,IAAI;;QAEA,MAAM,SAAS,GAAG,WAAW,CAAC,KAAK,CAAC,aAAa,CAAC;QAClD,IAAI,CAAC,SAAS,EAAE;AACZ,YAAA,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC;QACvD;QACA,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;AACrC,QAAA,UAAU,CAAC,IAAI,CAAC,4BAA4B,EAAE;YAC1C,MAAM,EAAE,IAAI,CAAC,EAAE;YACf,KAAK,EAAE,UAAU,CAAC,kBAAkB;AACpC,YAAA,YAAY,EAAE,CAAC,CAAC,UAAU,CAAC,SAAS;AACvC,SAAA,CAAC;IACN;IAAE,OAAO,KAAK,EAAE;AACZ,QAAA,UAAU,CAAC,KAAK,CAAC,+BAA+B,EAAE;YAC9C,MAAM,EAAE,IAAI,CAAC,EAAE;AACf,YAAA,KAAK,EAAE,KAAK,YAAY,KAAK,GAAG,KAAK,CAAC,OAAO,GAAG,MAAM,CAAC,KAAK,CAAC;YAC7D,OAAO,EAAE,WAAW,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC;AACzC,SAAA,CAAC;AACF,QAAA,SAAS,CAAC;AACN,YAAA,IAAI,EAAE,OAAO;AACb,YAAA,EAAE,EAAE,IAAI,CAAC,GAAG,EAAE;AACd,YAAA,OAAO,EAAE,CAAA,+BAAA,EACL,KAAK,YAAY,KAAK,GAAG,KAAK,CAAC,OAAO,GAAG,MAAM,CAAC,KAAK,CACzD,CAAA,CAAE;AACL,SAAA,CAAC;QACF,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,IAAI,EAAE,IAAI,EAAE;IAC9C;;AAGA,IAAA,IAAI,UAAU,CAAC,SAAS,IAAI,UAAU,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE;AACzD,QAAA,UAAU,CAAC,QAAQ,GAAG,KAAK;AAC3B,QAAA,UAAU,CAAC,OAAO,GAAG,SAAS;IAClC;;IAGA,MAAM,WAAW,CAAC,aAAa,CAAC,IAAI,CAAC,EAAE,EAAE,UAAU,CAAC;AACpD,IAAA,UAAU,CAAC,IAAI,CAAC,6BAA6B,EAAE;QAC3C,MAAM,EAAE,IAAI,CAAC,EAAE;QACf,KAAK,EAAE,UAAU,CAAC,kBAAkB;AACpC,QAAA,YAAY,EAAE,CAAC,CAAC,UAAU,CAAC,SAAS;AACvC,KAAA,CAAC;AAEF,IAAA,SAAS,CAAC;AACN,QAAA,IAAI,EAAE,UAAU;AAChB,QAAA,EAAE,EAAE,IAAI,CAAC,GAAG,EAAE;AACd,QAAA,IAAI,EAAE,qBAAqB;AAC3B,QAAA,OAAO,EAAE,UAAU;AACtB,KAAA,CAAC;AAEF,IAAA,MAAM,UAAU,CAAC,kBAAkB,EAAE;AACrC,IAAA,MAAM,sBAAsB,CAAC,OAAO,EAAE,IAAI,EAAE;AACxC,QAAA,aAAa,EAAE,CAAA,mBAAA,EAAsB,IAAI,CAAC,KAAK,CAAA,CAAE;AACpD,KAAA,CAAC;;AAGF,IAAA,IAAI,UAAU,CAAC,kBAAkB,GAAG,GAAG,IAAI,UAAU,CAAC,SAAS,IAAI,UAAU,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE;AAChG,QAAA,UAAU,CAAC,IAAI,CAAC,uDAAuD,EAAE;YACrE,MAAM,EAAE,IAAI,CAAC,EAAE;YACf,KAAK,EAAE,UAAU,CAAC,kBAAkB;AACpC,YAAA,aAAa,EAAE,UAAU,CAAC,SAAS,CAAC,MAAM;AAC7C,SAAA,CAAC;AAEF,QAAA,SAAS,CAAC;AACN,YAAA,IAAI,EAAE,UAAU;AAChB,YAAA,EAAE,EAAE,IAAI,CAAC,GAAG,EAAE;AACd,YAAA,IAAI,EAAE,oBAAoB;YAC1B,OAAO,EAAE,UAAU,CAAC,SAAS;AAChC,SAAA,CAAC;IACN;SAAO;AACH,QAAA,UAAU,CAAC,IAAI,CAAC,wDAAwD,EAAE;YACtE,MAAM,EAAE,IAAI,CAAC,EAAE;YACf,KAAK,EAAE,UAAU,CAAC,kBAAkB;AACvC,SAAA,CAAC;IACN;;IAGA,IAAI,CAAC,WAAW,EAAE;AACd,QAAA,SAAS,CAAC,OAAO,CAAC,iBAAiB,CAAC,gBAAgB,EAAE,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC;QAC7E,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,IAAI,EAAE,IAAI,EAAE;IAC9C;;IAGA,MAAM,YAAY,GAAG,MAAM,WAAW,CAAC,YAAY,CAAC,IAAI,CAAC,EAAE,CAAC;IAC5D,IAAI,YAAY,EAAE,SAAS,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE;;AAEnD,QAAA,SAAS,CAAC,OAAO,CAAC,iBAAiB,CAAC,gBAAgB,EAAE,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC;QAC7E,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,IAAI,EAAE,IAAI,EAAE;IAC9C;;AAGA,IAAA,SAAS,CAAC,OAAO,CAAC,iBAAiB,CAAC,gBAAgB,EAAE,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC;AAC7E,IAAA,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE;AAClC;;;;"}
+{"version":3,"file":"research.js","sources":["../../../../src/workflow/steps/research.ts"],"sourcesContent":["import { query } from '@anthropic-ai/claude-agent-sdk';\nimport { RESEARCH_SYSTEM_PROMPT } from '../../agents/research.js';\nimport type { WorkflowStepRunner } from '../types.js';\nimport type { ResearchEvaluation } from '../../types.js';\nimport { finalizeStepGitActions } from '../utils.js';\n\nexport const researchStep: WorkflowStepRunner = async ({ step, context }) => {\n    const {\n        task,\n        cwd,\n        isCloudMode,\n        options,\n        logger,\n        fileManager,\n        gitManager,\n        promptBuilder,\n        adapter,\n        mcpServers,\n        emitEvent,\n    } = context;\n\n    const stepLogger = logger.child('ResearchStep');\n\n    const existingResearch = await fileManager.readResearch(task.id);\n    if (existingResearch) {\n        stepLogger.info('Research already exists', { taskId: task.id, hasQuestions: !!existingResearch.questions, answered: existingResearch.answered });\n        \n        // If there are unanswered questions, re-emit them so UI can prompt user\n        if (existingResearch.questions && !existingResearch.answered) {\n            stepLogger.info('Re-emitting unanswered research questions', { \n                taskId: task.id,\n                questionCount: existingResearch.questions.length \n            });\n            \n            emitEvent({\n                type: 'artifact',\n                ts: Date.now(),\n                kind: 'research_questions',\n                content: existingResearch.questions,\n            });\n            \n            // In local mode, halt to allow user to answer\n            if (!isCloudMode) {\n                emitEvent(adapter.createStatusEvent('phase_complete', { phase: 'research' }));\n                return { status: 'skipped', halt: true };\n            }\n        }\n        \n        return { status: 'skipped' };\n    }\n\n    stepLogger.info('Starting research phase', { taskId: task.id });\n    emitEvent(adapter.createStatusEvent('phase_start', { phase: 'research' }));\n\n    const researchPrompt = await promptBuilder.buildResearchPrompt(task, cwd);\n    const fullPrompt = `${RESEARCH_SYSTEM_PROMPT}\\n\\n${researchPrompt}`;\n\n    const baseOptions: Record<string, any> = {\n        model: step.model,\n        cwd,\n        permissionMode: 'plan',\n        settingSources: ['local'],\n        mcpServers,\n        // Allow research tools: read-only operations, web search, and MCP resources\n        allowedTools: [\n            'Read',\n            'Glob',\n            'Grep',\n            'WebFetch',\n            'WebSearch',\n            'ListMcpResources',\n            'ReadMcpResource',\n            'TodoWrite',\n            'BashOutput',\n        ],\n    };\n\n    const response = query({\n        prompt: fullPrompt,\n        options: { ...baseOptions, ...(options.queryOverrides || {}) },\n    });\n\n    let jsonContent = '';\n    for await (const message of response) {\n        emitEvent(adapter.createRawSDKEvent(message));\n        const transformedEvents = adapter.transform(message);\n        for (const event of transformedEvents) {\n            emitEvent(event);\n        }\n        if (message.type === 'assistant' && message.message?.content) {\n            for (const c of message.message.content) {\n                if (c.type === 'text' && c.text) {\n                    jsonContent += c.text;\n                }\n            }\n        }\n    }\n\n    if (!jsonContent.trim()) {\n        stepLogger.error('No JSON output from research agent', { taskId: task.id });\n        emitEvent({\n            type: 'error',\n            ts: Date.now(),\n            message: 'Research agent returned no output',\n        });\n        return { status: 'completed', halt: true };\n    }\n\n    // Parse JSON response\n    let evaluation: ResearchEvaluation;\n    try {\n        // Extract JSON from potential markdown code blocks or other wrapping\n        const jsonMatch = jsonContent.match(/\\{[\\s\\S]*\\}/);\n        if (!jsonMatch) {\n            throw new Error('No JSON object found in response');\n        }\n        evaluation = JSON.parse(jsonMatch[0]);\n        stepLogger.info('Parsed research evaluation', {\n            taskId: task.id,\n            score: evaluation.actionabilityScore,\n            hasQuestions: !!evaluation.questions,\n        });\n    } catch (error) {\n        stepLogger.error('Failed to parse research JSON', {\n            taskId: task.id,\n            error: error instanceof Error ? error.message : String(error),\n            content: jsonContent.substring(0, 500),\n        });\n        emitEvent({\n            type: 'error',\n            ts: Date.now(),\n            message: `Failed to parse research JSON: ${\n                error instanceof Error ? error.message : String(error)\n            }`,\n        });\n        return { status: 'completed', halt: true };\n    }\n\n    // Add answered/answers fields to evaluation\n    if (evaluation.questions && evaluation.questions.length > 0) {\n        evaluation.answered = false;\n        evaluation.answers = undefined;\n    }\n\n    // Always write research.json\n    await fileManager.writeResearch(task.id, evaluation);\n    stepLogger.info('Research evaluation written', {\n        taskId: task.id,\n        score: evaluation.actionabilityScore,\n        hasQuestions: !!evaluation.questions,\n    });\n\n    emitEvent({\n        type: 'artifact',\n        ts: Date.now(),\n        kind: 'research_evaluation',\n        content: evaluation,\n    });\n\n    await gitManager.addAllPostHogFiles();\n    await finalizeStepGitActions(context, step, {\n        commitMessage: `Research phase for ${task.title}`,\n    });\n\n    // Log whether questions need answering\n    if (evaluation.actionabilityScore < 0.7 && evaluation.questions && evaluation.questions.length > 0) {\n        stepLogger.info('Actionability score below threshold, questions needed', {\n            taskId: task.id,\n            score: evaluation.actionabilityScore,\n            questionCount: evaluation.questions.length,\n        });\n        \n        emitEvent({\n            type: 'artifact',\n            ts: Date.now(),\n            kind: 'research_questions',\n            content: evaluation.questions,\n        });\n    } else {\n        stepLogger.info('Actionability score acceptable, proceeding to planning', {\n            taskId: task.id,\n            score: evaluation.actionabilityScore,\n        });\n    }\n\n    // In local mode, always halt after research for user review\n    if (!isCloudMode) {\n        emitEvent(adapter.createStatusEvent('phase_complete', { phase: 'research' }));\n        return { status: 'completed', halt: true };\n    }\n\n    // In cloud mode, check if questions need answering\n    const researchData = await fileManager.readResearch(task.id);\n    if (researchData?.questions && !researchData.answered) {\n        // Questions need answering - halt for user input in cloud mode too\n        emitEvent(adapter.createStatusEvent('phase_complete', { phase: 'research' }));\n        return { status: 'completed', halt: true };\n    }\n\n    // No questions or questions already answered - proceed to planning\n    emitEvent(adapter.createStatusEvent('phase_complete', { phase: 'research' }));\n    return { status: 'completed' };\n};\n"],"names":[],"mappings":";;;;AAMO,MAAM,YAAY,GAAuB,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,KAAI;IACxE,MAAM,EACF,IAAI,EACJ,GAAG,EACH,WAAW,EACX,OAAO,EACP,MAAM,EACN,WAAW,EACX,UAAU,EACV,aAAa,EACb,OAAO,EACP,UAAU,EACV,SAAS,GACZ,GAAG,OAAO;IAEX,MAAM,UAAU,GAAG,MAAM,CAAC,KAAK,CAAC,cAAc,CAAC;IAE/C,MAAM,gBAAgB,GAAG,MAAM,WAAW,CAAC,YAAY,CAAC,IAAI,CAAC,EAAE,CAAC;IAChE,IAAI,gBAAgB,EAAE;QAClB,UAAU,CAAC,IAAI,CAAC,yBAAyB,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,YAAY,EAAE,CAAC,CAAC,gBAAgB,CAAC,SAAS,EAAE,QAAQ,EAAE,gBAAgB,CAAC,QAAQ,EAAE,CAAC;;QAGhJ,IAAI,gBAAgB,CAAC,SAAS,IAAI,CAAC,gBAAgB,CAAC,QAAQ,EAAE;AAC1D,YAAA,UAAU,CAAC,IAAI,CAAC,2CAA2C,EAAE;gBACzD,MAAM,EAAE,IAAI,CAAC,EAAE;AACf,gBAAA,aAAa,EAAE,gBAAgB,CAAC,SAAS,CAAC;AAC7C,aAAA,CAAC;AAEF,YAAA,SAAS,CAAC;AACN,gBAAA,IAAI,EAAE,UAAU;AAChB,gBAAA,EAAE,EAAE,IAAI,CAAC,GAAG,EAAE;AACd,gBAAA,IAAI,EAAE,oBAAoB;gBAC1B,OAAO,EAAE,gBAAgB,CAAC,SAAS;AACtC,aAAA,CAAC;;YAGF,IAAI,CAAC,WAAW,EAAE;AACd,gBAAA,SAAS,CAAC,OAAO,CAAC,iBAAiB,CAAC,gBAAgB,EAAE,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC;gBAC7E,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE;YAC5C;QACJ;AAEA,QAAA,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE;IAChC;AAEA,IAAA,UAAU,CAAC,IAAI,CAAC,yBAAyB,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC;AAC/D,IAAA,SAAS,CAAC,OAAO,CAAC,iBAAiB,CAAC,aAAa,EAAE,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC;IAE1E,MAAM,cAAc,GAAG,MAAM,aAAa,CAAC,mBAAmB,CAAC,IAAI,EAAE,GAAG,CAAC;AACzE,IAAA,MAAM,UAAU,GAAG,CAAA,EAAG,sBAAsB,CAAA,IAAA,EAAO,cAAc,EAAE;AAEnE,IAAA,MAAM,WAAW,GAAwB;QACrC,KAAK,EAAE,IAAI,CAAC,KAAK;QACjB,GAAG;AACH,QAAA,cAAc,EAAE,MAAM;QACtB,cAAc,EAAE,CAAC,OAAO,CAAC;QACzB,UAAU;;AAEV,QAAA,YAAY,EAAE;YACV,MAAM;YACN,MAAM;YACN,MAAM;YACN,UAAU;YACV,WAAW;YACX,kBAAkB;YAClB,iBAAiB;YACjB,WAAW;YACX,YAAY;AACf,SAAA;KACJ;IAED,MAAM,QAAQ,GAAG,KAAK,CAAC;AACnB,QAAA,MAAM,EAAE,UAAU;AAClB,QAAA,OAAO,EAAE,EAAE,GAAG,WAAW,EAAE,IAAI,OAAO,CAAC,cAAc,IAAI,EAAE,CAAC,EAAE;AACjE,KAAA,CAAC;IAEF,IAAI,WAAW,GAAG,EAAE;AACpB,IAAA,WAAW,MAAM,OAAO,IAAI,QAAQ,EAAE;QAClC,SAAS,CAAC,OAAO,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;QAC7C,MAAM,iBAAiB,GAAG,OAAO,CAAC,SAAS,CAAC,OAAO,CAAC;AACpD,QAAA,KAAK,MAAM,KAAK,IAAI,iBAAiB,EAAE;YACnC,SAAS,CAAC,KAAK,CAAC;QACpB;AACA,QAAA,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,IAAI,OAAO,CAAC,OAAO,EAAE,OAAO,EAAE;YAC1D,KAAK,MAAM,CAAC,IAAI,OAAO,CAAC,OAAO,CAAC,OAAO,EAAE;gBACrC,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM,IAAI,CAAC,CAAC,IAAI,EAAE;AAC7B,oBAAA,WAAW,IAAI,CAAC,CAAC,IAAI;gBACzB;YACJ;QACJ;IACJ;AAEA,IAAA,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,EAAE;AACrB,QAAA,UAAU,CAAC,KAAK,CAAC,oCAAoC,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC;AAC3E,QAAA,SAAS,CAAC;AACN,YAAA,IAAI,EAAE,OAAO;AACb,YAAA,EAAE,EAAE,IAAI,CAAC,GAAG,EAAE;AACd,YAAA,OAAO,EAAE,mCAAmC;AAC/C,SAAA,CAAC;QACF,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,IAAI,EAAE,IAAI,EAAE;IAC9C;;AAGA,IAAA,IAAI,UAA8B;AAClC,IAAA,IAAI;;QAEA,MAAM,SAAS,GAAG,WAAW,CAAC,KAAK,CAAC,aAAa,CAAC;QAClD,IAAI,CAAC,SAAS,EAAE;AACZ,YAAA,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC;QACvD;QACA,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;AACrC,QAAA,UAAU,CAAC,IAAI,CAAC,4BAA4B,EAAE;YAC1C,MAAM,EAAE,IAAI,CAAC,EAAE;YACf,KAAK,EAAE,UAAU,CAAC,kBAAkB;AACpC,YAAA,YAAY,EAAE,CAAC,CAAC,UAAU,CAAC,SAAS;AACvC,SAAA,CAAC;IACN;IAAE,OAAO,KAAK,EAAE;AACZ,QAAA,UAAU,CAAC,KAAK,CAAC,+BAA+B,EAAE;YAC9C,MAAM,EAAE,IAAI,CAAC,EAAE;AACf,YAAA,KAAK,EAAE,KAAK,YAAY,KAAK,GAAG,KAAK,CAAC,OAAO,GAAG,MAAM,CAAC,KAAK,CAAC;YAC7D,OAAO,EAAE,WAAW,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC;AACzC,SAAA,CAAC;AACF,QAAA,SAAS,CAAC;AACN,YAAA,IAAI,EAAE,OAAO;AACb,YAAA,EAAE,EAAE,IAAI,CAAC,GAAG,EAAE;AACd,YAAA,OAAO,EAAE,CAAA,+BAAA,EACL,KAAK,YAAY,KAAK,GAAG,KAAK,CAAC,OAAO,GAAG,MAAM,CAAC,KAAK,CACzD,CAAA,CAAE;AACL,SAAA,CAAC;QACF,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,IAAI,EAAE,IAAI,EAAE;IAC9C;;AAGA,IAAA,IAAI,UAAU,CAAC,SAAS,IAAI,UAAU,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE;AACzD,QAAA,UAAU,CAAC,QAAQ,GAAG,KAAK;AAC3B,QAAA,UAAU,CAAC,OAAO,GAAG,SAAS;IAClC;;IAGA,MAAM,WAAW,CAAC,aAAa,CAAC,IAAI,CAAC,EAAE,EAAE,UAAU,CAAC;AACpD,IAAA,UAAU,CAAC,IAAI,CAAC,6BAA6B,EAAE;QAC3C,MAAM,EAAE,IAAI,CAAC,EAAE;QACf,KAAK,EAAE,UAAU,CAAC,kBAAkB;AACpC,QAAA,YAAY,EAAE,CAAC,CAAC,UAAU,CAAC,SAAS;AACvC,KAAA,CAAC;AAEF,IAAA,SAAS,CAAC;AACN,QAAA,IAAI,EAAE,UAAU;AAChB,QAAA,EAAE,EAAE,IAAI,CAAC,GAAG,EAAE;AACd,QAAA,IAAI,EAAE,qBAAqB;AAC3B,QAAA,OAAO,EAAE,UAAU;AACtB,KAAA,CAAC;AAEF,IAAA,MAAM,UAAU,CAAC,kBAAkB,EAAE;AACrC,IAAA,MAAM,sBAAsB,CAAC,OAAO,EAAE,IAAI,EAAE;AACxC,QAAA,aAAa,EAAE,CAAA,mBAAA,EAAsB,IAAI,CAAC,KAAK,CAAA,CAAE;AACpD,KAAA,CAAC;;AAGF,IAAA,IAAI,UAAU,CAAC,kBAAkB,GAAG,GAAG,IAAI,UAAU,CAAC,SAAS,IAAI,UAAU,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE;AAChG,QAAA,UAAU,CAAC,IAAI,CAAC,uDAAuD,EAAE;YACrE,MAAM,EAAE,IAAI,CAAC,EAAE;YACf,KAAK,EAAE,UAAU,CAAC,kBAAkB;AACpC,YAAA,aAAa,EAAE,UAAU,CAAC,SAAS,CAAC,MAAM;AAC7C,SAAA,CAAC;AAEF,QAAA,SAAS,CAAC;AACN,YAAA,IAAI,EAAE,UAAU;AAChB,YAAA,EAAE,EAAE,IAAI,CAAC,GAAG,EAAE;AACd,YAAA,IAAI,EAAE,oBAAoB;YAC1B,OAAO,EAAE,UAAU,CAAC,SAAS;AAChC,SAAA,CAAC;IACN;SAAO;AACH,QAAA,UAAU,CAAC,IAAI,CAAC,wDAAwD,EAAE;YACtE,MAAM,EAAE,IAAI,CAAC,EAAE;YACf,KAAK,EAAE,UAAU,CAAC,kBAAkB;AACvC,SAAA,CAAC;IACN;;IAGA,IAAI,CAAC,WAAW,EAAE;AACd,QAAA,SAAS,CAAC,OAAO,CAAC,iBAAiB,CAAC,gBAAgB,EAAE,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC;QAC7E,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,IAAI,EAAE,IAAI,EAAE;IAC9C;;IAGA,MAAM,YAAY,GAAG,MAAM,WAAW,CAAC,YAAY,CAAC,IAAI,CAAC,EAAE,CAAC;IAC5D,IAAI,YAAY,EAAE,SAAS,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE;;AAEnD,QAAA,SAAS,CAAC,OAAO,CAAC,iBAAiB,CAAC,gBAAgB,EAAE,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC;QAC7E,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,IAAI,EAAE,IAAI,EAAE;IAC9C;;AAGA,IAAA,SAAS,CAAC,OAAO,CAAC,iBAAiB,CAAC,gBAAgB,EAAE,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC;AAC7E,IAAA,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE;AAClC;;;;"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@posthog/agent",
-  "version": "1.19.0",
+  "version": "1.21.0",
   "description": "TypeScript agent framework wrapping Claude Agent SDK with Git-based task execution for PostHog",
   "main": "./dist/index.js",
   "module": "./dist/index.js",

package/src/adapters/claude/claude-adapter.ts

@@ -1,4 +1,4 @@
-import type { AgentEvent } from '../../types.js';
+import type { AgentEvent, ArtifactEvent, StatusEvent } from '../../types.js';
 import type { SDKMessage } from '@anthropic-ai/claude-agent-sdk';
 import type { ProviderAdapter } from '../types.js';
 import { ClaudeToolMapper } from './tool-mapper.js';
@@ -18,72 +18,71 @@ export class ClaudeAdapter implements ProviderAdapter {
     };
   }
 
-  transform(sdkMessage: SDKMessage): AgentEvent | null {
+  transform(sdkMessage: SDKMessage): AgentEvent[] {
     const baseEvent = { ts: Date.now() };
 
-    // Handle stream events
     if (sdkMessage.type === 'stream_event') {
       const event = sdkMessage.event;
 
       switch (event.type) {
         case 'message_start':
-          return {
+          return [{
             ...baseEvent,
             type: 'message_start',
             messageId: event.message?.id,
             model: event.message?.model
-          };
+          }];
 
         case 'content_block_start':
           const contentBlock = event.content_block;
-          if (!contentBlock) return null;
+          if (!contentBlock) return [];
 
-          return {
+          return [{
             ...baseEvent,
             type: 'content_block_start',
             index: event.index,
             contentType: contentBlock.type as 'text' | 'tool_use' | 'thinking',
             toolName: contentBlock.type === 'tool_use' ? contentBlock.name : undefined,
             toolId: contentBlock.type === 'tool_use' ? contentBlock.id : undefined
-          };
+          }];
 
         case 'content_block_delta':
           const delta = event.delta;
-          if (!delta) return null;
+          if (!delta) return [];
 
           if (delta.type === 'text_delta') {
-            return {
+            return [{
               ...baseEvent,
               type: 'token',
               content: delta.text,
               contentType: 'text'
-            };
+            }];
           } else if (delta.type === 'input_json_delta') {
-            return {
+            return [{
               ...baseEvent,
               type: 'token',
               content: delta.partial_json,
               contentType: 'tool_input'
-            };
+            }];
           } else if (delta.type === 'thinking_delta') {
-            return {
+            return [{
               ...baseEvent,
               type: 'token',
               content: delta.thinking,
               contentType: 'thinking'
-            };
+            }];
           }
-          return null;
+          return [];
 
         case 'content_block_stop':
-          return {
+          return [{
             ...baseEvent,
             type: 'content_block_stop',
             index: event.index
-          };
+          }];
 
         case 'message_delta':
-          return {
+          return [{
             ...baseEvent,
             type: 'message_delta',
             stopReason: event.delta?.stop_reason,
@@ -91,20 +90,20 @@ export class ClaudeAdapter implements ProviderAdapter {
             usage: event.usage ? {
               outputTokens: event.usage.output_tokens
             } : undefined
-          };
+          }];
 
         case 'message_stop':
-          return {
+          return [{
             ...baseEvent,
             type: 'message_stop'
-          };
+          }];
 
         case 'ping':
           // Ignore ping events
-          return null;
+          return [];
 
         case 'error':
-          return {
+          return [{
             ...baseEvent,
             type: 'error',
             message: event.error?.message || 'Unknown error',
@@ -115,18 +114,20 @@ export class ClaudeAdapter implements ProviderAdapter {
               code: event.error.code,
             } : undefined,
             sdkError: event.error
-          };
+          }];
 
         default:
-          return null;
+          return [];
       }
     }
 
     // Handle assistant messages (full message, not streaming)
     if (sdkMessage.type === 'assistant') {
       const message = sdkMessage.message;
+      const events: AgentEvent[] = [];
 
       // Extract tool calls from content blocks
+      // A single assistant message can contain multiple tool_use blocks
       if (message.content && Array.isArray(message.content)) {
         for (const block of message.content) {
           if (block.type === 'tool_use') {
@@ -139,31 +140,36 @@ export class ClaudeAdapter implements ProviderAdapter {
               args: block.input || {},
               parentToolUseId: sdkMessage.parent_tool_use_id
             };
-            // Enrich with tool metadata
-            return this.toolMapper.enrichToolCall(toolCallEvent);
+            // Enrich with tool metadata and add to events array
+            events.push(this.toolMapper.enrichToolCall(toolCallEvent));
           }
         }
       }
 
+      // If we found tool calls, return them
+      if (events.length > 0) {
+        return events;
+      }
+
       // If no tool calls, emit status event
-      return {
+      return [{
         ...baseEvent,
         type: 'status',
         phase: 'assistant_message',
         messageId: message.id,
         model: message.model
-      };
+      }];
     }
 
     // Handle user messages
     if (sdkMessage.type === 'user') {
       const message = sdkMessage.message;
+      const events: AgentEvent[] = [];
 
-      // Check for tool results in content blocks
-      if (message?.content && Array.isArray(message.content)) {
+      // Check for tool results in content blocks, A single user message can contain multiple tool_result blocks
+      if (message?.content && Array.isArray(message.content)) { 
         for (const block of message.content) {
           if (block.type === 'tool_result') {
-            // Create tool_result event and enrich with metadata
             const toolResultEvent = {
               ...baseEvent,
               type: 'tool_result' as const,
@@ -173,29 +179,33 @@ export class ClaudeAdapter implements ProviderAdapter {
               isError: block.is_error,
               parentToolUseId: sdkMessage.parent_tool_use_id
             };
-            // Enrich with tool metadata
-            return this.toolMapper.enrichToolResult(toolResultEvent);
+            events.push(this.toolMapper.enrichToolResult(toolResultEvent));
           }
         }
       }
 
+      // If we found tool results, return them
+      if (events.length > 0) {
+        return events;
+      }
+
       // Otherwise extract text content
       const textContent = this.extractUserContent(message?.content);
       if (!textContent) {
-        return null;
+        return [];
       }
-      return {
+      return [{
         ...baseEvent,
         type: 'user_message',
         content: textContent,
         isSynthetic: sdkMessage.isSynthetic
-      };
+      }];
     }
 
     // Handle result messages
     if (sdkMessage.type === 'result') {
       if (sdkMessage.subtype === 'success') {
-        return {
+        return [{
           ...baseEvent,
           type: 'done',
           result: sdkMessage.result,
@@ -206,9 +216,9 @@ export class ClaudeAdapter implements ProviderAdapter {
           usage: sdkMessage.usage,
           modelUsage: sdkMessage.modelUsage,
           permissionDenials: sdkMessage.permission_denials
-        };
+        }];
       } else {
-        return {
+        return [{
           ...baseEvent,
           type: 'error',
           message: `Execution failed: ${sdkMessage.subtype}`,
@@ -220,14 +230,14 @@ export class ClaudeAdapter implements ProviderAdapter {
             num_turns: sdkMessage.num_turns
           },
           sdkError: sdkMessage
-        };
+        }];
       }
     }
 
     // Handle system messages
     if (sdkMessage.type === 'system') {
       if (sdkMessage.subtype === 'init') {
-        return {
+        return [{
           ...baseEvent,
           type: 'init',
           model: sdkMessage.model,
@@ -239,21 +249,21 @@ export class ClaudeAdapter implements ProviderAdapter {
           slashCommands: sdkMessage.slash_commands,
           outputStyle: sdkMessage.output_style,
           mcpServers: sdkMessage.mcp_servers
-        };
+        }];
       } else if (sdkMessage.subtype === 'compact_boundary') {
-        return {
+        return [{
           ...baseEvent,
           type: 'compact_boundary',
           trigger: sdkMessage.compact_metadata.trigger,
           preTokens: sdkMessage.compact_metadata.pre_tokens
-        };
+        }];
       }
     }
 
-    return null;
+    return [];
   }
   
-  createStatusEvent(phase: string, additionalData?: any): import('../../types.js').StatusEvent {
+  createStatusEvent(phase: string, additionalData?: any): StatusEvent {
     return {
       type: 'status',
       ts: Date.now(),
@@ -262,6 +272,15 @@ export class ClaudeAdapter implements ProviderAdapter {
     };
   }
 
+  createArtifactEvent(kind: string, content: any): ArtifactEvent {
+    return {
+      type: 'artifact',
+      ts: Date.now(),
+      kind,
+      content
+    };
+  }
+
   private extractUserContent(content: unknown): string | null {
     if (!content) {
       return null;

package/src/adapters/types.ts

@@ -1,4 +1,4 @@
-import type { AgentEvent, StatusEvent } from '../types.js';
+import type { AgentEvent, StatusEvent, ArtifactEvent } from '../types.js';
 
 /**
  * Provider adapter interface for transforming provider-specific messages
@@ -12,10 +12,10 @@ export interface ProviderAdapter {
   name: string;
 
   /**
-   * Transform a provider-specific SDK message into an AgentEvent.
-   * Returns null if the message should be ignored.
+   * Transform a provider-specific SDK message into one or more AgentEvents.
+   * Returns an array of events (can be empty if the message should be ignored).
    */
-  transform(sdkMessage: unknown): AgentEvent | null;
+  transform(sdkMessage: unknown): AgentEvent[];
 
   /**
    * Create a standardized status event.
@@ -23,6 +23,12 @@ export interface ProviderAdapter {
    */
   createStatusEvent(phase: string, additionalData?: any): StatusEvent;
 
+  /**
+   * Create an artifact event for custom task artifacts (todos, etc).
+   * Used to emit structured artifacts for UI consumption.
+   */
+  createArtifactEvent(kind: string, content: any): ArtifactEvent;
+
   /**
    * Create a raw SDK event for debugging purposes.
    * Wraps the original SDK message in a raw_sdk_event.

package/src/agent.ts

@@ -214,10 +214,9 @@ export class Agent {
             this.logger.debug('Received message in direct run', message);
             // Emit raw SDK event
             this.emitEvent(this.adapter.createRawSDKEvent(message));
-            // Emit transformed event
-            const transformedEvent = this.adapter.transform(message);
-            if (transformedEvent) {
-                this.emitEvent(transformedEvent);
+            const transformedEvents = this.adapter.transform(message);
+            for (const event of transformedEvents) {
+                this.emitEvent(event);
             }
             results.push(message);
         }
@@ -308,11 +307,16 @@ export class Agent {
         return commitHash;
     }
 
-    async createPullRequest(taskId: string, branchName: string, taskTitle: string, taskDescription: string): Promise<string> {
+    async createPullRequest(
+        taskId: string,
+        branchName: string,
+        taskTitle: string,
+        taskDescription: string,
+        customBody?: string
+    ): Promise<string> {
         this.logger.info('Creating pull request', { taskId, branchName, taskTitle });
 
-        // Build PR body
-        const prBody = `## Task Details
+        const defaultBody = `## Task Details
 **Task ID**: ${taskId}
 **Description**: ${taskDescription}
 
@@ -320,6 +324,7 @@ export class Agent {
 This PR implements the changes described in the task.
 
 Generated by PostHog Agent`;
+        const prBody = customBody || defaultBody;
 
         const prUrl = await this.gitManager.createPullRequest(
             branchName,
@@ -447,11 +452,15 @@ Generated by PostHog Agent`;
         }
 
         const branchName = await this.gitManager.getCurrentBranch();
+        const finalizeResult = stepResults['finalize'];
+        const prBody = finalizeResult?.prBody;
+
         const prUrl = await this.createPullRequest(
             task.id,
             branchName,
             task.title,
-            task.description ?? ''
+            task.description ?? '',
+            prBody
         );
 
         this.emitEvent(this.adapter.createStatusEvent('pr_created', { prUrl }));

package/src/agents/research.ts

@@ -19,7 +19,14 @@ Calculate an actionabilityScore (0-1) based on:
 
 If actionabilityScore < 0.7, generate specific clarifying questions to increase confidence.
 
-DO NOT ask questions like "how should I fix this" — focus on missing information that prevents confident planning.
+Questions must present complete implementation choices, NOT request information from the user:
+options: array of strings
+- GOOD: options: ["Use Redux Toolkit (matches pattern in src/store/)", "Zustand (lighter weight)"]
+- BAD:  "Tell me which state management library to use"
+- GOOD: options: ["Place in Button.tsx (existing component)", "create NewButton.tsx (separate concerns)?"]
+- BAD: "Where should I put this code?"
+
+DO NOT ask questions like "how should I fix this" or "tell me the pattern" — present concrete options that can be directly chosen and acted upon.
 </objective>
 
 <process>
@@ -60,6 +67,9 @@ Rules:
 - questions: ONLY include if actionabilityScore < 0.7
 - Each question must have 2-3 options (maximum 3)
 - Max 3 questions total
+- Options must be complete, actionable choices that require NO additional user input
+- NEVER use options like "Tell me the pattern", "Show me examples", "Specify the approach"
+- Each option must be a full implementation decision that can be directly acted upon
 </output_format>
 
 <scoring_examples>
@@ -92,11 +102,25 @@ Questions needed: What feature? Which product area? What should it do?
   "id": "q1",
   "question": "Which caching layer should we use for API responses?",
   "options": [
-    "Redis (existing infrastructure, requires setup)",
-    "In-memory cache (simpler, but not distributed)",
-    "Browser-side caching only (minimal backend changes)"
+    "Redis with 1-hour TTL (existing infrastructure, requires Redis client setup)",
+    "In-memory LRU cache with 100MB limit (simpler, single-server only)",
+    "HTTP Cache-Control headers only (minimal backend changes, relies on browser/CDN)"
   ]
 }
+Reason: Each option is a complete, actionable decision with concrete details
+</good_example>
+
+<good_example>
+{
+  "id": "q2",
+  "question": "Where should the new analytics tracking code be placed?",
+  "options": [
+    "In the existing UserAnalytics.ts module alongside page view tracking",
+    "Create a new EventTracking.ts module in src/analytics/ for all event tracking",
+    "Add directly to each component that needs tracking (no centralized module)"
+  ]
+}
+Reason: Specific file paths and architectural patterns, no user input needed
 </good_example>
 
 <bad_example>
@@ -105,7 +129,33 @@ Questions needed: What feature? Which product area? What should it do?
   "question": "How should I implement this?",
   "options": ["One way", "Another way"]
 }
-Reason: Too vague, doesn't explain the tradeoffs
+Reason: Too vague, doesn't explain the tradeoffs or provide concrete details
+</bad_example>
+
+<bad_example>
+{
+  "id": "q2",
+  "question": "Which pattern should we follow for state management?",
+  "options": [
+    "Tell me which pattern the codebase currently uses",
+    "Show me examples of state management",
+    "Whatever you think is best"
+  ]
+}
+Reason: Options request user input instead of being actionable choices. Should be concrete patterns like "Zustand stores (matching existing patterns in src/stores/)" or "React Context (simpler, no new dependencies)"
+</bad_example>
+
+<bad_example>
+{
+  "id": "q3",
+  "question": "What color scheme should the button use?",
+  "options": [
+    "Use the existing theme colors",
+    "Let me specify custom colors",
+    "Match the design system"
+  ]
+}
+Reason: "Let me specify" requires user input. Should be "Primary blue (#0066FF, existing theme)" or "Secondary gray (#6B7280, existing theme)"
 </bad_example>
 </question_examples>`;