@posthog/agent 1.4.0 → 1.6.0

package/src/stage-executor.ts

@@ -71,7 +71,7 @@ export class StageExecutor {
   }
 
   private async runPlanning(task: Task, cwd: string, options: WorkflowExecutionOptions, stageKey: string): Promise<WorkflowStageExecutionResult> {
-    const contextPrompt = await this.promptBuilder.buildPlanningPrompt(task);
+    const contextPrompt = await this.promptBuilder.buildPlanningPrompt(task, cwd);
     let prompt = PLANNING_SYSTEM_PROMPT + '\n\n' + contextPrompt;
 
     const stageOverrides = options.stageOverrides?.[stageKey] || options.stageOverrides?.['plan'];
@@ -95,6 +95,10 @@ export class StageExecutor {
 
     let plan = '';
     for await (const message of response) {
+      // Emit raw SDK event first
+      this.eventHandler?.(this.eventTransformer.createRawSDKEvent(message));
+      
+      // Then emit transformed event
       const transformed = this.eventTransformer.transform(message);
       if (transformed) {
         if (transformed.type !== 'token') {
@@ -102,6 +106,7 @@ export class StageExecutor {
         }
         this.eventHandler?.(transformed);
       }
+      
       if (message.type === 'assistant' && message.message?.content) {
         for (const c of message.message.content) {
           if (c.type === 'text' && c.text) plan += c.text + '\n';
@@ -113,7 +118,7 @@ export class StageExecutor {
   }
 
   private async runExecution(task: Task, cwd: string, permissionMode: WorkflowExecutionOptions['permissionMode'], options: WorkflowExecutionOptions, stageKey: string): Promise<WorkflowStageExecutionResult> {
-    const contextPrompt = await this.promptBuilder.buildExecutionPrompt(task);
+    const contextPrompt = await this.promptBuilder.buildExecutionPrompt(task, cwd);
     let prompt = EXECUTION_SYSTEM_PROMPT + '\n\n' + contextPrompt;
 
     const stageOverrides = options.stageOverrides?.[stageKey];
@@ -136,6 +141,10 @@ export class StageExecutor {
     });
     const results: any[] = [];
     for await (const message of response) {
+      // Emit raw SDK event first
+      this.eventHandler?.(this.eventTransformer.createRawSDKEvent(message));
+      
+      // Then emit transformed event
       const transformed = this.eventTransformer.transform(message);
       if (transformed) {
         if (transformed.type !== 'token') {
@@ -143,6 +152,7 @@ export class StageExecutor {
         }
         this.eventHandler?.(transformed);
       }
+      
       results.push(message);
     }
     return { results };

package/src/task-progress-reporter.ts

@@ -120,15 +120,30 @@ export class TaskProgressReporter {
       case 'content_block_start':
       case 'content_block_stop':
       case 'compact_boundary':
-      case 'tool_call':
-      case 'tool_result':
       case 'message_start':
       case 'message_stop':
       case 'metric':
       case 'artifact':
+      case 'raw_sdk_event':
         // Skip verbose streaming artifacts from persistence
         return;
 
+      case 'tool_call': {
+        const logLine = this.formatToolCallEvent(event);
+        if (logLine) {
+          await this.appendLog(logLine);
+        }
+        return;
+      }
+
+      case 'tool_result': {
+        const logLine = this.formatToolResultEvent(event);
+        if (logLine) {
+          await this.appendLog(logLine);
+        }
+        return;
+      }
+
       case 'file_write':
         await this.appendLog(this.formatFileWriteEvent(event));
         return;
@@ -284,4 +299,31 @@ export class TaskProgressReporter {
     const compact = text.replace(/\s+/g, ' ').trim();
     return compact.length > max ? `${compact.slice(0, max)}…` : compact;
   }
+
+  private formatToolCallEvent(event: Extract<AgentEvent, { type: 'tool_call' }>): string | null {
+    // File operations to track
+    const fileOps = ['read_file', 'write', 'search_replace', 'delete_file', 'glob_file_search', 'file_search', 'list_dir', 'edit_notebook'];
+    // Terminal commands to track
+    const terminalOps = ['run_terminal_cmd', 'bash', 'shell'];
+
+    if (fileOps.includes(event.toolName)) {
+      // Extract file path from args
+      const path = event.args?.target_file || event.args?.file_path || event.args?.target_notebook || event.args?.target_directory || '';
+      return `[tool] ${event.toolName}${path ? `: ${path}` : ''}`;
+    } else if (terminalOps.includes(event.toolName)) {
+      // Extract command from args
+      const cmd = event.args?.command || '';
+      const truncated = cmd.length > 80 ? `${cmd.slice(0, 80)}…` : cmd;
+      return `[cmd] ${truncated}`;
+    }
+
+    // Skip other tools from persistence
+    return null;
+  }
+
+  private formatToolResultEvent(event: Extract<AgentEvent, { type: 'tool_result' }>): string | null {
+    // We don't need to log tool results separately - tool calls are sufficient
+    // This keeps the log concise
+    return null;
+  }
 }

package/src/types.ts

@@ -141,6 +141,8 @@ export interface ErrorEvent extends BaseEvent {
   message: string;
   error?: any;
   errorType?: string;
+  context?: Record<string, any>; // Partial error context for debugging
+  sdkError?: any; // Original SDK error object
 }
 
 // Legacy events (keeping for backwards compatibility)
@@ -170,6 +172,11 @@ export interface ArtifactEvent extends BaseEvent {
   content: any;
 }
 
+export interface RawSDKEvent extends BaseEvent {
+  type: 'raw_sdk_event';
+  sdkMessage: any; // Full SDK message for debugging
+}
+
 export type AgentEvent =
   | TokenEvent
   | ContentBlockStartEvent
@@ -188,7 +195,8 @@ export type AgentEvent =
   | DiffEvent
   | FileWriteEvent
   | MetricEvent
-  | ArtifactEvent;
+  | ArtifactEvent
+  | RawSDKEvent;
 
 export interface ExecutionResult {
   results: any[];