@posthog/agent 1.9.0 → 1.11.0

package/README.md

@@ -15,7 +15,7 @@ bun run example
 - **PostHog Integration**: Fetches existing tasks from PostHog API
 - **Configurable Workflows**: Execute tasks via PostHog-defined or local workflows
 - **Branch Management**: Automatic branch creation for planning and implementation
-- **Progress Tracking**: Execution status stored in PostHog `TaskProgress` records for easy polling
+- **Progress Tracking**: Execution status stored in PostHog `TaskRun` records for easy polling
 
 ## Usage
 
@@ -96,7 +96,7 @@ your-repo/
 
 ## Progress Updates
 
-Progress for each task execution is persisted to PostHog's `TaskProgress` model, so UIs can poll for updates without relying on streaming hooks:
+Progress for each task execution is persisted to PostHog's `TaskRun` model, so UIs can poll for updates without relying on streaming hooks:
 
 ```typescript
 const agent = new Agent({
@@ -106,9 +106,12 @@ const agent = new Agent({
 });
 
 const poller = setInterval(async () => {
-  const progress = await agent.getPostHogClient()?.getTaskProgress(taskId);
-  if (progress?.has_progress) {
-    renderProgress(progress.status, progress.current_step, progress.completed_steps, progress.total_steps);
+  const runs = await agent.getPostHogClient()?.listTaskRuns(taskId);
+  const latestRun = runs?.sort((a, b) =>
+    new Date(b.created_at).getTime() - new Date(a.created_at).getTime()
+  )[0];
+  if (latestRun) {
+    renderProgress(latestRun.status, latestRun.log, latestRun.current_stage);
   }
 }, 3000);
 

package/dist/index.d.ts

@@ -1,9 +1,11 @@
 export { Agent, } from './src/agent.js';
 export { PermissionMode, } from './src/types.js';
-export type { Task, SupportingFile, ExecutionResult, AgentConfig, McpServerConfig, AgentEvent } from './src/types.js';
+export type { Task, TaskRun, SupportingFile, ExecutionResult, AgentConfig, McpServerConfig, AgentEvent } from './src/types.js';
 export type { WorkflowDefinition, WorkflowStage, WorkflowExecutionOptions, AgentDefinition } from './src/workflow-types.js';
 export { Logger, LogLevel, } from './src/utils/logger.js';
 export type { LoggerConfig } from './src/utils/logger.js';
+export type { ExtractedQuestion, ExtractedQuestionWithAnswer, StructuredExtractor } from './src/structured-extraction.js';
+export type { QuestionData, AnswerData, QuestionsFile } from './src/file-manager.js';
 export type { ProviderAdapter } from './src/adapters/types.js';
 export { ClaudeAdapter } from './src/adapters/claude/claude-adapter.js';
 export type { Tool, ToolCategory, KnownTool, ReadTool, WriteTool, EditTool, GlobTool, NotebookEditTool, BashTool, BashOutputTool, KillShellTool, WebFetchTool, WebSearchTool, GrepTool, TaskTool, TodoWriteTool, ExitPlanModeTool, SlashCommandTool, } from './src/tools/types.js';

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../index.ts"],"names":[],"mappings":"AACA,OAAO,EACH,KAAK,GACR,MAAM,gBAAgB,CAAC;AAExB,OAAO,EACH,cAAc,GACjB,MAAM,gBAAgB,CAAC;AAExB,YAAY,EACR,IAAI,EACJ,cAAc,EACd,eAAe,EACf,WAAW,EACX,eAAe,EACf,UAAU,EACb,MAAM,gBAAgB,CAAC;AAExB,YAAY,EACV,kBAAkB,EAClB,aAAa,EACb,wBAAwB,EACxB,eAAe,EAChB,MAAM,yBAAyB,CAAC;AAEjC,OAAO,EACH,MAAM,EACN,QAAQ,GACX,MAAM,uBAAuB,CAAC;AAE/B,YAAY,EACR,YAAY,EACf,MAAM,uBAAuB,CAAC;AAG/B,YAAY,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAC/D,OAAO,EAAE,aAAa,EAAE,MAAM,yCAAyC,CAAC;AAGxE,YAAY,EACR,IAAI,EACJ,YAAY,EACZ,SAAS,EACT,QAAQ,EACR,SAAS,EACT,QAAQ,EACR,QAAQ,EACR,gBAAgB,EAChB,QAAQ,EACR,cAAc,EACd,aAAa,EACb,YAAY,EACZ,aAAa,EACb,QAAQ,EACR,QAAQ,EACR,aAAa,EACb,gBAAgB,EAChB,gBAAgB,GACnB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,YAAY,EAAE,MAAM,yBAAyB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../index.ts"],"names":[],"mappings":"AACA,OAAO,EACH,KAAK,GACR,MAAM,gBAAgB,CAAC;AAExB,OAAO,EACH,cAAc,GACjB,MAAM,gBAAgB,CAAC;AAExB,YAAY,EACR,IAAI,EACJ,OAAO,EACP,cAAc,EACd,eAAe,EACf,WAAW,EACX,eAAe,EACf,UAAU,EACb,MAAM,gBAAgB,CAAC;AAExB,YAAY,EACV,kBAAkB,EAClB,aAAa,EACb,wBAAwB,EACxB,eAAe,EAChB,MAAM,yBAAyB,CAAC;AAEjC,OAAO,EACH,MAAM,EACN,QAAQ,GACX,MAAM,uBAAuB,CAAC;AAE/B,YAAY,EACR,YAAY,EACf,MAAM,uBAAuB,CAAC;AAG/B,YAAY,EACR,iBAAiB,EACjB,2BAA2B,EAC3B,mBAAmB,EACtB,MAAM,gCAAgC,CAAC;AAGxC,YAAY,EACR,YAAY,EACZ,UAAU,EACV,aAAa,EAChB,MAAM,uBAAuB,CAAC;AAG/B,YAAY,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAC/D,OAAO,EAAE,aAAa,EAAE,MAAM,yCAAyC,CAAC;AAGxE,YAAY,EACR,IAAI,EACJ,YAAY,EACZ,SAAS,EACT,QAAQ,EACR,SAAS,EACT,QAAQ,EACR,QAAQ,EACR,gBAAgB,EAChB,QAAQ,EACR,cAAc,EACd,aAAa,EACb,YAAY,EACZ,aAAa,EACb,QAAQ,EACR,QAAQ,EACR,aAAa,EACb,gBAAgB,EAChB,gBAAgB,GACnB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,YAAY,EAAE,MAAM,yBAAyB,CAAC"}

package/dist/src/agent-registry.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"agent-registry.d.ts","sourceRoot":"","sources":["../../src/agent-registry.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,eAAe,EAAE,SAAS,EAAE,MAAM,qBAAqB,CAAC;AAEtE,qBAAa,aAAa;IACxB,OAAO,CAAC,YAAY,CAA2C;gBAEnD,WAAW,CAAC,EAAE,eAAe,EAAE;IAQ3C,MAAM,CAAC,gBAAgB,IAAI,eAAe,EAAE;IA6B5C,QAAQ,CAAC,GAAG,EAAE,eAAe,GAAG,IAAI;IAIpC,QAAQ,CAAC,IAAI,EAAE,MAAM,GAAG,eAAe,GAAG,SAAS;IAInD,UAAU,IAAI,eAAe,EAAE;IAI/B,gBAAgB,IAAI;QAAE,EAAE,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAC;QAAC,UAAU,EAAE,SAAS,CAAC;QAAC,WAAW,CAAC,EAAE,MAAM,CAAA;KAAE,EAAE;CAGhG"}
+{"version":3,"file":"agent-registry.d.ts","sourceRoot":"","sources":["../../src/agent-registry.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,eAAe,EAAE,SAAS,EAAE,MAAM,qBAAqB,CAAC;AAEtE,qBAAa,aAAa;IACxB,OAAO,CAAC,YAAY,CAA2C;gBAEnD,WAAW,CAAC,EAAE,eAAe,EAAE;IAQ3C,MAAM,CAAC,gBAAgB,IAAI,eAAe,EAAE;IAmC5C,QAAQ,CAAC,GAAG,EAAE,eAAe,GAAG,IAAI;IAIpC,QAAQ,CAAC,IAAI,EAAE,MAAM,GAAG,eAAe,GAAG,SAAS;IAInD,UAAU,IAAI,eAAe,EAAE;IAI/B,gBAAgB,IAAI;QAAE,EAAE,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAC;QAAC,UAAU,EAAE,SAAS,CAAC;QAAC,WAAW,CAAC,EAAE,MAAM,CAAA;KAAE,EAAE;CAGhG"}

package/dist/src/agent-registry.js

@@ -12,6 +12,12 @@ class AgentRegistry {
     }
     static getDefaultAgents() {
         return [
+            {
+                id: 'research',
+                name: 'research',
+                agent_type: 'research',
+                description: 'Explore codebase and generate clarifying questions',
+            },
             {
                 id: 'planning',
                 name: 'planning',

package/dist/src/agent-registry.js.map

@@ -1 +1 @@
-{"version":3,"file":"agent-registry.js","sources":["../../src/agent-registry.ts"],"sourcesContent":["import type { AgentDefinition, AgentType } from './workflow-types.js';\n\nexport class AgentRegistry {\n  private agentsByName: Map<string, AgentDefinition> = new Map();\n\n  constructor(definitions?: AgentDefinition[]) {\n    if (definitions) {\n      for (const def of definitions) this.register(def);\n    } else {\n      for (const def of AgentRegistry.getDefaultAgents()) this.register(def);\n    }\n  }\n\n  static getDefaultAgents(): AgentDefinition[] {\n    return [\n      {\n        id: 'planning',\n        name: 'planning',\n        agent_type: 'planning',\n        description: 'Analyze repo and produce implementation plan',\n      },\n      {\n        id: 'code_generation',\n        name: 'code_generation',\n        agent_type: 'execution',\n        description: 'Implements code changes using Claude SDK',\n      },\n      {\n        id: 'review',\n        name: 'review',\n        agent_type: 'review',\n        description: 'Reviews changes and suggests fixes',\n      },\n      {\n        id: 'testing',\n        name: 'testing',\n        agent_type: 'testing',\n        description: 'Runs tests and reports results',\n      },\n    ];\n  }\n\n  register(def: AgentDefinition): void {\n    this.agentsByName.set(def.name, def);\n  }\n\n  getAgent(name: string): AgentDefinition | undefined {\n    return this.agentsByName.get(name);\n  }\n\n  listAgents(): AgentDefinition[] {\n    return Array.from(this.agentsByName.values());\n  }\n\n  exportForPostHog(): { id: string; name: string; agent_type: AgentType; description?: string }[] {\n    return this.listAgents().map(({ id, name, agent_type, description }) => ({ id, name, agent_type, description }));\n  }\n}\n\n"],"names":[],"mappings":"MAEa,aAAa,CAAA;AAChB,IAAA,YAAY,GAAiC,IAAI,GAAG,EAAE;AAE9D,IAAA,WAAA,CAAY,WAA+B,EAAA;QACzC,IAAI,WAAW,EAAE;YACf,KAAK,MAAM,GAAG,IAAI,WAAW;AAAE,gBAAA,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC;QACnD;aAAO;AACL,YAAA,KAAK,MAAM,GAAG,IAAI,aAAa,CAAC,gBAAgB,EAAE;AAAE,gBAAA,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC;QACxE;IACF;AAEA,IAAA,OAAO,gBAAgB,GAAA;QACrB,OAAO;AACL,YAAA;AACE,gBAAA,EAAE,EAAE,UAAU;AACd,gBAAA,IAAI,EAAE,UAAU;AAChB,gBAAA,UAAU,EAAE,UAAU;AACtB,gBAAA,WAAW,EAAE,8CAA8C;AAC5D,aAAA;AACD,YAAA;AACE,gBAAA,EAAE,EAAE,iBAAiB;AACrB,gBAAA,IAAI,EAAE,iBAAiB;AACvB,gBAAA,UAAU,EAAE,WAAW;AACvB,gBAAA,WAAW,EAAE,0CAA0C;AACxD,aAAA;AACD,YAAA;AACE,gBAAA,EAAE,EAAE,QAAQ;AACZ,gBAAA,IAAI,EAAE,QAAQ;AACd,gBAAA,UAAU,EAAE,QAAQ;AACpB,gBAAA,WAAW,EAAE,oCAAoC;AAClD,aAAA;AACD,YAAA;AACE,gBAAA,EAAE,EAAE,SAAS;AACb,gBAAA,IAAI,EAAE,SAAS;AACf,gBAAA,UAAU,EAAE,SAAS;AACrB,gBAAA,WAAW,EAAE,gCAAgC;AAC9C,aAAA;SACF;IACH;AAEA,IAAA,QAAQ,CAAC,GAAoB,EAAA;QAC3B,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,EAAE,GAAG,CAAC;IACtC;AAEA,IAAA,QAAQ,CAAC,IAAY,EAAA;QACnB,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC;IACpC;IAEA,UAAU,GAAA;QACR,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,CAAC;IAC/C;IAEA,gBAAgB,GAAA;AACd,QAAA,OAAO,IAAI,CAAC,UAAU,EAAE,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,WAAW,EAAE,MAAM,EAAE,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,WAAW,EAAE,CAAC,CAAC;IAClH;AACD;;;;"}
+{"version":3,"file":"agent-registry.js","sources":["../../src/agent-registry.ts"],"sourcesContent":["import type { AgentDefinition, AgentType } from './workflow-types.js';\n\nexport class AgentRegistry {\n  private agentsByName: Map<string, AgentDefinition> = new Map();\n\n  constructor(definitions?: AgentDefinition[]) {\n    if (definitions) {\n      for (const def of definitions) this.register(def);\n    } else {\n      for (const def of AgentRegistry.getDefaultAgents()) this.register(def);\n    }\n  }\n\n  static getDefaultAgents(): AgentDefinition[] {\n    return [\n      {\n        id: 'research',\n        name: 'research',\n        agent_type: 'research',\n        description: 'Explore codebase and generate clarifying questions',\n      },\n      {\n        id: 'planning',\n        name: 'planning',\n        agent_type: 'planning',\n        description: 'Analyze repo and produce implementation plan',\n      },\n      {\n        id: 'code_generation',\n        name: 'code_generation',\n        agent_type: 'execution',\n        description: 'Implements code changes using Claude SDK',\n      },\n      {\n        id: 'review',\n        name: 'review',\n        agent_type: 'review',\n        description: 'Reviews changes and suggests fixes',\n      },\n      {\n        id: 'testing',\n        name: 'testing',\n        agent_type: 'testing',\n        description: 'Runs tests and reports results',\n      },\n    ];\n  }\n\n  register(def: AgentDefinition): void {\n    this.agentsByName.set(def.name, def);\n  }\n\n  getAgent(name: string): AgentDefinition | undefined {\n    return this.agentsByName.get(name);\n  }\n\n  listAgents(): AgentDefinition[] {\n    return Array.from(this.agentsByName.values());\n  }\n\n  exportForPostHog(): { id: string; name: string; agent_type: AgentType; description?: string }[] {\n    return this.listAgents().map(({ id, name, agent_type, description }) => ({ id, name, agent_type, description }));\n  }\n}\n\n"],"names":[],"mappings":"MAEa,aAAa,CAAA;AAChB,IAAA,YAAY,GAAiC,IAAI,GAAG,EAAE;AAE9D,IAAA,WAAA,CAAY,WAA+B,EAAA;QACzC,IAAI,WAAW,EAAE;YACf,KAAK,MAAM,GAAG,IAAI,WAAW;AAAE,gBAAA,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC;QACnD;aAAO;AACL,YAAA,KAAK,MAAM,GAAG,IAAI,aAAa,CAAC,gBAAgB,EAAE;AAAE,gBAAA,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC;QACxE;IACF;AAEA,IAAA,OAAO,gBAAgB,GAAA;QACrB,OAAO;AACL,YAAA;AACE,gBAAA,EAAE,EAAE,UAAU;AACd,gBAAA,IAAI,EAAE,UAAU;AAChB,gBAAA,UAAU,EAAE,UAAU;AACtB,gBAAA,WAAW,EAAE,oDAAoD;AAClE,aAAA;AACD,YAAA;AACE,gBAAA,EAAE,EAAE,UAAU;AACd,gBAAA,IAAI,EAAE,UAAU;AAChB,gBAAA,UAAU,EAAE,UAAU;AACtB,gBAAA,WAAW,EAAE,8CAA8C;AAC5D,aAAA;AACD,YAAA;AACE,gBAAA,EAAE,EAAE,iBAAiB;AACrB,gBAAA,IAAI,EAAE,iBAAiB;AACvB,gBAAA,UAAU,EAAE,WAAW;AACvB,gBAAA,WAAW,EAAE,0CAA0C;AACxD,aAAA;AACD,YAAA;AACE,gBAAA,EAAE,EAAE,QAAQ;AACZ,gBAAA,IAAI,EAAE,QAAQ;AACd,gBAAA,UAAU,EAAE,QAAQ;AACpB,gBAAA,WAAW,EAAE,oCAAoC;AAClD,aAAA;AACD,YAAA;AACE,gBAAA,EAAE,EAAE,SAAS;AACb,gBAAA,IAAI,EAAE,SAAS;AACf,gBAAA,UAAU,EAAE,SAAS;AACrB,gBAAA,WAAW,EAAE,gCAAgC;AAC9C,aAAA;SACF;IACH;AAEA,IAAA,QAAQ,CAAC,GAAoB,EAAA;QAC3B,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,EAAE,GAAG,CAAC;IACtC;AAEA,IAAA,QAAQ,CAAC,IAAY,EAAA;QACnB,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC;IACpC;IAEA,UAAU,GAAA;QACR,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,CAAC;IAC/C;IAEA,gBAAgB,GAAA;AACd,QAAA,OAAO,IAAI,CAAC,UAAU,EAAE,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,WAAW,EAAE,MAAM,EAAE,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,WAAW,EAAE,CAAC,CAAC;IAClH;AACD;;;;"}

package/dist/src/agent.d.ts

@@ -1,6 +1,7 @@
 import type { Task, ExecutionResult, AgentConfig } from './types.js';
 import type { WorkflowDefinition, WorkflowStage, WorkflowExecutionOptions } from './workflow-types.js';
 import { PostHogAPIClient } from './posthog-api.js';
+import { type ExtractedQuestion, type ExtractedQuestionWithAnswer } from './structured-extraction.js';
 export declare class Agent {
     private workingDirectory;
     private onEvent?;
@@ -15,6 +16,8 @@ export declare class Agent {
     private workflowRegistry;
     private stageExecutor;
     private progressReporter;
+    private promptBuilder;
+    private extractor?;
     private mcpServers?;
     debug: boolean;
     constructor(config?: AgentConfig);
@@ -31,6 +34,7 @@ export declare class Agent {
         workflow: WorkflowDefinition;
     }>;
     executeStage(task: Task, stage: WorkflowStage, options?: WorkflowExecutionOptions): Promise<void>;
+    runTask(taskOrId: Task | string, options?: import('./types.js').TaskExecutionOptions): Promise<void>;
     progressToNextStage(taskId: string, currentStageKey?: string): Promise<void>;
     run(prompt: string, options?: {
         repositoryPath?: string;
@@ -51,6 +55,7 @@ export declare class Agent {
     getTaskFiles(taskId: string): Promise<any[]>;
     writePlan(taskId: string, plan: string): Promise<void>;
     readPlan(taskId: string): Promise<string | null>;
+    extractQuestionsFromResearch(taskId: string, includeAnswers?: boolean): Promise<ExtractedQuestion[] | ExtractedQuestionWithAnswer[]>;
     createPlanningBranch(taskId: string): Promise<string>;
     commitPlan(taskId: string, taskTitle: string): Promise<string>;
     createImplementationBranch(taskId: string, planningBranchName?: string): Promise<string>;

package/dist/src/agent.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"agent.d.ts","sourceRoot":"","sources":["../../src/agent.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,IAAI,EAAE,eAAe,EAAc,WAAW,EAAE,MAAM,YAAY,CAAC;AACjF,OAAO,KAAK,EAAE,kBAAkB,EAAE,aAAa,EAAE,wBAAwB,EAAE,MAAM,qBAAqB,CAAC;AAEvG,OAAO,EAAE,gBAAgB,EAAE,MAAM,kBAAkB,CAAC;AAepD,qBAAa,KAAK;IACd,OAAO,CAAC,gBAAgB,CAAS;IACjC,OAAO,CAAC,OAAO,CAAC,CAAuB;IACvC,OAAO,CAAC,WAAW,CAAc;IACjC,OAAO,CAAC,UAAU,CAAC,CAAmB;IACtC,OAAO,CAAC,WAAW,CAAqB;IACxC,OAAO,CAAC,UAAU,CAAa;IAC/B,OAAO,CAAC,eAAe,CAAkB;IACzC,OAAO,CAAC,OAAO,CAAkB;IACjC,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,aAAa,CAAgB;IACrC,OAAO,CAAC,gBAAgB,CAAmB;IAC3C,OAAO,CAAC,aAAa,CAAgB;IACrC,OAAO,CAAC,gBAAgB,CAAuB;IAC/C,OAAO,CAAC,UAAU,CAAC,CAAsB;IAClC,KAAK,EAAE,OAAO,CAAC;gBAEV,MAAM,GAAE,WAAgB;IAuEpC;;OAEG;IACH,QAAQ,CAAC,OAAO,EAAE,OAAO;IAKzB;;OAEG;YACW,oBAAoB;IAwB5B,WAAW,CAAC,QAAQ,EAAE,IAAI,GAAG,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE,OAAO,GAAE,wBAA6B,GAAG,OAAO,CAAC;QAAE,IAAI,EAAE,IAAI,CAAC;QAAC,QAAQ,EAAE,kBAAkB,CAAA;KAAE,CAAC;IA0FvJ,YAAY,CAAC,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,aAAa,EAAE,OAAO,GAAE,wBAA6B,GAAG,OAAO,CAAC,IAAI,CAAC;IAuFrG,mBAAmB,CAAC,MAAM,EAAE,MAAM,EAAE,eAAe,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAoB5E,GAAG,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,GAAE;QAAE,cAAc,CAAC,EAAE,MAAM,CAAC;QAAC,cAAc,CAAC,EAAE,OAAO,YAAY,EAAE,cAAc,CAAC;QAAC,cAAc,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAA;KAAO,GAAG,OAAO,CAAC,eAAe,CAAC;IAgCpL,SAAS,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAU9C,gBAAgB,IAAI,gBAAgB,GAAG,SAAS;IAI1C,SAAS,CAAC,OAAO,CAAC,EAAE;QACtB,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB,YAAY,CAAC,EAAE,MAAM,CAAC;QACtB,cAAc,CAAC,EAAE,MAAM,CAAC;QACxB,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,aAAa,CAAC,EAAE,MAAM,CAAC;KAC1B,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC;IAQb,aAAa,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,IAAI,GAAE,MAAM,GAAG,SAAS,GAAG,WAAW,GAAG,QAAsB,GAAG,OAAO,CAAC,IAAI,CAAC;IAKhJ,YAAY,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC;IAKtE,YAAY,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;IAO5C,SAAS,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAKtD,QAAQ,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC;IAMhD,oBAAoB,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IASrD,UAAU,CAAC,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAO9D,0BAA0B,CAAC,MAAM,EAAE,MAAM,EAAE,kBAAkB,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAOxF,oBAAoB,CAAC,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,WAAW,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAO9F,iBAAiB,CAAC,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,eAAe,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAuBlH,uBAAuB,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,UAAU,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAa1F,gBAAgB,CAAC,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAczE,UAAU,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI;IAUhC,sBAAsB,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI;IAUrD,OAAO,CAAC,SAAS;CAapB;AAED,OAAO,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAC5C,YAAY,EAAE,IAAI,EAAE,cAAc,EAAE,eAAe,EAAE,WAAW,EAAE,MAAM,YAAY,CAAC;AACrF,YAAY,EAAE,kBAAkB,EAAE,aAAa,EAAE,wBAAwB,EAAE,MAAM,qBAAqB,CAAC"}
+{"version":3,"file":"agent.d.ts","sourceRoot":"","sources":["../../src/agent.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,IAAI,EAAE,eAAe,EAAc,WAAW,EAAE,MAAM,YAAY,CAAC;AACjF,OAAO,KAAK,EAAE,kBAAkB,EAAE,aAAa,EAAE,wBAAwB,EAAE,MAAM,qBAAqB,CAAC;AAEvG,OAAO,EAAE,gBAAgB,EAAE,MAAM,kBAAkB,CAAC;AAcpD,OAAO,EAAmB,KAAK,iBAAiB,EAAE,KAAK,2BAA2B,EAAE,MAAM,4BAA4B,CAAC;AAEvH,qBAAa,KAAK;IACd,OAAO,CAAC,gBAAgB,CAAS;IACjC,OAAO,CAAC,OAAO,CAAC,CAAuB;IACvC,OAAO,CAAC,WAAW,CAAc;IACjC,OAAO,CAAC,UAAU,CAAC,CAAmB;IACtC,OAAO,CAAC,WAAW,CAAqB;IACxC,OAAO,CAAC,UAAU,CAAa;IAC/B,OAAO,CAAC,eAAe,CAAkB;IACzC,OAAO,CAAC,OAAO,CAAkB;IACjC,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,aAAa,CAAgB;IACrC,OAAO,CAAC,gBAAgB,CAAmB;IAC3C,OAAO,CAAC,aAAa,CAAgB;IACrC,OAAO,CAAC,gBAAgB,CAAuB;IAC/C,OAAO,CAAC,aAAa,CAAgB;IACrC,OAAO,CAAC,SAAS,CAAC,CAAkB;IACpC,OAAO,CAAC,UAAU,CAAC,CAAsB;IAClC,KAAK,EAAE,OAAO,CAAC;gBAEV,MAAM,GAAE,WAAgB;IA4EpC;;OAEG;IACH,QAAQ,CAAC,OAAO,EAAE,OAAO;IAKzB;;OAEG;YACW,oBAAoB;IAwB5B,WAAW,CAAC,QAAQ,EAAE,IAAI,GAAG,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE,OAAO,GAAE,wBAA6B,GAAG,OAAO,CAAC;QAAE,IAAI,EAAE,IAAI,CAAC;QAAC,QAAQ,EAAE,kBAAkB,CAAA;KAAE,CAAC;IAqGvJ,YAAY,CAAC,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,aAAa,EAAE,OAAO,GAAE,wBAA6B,GAAG,OAAO,CAAC,IAAI,CAAC;IAwFrG,OAAO,CAAC,QAAQ,EAAE,IAAI,GAAG,MAAM,EAAE,OAAO,GAAE,OAAO,YAAY,EAAE,oBAAyB,GAAG,OAAO,CAAC,IAAI,CAAC;IA8UxG,mBAAmB,CAAC,MAAM,EAAE,MAAM,EAAE,eAAe,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAuB5E,GAAG,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,GAAE;QAAE,cAAc,CAAC,EAAE,MAAM,CAAC;QAAC,cAAc,CAAC,EAAE,OAAO,YAAY,EAAE,cAAc,CAAC;QAAC,cAAc,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAA;KAAO,GAAG,OAAO,CAAC,eAAe,CAAC;IAgCpL,SAAS,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAU9C,gBAAgB,IAAI,gBAAgB,GAAG,SAAS;IAI1C,SAAS,CAAC,OAAO,CAAC,EAAE;QACtB,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB,YAAY,CAAC,EAAE,MAAM,CAAC;QACtB,cAAc,CAAC,EAAE,MAAM,CAAC;QACxB,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,aAAa,CAAC,EAAE,MAAM,CAAC;KAC1B,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC;IAQb,aAAa,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,IAAI,GAAE,MAAM,GAAG,SAAS,GAAG,WAAW,GAAG,QAAsB,GAAG,OAAO,CAAC,IAAI,CAAC;IAKhJ,YAAY,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC;IAKtE,YAAY,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;IAO5C,SAAS,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAKtD,QAAQ,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC;IAKhD,4BAA4B,CAAC,MAAM,EAAE,MAAM,EAAE,cAAc,GAAE,OAAe,GAAG,OAAO,CAAC,iBAAiB,EAAE,GAAG,2BAA2B,EAAE,CAAC;IAoB3I,oBAAoB,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IASrD,UAAU,CAAC,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAO9D,0BAA0B,CAAC,MAAM,EAAE,MAAM,EAAE,kBAAkB,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAOxF,oBAAoB,CAAC,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,WAAW,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAO9F,iBAAiB,CAAC,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,eAAe,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAuBlH,uBAAuB,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,UAAU,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAoB1F,gBAAgB,CAAC,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAczE,UAAU,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI;IAUhC,sBAAsB,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI;IAUrD,OAAO,CAAC,SAAS;CAapB;AAED,OAAO,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAC5C,YAAY,EAAE,IAAI,EAAE,cAAc,EAAE,eAAe,EAAE,WAAW,EAAE,MAAM,YAAY,CAAC;AACrF,YAAY,EAAE,kBAAkB,EAAE,aAAa,EAAE,wBAAwB,EAAE,MAAM,qBAAqB,CAAC"}

package/dist/src/agent.js

@@ -11,6 +11,7 @@ import { WorkflowRegistry } from './workflow-registry.js';
 import { StageExecutor } from './stage-executor.js';
 import { PromptBuilder } from './prompt-builder.js';
 import { TaskProgressReporter } from './task-progress-reporter.js';
+import { OpenAIExtractor } from './structured-extraction.js';
 export { PermissionMode } from './types.js';
 
 class Agent {
@@ -27,6 +28,8 @@ class Agent {
     workflowRegistry;
     stageExecutor;
     progressReporter;
+    promptBuilder;
+    extractor;
     mcpServers;
     debug;
     constructor(config = {}) {
@@ -73,16 +76,20 @@ class Agent {
             });
         }
         this.workflowRegistry = new WorkflowRegistry(this.posthogAPI);
-        const promptBuilder = new PromptBuilder({
+        this.promptBuilder = new PromptBuilder({
             getTaskFiles: (taskId) => this.getTaskFiles(taskId),
             generatePlanTemplate: (vars) => this.templateManager.generatePlan(vars),
             posthogClient: this.posthogAPI,
             logger: this.logger.child('PromptBuilder')
         });
-        this.stageExecutor = new StageExecutor(this.agentRegistry, this.logger, promptBuilder, undefined, // eventHandler set via setEventHandler below
+        this.stageExecutor = new StageExecutor(this.agentRegistry, this.logger, this.promptBuilder, undefined, // eventHandler set via setEventHandler below
         this.mcpServers);
         this.stageExecutor.setEventHandler((event) => this.emitEvent(event));
         this.progressReporter = new TaskProgressReporter(this.posthogAPI, this.logger);
+        // Initialize OpenAI extractor if API key is available
+        if (process.env.OPENAI_API_KEY) {
+            this.extractor = new OpenAIExtractor(this.logger.child('OpenAIExtractor'));
+        }
     }
     /**
      * Enable or disable debug logging
@@ -130,24 +137,29 @@ class Agent {
                     await this.posthogAPI.updateTask(task.id, { workflow: workflowId });
                     task.workflow = workflowId;
                 }
-                if (!task.current_stage && workflow.stages.length > 0) {
-                    const firstStage = [...workflow.stages].sort((a, b) => a.position - b.position)[0];
-                    await this.posthogAPI.updateTaskStage(task.id, firstStage.id);
-                    task.current_stage = firstStage.id;
-                }
             }
             catch (e) {
-                this.logger.warn('Failed to sync task workflow/stage before execution', { error: e.message });
+                this.logger.warn('Failed to sync task workflow before execution', { error: e.message });
             }
         }
         const executionId = this.taskManager.generateExecutionId();
         this.logger.info('Starting workflow execution', { taskId: task.id, workflowId, executionId });
         this.taskManager.startExecution(task.id, 'plan_and_build', executionId);
         await this.progressReporter.start(task.id, {
-            workflowId,
-            workflowRunId: executionId,
             totalSteps: orderedStages.length,
         });
+        // Set initial stage on the newly created run
+        const firstStage = orderedStages[0];
+        if (this.posthogAPI && this.progressReporter.runId && firstStage) {
+            try {
+                await this.posthogAPI.updateTaskRun(task.id, this.progressReporter.runId, {
+                    current_stage: firstStage.id
+                });
+            }
+            catch (e) {
+                this.logger.warn('Failed to set initial stage on run', { error: e.message });
+            }
+        }
         try {
             let startIndex = 0;
             const currentStageId = task.current_stage;
@@ -169,15 +181,18 @@ class Agent {
                 if (idx >= 0)
                     startIndex = idx;
             }
-            // Align server-side stage when restarting from the beginning
-            if (this.posthogAPI) {
+            // Align server-side stage when restarting from a different stage
+            if (this.posthogAPI && this.progressReporter.runId) {
                 const targetStage = orderedStages[startIndex];
                 if (targetStage && targetStage.id !== currentStageId) {
                     try {
-                        await this.posthogAPI.updateTaskStage(task.id, targetStage.id);
-                        task.current_stage = targetStage.id;
+                        await this.posthogAPI.updateTaskRun(task.id, this.progressReporter.runId, {
+                            current_stage: targetStage.id
+                        });
+                    }
+                    catch (e) {
+                        this.logger.warn('Failed to update run stage', { error: e.message });
                     }
-                    catch { }
                 }
             }
             for (let i = startIndex; i < orderedStages.length; i++) {
@@ -284,16 +299,320 @@ class Agent {
         }
         this.emitEvent(this.adapter.createStatusEvent('stage_complete', { stage: stage.key }));
     }
+    // Adaptive task execution - 3-phase workflow (research → plan → build)
+    async runTask(taskOrId, options = {}) {
+        await this._configureLlmGateway();
+        const task = typeof taskOrId === 'string' ? await this.fetchTask(taskOrId) : taskOrId;
+        const cwd = options.repositoryPath || this.workingDirectory;
+        const isCloudMode = options.isCloudMode ?? false;
+        const taskSlug = task.slug || task.id;
+        this.logger.info('Starting adaptive task execution', { taskId: task.id, taskSlug, isCloudMode });
+        // Initialize progress reporter for task run tracking (needed for PR attachment)
+        await this.progressReporter.start(task.id, { totalSteps: 3 }); // 3 phases: research, plan, build
+        this.emitEvent(this.adapter.createStatusEvent('run_started', { runId: this.progressReporter.runId }));
+        // Phase 1: Branch check
+        const existingBranch = await this.gitManager.getTaskBranch(taskSlug);
+        if (!existingBranch) {
+            this.logger.info('Creating task branch', { taskSlug });
+            const branchName = `posthog/task-${taskSlug}`;
+            await this.gitManager.createOrSwitchToBranch(branchName);
+            this.emitEvent(this.adapter.createStatusEvent('branch_created', { branch: branchName }));
+            // Initial commit
+            await this.fileManager.ensureGitignore();
+            await this.gitManager.addAllPostHogFiles();
+            if (isCloudMode) {
+                await this.gitManager.commitAndPush(`Initialize task ${taskSlug}`, { allowEmpty: true });
+            }
+            else {
+                await this.gitManager.commitChanges(`Initialize task ${taskSlug}`);
+            }
+        }
+        else {
+            this.logger.info('Switching to existing task branch', { branch: existingBranch });
+            await this.gitManager.switchToBranch(existingBranch);
+        }
+        // Phase 2: Research
+        const researchExists = await this.fileManager.readResearch(task.id);
+        if (!researchExists) {
+            this.logger.info('Starting research phase', { taskId: task.id });
+            this.emitEvent(this.adapter.createStatusEvent('phase_start', { phase: 'research' }));
+            // Run research agent
+            const researchPrompt = await this.promptBuilder.buildResearchPrompt(task, cwd);
+            const { RESEARCH_SYSTEM_PROMPT } = await import('./agents/research.js');
+            const fullPrompt = RESEARCH_SYSTEM_PROMPT + '\n\n' + researchPrompt;
+            const baseOptions = {
+                model: 'claude-sonnet-4-5-20250929',
+                cwd,
+                permissionMode: 'plan',
+                settingSources: ['local'],
+                mcpServers: this.mcpServers,
+            };
+            const response = query({
+                prompt: fullPrompt,
+                options: { ...baseOptions, ...(options.queryOverrides || {}) },
+            });
+            let researchContent = '';
+            for await (const message of response) {
+                this.emitEvent(this.adapter.createRawSDKEvent(message));
+                const transformed = this.adapter.transform(message);
+                if (transformed) {
+                    this.emitEvent(transformed);
+                }
+                if (message.type === 'assistant' && message.message?.content) {
+                    for (const c of message.message.content) {
+                        if (c.type === 'text' && c.text)
+                            researchContent += c.text + '\n';
+                    }
+                }
+            }
+            // Write research.md
+            if (researchContent.trim()) {
+                await this.fileManager.writeResearch(task.id, researchContent.trim());
+                this.logger.info('Research completed', { taskId: task.id });
+            }
+            // Commit research
+            await this.gitManager.addAllPostHogFiles();
+            // Extract questions using structured output and save to questions.json
+            if (this.extractor) {
+                try {
+                    this.logger.info('Extracting questions from research.md', { taskId: task.id });
+                    const questions = await this.extractQuestionsFromResearch(task.id, false);
+                    this.logger.info('Questions extracted successfully', { taskId: task.id, count: questions.length });
+                    // Save questions.json
+                    await this.fileManager.writeQuestions(task.id, {
+                        questions,
+                        answered: false,
+                        answers: null,
+                    });
+                    this.logger.info('Questions saved to questions.json', { taskId: task.id });
+                    // Emit event for Array to pick up (local mode)
+                    if (!isCloudMode) {
+                        this.emitEvent({
+                            type: 'artifact',
+                            ts: Date.now(),
+                            kind: 'research_questions',
+                            content: questions,
+                        });
+                        this.logger.info('Emitted research_questions artifact event', { taskId: task.id });
+                    }
+                }
+                catch (error) {
+                    this.logger.error('Failed to extract questions', { error: error instanceof Error ? error.message : String(error) });
+                    this.emitEvent({
+                        type: 'error',
+                        ts: Date.now(),
+                        message: `Failed to extract questions: ${error instanceof Error ? error.message : String(error)}`,
+                    });
+                }
+            }
+            else {
+                this.logger.warn('OpenAI extractor not available (OPENAI_API_KEY not set), skipping question extraction');
+                this.emitEvent({
+                    type: 'status',
+                    ts: Date.now(),
+                    phase: 'extraction_skipped',
+                    message: 'Question extraction skipped - OPENAI_API_KEY not configured',
+                });
+            }
+            if (isCloudMode) {
+                await this.gitManager.commitAndPush(`Research phase for ${task.title}`);
+            }
+            else {
+                await this.gitManager.commitChanges(`Research phase for ${task.title}`);
+                this.emitEvent(this.adapter.createStatusEvent('phase_complete', { phase: 'research' }));
+                return; // Local mode: return to user
+            }
+        }
+        // Phase 3: Auto-answer questions (cloud mode only)
+        if (isCloudMode) {
+            const questionsData = await this.fileManager.readQuestions(task.id);
+            if (questionsData && !questionsData.answered) {
+                this.logger.info('Auto-answering research questions', { taskId: task.id });
+                // Extract questions with recommended answers using structured output
+                if (this.extractor) {
+                    const questionsWithAnswers = await this.extractQuestionsFromResearch(task.id, true);
+                    // Save answers to questions.json
+                    await this.fileManager.writeQuestions(task.id, {
+                        questions: questionsWithAnswers.map(qa => ({
+                            id: qa.id,
+                            question: qa.question,
+                            options: qa.options,
+                        })),
+                        answered: true,
+                        answers: questionsWithAnswers.map(qa => ({
+                            questionId: qa.id,
+                            selectedOption: qa.recommendedAnswer,
+                            customInput: qa.justification,
+                        })),
+                    });
+                    this.logger.info('Auto-answers saved to questions.json', { taskId: task.id });
+                    await this.gitManager.addAllPostHogFiles();
+                    await this.gitManager.commitAndPush(`Answer research questions for ${task.title}`);
+                }
+                else {
+                    this.logger.warn('OpenAI extractor not available, skipping auto-answer');
+                }
+            }
+        }
+        // Phase 4: Plan
+        const planExists = await this.readPlan(task.id);
+        if (!planExists) {
+            // Check if questions have been answered
+            const questionsData = await this.fileManager.readQuestions(task.id);
+            if (!questionsData || !questionsData.answered) {
+                this.logger.info('Waiting for user answers to research questions');
+                this.emitEvent(this.adapter.createStatusEvent('phase_complete', { phase: 'research_questions' }));
+                return; // Wait for user to answer questions
+            }
+            this.logger.info('Starting planning phase', { taskId: task.id });
+            this.emitEvent(this.adapter.createStatusEvent('phase_start', { phase: 'planning' }));
+            // Build context with research + questions + answers
+            const research = await this.fileManager.readResearch(task.id);
+            let researchContext = '';
+            if (research) {
+                researchContext += `## Research Analysis\n\n${research}\n\n`;
+            }
+            // Add questions and answers
+            researchContext += `## Implementation Decisions\n\n`;
+            const answers = questionsData.answers || [];
+            for (const question of questionsData.questions) {
+                const answer = answers.find((a) => a.questionId === question.id);
+                researchContext += `### ${question.question}\n\n`;
+                if (answer) {
+                    researchContext += `**Selected:** ${answer.selectedOption}\n`;
+                    if (answer.customInput) {
+                        researchContext += `**Details:** ${answer.customInput}\n`;
+                    }
+                }
+                else {
+                    this.logger.warn('No answer found for question', { questionId: question.id });
+                    researchContext += `**Selected:** Not answered\n`;
+                }
+                researchContext += '\n';
+            }
+            // Run planning agent with full context
+            const planningPrompt = await this.promptBuilder.buildPlanningPrompt(task, cwd);
+            const { PLANNING_SYSTEM_PROMPT } = await import('./agents/planning.js');
+            const fullPrompt = PLANNING_SYSTEM_PROMPT + '\n\n' + planningPrompt + '\n\n' + researchContext;
+            const baseOptions = {
+                model: 'claude-sonnet-4-5-20250929',
+                cwd,
+                permissionMode: 'plan',
+                settingSources: ['local'],
+                mcpServers: this.mcpServers,
+            };
+            const response = query({
+                prompt: fullPrompt,
+                options: { ...baseOptions, ...(options.queryOverrides || {}) },
+            });
+            let planContent = '';
+            for await (const message of response) {
+                this.emitEvent(this.adapter.createRawSDKEvent(message));
+                const transformed = this.adapter.transform(message);
+                if (transformed) {
+                    this.emitEvent(transformed);
+                }
+                if (message.type === 'assistant' && message.message?.content) {
+                    for (const c of message.message.content) {
+                        if (c.type === 'text' && c.text)
+                            planContent += c.text + '\n';
+                    }
+                }
+            }
+            // Write plan.md
+            if (planContent.trim()) {
+                await this.writePlan(task.id, planContent.trim());
+                this.logger.info('Plan completed', { taskId: task.id });
+            }
+            // Commit plan
+            await this.gitManager.addAllPostHogFiles();
+            if (isCloudMode) {
+                await this.gitManager.commitAndPush(`Planning phase for ${task.title}`);
+            }
+            else {
+                await this.gitManager.commitChanges(`Planning phase for ${task.title}`);
+                this.emitEvent(this.adapter.createStatusEvent('phase_complete', { phase: 'planning' }));
+                return; // Local mode: return to user
+            }
+        }
+        // Phase 5: Build
+        const latestRun = task.latest_run;
+        const prExists = latestRun?.output && latestRun.output.pr_url;
+        if (!prExists) {
+            this.logger.info('Starting build phase', { taskId: task.id });
+            this.emitEvent(this.adapter.createStatusEvent('phase_start', { phase: 'build' }));
+            // Run execution agent
+            const executionPrompt = await this.promptBuilder.buildExecutionPrompt(task, cwd);
+            const { EXECUTION_SYSTEM_PROMPT } = await import('./agents/execution.js');
+            const fullPrompt = EXECUTION_SYSTEM_PROMPT + '\n\n' + executionPrompt;
+            const { PermissionMode } = await import('./types.js');
+            const permissionMode = options.permissionMode || PermissionMode.ACCEPT_EDITS;
+            const baseOptions = {
+                model: 'claude-sonnet-4-5-20250929',
+                cwd,
+                permissionMode,
+                settingSources: ['local'],
+                mcpServers: this.mcpServers,
+            };
+            const response = query({
+                prompt: fullPrompt,
+                options: { ...baseOptions, ...(options.queryOverrides || {}) },
+            });
+            for await (const message of response) {
+                this.emitEvent(this.adapter.createRawSDKEvent(message));
+                const transformed = this.adapter.transform(message);
+                if (transformed) {
+                    this.emitEvent(transformed);
+                }
+            }
+            // Commit and push implementation
+            // Stage ALL changes (not just .posthog/)
+            const hasChanges = await this.gitManager.hasChanges();
+            if (hasChanges) {
+                await this.gitManager.addFiles(['.']); // Stage all changes
+                await this.gitManager.commitChanges(`Implementation for ${task.title}`);
+                // Push to origin
+                const branchName = await this.gitManager.getCurrentBranch();
+                await this.gitManager.pushBranch(branchName);
+                this.logger.info('Implementation committed and pushed', { taskId: task.id });
+            }
+            else {
+                this.logger.warn('No changes to commit in build phase', { taskId: task.id });
+            }
+            // Create PR
+            const branchName = await this.gitManager.getCurrentBranch();
+            const prUrl = await this.createPullRequest(task.id, branchName, task.title, task.description);
+            this.logger.info('Pull request created', { taskId: task.id, prUrl });
+            this.emitEvent(this.adapter.createStatusEvent('pr_created', { prUrl }));
+            // Attach PR to task run
+            try {
+                await this.attachPullRequestToTask(task.id, prUrl, branchName);
+                this.logger.info('PR attached to task successfully', { taskId: task.id, prUrl });
+            }
+            catch (error) {
+                this.logger.warn('Could not attach PR to task', { error: error instanceof Error ? error.message : String(error) });
+            }
+        }
+        else {
+            this.logger.info('PR already exists, skipping build phase', { taskId: task.id });
+        }
+        // Phase 6: Complete
+        await this.progressReporter.complete();
+        this.logger.info('Task execution complete', { taskId: task.id });
+        this.emitEvent(this.adapter.createStatusEvent('task_complete', { taskId: task.id }));
+    }
     async progressToNextStage(taskId, currentStageKey) {
-        if (!this.posthogAPI)
-            throw new Error('PostHog API not configured. Cannot progress stage.');
+        if (!this.posthogAPI || !this.progressReporter.runId) {
+            throw new Error('PostHog API not configured or no active run. Cannot progress stage.');
+        }
         try {
-            await this.posthogAPI.progressTask(taskId, { auto: true });
+            await this.posthogAPI.progressTaskRun(taskId, this.progressReporter.runId);
         }
         catch (error) {
             if (error instanceof Error && error.message.includes('No next stage available')) {
-                this.logger.warn('No next stage available when attempting to progress task', {
+                this.logger.warn('No next stage available when attempting to progress run', {
                     taskId,
+                    runId: this.progressReporter.runId,
                     stage: currentStageKey,
                     error: error.message,
                 });
@@ -374,6 +693,22 @@ class Agent {
         this.logger.debug('Reading plan', { taskId });
         return await this.fileManager.readPlan(taskId);
     }
+    async extractQuestionsFromResearch(taskId, includeAnswers = false) {
+        this.logger.info('Extracting questions from research.md', { taskId, includeAnswers });
+        if (!this.extractor) {
+            throw new Error('OpenAI extractor not initialized. Set OPENAI_API_KEY environment variable.');
+        }
+        const researchContent = await this.fileManager.readResearch(taskId);
+        if (!researchContent) {
+            throw new Error('research.md not found for task ' + taskId);
+        }
+        if (includeAnswers) {
+            return await this.extractor.extractQuestionsWithAnswers(researchContent);
+        }
+        else {
+            return await this.extractor.extractQuestions(researchContent);
+        }
+    }
     // Git operations for task workflow
     async createPlanningBranch(taskId) {
         this.logger.info('Creating planning branch', { taskId });
@@ -417,24 +752,30 @@ Generated by PostHog Agent`;
         return prUrl;
     }
     async attachPullRequestToTask(taskId, prUrl, branchName) {
-        this.logger.info('Attaching PR to task', { taskId, prUrl, branchName });
-        if (!this.posthogAPI) {
-            const error = new Error('PostHog API not configured. Cannot attach PR to task.');
+        this.logger.info('Attaching PR to task run', { taskId, prUrl, branchName });
+        if (!this.posthogAPI || !this.progressReporter.runId) {
+            const error = new Error('PostHog API not configured or no active run. Cannot attach PR to task.');
             this.logger.error('PostHog API not configured', error);
             throw error;
         }
-        await this.posthogAPI.attachTaskPullRequest(taskId, prUrl, branchName);
-        this.logger.debug('PR attached to task', { taskId, prUrl });
+        const updates = {
+            output: { pr_url: prUrl }
+        };
+        if (branchName) {
+            updates.branch = branchName;
+        }
+        await this.posthogAPI.updateTaskRun(taskId, this.progressReporter.runId, updates);
+        this.logger.debug('PR attached to task run', { taskId, runId: this.progressReporter.runId, prUrl });
     }
     async updateTaskBranch(taskId, branchName) {
-        this.logger.info('Updating task branch', { taskId, branchName });
-        if (!this.posthogAPI) {
-            const error = new Error('PostHog API not configured. Cannot update task branch.');
+        this.logger.info('Updating task run branch', { taskId, branchName });
+        if (!this.posthogAPI || !this.progressReporter.runId) {
+            const error = new Error('PostHog API not configured or no active run. Cannot update branch.');
             this.logger.error('PostHog API not configured', error);
             throw error;
         }
-        await this.posthogAPI.setTaskBranch(taskId, branchName);
-        this.logger.debug('Task branch updated', { taskId, branchName });
+        await this.posthogAPI.updateTaskRun(taskId, this.progressReporter.runId, { branch: branchName });
+        this.logger.debug('Task run branch updated', { taskId, runId: this.progressReporter.runId, branchName });
     }
     // Execution management
     cancelTask(taskId) {