@posthog/agent 1.22.0 → 1.24.0

package/src/workflow/config.ts

@@ -1,53 +1,53 @@
-import type { WorkflowDefinition } from './types.js';
-import { researchStep } from './steps/research.js';
-import { planStep } from './steps/plan.js';
-import { buildStep } from './steps/build.js';
-import { finalizeStep } from './steps/finalize.js';
+import { buildStep } from "./steps/build.js";
+import { finalizeStep } from "./steps/finalize.js";
+import { planStep } from "./steps/plan.js";
+import { researchStep } from "./steps/research.js";
+import type { WorkflowDefinition } from "./types.js";
 
 const MODELS = {
-    SONNET: "claude-sonnet-4-5",
-    HAIKU: "claude-haiku-4-5",
-}
+  SONNET: "claude-sonnet-4-5",
+  HAIKU: "claude-haiku-4-5",
+};
 
 export const TASK_WORKFLOW: WorkflowDefinition = [
-    {
-        id: 'research',
-        name: 'Research',
-        agent: 'research',
-        model: MODELS.HAIKU,
-        permissionMode: 'plan',
-        commit: true,
-        push: true,
-        run: researchStep,
-    },
-    {
-        id: 'plan',
-        name: 'Plan',
-        agent: 'planning',
-        model: MODELS.SONNET,
-        permissionMode: 'plan',
-        commit: true,
-        push: true,
-        run: planStep,
-    },
-    {
-        id: 'build',
-        name: 'Build',
-        agent: 'execution',
-        model: MODELS.SONNET,
-        permissionMode: 'acceptEdits',
-        commit: true,
-        push: true,
-        run: buildStep,
-    },
-    {
-        id: 'finalize',
-        name: 'Finalize',
-        agent: 'system', // not used
-        model: MODELS.HAIKU, // not used
-        permissionMode: 'plan', // not used
-        commit: true,
-        push: true,
-        run: finalizeStep,
-    },
+  {
+    id: "research",
+    name: "Research",
+    agent: "research",
+    model: MODELS.HAIKU,
+    permissionMode: "plan",
+    commit: true,
+    push: true,
+    run: researchStep,
+  },
+  {
+    id: "plan",
+    name: "Plan",
+    agent: "planning",
+    model: MODELS.SONNET,
+    permissionMode: "plan",
+    commit: true,
+    push: true,
+    run: planStep,
+  },
+  {
+    id: "build",
+    name: "Build",
+    agent: "execution",
+    model: MODELS.SONNET,
+    permissionMode: "acceptEdits",
+    commit: true,
+    push: true,
+    run: buildStep,
+  },
+  {
+    id: "finalize",
+    name: "Finalize",
+    agent: "system", // not used
+    model: MODELS.HAIKU, // not used
+    permissionMode: "plan", // not used
+    commit: true,
+    push: true,
+    run: finalizeStep,
+  },
 ];

package/src/workflow/steps/build.ts

@@ -1,121 +1,130 @@
-import { query } from '@anthropic-ai/claude-agent-sdk';
-import { EXECUTION_SYSTEM_PROMPT } from '../../agents/execution.js';
-import { PermissionMode } from '../../types.js';
-import type { WorkflowStepRunner } from '../types.js';
-import { finalizeStepGitActions } from '../utils.js';
-import { TodoManager } from '../../todo-manager.js';
+import { query } from "@anthropic-ai/claude-agent-sdk";
+import { EXECUTION_SYSTEM_PROMPT } from "../../agents/execution.js";
+import { TodoManager } from "../../todo-manager.js";
+import { PermissionMode } from "../../types.js";
+import type { WorkflowStepRunner } from "../types.js";
 
 export const buildStep: WorkflowStepRunner = async ({ step, context }) => {
-    const {
-        task,
-        cwd,
-        options,
-        logger,
-        promptBuilder,
-        adapter,
-        mcpServers,
-        gitManager,
-        emitEvent,
-    } = context;
-
-    const stepLogger = logger.child('BuildStep');
-
-    const latestRun = task.latest_run;
-    const prExists =
-        latestRun?.output && typeof latestRun.output === 'object'
-            ? (latestRun.output as any).pr_url
-            : null;
-
-    if (prExists) {
-        stepLogger.info('PR already exists, skipping build phase', { taskId: task.id });
-        return { status: 'skipped' };
-    }
-
-    stepLogger.info('Starting build phase', { taskId: task.id });
-    emitEvent(adapter.createStatusEvent('phase_start', { phase: 'build' }));
-
-    const executionPrompt = await promptBuilder.buildExecutionPrompt(task, cwd);
-    const fullPrompt = `${EXECUTION_SYSTEM_PROMPT}\n\n${executionPrompt}`;
-
-    const configuredPermissionMode =
-        options.permissionMode ??
-        (typeof step.permissionMode === 'string'
-            ? (step.permissionMode as PermissionMode)
-            : step.permissionMode) ??
-        PermissionMode.ACCEPT_EDITS;
-
-    const baseOptions: Record<string, any> = {
-        model: step.model,
-        cwd,
-        permissionMode: configuredPermissionMode,
-        settingSources: ['local'],
-        mcpServers,
-        // Allow all tools for build phase - full read/write access needed for implementation
-        allowedTools: [
-            'Task',
-            'Bash',
-            'BashOutput',
-            'KillBash',
-            'Edit',
-            'Read',
-            'Write',
-            'Glob',
-            'Grep',
-            'NotebookEdit',
-            'WebFetch',
-            'WebSearch',
-            'ListMcpResources',
-            'ReadMcpResource',
-            'TodoWrite',
-        ],
-    };
-
-    // Add fine-grained permission hook if provided
-    if (options.canUseTool) {
-        baseOptions.canUseTool = options.canUseTool;
-    }
-
-    const response = query({
-        prompt: fullPrompt,
-        options: { ...baseOptions, ...(options.queryOverrides || {}) },
+  const {
+    task,
+    cwd,
+    options,
+    logger,
+    promptBuilder,
+    adapter,
+    mcpServers,
+    gitManager,
+    emitEvent,
+  } = context;
+
+  const stepLogger = logger.child("BuildStep");
+
+  const latestRun = task.latest_run;
+  const prExists =
+    latestRun?.output && typeof latestRun.output === "object"
+      ? (latestRun.output as Record<string, unknown>).pr_url
+      : null;
+
+  if (prExists) {
+    stepLogger.info("PR already exists, skipping build phase", {
+      taskId: task.id,
     });
-
-    // Track commits made during Claude Code execution
-    const commitTracker = await gitManager.trackCommitsDuring();
-
-    // Track todos from TodoWrite tool calls
-    const todoManager = new TodoManager(context.fileManager, stepLogger);
-
+    return { status: "skipped" };
+  }
+
+  stepLogger.info("Starting build phase", { taskId: task.id });
+  emitEvent(adapter.createStatusEvent("phase_start", { phase: "build" }));
+
+  const executionPrompt = await promptBuilder.buildExecutionPrompt(task, cwd);
+  const fullPrompt = `${EXECUTION_SYSTEM_PROMPT}\n\n${executionPrompt}`;
+
+  const configuredPermissionMode =
+    options.permissionMode ??
+    (typeof step.permissionMode === "string"
+      ? (step.permissionMode as PermissionMode)
+      : step.permissionMode) ??
+    PermissionMode.ACCEPT_EDITS;
+
+  const baseOptions: Record<string, unknown> = {
+    model: step.model,
+    cwd,
+    permissionMode: configuredPermissionMode,
+    settingSources: ["local"],
+    mcpServers,
+    // Allow all tools for build phase - full read/write access needed for implementation
+    allowedTools: [
+      "Task",
+      "Bash",
+      "BashOutput",
+      "KillBash",
+      "Edit",
+      "Read",
+      "Write",
+      "Glob",
+      "Grep",
+      "NotebookEdit",
+      "WebFetch",
+      "WebSearch",
+      "ListMcpResources",
+      "ReadMcpResource",
+      "TodoWrite",
+    ],
+  };
+
+  // Add fine-grained permission hook if provided
+  if (options.canUseTool) {
+    baseOptions.canUseTool = options.canUseTool;
+  }
+
+  const response = query({
+    prompt: fullPrompt,
+    options: { ...baseOptions, ...(options.queryOverrides || {}) },
+  });
+
+  // Track commits made during Claude Code execution
+  const commitTracker = await gitManager.trackCommitsDuring();
+
+  // Track todos from TodoWrite tool calls
+  const todoManager = new TodoManager(context.fileManager, stepLogger);
+
+  try {
     for await (const message of response) {
-        emitEvent(adapter.createRawSDKEvent(message));
-        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));
-        }
+      emitEvent(adapter.createRawSDKEvent(message));
+      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));
+      }
     }
-
-    // Finalize: commit any remaining changes and optionally push
-    const { commitCreated, pushedBranch } = await commitTracker.finalize({
-        commitMessage: `Implementation for ${task.title}`,
-        push: step.push,
+  } catch (error) {
+    stepLogger.error("Error during build step query", error);
+    throw error;
+  }
+
+  // Finalize: commit any remaining changes and optionally push
+  const { commitCreated, pushedBranch } = await commitTracker.finalize({
+    commitMessage: `Implementation for ${task.title}`,
+    push: step.push,
+  });
+
+  context.stepResults[step.id] = { commitCreated };
+
+  if (!commitCreated) {
+    stepLogger.warn("No changes to commit in build phase", { taskId: task.id });
+  } else {
+    stepLogger.info("Build commits finalized", {
+      taskId: task.id,
+      pushedBranch,
     });
+  }
 
-    context.stepResults[step.id] = { commitCreated };
-
-    if (!commitCreated) {
-        stepLogger.warn('No changes to commit in build phase', { taskId: task.id });
-    } else {
-        stepLogger.info('Build commits finalized', {
-            taskId: task.id,
-            pushedBranch
-        });
-    }
-
-    emitEvent(adapter.createStatusEvent('phase_complete', { phase: 'build' }));
-    return { status: 'completed' };
+  emitEvent(adapter.createStatusEvent("phase_complete", { phase: "build" }));
+  return { status: "completed" };
 };