@posthog/agent 1.24.0 → 1.24.2

package/src/workflow/config.ts

@@ -1,53 +1,53 @@
-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";
+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';
 
 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,130 +1,121 @@
-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";
+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';
 
 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 Record<string, unknown>).pr_url
-      : null;
-
-  if (prExists) {
-    stepLogger.info("PR already exists, skipping build phase", {
-      taskId: task.id,
+    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 || {}) },
     });
-    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 {
+
+    // 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);
+
     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));
+        }
     }
-  } 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,
+
+    // Finalize: commit any remaining changes and optionally push
+    const { commitCreated, pushedBranch } = await commitTracker.finalize({
+        commitMessage: `Implementation for ${task.title}`,
+        push: step.push,
     });
-  }
 
-  emitEvent(adapter.createStatusEvent("phase_complete", { phase: "build" }));
-  return { status: "completed" };
+    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' };
 };