@posthog/agent 1.24.0 → 1.24.2

package/src/tools/types.ts

@@ -3,12 +3,12 @@
  * Makes it easier for UIs to filter and display tools by function.
  */
 export type ToolCategory =
-  | "filesystem" // File operations: Read, Write, Edit, Glob, NotebookEdit
-  | "shell" // Shell operations: Bash, BashOutput, KillShell
-  | "web" // Web operations: WebFetch, WebSearch
-  | "assistant" // Assistant operations: Task, TodoWrite, ExitPlanMode
-  | "search" // Search operations: Grep
-  | "unknown"; // Unknown or unrecognized tools
+  | 'filesystem'  // File operations: Read, Write, Edit, Glob, NotebookEdit
+  | 'shell'       // Shell operations: Bash, BashOutput, KillShell
+  | 'web'         // Web operations: WebFetch, WebSearch
+  | 'assistant'   // Assistant operations: Task, TodoWrite, ExitPlanMode
+  | 'search'      // Search operations: Grep
+  | 'unknown';    // Unknown or unrecognized tools
 
 /**
  * Base tool interface representing a tool that can be called by the agent.
@@ -23,86 +23,86 @@ export interface Tool {
 // Filesystem tools
 
 export interface ReadTool extends Tool {
-  name: "Read";
-  category: "filesystem";
+  name: 'Read';
+  category: 'filesystem';
 }
 
 export interface WriteTool extends Tool {
-  name: "Write";
-  category: "filesystem";
+  name: 'Write';
+  category: 'filesystem';
 }
 
 export interface EditTool extends Tool {
-  name: "Edit";
-  category: "filesystem";
+  name: 'Edit';
+  category: 'filesystem';
 }
 
 export interface GlobTool extends Tool {
-  name: "Glob";
-  category: "filesystem";
+  name: 'Glob';
+  category: 'filesystem';
 }
 
 export interface NotebookEditTool extends Tool {
-  name: "NotebookEdit";
-  category: "filesystem";
+  name: 'NotebookEdit';
+  category: 'filesystem';
 }
 
 // Shell tools
 
 export interface BashTool extends Tool {
-  name: "Bash";
-  category: "shell";
+  name: 'Bash';
+  category: 'shell';
 }
 
 export interface BashOutputTool extends Tool {
-  name: "BashOutput";
-  category: "shell";
+  name: 'BashOutput';
+  category: 'shell';
 }
 
 export interface KillShellTool extends Tool {
-  name: "KillShell";
-  category: "shell";
+  name: 'KillShell';
+  category: 'shell';
 }
 
 // Web tools
 
 export interface WebFetchTool extends Tool {
-  name: "WebFetch";
-  category: "web";
+  name: 'WebFetch';
+  category: 'web';
 }
 
 export interface WebSearchTool extends Tool {
-  name: "WebSearch";
-  category: "web";
+  name: 'WebSearch';
+  category: 'web';
 }
 
 // Search tools
 
 export interface GrepTool extends Tool {
-  name: "Grep";
-  category: "search";
+  name: 'Grep';
+  category: 'search';
 }
 
 // Assistant tools
 
 export interface TaskTool extends Tool {
-  name: "Task";
-  category: "assistant";
+  name: 'Task';
+  category: 'assistant';
 }
 
 export interface TodoWriteTool extends Tool {
-  name: "TodoWrite";
-  category: "assistant";
+  name: 'TodoWrite';
+  category: 'assistant';
 }
 
 export interface ExitPlanModeTool extends Tool {
-  name: "ExitPlanMode";
-  category: "assistant";
+  name: 'ExitPlanMode';
+  category: 'assistant';
 }
 
 export interface SlashCommandTool extends Tool {
-  name: "SlashCommand";
-  category: "assistant";
+  name: 'SlashCommand';
+  category: 'assistant';
 }
 
 /**

package/src/types.ts

@@ -1,8 +1,6 @@
+
 // import and export to keep a single type file
-import type {
-  CanUseTool,
-  PermissionResult,
-} from "@anthropic-ai/claude-agent-sdk";
+import type { CanUseTool, PermissionResult } from '@anthropic-ai/claude-agent-sdk';
 export type { CanUseTool, PermissionResult };
 
 // PostHog Task model (matches Array's OpenAPI schema)
@@ -10,12 +8,7 @@ export interface Task {
   id: string;
   title: string;
   description: string;
-  origin_product:
-    | "error_tracking"
-    | "eval_clusters"
-    | "user_created"
-    | "support_queue"
-    | "session_summaries";
+  origin_product: 'error_tracking' | 'eval_clusters' | 'user_created' | 'support_queue' | 'session_summaries';
   position?: number;
   github_integration?: number | null;
   repository_config?: unknown; // JSONField
@@ -39,12 +32,7 @@ export interface LogEntry {
   [key: string]: unknown; // Allow additional fields
 }
 
-export type ArtifactType =
-  | "plan"
-  | "context"
-  | "reference"
-  | "output"
-  | "artifact";
+export type ArtifactType = 'plan' | 'context' | 'reference' | 'output' | 'artifact';
 
 export interface TaskRunArtifact {
   name: string;
@@ -61,7 +49,7 @@ export interface TaskRun {
   task: string; // Task ID
   team: number;
   branch: string | null;
-  status: "started" | "in_progress" | "completed" | "failed";
+  status: 'started' | 'in_progress' | 'completed' | 'failed';
   log_url?: string; // Presigned S3 URL for log access (valid for 1 hour)
   error_message: string | null;
   output: Record<string, unknown> | null; // Structured output (PR URL, commit SHA, etc.)
@@ -90,7 +78,7 @@ export enum PermissionMode {
   PLAN = "plan",
   DEFAULT = "default",
   ACCEPT_EDITS = "acceptEdits",
-  BYPASS = "bypassPermissions",
+  BYPASS = "bypassPermissions"
 }
 
 export interface ExecutionOptions {
@@ -117,57 +105,57 @@ interface BaseEvent {
 
 // Streaming content events
 export interface TokenEvent extends BaseEvent {
-  type: "token";
+  type: 'token';
   content: string;
-  contentType?: "text" | "thinking" | "tool_input";
+  contentType?: 'text' | 'thinking' | 'tool_input';
 }
 
 export interface ContentBlockStartEvent extends BaseEvent {
-  type: "content_block_start";
+  type: 'content_block_start';
   index: number;
-  contentType: "text" | "tool_use" | "thinking";
+  contentType: 'text' | 'tool_use' | 'thinking';
   toolName?: string;
   toolId?: string;
 }
 
 export interface ContentBlockStopEvent extends BaseEvent {
-  type: "content_block_stop";
+  type: 'content_block_stop';
   index: number;
 }
 
 // Tool events
 export interface ToolCallEvent extends BaseEvent {
-  type: "tool_call";
+  type: 'tool_call';
   toolName: string;
   callId: string;
   args: Record<string, any>;
-  parentToolUseId?: string | null; // For nested tool calls (subagents)
+  parentToolUseId?: string | null;  // For nested tool calls (subagents)
   // Tool metadata (enriched by adapter for UI consumption)
-  tool?: import("./tools/types.js").Tool;
-  category?: import("./tools/types.js").ToolCategory;
+  tool?: import('./tools/types.js').Tool;
+  category?: import('./tools/types.js').ToolCategory;
 }
 
 export interface ToolResultEvent extends BaseEvent {
-  type: "tool_result";
+  type: 'tool_result';
   toolName: string;
   callId: string;
   result: any;
-  isError?: boolean; // Whether the tool execution failed
-  parentToolUseId?: string | null; // For nested tool calls (subagents)
+  isError?: boolean;                // Whether the tool execution failed
+  parentToolUseId?: string | null;  // For nested tool calls (subagents)
   // Tool metadata (enriched by adapter for UI consumption)
-  tool?: import("./tools/types.js").Tool;
-  category?: import("./tools/types.js").ToolCategory;
+  tool?: import('./tools/types.js').Tool;
+  category?: import('./tools/types.js').ToolCategory;
 }
 
 // Message lifecycle events
 export interface MessageStartEvent extends BaseEvent {
-  type: "message_start";
+  type: 'message_start';
   messageId?: string;
   model?: string;
 }
 
 export interface MessageDeltaEvent extends BaseEvent {
-  type: "message_delta";
+  type: 'message_delta';
   stopReason?: string;
   stopSequence?: string;
   usage?: {
@@ -176,32 +164,32 @@ export interface MessageDeltaEvent extends BaseEvent {
 }
 
 export interface MessageStopEvent extends BaseEvent {
-  type: "message_stop";
+  type: 'message_stop';
 }
 
 // User message events
 export interface UserMessageEvent extends BaseEvent {
-  type: "user_message";
+  type: 'user_message';
   content: string;
   isSynthetic?: boolean;
 }
 
 // System events
 export interface StatusEvent extends BaseEvent {
-  type: "status";
+  type: 'status';
   phase: string;
   // Common optional fields (varies by phase):
-  kind?: string; // Kind of status (plan, implementation)
-  branch?: string; // Git branch name
-  prUrl?: string; // Pull request URL
-  taskId?: string; // Task identifier
-  messageId?: string; // Claude message ID
-  model?: string; // Model name
-  [key: string]: any; // Allow additional fields
+  kind?: string;            // Kind of status (plan, implementation)
+  branch?: string;          // Git branch name
+  prUrl?: string;           // Pull request URL
+  taskId?: string;          // Task identifier
+  messageId?: string;       // Claude message ID
+  model?: string;           // Model name
+  [key: string]: any;       // Allow additional fields
 }
 
 export interface InitEvent extends BaseEvent {
-  type: "init";
+  type: 'init';
   model: string;
   tools: string[];
   permissionMode: string;
@@ -214,22 +202,21 @@ export interface InitEvent extends BaseEvent {
 }
 
 export interface CompactBoundaryEvent extends BaseEvent {
-  type: "compact_boundary";
-  trigger: "manual" | "auto";
+  type: 'compact_boundary';
+  trigger: 'manual' | 'auto';
   preTokens: number;
 }
 
 // Result events
 export interface DoneEvent extends BaseEvent {
-  type: "done";
-  result?: string; // Final summary text from Claude
+  type: 'done';
+  result?: string;              // Final summary text from Claude
   durationMs?: number;
-  durationApiMs?: number; // API-only duration (excluding local processing)
+  durationApiMs?: number;       // API-only duration (excluding local processing)
   numTurns?: number;
   totalCostUsd?: number;
   usage?: any;
-  modelUsage?: {
-    // Per-model usage breakdown
+  modelUsage?: {                // Per-model usage breakdown
     [modelName: string]: {
       inputTokens: number;
       outputTokens: number;
@@ -240,8 +227,7 @@ export interface DoneEvent extends BaseEvent {
       contextWindow: number;
     };
   };
-  permissionDenials?: Array<{
-    // Tools that were denied by permissions
+  permissionDenials?: Array<{   // Tools that were denied by permissions
     tool_name: string;
     tool_use_id: string;
     tool_input: Record<string, unknown>;
@@ -249,7 +235,7 @@ export interface DoneEvent extends BaseEvent {
 }
 
 export interface ErrorEvent extends BaseEvent {
-  type: "error";
+  type: 'error';
   message: string;
   error?: any;
   errorType?: string;
@@ -259,20 +245,20 @@ export interface ErrorEvent extends BaseEvent {
 
 // Metric and artifact events (general purpose, not tool-specific)
 export interface MetricEvent extends BaseEvent {
-  type: "metric";
+  type: 'metric';
   key: string;
   value: number;
   unit?: string;
 }
 
 export interface ArtifactEvent extends BaseEvent {
-  type: "artifact";
+  type: 'artifact';
   kind: string;
   content: any;
 }
 
 export interface RawSDKEvent extends BaseEvent {
-  type: "raw_sdk_event";
+  type: 'raw_sdk_event';
   sdkMessage: any; // Full SDK message for debugging
 }
 
@@ -310,28 +296,24 @@ export interface TaskExecutionResult {
 }
 
 // MCP Server configuration types (re-exported from Claude SDK for convenience)
-export type McpServerConfig =
-  | {
-      type?: "stdio";
-      command: string;
-      args?: string[];
-      env?: Record<string, string>;
-    }
-  | {
-      type: "sse";
-      url: string;
-      headers?: Record<string, string>;
-    }
-  | {
-      type: "http";
-      url: string;
-      headers?: Record<string, string>;
-    }
-  | {
-      type: "sdk";
-      name: string;
-      instance?: any; // McpServer instance
-    };
+export type McpServerConfig = {
+  type?: 'stdio';
+  command: string;
+  args?: string[];
+  env?: Record<string, string>;
+} | {
+  type: 'sse';
+  url: string;
+  headers?: Record<string, string>;
+} | {
+  type: 'http';
+  url: string;
+  headers?: Record<string, string>;
+} | {
+  type: 'sdk';
+  name: string;
+  instance?: any; // McpServer instance
+};
 
 export interface AgentConfig {
   workingDirectory?: string;
@@ -345,6 +327,7 @@ export interface AgentConfig {
   // PostHog MCP configuration
   posthogMcpUrl?: string;
 
+
   // MCP Server configuration
   // Additional MCP servers (PostHog MCP is always included by default)
   // You can override the PostHog MCP config by providing mcpServers.posthog
@@ -365,12 +348,7 @@ export interface PostHogAPIConfig {
 }
 
 // URL mention types
-export type ResourceType =
-  | "error"
-  | "experiment"
-  | "insight"
-  | "feature_flag"
-  | "generic";
+export type ResourceType = 'error' | 'experiment' | 'insight' | 'feature_flag' | 'generic';
 
 export interface PostHogResource {
   type: ResourceType;
@@ -402,11 +380,11 @@ export interface ResearchAnswer {
 }
 
 export interface ResearchEvaluation {
-  actionabilityScore: number; // 0-1 confidence score
-  context: string; // brief summary for planning
-  keyFiles: string[]; // files needing modification
-  blockers?: string[]; // what's preventing full confidence
+  actionabilityScore: number;    // 0-1 confidence score
+  context: string;               // brief summary for planning
+  keyFiles: string[];            // files needing modification
+  blockers?: string[];           // what's preventing full confidence
   questions?: ResearchQuestion[]; // only if score < 0.7
-  answered?: boolean; // whether questions have been answered
-  answers?: ResearchAnswer[]; // user's answers to questions
-}
+  answered?: boolean;            // whether questions have been answered
+  answers?: ResearchAnswer[];    // user's answers to questions
+}

package/src/utils/logger.ts

@@ -2,84 +2,78 @@
  * Simple logger utility with configurable debug mode
  */
 export enum LogLevel {
-  ERROR = 0,
-  WARN = 1,
-  INFO = 2,
-  DEBUG = 3,
+    ERROR = 0,
+    WARN = 1,
+    INFO = 2,
+    DEBUG = 3
 }
 
 export interface LoggerConfig {
-  debug?: boolean;
-  prefix?: string;
+    debug?: boolean;
+    prefix?: string;
 }
 
 export class Logger {
-  private debugEnabled: boolean;
-  private prefix: string;
+    private debugEnabled: boolean;
+    private prefix: string;
 
-  constructor(config: LoggerConfig = {}) {
-    this.debugEnabled = config.debug ?? false;
-    this.prefix = config.prefix ?? "[PostHog Agent]";
-  }
-
-  setDebug(enabled: boolean) {
-    this.debugEnabled = enabled;
-  }
-
-  private formatMessage(
-    level: string,
-    message: string,
-    data?: unknown,
-  ): string {
-    const timestamp = new Date().toISOString();
-    const base = `${timestamp} ${this.prefix} ${level} ${message}`;
+    constructor(config: LoggerConfig = {}) {
+        this.debugEnabled = config.debug ?? false;
+        this.prefix = config.prefix ?? '[PostHog Agent]';
+    }
 
-    if (data !== undefined) {
-      return `${base} ${JSON.stringify(data, null, 2)}`;
+    setDebug(enabled: boolean) {
+        this.debugEnabled = enabled;
     }
 
-    return base;
-  }
+    private formatMessage(level: string, message: string, data?: any): string {
+        const timestamp = new Date().toISOString();
+        const base = `${timestamp} ${this.prefix} ${level} ${message}`;
+        
+        if (data !== undefined) {
+            return `${base} ${JSON.stringify(data, null, 2)}`;
+        }
+        
+        return base;
+    }
 
-  error(message: string, error?: Error | unknown) {
-    // Always log errors
-    if (error instanceof Error) {
-      console.error(
-        this.formatMessage("[ERROR]", message, {
-          message: error.message,
-          stack: error.stack,
-        }),
-      );
-    } else {
-      console.error(this.formatMessage("[ERROR]", message, error));
+    error(message: string, error?: Error | any) {
+        // Always log errors
+        if (error instanceof Error) {
+            console.error(this.formatMessage('[ERROR]', message, {
+                message: error.message,
+                stack: error.stack
+            }));
+        } else {
+            console.error(this.formatMessage('[ERROR]', message, error));
+        }
     }
-  }
 
-  warn(message: string, data?: unknown) {
-    if (this.debugEnabled) {
-      console.warn(this.formatMessage("[WARN]", message, data));
+    warn(message: string, data?: any) {
+        if (this.debugEnabled) {
+            console.warn(this.formatMessage('[WARN]', message, data));
+        }
     }
-  }
 
-  info(message: string, data?: unknown) {
-    if (this.debugEnabled) {
-      console.log(this.formatMessage("[INFO]", message, data));
+    info(message: string, data?: any) {
+        if (this.debugEnabled) {
+            console.log(this.formatMessage('[INFO]', message, data));
+        }
     }
-  }
 
-  debug(message: string, data?: unknown) {
-    if (this.debugEnabled) {
-      console.log(this.formatMessage("[DEBUG]", message, data));
+    debug(message: string, data?: any) {
+        if (this.debugEnabled) {
+            console.log(this.formatMessage('[DEBUG]', message, data));
+        }
     }
-  }
 
-  /**
-   * Create a child logger with additional prefix
-   */
-  child(childPrefix: string): Logger {
-    return new Logger({
-      debug: this.debugEnabled,
-      prefix: `${this.prefix} [${childPrefix}]`,
-    });
-  }
-}
+    /**
+     * Create a child logger with additional prefix
+     */
+    child(childPrefix: string): Logger {
+        return new Logger({
+            debug: this.debugEnabled,
+            prefix: `${this.prefix} [${childPrefix}]`
+        });
+    }
+}