@posthog/agent 1.1.0 → 1.2.0

package/dist/src/task-progress-reporter.js

@@ -0,0 +1,234 @@
+/**
+ * Persists task execution progress to PostHog so clients can poll for updates.
+ *
+ * The reporter is intentionally best-effort – failures are logged but never
+ * allowed to break the agent execution flow.
+ */
+class TaskProgressReporter {
+    posthogAPI;
+    logger;
+    progressRecord;
+    taskId;
+    outputLog = [];
+    totalSteps;
+    lastLogEntry;
+    constructor(posthogAPI, logger) {
+        this.posthogAPI = posthogAPI;
+        this.logger = logger.child('TaskProgressReporter');
+    }
+    get progressId() {
+        return this.progressRecord?.id;
+    }
+    async start(taskId, metadata = {}) {
+        if (!this.posthogAPI) {
+            return;
+        }
+        this.taskId = taskId;
+        this.totalSteps = metadata.totalSteps;
+        try {
+            const record = await this.posthogAPI.createTaskProgress(taskId, {
+                status: 'started',
+                current_step: 'initializing',
+                total_steps: metadata.totalSteps ?? 0,
+                completed_steps: 0,
+                workflow_id: metadata.workflowId,
+                workflow_run_id: metadata.workflowRunId,
+                activity_id: metadata.activityId,
+                output_log: '',
+            });
+            this.progressRecord = record;
+            this.outputLog = record.output_log ? record.output_log.split('\n') : [];
+            this.logger.debug('Created task progress record', { taskId, progressId: record.id });
+        }
+        catch (error) {
+            this.logger.warn('Failed to create task progress record', { taskId, error: error.message });
+        }
+    }
+    async stageStarted(stageKey, stageIndex) {
+        await this.update({
+            status: 'in_progress',
+            current_step: stageKey,
+            completed_steps: Math.min(stageIndex, this.totalSteps ?? stageIndex),
+        }, `Stage started: ${stageKey}`);
+    }
+    async stageCompleted(stageKey, completedStages) {
+        await this.update({
+            status: 'in_progress',
+            current_step: stageKey,
+            completed_steps: Math.min(completedStages, this.totalSteps ?? completedStages),
+        }, `Stage completed: ${stageKey}`);
+    }
+    async branchCreated(stageKey, branchName) {
+        await this.appendLog(`Branch created (${stageKey}): ${branchName}`);
+    }
+    async commitMade(stageKey, kind) {
+        await this.appendLog(`Commit made (${stageKey}, ${kind})`);
+    }
+    async pullRequestCreated(stageKey, prUrl) {
+        await this.appendLog(`Pull request created (${stageKey}): ${prUrl}`);
+    }
+    async noNextStage(stageKey) {
+        await this.appendLog(stageKey
+            ? `No next stage available after '${stageKey}'. Execution halted.`
+            : 'No next stage available. Execution halted.');
+    }
+    async complete() {
+        await this.update({ status: 'completed', completed_steps: this.totalSteps }, 'Workflow execution completed');
+    }
+    async fail(error) {
+        const message = typeof error === 'string' ? error : error.message;
+        await this.update({ status: 'failed', error_message: message }, `Workflow execution failed: ${message}`);
+    }
+    async appendLog(line) {
+        await this.update({}, line);
+    }
+    async recordEvent(event) {
+        if (!this.posthogAPI || !this.progressId || !this.taskId) {
+            return;
+        }
+        switch (event.type) {
+            case 'token':
+            case 'message_delta':
+            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':
+                // Skip verbose streaming artifacts from persistence
+                return;
+            case 'file_write':
+                await this.appendLog(this.formatFileWriteEvent(event));
+                return;
+            case 'diff':
+                await this.appendLog(this.formatDiffEvent(event));
+                return;
+            case 'status':
+                // Status events are covered by dedicated progress updates
+                return;
+            case 'error':
+                await this.appendLog(`[error] ${event.message}`);
+                return;
+            case 'done': {
+                const cost = event.totalCostUsd !== undefined ? ` cost=$${event.totalCostUsd.toFixed(2)}` : '';
+                await this.appendLog(`[done] duration=${event.durationMs ?? 'unknown'}ms turns=${event.numTurns ?? 'unknown'}${cost}`);
+                return;
+            }
+            case 'init':
+                // Omit verbose init messages from persisted log
+                return;
+            case 'user_message': {
+                const summary = this.summarizeUserMessage(event.content);
+                if (summary) {
+                    await this.appendLog(summary);
+                }
+                return;
+            }
+            default:
+                // For any unfamiliar event types, avoid spamming the log.
+                return;
+        }
+    }
+    async update(update, logLine) {
+        if (!this.posthogAPI || !this.progressId || !this.taskId) {
+            return;
+        }
+        if (logLine) {
+            if (logLine !== this.lastLogEntry) {
+                this.outputLog.push(logLine);
+                this.lastLogEntry = logLine;
+            }
+            update.output_log = this.outputLog.join('\n');
+        }
+        try {
+            const record = await this.posthogAPI.updateTaskProgress(this.taskId, this.progressId, update);
+            // Sync local cache with server response to avoid drift if server modifies values
+            this.progressRecord = record;
+            if (record.output_log !== undefined && record.output_log !== null) {
+                this.outputLog = record.output_log ? record.output_log.split('\n') : [];
+            }
+        }
+        catch (error) {
+            this.logger.warn('Failed to update task progress record', {
+                taskId: this.taskId,
+                progressId: this.progressId,
+                error: error.message,
+            });
+        }
+    }
+    summarizeUserMessage(content) {
+        if (!content) {
+            return null;
+        }
+        const trimmed = content.trim();
+        if (!trimmed) {
+            return null;
+        }
+        const fileUpdateMatch = trimmed.match(/The file\s+([^\s]+)\s+has been updated/i);
+        if (fileUpdateMatch) {
+            return `[user] file updated: ${fileUpdateMatch[1]}`;
+        }
+        if (/Todos have been modified/i.test(trimmed)) {
+            return '[todo] list updated';
+        }
+        const diffMatch = trimmed.match(/diff --git a\/([^\s]+) b\/([^\s]+)/);
+        if (diffMatch) {
+            return `[diff] ${diffMatch[2] ?? diffMatch[1]}`;
+        }
+        const gitStatusMatch = trimmed.match(/^On branch ([^\n]+)/);
+        if (gitStatusMatch) {
+            return `[git] status ${gitStatusMatch[1]}`;
+        }
+        if (/This Bash command contains multiple operations/i.test(trimmed)) {
+            return '[approval] multi-step command pending';
+        }
+        if (/This command requires approval/i.test(trimmed)) {
+            return '[approval] command awaiting approval';
+        }
+        if (/^Exit plan mode\?/i.test(trimmed)) {
+            return null;
+        }
+        if (trimmed.includes('node_modules')) {
+            return null;
+        }
+        if (trimmed.includes('total ') && trimmed.includes('drwx')) {
+            return null;
+        }
+        if (trimmed.includes('→')) {
+            return null;
+        }
+        if (trimmed.split('\n').length > 2) {
+            return null;
+        }
+        const normalized = trimmed.replace(/\s+/g, ' ');
+        const maxLen = 120;
+        if (!normalized) {
+            return null;
+        }
+        const preview = normalized.length > maxLen ? `${normalized.slice(0, maxLen)}…` : normalized;
+        return `[user] ${preview}`;
+    }
+    formatFileWriteEvent(event) {
+        const size = event.bytes !== undefined ? ` (${event.bytes} bytes)` : '';
+        return `[file] wrote ${event.path}${size}`;
+    }
+    formatDiffEvent(event) {
+        const summary = event.summary
+            ? event.summary.trim()
+            : this.truncateMultiline(event.patch ?? '', 160);
+        return `[diff] ${event.file}${summary ? ` | ${summary}` : ''}`;
+    }
+    truncateMultiline(text, max = 160) {
+        if (!text) {
+            return '';
+        }
+        const compact = text.replace(/\s+/g, ' ').trim();
+        return compact.length > max ? `${compact.slice(0, max)}…` : compact;
+    }
+}
+
+export { TaskProgressReporter };
+//# sourceMappingURL=task-progress-reporter.js.map

package/dist/src/task-progress-reporter.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"task-progress-reporter.js","sources":["../../src/task-progress-reporter.ts"],"sourcesContent":["import type { Logger } from './utils/logger.js';\nimport type { PostHogAPIClient, TaskProgressRecord, TaskProgressUpdate } from './posthog-api.js';\nimport type { AgentEvent } from './types.js';\n\ninterface ProgressMetadata {\n  workflowId?: string;\n  workflowRunId?: string;\n  activityId?: string;\n  totalSteps?: number;\n}\n\n/**\n * Persists task execution progress to PostHog so clients can poll for updates.\n *\n * The reporter is intentionally best-effort – failures are logged but never\n * allowed to break the agent execution flow.\n */\nexport class TaskProgressReporter {\n  private posthogAPI?: PostHogAPIClient;\n  private logger: Logger;\n  private progressRecord?: TaskProgressRecord;\n  private taskId?: string;\n  private outputLog: string[] = [];\n  private totalSteps?: number;\n  private lastLogEntry?: string;\n\n  constructor(posthogAPI: PostHogAPIClient | undefined, logger: Logger) {\n    this.posthogAPI = posthogAPI;\n    this.logger = logger.child('TaskProgressReporter');\n  }\n\n  get progressId(): string | undefined {\n    return this.progressRecord?.id;\n  }\n\n  async start(taskId: string, metadata: ProgressMetadata = {}): Promise<void> {\n    if (!this.posthogAPI) {\n      return;\n    }\n\n    this.taskId = taskId;\n    this.totalSteps = metadata.totalSteps;\n\n    try {\n      const record = await this.posthogAPI.createTaskProgress(taskId, {\n        status: 'started',\n        current_step: 'initializing',\n        total_steps: metadata.totalSteps ?? 0,\n        completed_steps: 0,\n        workflow_id: metadata.workflowId,\n        workflow_run_id: metadata.workflowRunId,\n        activity_id: metadata.activityId,\n        output_log: '',\n      });\n      this.progressRecord = record;\n      this.outputLog = record.output_log ? record.output_log.split('\\n') : [];\n      this.logger.debug('Created task progress record', { taskId, progressId: record.id });\n    } catch (error) {\n      this.logger.warn('Failed to create task progress record', { taskId, error: (error as Error).message });\n    }\n  }\n\n  async stageStarted(stageKey: string, stageIndex: number): Promise<void> {\n    await this.update({\n      status: 'in_progress',\n      current_step: stageKey,\n      completed_steps: Math.min(stageIndex, this.totalSteps ?? stageIndex),\n    }, `Stage started: ${stageKey}`);\n  }\n\n  async stageCompleted(stageKey: string, completedStages: number): Promise<void> {\n    await this.update({\n      status: 'in_progress',\n      current_step: stageKey,\n      completed_steps: Math.min(completedStages, this.totalSteps ?? completedStages),\n    }, `Stage completed: ${stageKey}`);\n  }\n\n  async branchCreated(stageKey: string, branchName: string): Promise<void> {\n    await this.appendLog(`Branch created (${stageKey}): ${branchName}`);\n  }\n\n  async commitMade(stageKey: string, kind: 'plan' | 'implementation'): Promise<void> {\n    await this.appendLog(`Commit made (${stageKey}, ${kind})`);\n  }\n\n  async pullRequestCreated(stageKey: string, prUrl: string): Promise<void> {\n    await this.appendLog(`Pull request created (${stageKey}): ${prUrl}`);\n  }\n\n  async noNextStage(stageKey?: string): Promise<void> {\n    await this.appendLog(\n      stageKey\n        ? `No next stage available after '${stageKey}'. Execution halted.`\n        : 'No next stage available. Execution halted.'\n    );\n  }\n\n  async complete(): Promise<void> {\n    await this.update({ status: 'completed', completed_steps: this.totalSteps }, 'Workflow execution completed');\n  }\n\n  async fail(error: Error | string): Promise<void> {\n    const message = typeof error === 'string' ? error : error.message;\n    await this.update({ status: 'failed', error_message: message }, `Workflow execution failed: ${message}`);\n  }\n\n  async appendLog(line: string): Promise<void> {\n    await this.update({}, line);\n  }\n\n  async recordEvent(event: AgentEvent): Promise<void> {\n    if (!this.posthogAPI || !this.progressId || !this.taskId) {\n      return;\n    }\n\n    switch (event.type) {\n      case 'token':\n      case 'message_delta':\n      case 'content_block_start':\n      case 'content_block_stop':\n      case 'compact_boundary':\n      case 'tool_call':\n      case 'tool_result':\n      case 'message_start':\n      case 'message_stop':\n      case 'metric':\n      case 'artifact':\n        // Skip verbose streaming artifacts from persistence\n        return;\n\n      case 'file_write':\n        await this.appendLog(this.formatFileWriteEvent(event));\n        return;\n\n      case 'diff':\n        await this.appendLog(this.formatDiffEvent(event));\n        return;\n\n      case 'status':\n        // Status events are covered by dedicated progress updates\n        return;\n\n      case 'error':\n        await this.appendLog(`[error] ${event.message}`);\n        return;\n\n      case 'done': {\n        const cost = event.totalCostUsd !== undefined ? ` cost=$${event.totalCostUsd.toFixed(2)}` : '';\n        await this.appendLog(\n          `[done] duration=${event.durationMs ?? 'unknown'}ms turns=${event.numTurns ?? 'unknown'}${cost}`\n        );\n        return;\n      }\n\n      case 'init':\n        // Omit verbose init messages from persisted log\n        return;\n\n      case 'user_message': {\n        const summary = this.summarizeUserMessage(event.content);\n        if (summary) {\n          await this.appendLog(summary);\n        }\n        return;\n      }\n\n      default:\n        // For any unfamiliar event types, avoid spamming the log.\n        return;\n    }\n  }\n\n  private async update(update: TaskProgressUpdate, logLine?: string): Promise<void> {\n    if (!this.posthogAPI || !this.progressId || !this.taskId) {\n      return;\n    }\n\n    if (logLine) {\n      if (logLine !== this.lastLogEntry) {\n        this.outputLog.push(logLine);\n        this.lastLogEntry = logLine;\n      }\n      update.output_log = this.outputLog.join('\\n');\n    }\n\n    try {\n      const record = await this.posthogAPI.updateTaskProgress(this.taskId, this.progressId, update);\n      // Sync local cache with server response to avoid drift if server modifies values\n      this.progressRecord = record;\n      if (record.output_log !== undefined && record.output_log !== null) {\n        this.outputLog = record.output_log ? record.output_log.split('\\n') : [];\n      }\n    } catch (error) {\n      this.logger.warn('Failed to update task progress record', {\n        taskId: this.taskId,\n        progressId: this.progressId,\n        error: (error as Error).message,\n      });\n    }\n  }\n\n  private summarizeUserMessage(content?: string): string | null {\n    if (!content) {\n      return null;\n    }\n    const trimmed = content.trim();\n    if (!trimmed) {\n      return null;\n    }\n\n    const fileUpdateMatch = trimmed.match(/The file\\s+([^\\s]+)\\s+has been updated/i);\n    if (fileUpdateMatch) {\n      return `[user] file updated: ${fileUpdateMatch[1]}`;\n    }\n\n    if (/Todos have been modified/i.test(trimmed)) {\n      return '[todo] list updated';\n    }\n\n    const diffMatch = trimmed.match(/diff --git a\\/([^\\s]+) b\\/([^\\s]+)/);\n    if (diffMatch) {\n      return `[diff] ${diffMatch[2] ?? diffMatch[1]}`;\n    }\n\n    const gitStatusMatch = trimmed.match(/^On branch ([^\\n]+)/);\n    if (gitStatusMatch) {\n      return `[git] status ${gitStatusMatch[1]}`;\n    }\n\n    if (/This Bash command contains multiple operations/i.test(trimmed)) {\n      return '[approval] multi-step command pending';\n    }\n\n    if (/This command requires approval/i.test(trimmed)) {\n      return '[approval] command awaiting approval';\n    }\n\n    if (/^Exit plan mode\\?/i.test(trimmed)) {\n      return null;\n    }\n\n    if (trimmed.includes('node_modules')) {\n      return null;\n    }\n\n    if (trimmed.includes('total ') && trimmed.includes('drwx')) {\n      return null;\n    }\n\n    if (trimmed.includes('→')) {\n      return null;\n    }\n\n    if (trimmed.split('\\n').length > 2) {\n      return null;\n    }\n\n    const normalized = trimmed.replace(/\\s+/g, ' ');\n    const maxLen = 120;\n    if (!normalized) {\n      return null;\n    }\n    const preview = normalized.length > maxLen ? `${normalized.slice(0, maxLen)}…` : normalized;\n    return `[user] ${preview}`;\n  }\n\n  private formatFileWriteEvent(event: Extract<AgentEvent, { type: 'file_write' }>): string {\n    const size = event.bytes !== undefined ? ` (${event.bytes} bytes)` : '';\n    return `[file] wrote ${event.path}${size}`;\n  }\n\n  private formatDiffEvent(event: Extract<AgentEvent, { type: 'diff' }>): string {\n    const summary = event.summary\n      ? event.summary.trim()\n      : this.truncateMultiline(event.patch ?? '', 160);\n    return `[diff] ${event.file}${summary ? ` | ${summary}` : ''}`;\n  }\n\n  private truncateMultiline(text: string, max = 160): string {\n    if (!text) {\n      return '';\n    }\n    const compact = text.replace(/\\s+/g, ' ').trim();\n    return compact.length > max ? `${compact.slice(0, max)}…` : compact;\n  }\n}\n"],"names":[],"mappings":"AAWA;;;;;AAKG;MACU,oBAAoB,CAAA;AACvB,IAAA,UAAU;AACV,IAAA,MAAM;AACN,IAAA,cAAc;AACd,IAAA,MAAM;IACN,SAAS,GAAa,EAAE;AACxB,IAAA,UAAU;AACV,IAAA,YAAY;IAEpB,WAAA,CAAY,UAAwC,EAAE,MAAc,EAAA;AAClE,QAAA,IAAI,CAAC,UAAU,GAAG,UAAU;QAC5B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,sBAAsB,CAAC;IACpD;AAEA,IAAA,IAAI,UAAU,GAAA;AACZ,QAAA,OAAO,IAAI,CAAC,cAAc,EAAE,EAAE;IAChC;AAEA,IAAA,MAAM,KAAK,CAAC,MAAc,EAAE,WAA6B,EAAE,EAAA;AACzD,QAAA,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE;YACpB;QACF;AAEA,QAAA,IAAI,CAAC,MAAM,GAAG,MAAM;AACpB,QAAA,IAAI,CAAC,UAAU,GAAG,QAAQ,CAAC,UAAU;AAErC,QAAA,IAAI;YACF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,kBAAkB,CAAC,MAAM,EAAE;AAC9D,gBAAA,MAAM,EAAE,SAAS;AACjB,gBAAA,YAAY,EAAE,cAAc;AAC5B,gBAAA,WAAW,EAAE,QAAQ,CAAC,UAAU,IAAI,CAAC;AACrC,gBAAA,eAAe,EAAE,CAAC;gBAClB,WAAW,EAAE,QAAQ,CAAC,UAAU;gBAChC,eAAe,EAAE,QAAQ,CAAC,aAAa;gBACvC,WAAW,EAAE,QAAQ,CAAC,UAAU;AAChC,gBAAA,UAAU,EAAE,EAAE;AACf,aAAA,CAAC;AACF,YAAA,IAAI,CAAC,cAAc,GAAG,MAAM;YAC5B,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,UAAU,GAAG,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE;AACvE,YAAA,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,8BAA8B,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,CAAC,EAAE,EAAE,CAAC;QACtF;QAAE,OAAO,KAAK,EAAE;AACd,YAAA,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,uCAAuC,EAAE,EAAE,MAAM,EAAE,KAAK,EAAG,KAAe,CAAC,OAAO,EAAE,CAAC;QACxG;IACF;AAEA,IAAA,MAAM,YAAY,CAAC,QAAgB,EAAE,UAAkB,EAAA;QACrD,MAAM,IAAI,CAAC,MAAM,CAAC;AAChB,YAAA,MAAM,EAAE,aAAa;AACrB,YAAA,YAAY,EAAE,QAAQ;AACtB,YAAA,eAAe,EAAE,IAAI,CAAC,GAAG,CAAC,UAAU,EAAE,IAAI,CAAC,UAAU,IAAI,UAAU,CAAC;AACrE,SAAA,EAAE,CAAA,eAAA,EAAkB,QAAQ,CAAA,CAAE,CAAC;IAClC;AAEA,IAAA,MAAM,cAAc,CAAC,QAAgB,EAAE,eAAuB,EAAA;QAC5D,MAAM,IAAI,CAAC,MAAM,CAAC;AAChB,YAAA,MAAM,EAAE,aAAa;AACrB,YAAA,YAAY,EAAE,QAAQ;AACtB,YAAA,eAAe,EAAE,IAAI,CAAC,GAAG,CAAC,eAAe,EAAE,IAAI,CAAC,UAAU,IAAI,eAAe,CAAC;AAC/E,SAAA,EAAE,CAAA,iBAAA,EAAoB,QAAQ,CAAA,CAAE,CAAC;IACpC;AAEA,IAAA,MAAM,aAAa,CAAC,QAAgB,EAAE,UAAkB,EAAA;QACtD,MAAM,IAAI,CAAC,SAAS,CAAC,CAAA,gBAAA,EAAmB,QAAQ,CAAA,GAAA,EAAM,UAAU,CAAA,CAAE,CAAC;IACrE;AAEA,IAAA,MAAM,UAAU,CAAC,QAAgB,EAAE,IAA+B,EAAA;QAChE,MAAM,IAAI,CAAC,SAAS,CAAC,CAAA,aAAA,EAAgB,QAAQ,CAAA,EAAA,EAAK,IAAI,CAAA,CAAA,CAAG,CAAC;IAC5D;AAEA,IAAA,MAAM,kBAAkB,CAAC,QAAgB,EAAE,KAAa,EAAA;QACtD,MAAM,IAAI,CAAC,SAAS,CAAC,CAAA,sBAAA,EAAyB,QAAQ,CAAA,GAAA,EAAM,KAAK,CAAA,CAAE,CAAC;IACtE;IAEA,MAAM,WAAW,CAAC,QAAiB,EAAA;AACjC,QAAA,MAAM,IAAI,CAAC,SAAS,CAClB;cACI,CAAA,+BAAA,EAAkC,QAAQ,CAAA,oBAAA;cAC1C,4CAA4C,CACjD;IACH;AAEA,IAAA,MAAM,QAAQ,GAAA;AACZ,QAAA,MAAM,IAAI,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,WAAW,EAAE,eAAe,EAAE,IAAI,CAAC,UAAU,EAAE,EAAE,8BAA8B,CAAC;IAC9G;IAEA,MAAM,IAAI,CAAC,KAAqB,EAAA;AAC9B,QAAA,MAAM,OAAO,GAAG,OAAO,KAAK,KAAK,QAAQ,GAAG,KAAK,GAAG,KAAK,CAAC,OAAO;AACjE,QAAA,MAAM,IAAI,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,aAAa,EAAE,OAAO,EAAE,EAAE,8BAA8B,OAAO,CAAA,CAAE,CAAC;IAC1G;IAEA,MAAM,SAAS,CAAC,IAAY,EAAA;QAC1B,MAAM,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,IAAI,CAAC;IAC7B;IAEA,MAAM,WAAW,CAAC,KAAiB,EAAA;AACjC,QAAA,IAAI,CAAC,IAAI,CAAC,UAAU,IAAI,CAAC,IAAI,CAAC,UAAU,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE;YACxD;QACF;AAEA,QAAA,QAAQ,KAAK,CAAC,IAAI;AAChB,YAAA,KAAK,OAAO;AACZ,YAAA,KAAK,eAAe;AACpB,YAAA,KAAK,qBAAqB;AAC1B,YAAA,KAAK,oBAAoB;AACzB,YAAA,KAAK,kBAAkB;AACvB,YAAA,KAAK,WAAW;AAChB,YAAA,KAAK,aAAa;AAClB,YAAA,KAAK,eAAe;AACpB,YAAA,KAAK,cAAc;AACnB,YAAA,KAAK,QAAQ;AACb,YAAA,KAAK,UAAU;;gBAEb;AAEF,YAAA,KAAK,YAAY;gBACf,MAAM,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;gBACtD;AAEF,YAAA,KAAK,MAAM;gBACT,MAAM,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;gBACjD;AAEF,YAAA,KAAK,QAAQ;;gBAEX;AAEF,YAAA,KAAK,OAAO;gBACV,MAAM,IAAI,CAAC,SAAS,CAAC,CAAA,QAAA,EAAW,KAAK,CAAC,OAAO,CAAA,CAAE,CAAC;gBAChD;YAEF,KAAK,MAAM,EAAE;gBACX,MAAM,IAAI,GAAG,KAAK,CAAC,YAAY,KAAK,SAAS,GAAG,CAAA,OAAA,EAAU,KAAK,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE;gBAC9F,MAAM,IAAI,CAAC,SAAS,CAClB,mBAAmB,KAAK,CAAC,UAAU,IAAI,SAAS,YAAY,KAAK,CAAC,QAAQ,IAAI,SAAS,GAAG,IAAI,CAAA,CAAE,CACjG;gBACD;YACF;AAEA,YAAA,KAAK,MAAM;;gBAET;YAEF,KAAK,cAAc,EAAE;gBACnB,MAAM,OAAO,GAAG,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC,OAAO,CAAC;gBACxD,IAAI,OAAO,EAAE;AACX,oBAAA,MAAM,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;gBAC/B;gBACA;YACF;AAEA,YAAA;;gBAEE;;IAEN;AAEQ,IAAA,MAAM,MAAM,CAAC,MAA0B,EAAE,OAAgB,EAAA;AAC/D,QAAA,IAAI,CAAC,IAAI,CAAC,UAAU,IAAI,CAAC,IAAI,CAAC,UAAU,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE;YACxD;QACF;QAEA,IAAI,OAAO,EAAE;AACX,YAAA,IAAI,OAAO,KAAK,IAAI,CAAC,YAAY,EAAE;AACjC,gBAAA,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC;AAC5B,gBAAA,IAAI,CAAC,YAAY,GAAG,OAAO;YAC7B;YACA,MAAM,CAAC,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC;QAC/C;AAEA,QAAA,IAAI;AACF,YAAA,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,kBAAkB,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,UAAU,EAAE,MAAM,CAAC;;AAE7F,YAAA,IAAI,CAAC,cAAc,GAAG,MAAM;AAC5B,YAAA,IAAI,MAAM,CAAC,UAAU,KAAK,SAAS,IAAI,MAAM,CAAC,UAAU,KAAK,IAAI,EAAE;gBACjE,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,UAAU,GAAG,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE;YACzE;QACF;QAAE,OAAO,KAAK,EAAE;AACd,YAAA,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,uCAAuC,EAAE;gBACxD,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,UAAU,EAAE,IAAI,CAAC,UAAU;gBAC3B,KAAK,EAAG,KAAe,CAAC,OAAO;AAChC,aAAA,CAAC;QACJ;IACF;AAEQ,IAAA,oBAAoB,CAAC,OAAgB,EAAA;QAC3C,IAAI,CAAC,OAAO,EAAE;AACZ,YAAA,OAAO,IAAI;QACb;AACA,QAAA,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,EAAE;QAC9B,IAAI,CAAC,OAAO,EAAE;AACZ,YAAA,OAAO,IAAI;QACb;QAEA,MAAM,eAAe,GAAG,OAAO,CAAC,KAAK,CAAC,yCAAyC,CAAC;QAChF,IAAI,eAAe,EAAE;AACnB,YAAA,OAAO,wBAAwB,eAAe,CAAC,CAAC,CAAC,EAAE;QACrD;AAEA,QAAA,IAAI,2BAA2B,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE;AAC7C,YAAA,OAAO,qBAAqB;QAC9B;QAEA,MAAM,SAAS,GAAG,OAAO,CAAC,KAAK,CAAC,oCAAoC,CAAC;QACrE,IAAI,SAAS,EAAE;YACb,OAAO,CAAA,OAAA,EAAU,SAAS,CAAC,CAAC,CAAC,IAAI,SAAS,CAAC,CAAC,CAAC,CAAA,CAAE;QACjD;QAEA,MAAM,cAAc,GAAG,OAAO,CAAC,KAAK,CAAC,qBAAqB,CAAC;QAC3D,IAAI,cAAc,EAAE;AAClB,YAAA,OAAO,gBAAgB,cAAc,CAAC,CAAC,CAAC,EAAE;QAC5C;AAEA,QAAA,IAAI,iDAAiD,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE;AACnE,YAAA,OAAO,uCAAuC;QAChD;AAEA,QAAA,IAAI,iCAAiC,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE;AACnD,YAAA,OAAO,sCAAsC;QAC/C;AAEA,QAAA,IAAI,oBAAoB,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE;AACtC,YAAA,OAAO,IAAI;QACb;AAEA,QAAA,IAAI,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAC,EAAE;AACpC,YAAA,OAAO,IAAI;QACb;AAEA,QAAA,IAAI,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE;AAC1D,YAAA,OAAO,IAAI;QACb;AAEA,QAAA,IAAI,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE;AACzB,YAAA,OAAO,IAAI;QACb;QAEA,IAAI,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE;AAClC,YAAA,OAAO,IAAI;QACb;QAEA,MAAM,UAAU,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC;QAC/C,MAAM,MAAM,GAAG,GAAG;QAClB,IAAI,CAAC,UAAU,EAAE;AACf,YAAA,OAAO,IAAI;QACb;QACA,MAAM,OAAO,GAAG,UAAU,CAAC,MAAM,GAAG,MAAM,GAAG,CAAA,EAAG,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,MAAM,CAAC,GAAG,GAAG,UAAU;QAC3F,OAAO,CAAA,OAAA,EAAU,OAAO,CAAA,CAAE;IAC5B;AAEQ,IAAA,oBAAoB,CAAC,KAAkD,EAAA;AAC7E,QAAA,MAAM,IAAI,GAAG,KAAK,CAAC,KAAK,KAAK,SAAS,GAAG,CAAA,EAAA,EAAK,KAAK,CAAC,KAAK,CAAA,OAAA,CAAS,GAAG,EAAE;AACvE,QAAA,OAAO,gBAAgB,KAAK,CAAC,IAAI,CAAA,EAAG,IAAI,EAAE;IAC5C;AAEQ,IAAA,eAAe,CAAC,KAA4C,EAAA;AAClE,QAAA,MAAM,OAAO,GAAG,KAAK,CAAC;AACpB,cAAE,KAAK,CAAC,OAAO,CAAC,IAAI;AACpB,cAAE,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,KAAK,IAAI,EAAE,EAAE,GAAG,CAAC;AAClD,QAAA,OAAO,UAAU,KAAK,CAAC,IAAI,CAAA,EAAG,OAAO,GAAG,CAAA,GAAA,EAAM,OAAO,CAAA,CAAE,GAAG,EAAE,EAAE;IAChE;AAEQ,IAAA,iBAAiB,CAAC,IAAY,EAAE,GAAG,GAAG,GAAG,EAAA;QAC/C,IAAI,CAAC,IAAI,EAAE;AACT,YAAA,OAAO,EAAE;QACX;AACA,QAAA,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE;QAChD,OAAO,OAAO,CAAC,MAAM,GAAG,GAAG,GAAG,CAAA,EAAG,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG,GAAG,OAAO;IACrE;AACD;;;;"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@posthog/agent",
-  "version": "1.1.0",
+  "version": "1.2.0",
   "description": "TypeScript agent framework wrapping Claude Agent SDK with Git-based workflow for PostHog tasks",
   "main": "./dist/index.js",
   "module": "./dist/index.js",

package/src/agent.ts

@@ -14,6 +14,7 @@ import { AgentRegistry } from './agent-registry.js';
 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';
 
 export class Agent {
     private workingDirectory: string;
@@ -28,6 +29,7 @@ export class Agent {
     private agentRegistry: AgentRegistry;
     private workflowRegistry: WorkflowRegistry;
     private stageExecutor: StageExecutor;
+    private progressReporter: TaskProgressReporter;
     public debug: boolean;
 
     constructor(config: AgentConfig = {}) {
@@ -64,6 +66,8 @@ export class Agent {
             logger: this.logger.child('PromptBuilder')
         });
         this.stageExecutor = new StageExecutor(this.agentRegistry, this.logger, promptBuilder);
+        this.stageExecutor.setEventHandler((event) => this.emitEvent(event));
+        this.progressReporter = new TaskProgressReporter(this.posthogAPI, this.logger);
     }
 
     /**
@@ -82,6 +86,7 @@ export class Agent {
         if (!workflow) {
             throw new Error(`Workflow ${workflowId} not found`);
         }
+        const orderedStages = [...workflow.stages].sort((a, b) => a.position - b.position);
 
         // Ensure task is assigned to workflow and positioned at first stage
         if (this.posthogAPI) {
@@ -103,9 +108,13 @@ export class Agent {
         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,
+        });
 
         try {
-            const orderedStages = [...workflow.stages].sort((a, b) => a.position - b.position);
             let startIndex = 0;
             const currentStageId = (task as any).current_stage as string | undefined;
 
@@ -114,7 +123,10 @@ export class Agent {
                 const currIdx = orderedStages.findIndex(s => s.id === currentStageId);
                 const atLastStage = currIdx >= 0 && currIdx === orderedStages.length - 1;
                 if (atLastStage) {
-                    this.emitEvent(this.eventTransformer.createStatusEvent('no_next_stage', { stage: orderedStages[currIdx].key }));
+                    const finalStageKey = orderedStages[currIdx]?.key;
+                    this.emitEvent(this.eventTransformer.createStatusEvent('no_next_stage', { stage: finalStageKey }));
+                    await this.progressReporter.noNextStage(finalStageKey);
+                    await this.progressReporter.complete();
                     this.taskManager.completeExecution(executionId, { task, workflow });
                     return { task, workflow };
                 }
@@ -135,7 +147,9 @@ export class Agent {
 
             for (let i = startIndex; i < orderedStages.length; i++) {
                 const stage = orderedStages[i];
+                await this.progressReporter.stageStarted(stage.key, i);
                 await this.executeStage(task, stage, options);
+                await this.progressReporter.stageCompleted(stage.key, i + 1);
                 if (options.autoProgress) {
                     const hasNext = i < orderedStages.length - 1;
                     if (hasNext) {
@@ -143,9 +157,11 @@ export class Agent {
                     }
                 }
             }
+            await this.progressReporter.complete();
             this.taskManager.completeExecution(executionId, { task, workflow });
             return { task, workflow };
         } catch (error) {
+            await this.progressReporter.fail(error as Error);
             this.taskManager.failExecution(executionId, error as Error);
             throw error;
         }
@@ -167,10 +183,12 @@ export class Agent {
             const planningBranch = await this.createPlanningBranch(task.id);
             await this.updateTaskBranch(task.id, planningBranch);
             this.emitEvent(this.eventTransformer.createStatusEvent('branch_created', { stage: stage.key, branch: planningBranch }));
+            await this.progressReporter.branchCreated(stage.key, planningBranch);
         } else if (!isPlanning && !isManual && shouldCreateImplBranch) {
             const implBranch = await this.createImplementationBranch(task.id);
             await this.updateTaskBranch(task.id, implBranch);
             this.emitEvent(this.eventTransformer.createStatusEvent('branch_created', { stage: stage.key, branch: implBranch }));
+            await this.progressReporter.branchCreated(stage.key, implBranch);
         }
 
         const result = await this.stageExecutor.execute(task, stage, options);
@@ -179,6 +197,7 @@ export class Agent {
             await this.writePlan(task.id, result.plan);
             await this.commitPlan(task.id, task.title);
             this.emitEvent(this.eventTransformer.createStatusEvent('commit_made', { stage: stage.key, kind: 'plan' }));
+            await this.progressReporter.commitMade(stage.key, 'plan');
         }
 
         if (isManual) {
@@ -193,12 +212,14 @@ export class Agent {
                     await this.updateTaskBranch(task.id, implBranch);
                     branchName = implBranch;
                     this.emitEvent(this.eventTransformer.createStatusEvent('branch_created', { stage: stage.key, branch: implBranch }));
+                    await this.progressReporter.branchCreated(stage.key, implBranch);
                 }
                 try {
                     const prUrl = await this.createPullRequest(task.id, branchName, task.title, task.description);
                     await this.updateTaskBranch(task.id, branchName);
-                    await this.attachPullRequestToTask(task.id, prUrl);
+                    await this.attachPullRequestToTask(task.id, prUrl, branchName);
                     this.emitEvent(this.eventTransformer.createStatusEvent('pr_created', { stage: stage.key, prUrl }));
+                    await this.progressReporter.pullRequestCreated(stage.key, prUrl);
                 } catch {}
             }
             // Do not auto-progress on manual stages
@@ -211,6 +232,7 @@ export class Agent {
             const planSummary = existingPlan ? existingPlan.split('\n')[0] : undefined;
             await this.commitImplementation(task.id, task.title, planSummary);
             this.emitEvent(this.eventTransformer.createStatusEvent('commit_made', { stage: stage.key, kind: 'implementation' }));
+            await this.progressReporter.commitMade(stage.key, 'implementation');
         }
 
         // PR creation on complete stage (or if explicitly requested), regardless of whether edits occurred
@@ -222,8 +244,9 @@ export class Agent {
                 try {
                     const prUrl = await this.createPullRequest(task.id, branchName, task.title, task.description);
                     await this.updateTaskBranch(task.id, branchName);
-                    await this.attachPullRequestToTask(task.id, prUrl);
+                    await this.attachPullRequestToTask(task.id, prUrl, branchName);
                     this.emitEvent(this.eventTransformer.createStatusEvent('pr_created', { stage: stage.key, prUrl }));
+                    await this.progressReporter.pullRequestCreated(stage.key, prUrl);
                 } catch {}
             }
         }
@@ -271,6 +294,10 @@ export class Agent {
         }
         return this.posthogAPI.fetchTask(taskId);
     }
+
+    getPostHogClient(): PostHogAPIClient | undefined {
+        return this.posthogAPI;
+    }
     
     async listTasks(filters?: {
         repository?: string;
@@ -367,8 +394,8 @@ Generated by PostHog Agent`;
         return prUrl;
     }
 
-    async attachPullRequestToTask(taskId: string, prUrl: string): Promise<void> {
-        this.logger.info('Attaching PR to task', { taskId, prUrl });
+    async attachPullRequestToTask(taskId: string, prUrl: string, branchName?: string): Promise<void> {
+        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.');
@@ -376,7 +403,7 @@ Generated by PostHog Agent`;
             throw error;
         }
 
-        await this.posthogAPI.updateTask(taskId, { github_pr_url: prUrl });
+        await this.posthogAPI.attachTaskPullRequest(taskId, prUrl, branchName);
         this.logger.debug('PR attached to task', { taskId, prUrl });
     }
 
@@ -389,7 +416,7 @@ Generated by PostHog Agent`;
             throw error;
         }
 
-        await this.posthogAPI.updateTask(taskId, { github_branch: branchName });
+        await this.posthogAPI.setTaskBranch(taskId, branchName);
         this.logger.debug('Task branch updated', { taskId, branchName });
     }
 
@@ -419,6 +446,12 @@ Generated by PostHog Agent`;
             // Log all events except tokens (too verbose)
             this.logger.debug('Emitting event', { type: event.type, ts: event.ts });
         }
+        const persistPromise = this.progressReporter.recordEvent(event);
+        if (persistPromise && typeof persistPromise.then === 'function') {
+            persistPromise.catch((error: Error) =>
+                this.logger.debug('Failed to persist agent event', { message: error.message })
+            );
+        }
         this.onEvent?.(event);
     }
 }

package/src/event-transformer.ts

@@ -118,15 +118,14 @@ export class EventTransformer {
 
     // Handle user messages
     if (sdkMessage.type === 'user') {
-      const content = sdkMessage.message.content;
-      const textContent = Array.isArray(content)
-        ? content.find(c => c.type === 'text')?.text
-        : typeof content === 'string' ? content : '';
-
+      const textContent = this.extractUserContent(sdkMessage.message?.content);
+      if (!textContent) {
+        return null;
+      }
       return {
         ...baseEvent,
         type: 'user_message',
-        content: textContent || '',
+        content: textContent,
         isSynthetic: sdkMessage.isSynthetic
       };
     }
@@ -186,4 +185,59 @@ export class EventTransformer {
       ...additionalData
     };
   }
-}
+
+  private extractUserContent(content: unknown): string | null {
+    if (!content) {
+      return null;
+    }
+
+    if (typeof content === 'string') {
+      const trimmed = content.trim();
+      return trimmed.length > 0 ? trimmed : null;
+    }
+
+    if (Array.isArray(content)) {
+      const parts: string[] = [];
+      for (const block of content) {
+        const extracted = this.extractUserContent(block);
+        if (extracted) {
+          parts.push(extracted);
+        } else if (block && typeof block === 'object') {
+          const candidate = this.extractFromObject(block as Record<string, unknown>);
+          if (candidate) {
+            parts.push(candidate);
+          }
+        }
+      }
+      const text = parts.join('\n').trim();
+      return text.length > 0 ? text : null;
+    }
+
+    if (typeof content === 'object') {
+      return this.extractFromObject(content as Record<string, unknown>);
+    }
+
+    return null;
+  }
+
+  private extractFromObject(value: Record<string, unknown>): string | null {
+    const preferredKeys = ['text', 'input_text', 'input', 'markdown', 'content', 'message'];
+    for (const key of preferredKeys) {
+      if (typeof value[key] === 'string') {
+        const trimmed = (value[key] as string).trim();
+        if (trimmed.length > 0) {
+          return trimmed;
+        }
+      }
+    }
+
+    for (const entry of Object.values(value)) {
+      const extracted = this.extractUserContent(entry);
+      if (extracted) {
+        return extracted;
+      }
+    }
+
+    return null;
+  }
+}

package/src/posthog-api.ts

@@ -26,6 +26,36 @@ interface TaskProgressResponse {
   message?: string;
 }
 
+export interface TaskProgressRecord {
+  id: string;
+  task: string;
+  status: "started" | "in_progress" | "completed" | "failed";
+  current_step?: string | null;
+  completed_steps?: number | null;
+  total_steps?: number | null;
+  progress_percentage?: number | null;
+  output_log?: string | null;
+  error_message?: string | null;
+  workflow_id?: string | null;
+  workflow_run_id?: string | null;
+  activity_id?: string | null;
+  created_at: string;
+  updated_at: string;
+  completed_at?: string | null;
+}
+
+export interface TaskProgressUpdate {
+  status?: TaskProgressRecord["status"];
+  current_step?: string | null;
+  completed_steps?: number | null;
+  total_steps?: number | null;
+  output_log?: string | null;
+  error_message?: string | null;
+  workflow_id?: string | null;
+  workflow_run_id?: string | null;
+  activity_id?: string | null;
+}
+
 export class PostHogAPIClient {
   private config: PostHogAPIConfig;
   private _teamId: number | null = null;
@@ -137,11 +167,60 @@ export class PostHogAPIClient {
     });
   }
 
+  async setTaskBranch(taskId: string, branch: string): Promise<Task> {
+    const teamId = await this.getTeamId();
+    return this.apiRequest<Task>(`/api/projects/${teamId}/tasks/${taskId}/set_branch/`, {
+      method: "POST",
+      body: JSON.stringify({ branch }),
+    });
+  }
+
+  async attachTaskPullRequest(taskId: string, prUrl: string, branch?: string): Promise<Task> {
+    const teamId = await this.getTeamId();
+    const payload: Record<string, string> = { pr_url: prUrl };
+    if (branch) {
+      payload.branch = branch;
+    }
+    return this.apiRequest<Task>(`/api/projects/${teamId}/tasks/${taskId}/attach_pr/`, {
+      method: "POST",
+      body: JSON.stringify(payload),
+    });
+  }
+
   async getTaskProgress(taskId: string): Promise<TaskProgressResponse> {
     const teamId = await this.getTeamId();
     return this.apiRequest<TaskProgressResponse>(`/api/projects/${teamId}/tasks/${taskId}/progress/`);
   }
 
+  async createTaskProgress(
+    taskId: string,
+    payload: TaskProgressUpdate & { status: TaskProgressRecord["status"] }
+  ): Promise<TaskProgressRecord> {
+    const teamId = await this.getTeamId();
+    return this.apiRequest<TaskProgressRecord>(`/api/projects/${teamId}/task_progress/`, {
+      method: "POST",
+      body: JSON.stringify({
+        ...payload,
+        task: taskId,
+      }),
+    });
+  }
+
+  async updateTaskProgress(
+    taskId: string,
+    progressId: string,
+    payload: TaskProgressUpdate
+  ): Promise<TaskProgressRecord> {
+    const teamId = await this.getTeamId();
+    return this.apiRequest<TaskProgressRecord>(`/api/projects/${teamId}/task_progress/${progressId}/`, {
+      method: "PATCH",
+      body: JSON.stringify({
+        ...payload,
+        task: taskId,
+      }),
+    });
+  }
+
   // Workflow endpoints
   async fetchWorkflow(workflowId: string): Promise<WorkflowDefinition> {
     const teamId = await this.getTeamId();