@posthog/agent 1.9.0 → 1.11.0

package/dist/src/prompt-builder.js.map

@@ -1 +1 @@
-{"version":3,"file":"prompt-builder.js","sources":["../../src/prompt-builder.ts"],"sourcesContent":["import type { Task, UrlMention, PostHogResource } from './types.js';\nimport type { TemplateVariables } from './template-manager.js';\nimport { Logger } from './utils/logger.js';\nimport { promises as fs } from 'fs';\nimport { join } from 'path';\n\nexport interface PromptBuilderDeps {\n  getTaskFiles: (taskId: string) => Promise<any[]>;\n  generatePlanTemplate: (vars: TemplateVariables) => Promise<string>;\n  posthogClient?: { fetchResourceByUrl: (mention: UrlMention) => Promise<PostHogResource> };\n  logger?: Logger;\n}\n\nexport class PromptBuilder {\n  private getTaskFiles: PromptBuilderDeps['getTaskFiles'];\n  private generatePlanTemplate: PromptBuilderDeps['generatePlanTemplate'];\n  private posthogClient?: PromptBuilderDeps['posthogClient'];\n  private logger: Logger;\n\n  constructor(deps: PromptBuilderDeps) {\n    this.getTaskFiles = deps.getTaskFiles;\n    this.generatePlanTemplate = deps.generatePlanTemplate;\n    this.posthogClient = deps.posthogClient;\n    this.logger = deps.logger || new Logger({ debug: false, prefix: '[PromptBuilder]' });\n  }\n\n  /**\n   * Extract file paths from XML tags in description\n   * Format: <file path=\"relative/path.ts\" />\n   */\n  private extractFilePaths(description: string): string[] {\n    const fileTagRegex = /<file\\s+path=\"([^\"]+)\"\\s*\\/>/g;\n    const paths: string[] = [];\n    let match: RegExpExecArray | null;\n\n    while ((match = fileTagRegex.exec(description)) !== null) {\n      paths.push(match[1]);\n    }\n\n    return paths;\n  }\n\n  /**\n   * Read file contents from repository\n   */\n  private async readFileContent(repositoryPath: string, filePath: string): Promise<string | null> {\n    try {\n      const fullPath = join(repositoryPath, filePath);\n      const content = await fs.readFile(fullPath, 'utf8');\n      return content;\n    } catch (error) {\n      this.logger.warn(`Failed to read referenced file: ${filePath}`, { error });\n      return null;\n    }\n  }\n\n  /**\n   * Extract URL mentions from XML tags in description\n   * Formats: <error id=\"...\" />, <experiment id=\"...\" />, <url href=\"...\" />\n   */\n  private extractUrlMentions(description: string): UrlMention[] {\n    const mentions: UrlMention[] = [];\n    \n    // PostHog resource mentions: <error id=\"...\" />, <experiment id=\"...\" />, etc.\n    const resourceRegex = /<(error|experiment|insight|feature_flag)\\s+id=\"([^\"]+)\"\\s*\\/>/g;\n    let match: RegExpExecArray | null;\n\n    while ((match = resourceRegex.exec(description)) !== null) {\n      const [, type, id] = match;\n      mentions.push({\n        url: '', // Will be reconstructed if needed\n        type: type as any,\n        id,\n        label: this.generateUrlLabel('', type as any),\n      });\n    }\n\n    // Generic URL mentions: <url href=\"...\" />\n    const urlRegex = /<url\\s+href=\"([^\"]+)\"\\s*\\/>/g;\n    while ((match = urlRegex.exec(description)) !== null) {\n      const [, url] = match;\n      mentions.push({\n        url,\n        type: 'generic',\n        label: this.generateUrlLabel(url, 'generic'),\n      });\n    }\n\n    return mentions;\n  }\n\n  /**\n   * Generate a display label for a URL mention\n   */\n  private generateUrlLabel(url: string, type: string): string {\n    try {\n      const urlObj = new URL(url);\n      switch (type) {\n        case 'error':\n          const errorMatch = url.match(/error_tracking\\/([a-f0-9-]+)/);\n          return errorMatch ? `Error ${errorMatch[1].slice(0, 8)}...` : 'Error';\n        case 'experiment':\n          const expMatch = url.match(/experiments\\/(\\d+)/);\n          return expMatch ? `Experiment #${expMatch[1]}` : 'Experiment';\n        case 'insight':\n          return 'Insight';\n        case 'feature_flag':\n          return 'Feature Flag';\n        default:\n          return urlObj.hostname;\n      }\n    } catch {\n      return 'URL';\n    }\n  }\n\n  /**\n   * Process URL references and fetch their content\n   */\n  private async processUrlReferences(\n    description: string\n  ): Promise<{ description: string; referencedResources: PostHogResource[] }> {\n    const urlMentions = this.extractUrlMentions(description);\n    const referencedResources: PostHogResource[] = [];\n\n    if (urlMentions.length === 0 || !this.posthogClient) {\n      return { description, referencedResources };\n    }\n\n    // Fetch all referenced resources\n    for (const mention of urlMentions) {\n      try {\n        const resource = await this.posthogClient.fetchResourceByUrl(mention);\n        referencedResources.push(resource);\n      } catch (error) {\n        this.logger.warn(`Failed to fetch resource from URL: ${mention.url}`, { error });\n        // Add a placeholder resource for failed fetches\n        referencedResources.push({\n          type: mention.type,\n          id: mention.id || '',\n          url: mention.url,\n          title: mention.label || 'Unknown Resource',\n          content: `Failed to fetch resource from ${mention.url}: ${error}`,\n          metadata: {},\n        });\n      }\n    }\n\n    // Replace URL tags with just the label for readability\n    let processedDescription = description;\n    for (const mention of urlMentions) {\n      if (mention.type === 'generic') {\n        // Generic URLs: <url href=\"...\" />\n        const escapedUrl = mention.url.replace(/[.*+?^${}()|[\\]\\\\]/g, '\\\\$&');\n        processedDescription = processedDescription.replace(\n          new RegExp(`<url\\\\s+href=\"${escapedUrl}\"\\\\s*/>`, 'g'),\n          `@${mention.label}`\n        );\n      } else {\n        // PostHog resources: <error id=\"...\" />, <experiment id=\"...\" />, etc.\n        const escapedType = mention.type.replace(/[.*+?^${}()|[\\]\\\\]/g, '\\\\$&');\n        const escapedId = mention.id ? mention.id.replace(/[.*+?^${}()|[\\]\\\\]/g, '\\\\$&') : '';\n        processedDescription = processedDescription.replace(\n          new RegExp(`<${escapedType}\\\\s+id=\"${escapedId}\"\\\\s*/>`, 'g'),\n          `@${mention.label}`\n        );\n      }\n    }\n\n    return { description: processedDescription, referencedResources };\n  }\n\n  /**\n   * Process description to extract file tags and read contents\n   * Returns processed description and referenced file contents\n   */\n  private async processFileReferences(\n    description: string,\n    repositoryPath?: string\n  ): Promise<{ description: string; referencedFiles: Array<{ path: string; content: string }> }> {\n    const filePaths = this.extractFilePaths(description);\n    const referencedFiles: Array<{ path: string; content: string }> = [];\n\n    if (filePaths.length === 0 || !repositoryPath) {\n      return { description, referencedFiles };\n    }\n\n    // Read all referenced files\n    for (const filePath of filePaths) {\n      const content = await this.readFileContent(repositoryPath, filePath);\n      if (content !== null) {\n        referencedFiles.push({ path: filePath, content });\n      }\n    }\n\n    // Replace file tags with just the filename for readability\n    let processedDescription = description;\n    for (const filePath of filePaths) {\n      const fileName = filePath.split('/').pop() || filePath;\n      processedDescription = processedDescription.replace(\n        new RegExp(`<file\\\\s+path=\"${filePath.replace(/[.*+?^${}()|[\\]\\\\]/g, '\\\\$&')}\"\\\\s*/>`, 'g'),\n        `@${fileName}`\n      );\n    }\n\n    return { description: processedDescription, referencedFiles };\n  }\n\n  async buildPlanningPrompt(task: Task, repositoryPath?: string): Promise<string> {\n    // Process file references in description\n    const { description: descriptionAfterFiles, referencedFiles } = await this.processFileReferences(\n      task.description,\n      repositoryPath\n    );\n\n    // Process URL references in description  \n    const { description: processedDescription, referencedResources } = await this.processUrlReferences(\n      descriptionAfterFiles\n    );\n\n    let prompt = '';\n    prompt += `## Current Task\\n\\n**Task**: ${task.title}\\n**Description**: ${processedDescription}`;\n\n    if ((task as any).primary_repository) {\n      prompt += `\\n**Repository**: ${(task as any).primary_repository}`;\n    }\n\n    // Add referenced files from @ mentions\n    if (referencedFiles.length > 0) {\n      prompt += `\\n\\n## Referenced Files\\n\\n`;\n      for (const file of referencedFiles) {\n        prompt += `### ${file.path}\\n\\`\\`\\`\\n${file.content}\\n\\`\\`\\`\\n\\n`;\n      }\n    }\n\n    // Add referenced resources from URL mentions\n    if (referencedResources.length > 0) {\n      prompt += `\\n\\n## Referenced Resources\\n\\n`;\n      for (const resource of referencedResources) {\n        prompt += `### ${resource.title} (${resource.type})\\n**URL**: ${resource.url}\\n\\n${resource.content}\\n\\n`;\n      }\n    }\n\n    try {\n      const taskFiles = await this.getTaskFiles(task.id);\n      const contextFiles = taskFiles.filter((f: any) => f.type === 'context' || f.type === 'reference');\n      if (contextFiles.length > 0) {\n        prompt += `\\n\\n## Supporting Files`;\n        for (const file of contextFiles) {\n          prompt += `\\n\\n### ${file.name} (${file.type})\\n${file.content}`;\n        }\n      }\n    } catch (error) {\n      this.logger.debug('No existing task files found for planning', { taskId: task.id });\n    }\n\n    const templateVariables = {\n      task_id: task.id,\n      task_title: task.title,\n      task_description: processedDescription,\n      date: new Date().toISOString().split('T')[0],\n      repository: ((task as any).primary_repository || '') as string,\n    };\n\n    const planTemplate = await this.generatePlanTemplate(templateVariables);\n\n    prompt += `\\n\\nPlease analyze the codebase and create a detailed implementation plan for this task. Use the following template structure for your plan:\\n\\n${planTemplate}\\n\\nFill in each section with specific, actionable information based on your analysis. Replace all placeholder content with actual details about this task.`;\n\n    return prompt;\n  }\n\n  async buildExecutionPrompt(task: Task, repositoryPath?: string): Promise<string> {\n    // Process file references in description\n    const { description: descriptionAfterFiles, referencedFiles } = await this.processFileReferences(\n      task.description,\n      repositoryPath\n    );\n\n    // Process URL references in description  \n    const { description: processedDescription, referencedResources } = await this.processUrlReferences(\n      descriptionAfterFiles\n    );\n\n    let prompt = '';\n    prompt += `## Current Task\\n\\n**Task**: ${task.title}\\n**Description**: ${processedDescription}`;\n\n    if ((task as any).primary_repository) {\n      prompt += `\\n**Repository**: ${(task as any).primary_repository}`;\n    }\n\n    // Add referenced files from @ mentions\n    if (referencedFiles.length > 0) {\n      prompt += `\\n\\n## Referenced Files\\n\\n`;\n      for (const file of referencedFiles) {\n        prompt += `### ${file.path}\\n\\`\\`\\`\\n${file.content}\\n\\`\\`\\`\\n\\n`;\n      }\n    }\n\n    // Add referenced resources from URL mentions\n    if (referencedResources.length > 0) {\n      prompt += `\\n\\n## Referenced Resources\\n\\n`;\n      for (const resource of referencedResources) {\n        prompt += `### ${resource.title} (${resource.type})\\n**URL**: ${resource.url}\\n\\n${resource.content}\\n\\n`;\n      }\n    }\n\n    try {\n      const taskFiles = await this.getTaskFiles(task.id);\n      const hasPlan = taskFiles.some((f: any) => f.type === 'plan');\n      if (taskFiles.length > 0) {\n        prompt += `\\n\\n## Context and Supporting Information`;\n        for (const file of taskFiles) {\n          if (file.type === 'plan') {\n            prompt += `\\n\\n### Execution Plan\\n${file.content}`;\n          } else {\n            prompt += `\\n\\n### ${file.name} (${file.type})\\n${file.content}`;\n          }\n        }\n      }\n      if (hasPlan) {\n        prompt += `\\n\\nPlease implement the changes described in the execution plan above. Follow the plan step-by-step and make the necessary file modifications. You must actually edit files and make changes - do not just analyze or review.`;\n      } else {\n        prompt += `\\n\\nPlease implement the changes described in the task above. You must actually edit files and make changes - do not just analyze or review.`;\n      }\n    } catch (error) {\n      this.logger.debug('No supporting files found for execution', { taskId: task.id });\n      prompt += `\\n\\nPlease implement the changes described in the task above.`;\n    }\n    return prompt;\n  }\n}\n\n\n"],"names":["fs"],"mappings":";;;;MAaa,aAAa,CAAA;AAChB,IAAA,YAAY;AACZ,IAAA,oBAAoB;AACpB,IAAA,aAAa;AACb,IAAA,MAAM;AAEd,IAAA,WAAA,CAAY,IAAuB,EAAA;AACjC,QAAA,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,YAAY;AACrC,QAAA,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC,oBAAoB;AACrD,QAAA,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,aAAa;QACvC,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,IAAI,IAAI,MAAM,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,iBAAiB,EAAE,CAAC;IACtF;AAEA;;;AAGG;AACK,IAAA,gBAAgB,CAAC,WAAmB,EAAA;QAC1C,MAAM,YAAY,GAAG,+BAA+B;QACpD,MAAM,KAAK,GAAa,EAAE;AAC1B,QAAA,IAAI,KAA6B;AAEjC,QAAA,OAAO,CAAC,KAAK,GAAG,YAAY,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,IAAI,EAAE;YACxD,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACtB;AAEA,QAAA,OAAO,KAAK;IACd;AAEA;;AAEG;AACK,IAAA,MAAM,eAAe,CAAC,cAAsB,EAAE,QAAgB,EAAA;AACpE,QAAA,IAAI;YACF,MAAM,QAAQ,GAAG,IAAI,CAAC,cAAc,EAAE,QAAQ,CAAC;YAC/C,MAAM,OAAO,GAAG,MAAMA,QAAE,CAAC,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC;AACnD,YAAA,OAAO,OAAO;QAChB;QAAE,OAAO,KAAK,EAAE;AACd,YAAA,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAA,gCAAA,EAAmC,QAAQ,CAAA,CAAE,EAAE,EAAE,KAAK,EAAE,CAAC;AAC1E,YAAA,OAAO,IAAI;QACb;IACF;AAEA;;;AAGG;AACK,IAAA,kBAAkB,CAAC,WAAmB,EAAA;QAC5C,MAAM,QAAQ,GAAiB,EAAE;;QAGjC,MAAM,aAAa,GAAG,gEAAgE;AACtF,QAAA,IAAI,KAA6B;AAEjC,QAAA,OAAO,CAAC,KAAK,GAAG,aAAa,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,IAAI,EAAE;YACzD,MAAM,GAAG,IAAI,EAAE,EAAE,CAAC,GAAG,KAAK;YAC1B,QAAQ,CAAC,IAAI,CAAC;gBACZ,GAAG,EAAE,EAAE;AACP,gBAAA,IAAI,EAAE,IAAW;gBACjB,EAAE;gBACF,KAAK,EAAE,IAAI,CAAC,gBAAgB,CAAC,EAAE,EAAE,IAAW,CAAC;AAC9C,aAAA,CAAC;QACJ;;QAGA,MAAM,QAAQ,GAAG,8BAA8B;AAC/C,QAAA,OAAO,CAAC,KAAK,GAAG,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,IAAI,EAAE;AACpD,YAAA,MAAM,GAAG,GAAG,CAAC,GAAG,KAAK;YACrB,QAAQ,CAAC,IAAI,CAAC;gBACZ,GAAG;AACH,gBAAA,IAAI,EAAE,SAAS;gBACf,KAAK,EAAE,IAAI,CAAC,gBAAgB,CAAC,GAAG,EAAE,SAAS,CAAC;AAC7C,aAAA,CAAC;QACJ;AAEA,QAAA,OAAO,QAAQ;IACjB;AAEA;;AAEG;IACK,gBAAgB,CAAC,GAAW,EAAE,IAAY,EAAA;AAChD,QAAA,IAAI;AACF,YAAA,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,GAAG,CAAC;YAC3B,QAAQ,IAAI;AACV,gBAAA,KAAK,OAAO;oBACV,MAAM,UAAU,GAAG,GAAG,CAAC,KAAK,CAAC,8BAA8B,CAAC;oBAC5D,OAAO,UAAU,GAAG,SAAS,UAAU,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,KAAK,GAAG,OAAO;AACvE,gBAAA,KAAK,YAAY;oBACf,MAAM,QAAQ,GAAG,GAAG,CAAC,KAAK,CAAC,oBAAoB,CAAC;AAChD,oBAAA,OAAO,QAAQ,GAAG,eAAe,QAAQ,CAAC,CAAC,CAAC,CAAA,CAAE,GAAG,YAAY;AAC/D,gBAAA,KAAK,SAAS;AACZ,oBAAA,OAAO,SAAS;AAClB,gBAAA,KAAK,cAAc;AACjB,oBAAA,OAAO,cAAc;AACvB,gBAAA;oBACE,OAAO,MAAM,CAAC,QAAQ;;QAE5B;AAAE,QAAA,MAAM;AACN,YAAA,OAAO,KAAK;QACd;IACF;AAEA;;AAEG;IACK,MAAM,oBAAoB,CAChC,WAAmB,EAAA;QAEnB,MAAM,WAAW,GAAG,IAAI,CAAC,kBAAkB,CAAC,WAAW,CAAC;QACxD,MAAM,mBAAmB,GAAsB,EAAE;QAEjD,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE;AACnD,YAAA,OAAO,EAAE,WAAW,EAAE,mBAAmB,EAAE;QAC7C;;AAGA,QAAA,KAAK,MAAM,OAAO,IAAI,WAAW,EAAE;AACjC,YAAA,IAAI;gBACF,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,kBAAkB,CAAC,OAAO,CAAC;AACrE,gBAAA,mBAAmB,CAAC,IAAI,CAAC,QAAQ,CAAC;YACpC;YAAE,OAAO,KAAK,EAAE;AACd,gBAAA,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAA,mCAAA,EAAsC,OAAO,CAAC,GAAG,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC;;gBAEhF,mBAAmB,CAAC,IAAI,CAAC;oBACvB,IAAI,EAAE,OAAO,CAAC,IAAI;AAClB,oBAAA,EAAE,EAAE,OAAO,CAAC,EAAE,IAAI,EAAE;oBACpB,GAAG,EAAE,OAAO,CAAC,GAAG;AAChB,oBAAA,KAAK,EAAE,OAAO,CAAC,KAAK,IAAI,kBAAkB;AAC1C,oBAAA,OAAO,EAAE,CAAA,8BAAA,EAAiC,OAAO,CAAC,GAAG,CAAA,EAAA,EAAK,KAAK,CAAA,CAAE;AACjE,oBAAA,QAAQ,EAAE,EAAE;AACb,iBAAA,CAAC;YACJ;QACF;;QAGA,IAAI,oBAAoB,GAAG,WAAW;AACtC,QAAA,KAAK,MAAM,OAAO,IAAI,WAAW,EAAE;AACjC,YAAA,IAAI,OAAO,CAAC,IAAI,KAAK,SAAS,EAAE;;AAE9B,gBAAA,MAAM,UAAU,GAAG,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,qBAAqB,EAAE,MAAM,CAAC;gBACrE,oBAAoB,GAAG,oBAAoB,CAAC,OAAO,CACjD,IAAI,MAAM,CAAC,CAAA,cAAA,EAAiB,UAAU,SAAS,EAAE,GAAG,CAAC,EACrD,CAAA,CAAA,EAAI,OAAO,CAAC,KAAK,CAAA,CAAE,CACpB;YACH;iBAAO;;AAEL,gBAAA,MAAM,WAAW,GAAG,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,qBAAqB,EAAE,MAAM,CAAC;gBACvE,MAAM,SAAS,GAAG,OAAO,CAAC,EAAE,GAAG,OAAO,CAAC,EAAE,CAAC,OAAO,CAAC,qBAAqB,EAAE,MAAM,CAAC,GAAG,EAAE;gBACrF,oBAAoB,GAAG,oBAAoB,CAAC,OAAO,CACjD,IAAI,MAAM,CAAC,CAAA,CAAA,EAAI,WAAW,CAAA,QAAA,EAAW,SAAS,CAAA,OAAA,CAAS,EAAE,GAAG,CAAC,EAC7D,CAAA,CAAA,EAAI,OAAO,CAAC,KAAK,CAAA,CAAE,CACpB;YACH;QACF;AAEA,QAAA,OAAO,EAAE,WAAW,EAAE,oBAAoB,EAAE,mBAAmB,EAAE;IACnE;AAEA;;;AAGG;AACK,IAAA,MAAM,qBAAqB,CACjC,WAAmB,EACnB,cAAuB,EAAA;QAEvB,MAAM,SAAS,GAAG,IAAI,CAAC,gBAAgB,CAAC,WAAW,CAAC;QACpD,MAAM,eAAe,GAA6C,EAAE;QAEpE,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,IAAI,CAAC,cAAc,EAAE;AAC7C,YAAA,OAAO,EAAE,WAAW,EAAE,eAAe,EAAE;QACzC;;AAGA,QAAA,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE;YAChC,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,cAAc,EAAE,QAAQ,CAAC;AACpE,YAAA,IAAI,OAAO,KAAK,IAAI,EAAE;gBACpB,eAAe,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC;YACnD;QACF;;QAGA,IAAI,oBAAoB,GAAG,WAAW;AACtC,QAAA,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE;AAChC,YAAA,MAAM,QAAQ,GAAG,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,IAAI,QAAQ;YACtD,oBAAoB,GAAG,oBAAoB,CAAC,OAAO,CACjD,IAAI,MAAM,CAAC,CAAA,eAAA,EAAkB,QAAQ,CAAC,OAAO,CAAC,qBAAqB,EAAE,MAAM,CAAC,CAAA,OAAA,CAAS,EAAE,GAAG,CAAC,EAC3F,CAAA,CAAA,EAAI,QAAQ,CAAA,CAAE,CACf;QACH;AAEA,QAAA,OAAO,EAAE,WAAW,EAAE,oBAAoB,EAAE,eAAe,EAAE;IAC/D;AAEA,IAAA,MAAM,mBAAmB,CAAC,IAAU,EAAE,cAAuB,EAAA;;AAE3D,QAAA,MAAM,EAAE,WAAW,EAAE,qBAAqB,EAAE,eAAe,EAAE,GAAG,MAAM,IAAI,CAAC,qBAAqB,CAC9F,IAAI,CAAC,WAAW,EAChB,cAAc,CACf;;AAGD,QAAA,MAAM,EAAE,WAAW,EAAE,oBAAoB,EAAE,mBAAmB,EAAE,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAChG,qBAAqB,CACtB;QAED,IAAI,MAAM,GAAG,EAAE;QACf,MAAM,IAAI,gCAAgC,IAAI,CAAC,KAAK,CAAA,mBAAA,EAAsB,oBAAoB,EAAE;AAEhG,QAAA,IAAK,IAAY,CAAC,kBAAkB,EAAE;AACpC,YAAA,MAAM,IAAI,CAAA,kBAAA,EAAsB,IAAY,CAAC,kBAAkB,EAAE;QACnE;;AAGA,QAAA,IAAI,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE;YAC9B,MAAM,IAAI,6BAA6B;AACvC,YAAA,KAAK,MAAM,IAAI,IAAI,eAAe,EAAE;gBAClC,MAAM,IAAI,CAAA,IAAA,EAAO,IAAI,CAAC,IAAI,aAAa,IAAI,CAAC,OAAO,CAAA,YAAA,CAAc;YACnE;QACF;;AAGA,QAAA,IAAI,mBAAmB,CAAC,MAAM,GAAG,CAAC,EAAE;YAClC,MAAM,IAAI,iCAAiC;AAC3C,YAAA,KAAK,MAAM,QAAQ,IAAI,mBAAmB,EAAE;AAC1C,gBAAA,MAAM,IAAI,CAAA,IAAA,EAAO,QAAQ,CAAC,KAAK,CAAA,EAAA,EAAK,QAAQ,CAAC,IAAI,CAAA,YAAA,EAAe,QAAQ,CAAC,GAAG,CAAA,IAAA,EAAO,QAAQ,CAAC,OAAO,MAAM;YAC3G;QACF;AAEA,QAAA,IAAI;YACF,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,EAAE,CAAC;YAClD,MAAM,YAAY,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC,CAAM,KAAK,CAAC,CAAC,IAAI,KAAK,SAAS,IAAI,CAAC,CAAC,IAAI,KAAK,WAAW,CAAC;AACjG,YAAA,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE;gBAC3B,MAAM,IAAI,yBAAyB;AACnC,gBAAA,KAAK,MAAM,IAAI,IAAI,YAAY,EAAE;AAC/B,oBAAA,MAAM,IAAI,CAAA,QAAA,EAAW,IAAI,CAAC,IAAI,CAAA,EAAA,EAAK,IAAI,CAAC,IAAI,CAAA,GAAA,EAAM,IAAI,CAAC,OAAO,EAAE;gBAClE;YACF;QACF;QAAE,OAAO,KAAK,EAAE;AACd,YAAA,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,2CAA2C,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC;QACrF;AAEA,QAAA,MAAM,iBAAiB,GAAG;YACxB,OAAO,EAAE,IAAI,CAAC,EAAE;YAChB,UAAU,EAAE,IAAI,CAAC,KAAK;AACtB,YAAA,gBAAgB,EAAE,oBAAoB;AACtC,YAAA,IAAI,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;AAC5C,YAAA,UAAU,GAAI,IAAY,CAAC,kBAAkB,IAAI,EAAE,CAAW;SAC/D;QAED,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,iBAAiB,CAAC;AAEvE,QAAA,MAAM,IAAI,CAAA,gJAAA,EAAmJ,YAAY,CAAA,2JAAA,CAA6J;AAEtU,QAAA,OAAO,MAAM;IACf;AAEA,IAAA,MAAM,oBAAoB,CAAC,IAAU,EAAE,cAAuB,EAAA;;AAE5D,QAAA,MAAM,EAAE,WAAW,EAAE,qBAAqB,EAAE,eAAe,EAAE,GAAG,MAAM,IAAI,CAAC,qBAAqB,CAC9F,IAAI,CAAC,WAAW,EAChB,cAAc,CACf;;AAGD,QAAA,MAAM,EAAE,WAAW,EAAE,oBAAoB,EAAE,mBAAmB,EAAE,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAChG,qBAAqB,CACtB;QAED,IAAI,MAAM,GAAG,EAAE;QACf,MAAM,IAAI,gCAAgC,IAAI,CAAC,KAAK,CAAA,mBAAA,EAAsB,oBAAoB,EAAE;AAEhG,QAAA,IAAK,IAAY,CAAC,kBAAkB,EAAE;AACpC,YAAA,MAAM,IAAI,CAAA,kBAAA,EAAsB,IAAY,CAAC,kBAAkB,EAAE;QACnE;;AAGA,QAAA,IAAI,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE;YAC9B,MAAM,IAAI,6BAA6B;AACvC,YAAA,KAAK,MAAM,IAAI,IAAI,eAAe,EAAE;gBAClC,MAAM,IAAI,CAAA,IAAA,EAAO,IAAI,CAAC,IAAI,aAAa,IAAI,CAAC,OAAO,CAAA,YAAA,CAAc;YACnE;QACF;;AAGA,QAAA,IAAI,mBAAmB,CAAC,MAAM,GAAG,CAAC,EAAE;YAClC,MAAM,IAAI,iCAAiC;AAC3C,YAAA,KAAK,MAAM,QAAQ,IAAI,mBAAmB,EAAE;AAC1C,gBAAA,MAAM,IAAI,CAAA,IAAA,EAAO,QAAQ,CAAC,KAAK,CAAA,EAAA,EAAK,QAAQ,CAAC,IAAI,CAAA,YAAA,EAAe,QAAQ,CAAC,GAAG,CAAA,IAAA,EAAO,QAAQ,CAAC,OAAO,MAAM;YAC3G;QACF;AAEA,QAAA,IAAI;YACF,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,EAAE,CAAC;AAClD,YAAA,MAAM,OAAO,GAAG,SAAS,CAAC,IAAI,CAAC,CAAC,CAAM,KAAK,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC;AAC7D,YAAA,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE;gBACxB,MAAM,IAAI,2CAA2C;AACrD,gBAAA,KAAK,MAAM,IAAI,IAAI,SAAS,EAAE;AAC5B,oBAAA,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE;AACxB,wBAAA,MAAM,IAAI,CAAA,wBAAA,EAA2B,IAAI,CAAC,OAAO,EAAE;oBACrD;yBAAO;AACL,wBAAA,MAAM,IAAI,CAAA,QAAA,EAAW,IAAI,CAAC,IAAI,CAAA,EAAA,EAAK,IAAI,CAAC,IAAI,CAAA,GAAA,EAAM,IAAI,CAAC,OAAO,EAAE;oBAClE;gBACF;YACF;YACA,IAAI,OAAO,EAAE;gBACX,MAAM,IAAI,gOAAgO;YAC5O;iBAAO;gBACL,MAAM,IAAI,8IAA8I;YAC1J;QACF;QAAE,OAAO,KAAK,EAAE;AACd,YAAA,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,yCAAyC,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC;YACjF,MAAM,IAAI,+DAA+D;QAC3E;AACA,QAAA,OAAO,MAAM;IACf;AACD;;;;"}
+{"version":3,"file":"prompt-builder.js","sources":["../../src/prompt-builder.ts"],"sourcesContent":["import type { Task, UrlMention, PostHogResource } from './types.js';\nimport type { TemplateVariables } from './template-manager.js';\nimport { Logger } from './utils/logger.js';\nimport { promises as fs } from 'fs';\nimport { join } from 'path';\n\nexport interface PromptBuilderDeps {\n  getTaskFiles: (taskId: string) => Promise<any[]>;\n  generatePlanTemplate: (vars: TemplateVariables) => Promise<string>;\n  posthogClient?: { fetchResourceByUrl: (mention: UrlMention) => Promise<PostHogResource> };\n  logger?: Logger;\n}\n\nexport class PromptBuilder {\n  private getTaskFiles: PromptBuilderDeps['getTaskFiles'];\n  private generatePlanTemplate: PromptBuilderDeps['generatePlanTemplate'];\n  private posthogClient?: PromptBuilderDeps['posthogClient'];\n  private logger: Logger;\n\n  constructor(deps: PromptBuilderDeps) {\n    this.getTaskFiles = deps.getTaskFiles;\n    this.generatePlanTemplate = deps.generatePlanTemplate;\n    this.posthogClient = deps.posthogClient;\n    this.logger = deps.logger || new Logger({ debug: false, prefix: '[PromptBuilder]' });\n  }\n\n  /**\n   * Extract file paths from XML tags in description\n   * Format: <file path=\"relative/path.ts\" />\n   */\n  private extractFilePaths(description: string): string[] {\n    const fileTagRegex = /<file\\s+path=\"([^\"]+)\"\\s*\\/>/g;\n    const paths: string[] = [];\n    let match: RegExpExecArray | null;\n\n    while ((match = fileTagRegex.exec(description)) !== null) {\n      paths.push(match[1]);\n    }\n\n    return paths;\n  }\n\n  /**\n   * Read file contents from repository\n   */\n  private async readFileContent(repositoryPath: string, filePath: string): Promise<string | null> {\n    try {\n      const fullPath = join(repositoryPath, filePath);\n      const content = await fs.readFile(fullPath, 'utf8');\n      return content;\n    } catch (error) {\n      this.logger.warn(`Failed to read referenced file: ${filePath}`, { error });\n      return null;\n    }\n  }\n\n  /**\n   * Extract URL mentions from XML tags in description\n   * Formats: <error id=\"...\" />, <experiment id=\"...\" />, <url href=\"...\" />\n   */\n  private extractUrlMentions(description: string): UrlMention[] {\n    const mentions: UrlMention[] = [];\n    \n    // PostHog resource mentions: <error id=\"...\" />, <experiment id=\"...\" />, etc.\n    const resourceRegex = /<(error|experiment|insight|feature_flag)\\s+id=\"([^\"]+)\"\\s*\\/>/g;\n    let match: RegExpExecArray | null;\n\n    while ((match = resourceRegex.exec(description)) !== null) {\n      const [, type, id] = match;\n      mentions.push({\n        url: '', // Will be reconstructed if needed\n        type: type as any,\n        id,\n        label: this.generateUrlLabel('', type as any),\n      });\n    }\n\n    // Generic URL mentions: <url href=\"...\" />\n    const urlRegex = /<url\\s+href=\"([^\"]+)\"\\s*\\/>/g;\n    while ((match = urlRegex.exec(description)) !== null) {\n      const [, url] = match;\n      mentions.push({\n        url,\n        type: 'generic',\n        label: this.generateUrlLabel(url, 'generic'),\n      });\n    }\n\n    return mentions;\n  }\n\n  /**\n   * Generate a display label for a URL mention\n   */\n  private generateUrlLabel(url: string, type: string): string {\n    try {\n      const urlObj = new URL(url);\n      switch (type) {\n        case 'error':\n          const errorMatch = url.match(/error_tracking\\/([a-f0-9-]+)/);\n          return errorMatch ? `Error ${errorMatch[1].slice(0, 8)}...` : 'Error';\n        case 'experiment':\n          const expMatch = url.match(/experiments\\/(\\d+)/);\n          return expMatch ? `Experiment #${expMatch[1]}` : 'Experiment';\n        case 'insight':\n          return 'Insight';\n        case 'feature_flag':\n          return 'Feature Flag';\n        default:\n          return urlObj.hostname;\n      }\n    } catch {\n      return 'URL';\n    }\n  }\n\n  /**\n   * Process URL references and fetch their content\n   */\n  private async processUrlReferences(\n    description: string\n  ): Promise<{ description: string; referencedResources: PostHogResource[] }> {\n    const urlMentions = this.extractUrlMentions(description);\n    const referencedResources: PostHogResource[] = [];\n\n    if (urlMentions.length === 0 || !this.posthogClient) {\n      return { description, referencedResources };\n    }\n\n    // Fetch all referenced resources\n    for (const mention of urlMentions) {\n      try {\n        const resource = await this.posthogClient.fetchResourceByUrl(mention);\n        referencedResources.push(resource);\n      } catch (error) {\n        this.logger.warn(`Failed to fetch resource from URL: ${mention.url}`, { error });\n        // Add a placeholder resource for failed fetches\n        referencedResources.push({\n          type: mention.type,\n          id: mention.id || '',\n          url: mention.url,\n          title: mention.label || 'Unknown Resource',\n          content: `Failed to fetch resource from ${mention.url}: ${error}`,\n          metadata: {},\n        });\n      }\n    }\n\n    // Replace URL tags with just the label for readability\n    let processedDescription = description;\n    for (const mention of urlMentions) {\n      if (mention.type === 'generic') {\n        // Generic URLs: <url href=\"...\" />\n        const escapedUrl = mention.url.replace(/[.*+?^${}()|[\\]\\\\]/g, '\\\\$&');\n        processedDescription = processedDescription.replace(\n          new RegExp(`<url\\\\s+href=\"${escapedUrl}\"\\\\s*/>`, 'g'),\n          `@${mention.label}`\n        );\n      } else {\n        // PostHog resources: <error id=\"...\" />, <experiment id=\"...\" />, etc.\n        const escapedType = mention.type.replace(/[.*+?^${}()|[\\]\\\\]/g, '\\\\$&');\n        const escapedId = mention.id ? mention.id.replace(/[.*+?^${}()|[\\]\\\\]/g, '\\\\$&') : '';\n        processedDescription = processedDescription.replace(\n          new RegExp(`<${escapedType}\\\\s+id=\"${escapedId}\"\\\\s*/>`, 'g'),\n          `@${mention.label}`\n        );\n      }\n    }\n\n    return { description: processedDescription, referencedResources };\n  }\n\n  /**\n   * Process description to extract file tags and read contents\n   * Returns processed description and referenced file contents\n   */\n  private async processFileReferences(\n    description: string,\n    repositoryPath?: string\n  ): Promise<{ description: string; referencedFiles: Array<{ path: string; content: string }> }> {\n    const filePaths = this.extractFilePaths(description);\n    const referencedFiles: Array<{ path: string; content: string }> = [];\n\n    if (filePaths.length === 0 || !repositoryPath) {\n      return { description, referencedFiles };\n    }\n\n    // Read all referenced files\n    for (const filePath of filePaths) {\n      const content = await this.readFileContent(repositoryPath, filePath);\n      if (content !== null) {\n        referencedFiles.push({ path: filePath, content });\n      }\n    }\n\n    // Replace file tags with just the filename for readability\n    let processedDescription = description;\n    for (const filePath of filePaths) {\n      const fileName = filePath.split('/').pop() || filePath;\n      processedDescription = processedDescription.replace(\n        new RegExp(`<file\\\\s+path=\"${filePath.replace(/[.*+?^${}()|[\\]\\\\]/g, '\\\\$&')}\"\\\\s*/>`, 'g'),\n        `@${fileName}`\n      );\n    }\n\n    return { description: processedDescription, referencedFiles };\n  }\n\n  async buildResearchPrompt(task: Task, repositoryPath?: string): Promise<string> {\n    // Process file references in description\n    const { description: descriptionAfterFiles, referencedFiles } = await this.processFileReferences(\n      task.description,\n      repositoryPath\n    );\n\n    // Process URL references in description  \n    const { description: processedDescription, referencedResources } = await this.processUrlReferences(\n      descriptionAfterFiles\n    );\n\n    let prompt = '';\n    prompt += `## Current Task\\n\\n**Task**: ${task.title}\\n**Description**: ${processedDescription}`;\n\n    if ((task as any).primary_repository) {\n      prompt += `\\n**Repository**: ${(task as any).primary_repository}`;\n    }\n\n    // Add referenced files from @ mentions\n    if (referencedFiles.length > 0) {\n      prompt += `\\n\\n## Referenced Files\\n\\n`;\n      for (const file of referencedFiles) {\n        prompt += `### ${file.path}\\n\\`\\`\\`\\n${file.content}\\n\\`\\`\\`\\n\\n`;\n      }\n    }\n\n    // Add referenced resources from URL mentions\n    if (referencedResources.length > 0) {\n      prompt += `\\n\\n## Referenced Resources\\n\\n`;\n      for (const resource of referencedResources) {\n        prompt += `### ${resource.title} (${resource.type})\\n**URL**: ${resource.url}\\n\\n${resource.content}\\n\\n`;\n      }\n    }\n\n    try {\n      const taskFiles = await this.getTaskFiles(task.id);\n      const contextFiles = taskFiles.filter((f: any) => f.type === 'context' || f.type === 'reference');\n      if (contextFiles.length > 0) {\n        prompt += `\\n\\n## Supporting Files`;\n        for (const file of contextFiles) {\n          prompt += `\\n\\n### ${file.name} (${file.type})\\n${file.content}`;\n        }\n      }\n    } catch (error) {\n      this.logger.debug('No existing task files found for research', { taskId: task.id });\n    }\n\n    prompt += `\\n\\nPlease explore the codebase thoroughly and generate 3-5 clarifying questions that will help guide the implementation of this task. Use the \\`create_plan\\` tool to create a research.md artifact with your questions in the markdown format specified in your system prompt.`;\n\n    return prompt;\n  }\n\n  async buildPlanningPrompt(task: Task, repositoryPath?: string): Promise<string> {\n    // Process file references in description\n    const { description: descriptionAfterFiles, referencedFiles } = await this.processFileReferences(\n      task.description,\n      repositoryPath\n    );\n\n    // Process URL references in description  \n    const { description: processedDescription, referencedResources } = await this.processUrlReferences(\n      descriptionAfterFiles\n    );\n\n    let prompt = '';\n    prompt += `## Current Task\\n\\n**Task**: ${task.title}\\n**Description**: ${processedDescription}`;\n\n    if ((task as any).primary_repository) {\n      prompt += `\\n**Repository**: ${(task as any).primary_repository}`;\n    }\n\n    // Add referenced files from @ mentions\n    if (referencedFiles.length > 0) {\n      prompt += `\\n\\n## Referenced Files\\n\\n`;\n      for (const file of referencedFiles) {\n        prompt += `### ${file.path}\\n\\`\\`\\`\\n${file.content}\\n\\`\\`\\`\\n\\n`;\n      }\n    }\n\n    // Add referenced resources from URL mentions\n    if (referencedResources.length > 0) {\n      prompt += `\\n\\n## Referenced Resources\\n\\n`;\n      for (const resource of referencedResources) {\n        prompt += `### ${resource.title} (${resource.type})\\n**URL**: ${resource.url}\\n\\n${resource.content}\\n\\n`;\n      }\n    }\n\n    try {\n      const taskFiles = await this.getTaskFiles(task.id);\n      const contextFiles = taskFiles.filter((f: any) => f.type === 'context' || f.type === 'reference');\n      if (contextFiles.length > 0) {\n        prompt += `\\n\\n## Supporting Files`;\n        for (const file of contextFiles) {\n          prompt += `\\n\\n### ${file.name} (${file.type})\\n${file.content}`;\n        }\n      }\n    } catch (error) {\n      this.logger.debug('No existing task files found for planning', { taskId: task.id });\n    }\n\n    const templateVariables = {\n      task_id: task.id,\n      task_title: task.title,\n      task_description: processedDescription,\n      date: new Date().toISOString().split('T')[0],\n      repository: ((task as any).primary_repository || '') as string,\n    };\n\n    const planTemplate = await this.generatePlanTemplate(templateVariables);\n\n    prompt += `\\n\\nPlease analyze the codebase and create a detailed implementation plan for this task. Use the following template structure for your plan:\\n\\n${planTemplate}\\n\\nFill in each section with specific, actionable information based on your analysis. Replace all placeholder content with actual details about this task.`;\n\n    return prompt;\n  }\n\n  async buildExecutionPrompt(task: Task, repositoryPath?: string): Promise<string> {\n    // Process file references in description\n    const { description: descriptionAfterFiles, referencedFiles } = await this.processFileReferences(\n      task.description,\n      repositoryPath\n    );\n\n    // Process URL references in description  \n    const { description: processedDescription, referencedResources } = await this.processUrlReferences(\n      descriptionAfterFiles\n    );\n\n    let prompt = '';\n    prompt += `## Current Task\\n\\n**Task**: ${task.title}\\n**Description**: ${processedDescription}`;\n\n    if ((task as any).primary_repository) {\n      prompt += `\\n**Repository**: ${(task as any).primary_repository}`;\n    }\n\n    // Add referenced files from @ mentions\n    if (referencedFiles.length > 0) {\n      prompt += `\\n\\n## Referenced Files\\n\\n`;\n      for (const file of referencedFiles) {\n        prompt += `### ${file.path}\\n\\`\\`\\`\\n${file.content}\\n\\`\\`\\`\\n\\n`;\n      }\n    }\n\n    // Add referenced resources from URL mentions\n    if (referencedResources.length > 0) {\n      prompt += `\\n\\n## Referenced Resources\\n\\n`;\n      for (const resource of referencedResources) {\n        prompt += `### ${resource.title} (${resource.type})\\n**URL**: ${resource.url}\\n\\n${resource.content}\\n\\n`;\n      }\n    }\n\n    try {\n      const taskFiles = await this.getTaskFiles(task.id);\n      const hasPlan = taskFiles.some((f: any) => f.type === 'plan');\n      if (taskFiles.length > 0) {\n        prompt += `\\n\\n## Context and Supporting Information`;\n        for (const file of taskFiles) {\n          if (file.type === 'plan') {\n            prompt += `\\n\\n### Execution Plan\\n${file.content}`;\n          } else {\n            prompt += `\\n\\n### ${file.name} (${file.type})\\n${file.content}`;\n          }\n        }\n      }\n      if (hasPlan) {\n        prompt += `\\n\\nPlease implement the changes described in the execution plan above. Follow the plan step-by-step and make the necessary file modifications. You must actually edit files and make changes - do not just analyze or review.`;\n      } else {\n        prompt += `\\n\\nPlease implement the changes described in the task above. You must actually edit files and make changes - do not just analyze or review.`;\n      }\n    } catch (error) {\n      this.logger.debug('No supporting files found for execution', { taskId: task.id });\n      prompt += `\\n\\nPlease implement the changes described in the task above.`;\n    }\n    return prompt;\n  }\n}\n\n\n"],"names":["fs"],"mappings":";;;;MAaa,aAAa,CAAA;AAChB,IAAA,YAAY;AACZ,IAAA,oBAAoB;AACpB,IAAA,aAAa;AACb,IAAA,MAAM;AAEd,IAAA,WAAA,CAAY,IAAuB,EAAA;AACjC,QAAA,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,YAAY;AACrC,QAAA,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC,oBAAoB;AACrD,QAAA,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,aAAa;QACvC,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,IAAI,IAAI,MAAM,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,iBAAiB,EAAE,CAAC;IACtF;AAEA;;;AAGG;AACK,IAAA,gBAAgB,CAAC,WAAmB,EAAA;QAC1C,MAAM,YAAY,GAAG,+BAA+B;QACpD,MAAM,KAAK,GAAa,EAAE;AAC1B,QAAA,IAAI,KAA6B;AAEjC,QAAA,OAAO,CAAC,KAAK,GAAG,YAAY,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,IAAI,EAAE;YACxD,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACtB;AAEA,QAAA,OAAO,KAAK;IACd;AAEA;;AAEG;AACK,IAAA,MAAM,eAAe,CAAC,cAAsB,EAAE,QAAgB,EAAA;AACpE,QAAA,IAAI;YACF,MAAM,QAAQ,GAAG,IAAI,CAAC,cAAc,EAAE,QAAQ,CAAC;YAC/C,MAAM,OAAO,GAAG,MAAMA,QAAE,CAAC,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC;AACnD,YAAA,OAAO,OAAO;QAChB;QAAE,OAAO,KAAK,EAAE;AACd,YAAA,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAA,gCAAA,EAAmC,QAAQ,CAAA,CAAE,EAAE,EAAE,KAAK,EAAE,CAAC;AAC1E,YAAA,OAAO,IAAI;QACb;IACF;AAEA;;;AAGG;AACK,IAAA,kBAAkB,CAAC,WAAmB,EAAA;QAC5C,MAAM,QAAQ,GAAiB,EAAE;;QAGjC,MAAM,aAAa,GAAG,gEAAgE;AACtF,QAAA,IAAI,KAA6B;AAEjC,QAAA,OAAO,CAAC,KAAK,GAAG,aAAa,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,IAAI,EAAE;YACzD,MAAM,GAAG,IAAI,EAAE,EAAE,CAAC,GAAG,KAAK;YAC1B,QAAQ,CAAC,IAAI,CAAC;gBACZ,GAAG,EAAE,EAAE;AACP,gBAAA,IAAI,EAAE,IAAW;gBACjB,EAAE;gBACF,KAAK,EAAE,IAAI,CAAC,gBAAgB,CAAC,EAAE,EAAE,IAAW,CAAC;AAC9C,aAAA,CAAC;QACJ;;QAGA,MAAM,QAAQ,GAAG,8BAA8B;AAC/C,QAAA,OAAO,CAAC,KAAK,GAAG,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,IAAI,EAAE;AACpD,YAAA,MAAM,GAAG,GAAG,CAAC,GAAG,KAAK;YACrB,QAAQ,CAAC,IAAI,CAAC;gBACZ,GAAG;AACH,gBAAA,IAAI,EAAE,SAAS;gBACf,KAAK,EAAE,IAAI,CAAC,gBAAgB,CAAC,GAAG,EAAE,SAAS,CAAC;AAC7C,aAAA,CAAC;QACJ;AAEA,QAAA,OAAO,QAAQ;IACjB;AAEA;;AAEG;IACK,gBAAgB,CAAC,GAAW,EAAE,IAAY,EAAA;AAChD,QAAA,IAAI;AACF,YAAA,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,GAAG,CAAC;YAC3B,QAAQ,IAAI;AACV,gBAAA,KAAK,OAAO;oBACV,MAAM,UAAU,GAAG,GAAG,CAAC,KAAK,CAAC,8BAA8B,CAAC;oBAC5D,OAAO,UAAU,GAAG,SAAS,UAAU,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,KAAK,GAAG,OAAO;AACvE,gBAAA,KAAK,YAAY;oBACf,MAAM,QAAQ,GAAG,GAAG,CAAC,KAAK,CAAC,oBAAoB,CAAC;AAChD,oBAAA,OAAO,QAAQ,GAAG,eAAe,QAAQ,CAAC,CAAC,CAAC,CAAA,CAAE,GAAG,YAAY;AAC/D,gBAAA,KAAK,SAAS;AACZ,oBAAA,OAAO,SAAS;AAClB,gBAAA,KAAK,cAAc;AACjB,oBAAA,OAAO,cAAc;AACvB,gBAAA;oBACE,OAAO,MAAM,CAAC,QAAQ;;QAE5B;AAAE,QAAA,MAAM;AACN,YAAA,OAAO,KAAK;QACd;IACF;AAEA;;AAEG;IACK,MAAM,oBAAoB,CAChC,WAAmB,EAAA;QAEnB,MAAM,WAAW,GAAG,IAAI,CAAC,kBAAkB,CAAC,WAAW,CAAC;QACxD,MAAM,mBAAmB,GAAsB,EAAE;QAEjD,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE;AACnD,YAAA,OAAO,EAAE,WAAW,EAAE,mBAAmB,EAAE;QAC7C;;AAGA,QAAA,KAAK,MAAM,OAAO,IAAI,WAAW,EAAE;AACjC,YAAA,IAAI;gBACF,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,kBAAkB,CAAC,OAAO,CAAC;AACrE,gBAAA,mBAAmB,CAAC,IAAI,CAAC,QAAQ,CAAC;YACpC;YAAE,OAAO,KAAK,EAAE;AACd,gBAAA,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAA,mCAAA,EAAsC,OAAO,CAAC,GAAG,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC;;gBAEhF,mBAAmB,CAAC,IAAI,CAAC;oBACvB,IAAI,EAAE,OAAO,CAAC,IAAI;AAClB,oBAAA,EAAE,EAAE,OAAO,CAAC,EAAE,IAAI,EAAE;oBACpB,GAAG,EAAE,OAAO,CAAC,GAAG;AAChB,oBAAA,KAAK,EAAE,OAAO,CAAC,KAAK,IAAI,kBAAkB;AAC1C,oBAAA,OAAO,EAAE,CAAA,8BAAA,EAAiC,OAAO,CAAC,GAAG,CAAA,EAAA,EAAK,KAAK,CAAA,CAAE;AACjE,oBAAA,QAAQ,EAAE,EAAE;AACb,iBAAA,CAAC;YACJ;QACF;;QAGA,IAAI,oBAAoB,GAAG,WAAW;AACtC,QAAA,KAAK,MAAM,OAAO,IAAI,WAAW,EAAE;AACjC,YAAA,IAAI,OAAO,CAAC,IAAI,KAAK,SAAS,EAAE;;AAE9B,gBAAA,MAAM,UAAU,GAAG,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,qBAAqB,EAAE,MAAM,CAAC;gBACrE,oBAAoB,GAAG,oBAAoB,CAAC,OAAO,CACjD,IAAI,MAAM,CAAC,CAAA,cAAA,EAAiB,UAAU,SAAS,EAAE,GAAG,CAAC,EACrD,CAAA,CAAA,EAAI,OAAO,CAAC,KAAK,CAAA,CAAE,CACpB;YACH;iBAAO;;AAEL,gBAAA,MAAM,WAAW,GAAG,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,qBAAqB,EAAE,MAAM,CAAC;gBACvE,MAAM,SAAS,GAAG,OAAO,CAAC,EAAE,GAAG,OAAO,CAAC,EAAE,CAAC,OAAO,CAAC,qBAAqB,EAAE,MAAM,CAAC,GAAG,EAAE;gBACrF,oBAAoB,GAAG,oBAAoB,CAAC,OAAO,CACjD,IAAI,MAAM,CAAC,CAAA,CAAA,EAAI,WAAW,CAAA,QAAA,EAAW,SAAS,CAAA,OAAA,CAAS,EAAE,GAAG,CAAC,EAC7D,CAAA,CAAA,EAAI,OAAO,CAAC,KAAK,CAAA,CAAE,CACpB;YACH;QACF;AAEA,QAAA,OAAO,EAAE,WAAW,EAAE,oBAAoB,EAAE,mBAAmB,EAAE;IACnE;AAEA;;;AAGG;AACK,IAAA,MAAM,qBAAqB,CACjC,WAAmB,EACnB,cAAuB,EAAA;QAEvB,MAAM,SAAS,GAAG,IAAI,CAAC,gBAAgB,CAAC,WAAW,CAAC;QACpD,MAAM,eAAe,GAA6C,EAAE;QAEpE,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,IAAI,CAAC,cAAc,EAAE;AAC7C,YAAA,OAAO,EAAE,WAAW,EAAE,eAAe,EAAE;QACzC;;AAGA,QAAA,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE;YAChC,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,cAAc,EAAE,QAAQ,CAAC;AACpE,YAAA,IAAI,OAAO,KAAK,IAAI,EAAE;gBACpB,eAAe,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC;YACnD;QACF;;QAGA,IAAI,oBAAoB,GAAG,WAAW;AACtC,QAAA,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE;AAChC,YAAA,MAAM,QAAQ,GAAG,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,IAAI,QAAQ;YACtD,oBAAoB,GAAG,oBAAoB,CAAC,OAAO,CACjD,IAAI,MAAM,CAAC,CAAA,eAAA,EAAkB,QAAQ,CAAC,OAAO,CAAC,qBAAqB,EAAE,MAAM,CAAC,CAAA,OAAA,CAAS,EAAE,GAAG,CAAC,EAC3F,CAAA,CAAA,EAAI,QAAQ,CAAA,CAAE,CACf;QACH;AAEA,QAAA,OAAO,EAAE,WAAW,EAAE,oBAAoB,EAAE,eAAe,EAAE;IAC/D;AAEA,IAAA,MAAM,mBAAmB,CAAC,IAAU,EAAE,cAAuB,EAAA;;AAE3D,QAAA,MAAM,EAAE,WAAW,EAAE,qBAAqB,EAAE,eAAe,EAAE,GAAG,MAAM,IAAI,CAAC,qBAAqB,CAC9F,IAAI,CAAC,WAAW,EAChB,cAAc,CACf;;AAGD,QAAA,MAAM,EAAE,WAAW,EAAE,oBAAoB,EAAE,mBAAmB,EAAE,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAChG,qBAAqB,CACtB;QAED,IAAI,MAAM,GAAG,EAAE;QACf,MAAM,IAAI,gCAAgC,IAAI,CAAC,KAAK,CAAA,mBAAA,EAAsB,oBAAoB,EAAE;AAEhG,QAAA,IAAK,IAAY,CAAC,kBAAkB,EAAE;AACpC,YAAA,MAAM,IAAI,CAAA,kBAAA,EAAsB,IAAY,CAAC,kBAAkB,EAAE;QACnE;;AAGA,QAAA,IAAI,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE;YAC9B,MAAM,IAAI,6BAA6B;AACvC,YAAA,KAAK,MAAM,IAAI,IAAI,eAAe,EAAE;gBAClC,MAAM,IAAI,CAAA,IAAA,EAAO,IAAI,CAAC,IAAI,aAAa,IAAI,CAAC,OAAO,CAAA,YAAA,CAAc;YACnE;QACF;;AAGA,QAAA,IAAI,mBAAmB,CAAC,MAAM,GAAG,CAAC,EAAE;YAClC,MAAM,IAAI,iCAAiC;AAC3C,YAAA,KAAK,MAAM,QAAQ,IAAI,mBAAmB,EAAE;AAC1C,gBAAA,MAAM,IAAI,CAAA,IAAA,EAAO,QAAQ,CAAC,KAAK,CAAA,EAAA,EAAK,QAAQ,CAAC,IAAI,CAAA,YAAA,EAAe,QAAQ,CAAC,GAAG,CAAA,IAAA,EAAO,QAAQ,CAAC,OAAO,MAAM;YAC3G;QACF;AAEA,QAAA,IAAI;YACF,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,EAAE,CAAC;YAClD,MAAM,YAAY,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC,CAAM,KAAK,CAAC,CAAC,IAAI,KAAK,SAAS,IAAI,CAAC,CAAC,IAAI,KAAK,WAAW,CAAC;AACjG,YAAA,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE;gBAC3B,MAAM,IAAI,yBAAyB;AACnC,gBAAA,KAAK,MAAM,IAAI,IAAI,YAAY,EAAE;AAC/B,oBAAA,MAAM,IAAI,CAAA,QAAA,EAAW,IAAI,CAAC,IAAI,CAAA,EAAA,EAAK,IAAI,CAAC,IAAI,CAAA,GAAA,EAAM,IAAI,CAAC,OAAO,EAAE;gBAClE;YACF;QACF;QAAE,OAAO,KAAK,EAAE;AACd,YAAA,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,2CAA2C,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC;QACrF;QAEA,MAAM,IAAI,kRAAkR;AAE5R,QAAA,OAAO,MAAM;IACf;AAEA,IAAA,MAAM,mBAAmB,CAAC,IAAU,EAAE,cAAuB,EAAA;;AAE3D,QAAA,MAAM,EAAE,WAAW,EAAE,qBAAqB,EAAE,eAAe,EAAE,GAAG,MAAM,IAAI,CAAC,qBAAqB,CAC9F,IAAI,CAAC,WAAW,EAChB,cAAc,CACf;;AAGD,QAAA,MAAM,EAAE,WAAW,EAAE,oBAAoB,EAAE,mBAAmB,EAAE,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAChG,qBAAqB,CACtB;QAED,IAAI,MAAM,GAAG,EAAE;QACf,MAAM,IAAI,gCAAgC,IAAI,CAAC,KAAK,CAAA,mBAAA,EAAsB,oBAAoB,EAAE;AAEhG,QAAA,IAAK,IAAY,CAAC,kBAAkB,EAAE;AACpC,YAAA,MAAM,IAAI,CAAA,kBAAA,EAAsB,IAAY,CAAC,kBAAkB,EAAE;QACnE;;AAGA,QAAA,IAAI,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE;YAC9B,MAAM,IAAI,6BAA6B;AACvC,YAAA,KAAK,MAAM,IAAI,IAAI,eAAe,EAAE;gBAClC,MAAM,IAAI,CAAA,IAAA,EAAO,IAAI,CAAC,IAAI,aAAa,IAAI,CAAC,OAAO,CAAA,YAAA,CAAc;YACnE;QACF;;AAGA,QAAA,IAAI,mBAAmB,CAAC,MAAM,GAAG,CAAC,EAAE;YAClC,MAAM,IAAI,iCAAiC;AAC3C,YAAA,KAAK,MAAM,QAAQ,IAAI,mBAAmB,EAAE;AAC1C,gBAAA,MAAM,IAAI,CAAA,IAAA,EAAO,QAAQ,CAAC,KAAK,CAAA,EAAA,EAAK,QAAQ,CAAC,IAAI,CAAA,YAAA,EAAe,QAAQ,CAAC,GAAG,CAAA,IAAA,EAAO,QAAQ,CAAC,OAAO,MAAM;YAC3G;QACF;AAEA,QAAA,IAAI;YACF,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,EAAE,CAAC;YAClD,MAAM,YAAY,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC,CAAM,KAAK,CAAC,CAAC,IAAI,KAAK,SAAS,IAAI,CAAC,CAAC,IAAI,KAAK,WAAW,CAAC;AACjG,YAAA,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE;gBAC3B,MAAM,IAAI,yBAAyB;AACnC,gBAAA,KAAK,MAAM,IAAI,IAAI,YAAY,EAAE;AAC/B,oBAAA,MAAM,IAAI,CAAA,QAAA,EAAW,IAAI,CAAC,IAAI,CAAA,EAAA,EAAK,IAAI,CAAC,IAAI,CAAA,GAAA,EAAM,IAAI,CAAC,OAAO,EAAE;gBAClE;YACF;QACF;QAAE,OAAO,KAAK,EAAE;AACd,YAAA,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,2CAA2C,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC;QACrF;AAEA,QAAA,MAAM,iBAAiB,GAAG;YACxB,OAAO,EAAE,IAAI,CAAC,EAAE;YAChB,UAAU,EAAE,IAAI,CAAC,KAAK;AACtB,YAAA,gBAAgB,EAAE,oBAAoB;AACtC,YAAA,IAAI,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;AAC5C,YAAA,UAAU,GAAI,IAAY,CAAC,kBAAkB,IAAI,EAAE,CAAW;SAC/D;QAED,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,iBAAiB,CAAC;AAEvE,QAAA,MAAM,IAAI,CAAA,gJAAA,EAAmJ,YAAY,CAAA,2JAAA,CAA6J;AAEtU,QAAA,OAAO,MAAM;IACf;AAEA,IAAA,MAAM,oBAAoB,CAAC,IAAU,EAAE,cAAuB,EAAA;;AAE5D,QAAA,MAAM,EAAE,WAAW,EAAE,qBAAqB,EAAE,eAAe,EAAE,GAAG,MAAM,IAAI,CAAC,qBAAqB,CAC9F,IAAI,CAAC,WAAW,EAChB,cAAc,CACf;;AAGD,QAAA,MAAM,EAAE,WAAW,EAAE,oBAAoB,EAAE,mBAAmB,EAAE,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAChG,qBAAqB,CACtB;QAED,IAAI,MAAM,GAAG,EAAE;QACf,MAAM,IAAI,gCAAgC,IAAI,CAAC,KAAK,CAAA,mBAAA,EAAsB,oBAAoB,EAAE;AAEhG,QAAA,IAAK,IAAY,CAAC,kBAAkB,EAAE;AACpC,YAAA,MAAM,IAAI,CAAA,kBAAA,EAAsB,IAAY,CAAC,kBAAkB,EAAE;QACnE;;AAGA,QAAA,IAAI,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE;YAC9B,MAAM,IAAI,6BAA6B;AACvC,YAAA,KAAK,MAAM,IAAI,IAAI,eAAe,EAAE;gBAClC,MAAM,IAAI,CAAA,IAAA,EAAO,IAAI,CAAC,IAAI,aAAa,IAAI,CAAC,OAAO,CAAA,YAAA,CAAc;YACnE;QACF;;AAGA,QAAA,IAAI,mBAAmB,CAAC,MAAM,GAAG,CAAC,EAAE;YAClC,MAAM,IAAI,iCAAiC;AAC3C,YAAA,KAAK,MAAM,QAAQ,IAAI,mBAAmB,EAAE;AAC1C,gBAAA,MAAM,IAAI,CAAA,IAAA,EAAO,QAAQ,CAAC,KAAK,CAAA,EAAA,EAAK,QAAQ,CAAC,IAAI,CAAA,YAAA,EAAe,QAAQ,CAAC,GAAG,CAAA,IAAA,EAAO,QAAQ,CAAC,OAAO,MAAM;YAC3G;QACF;AAEA,QAAA,IAAI;YACF,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,EAAE,CAAC;AAClD,YAAA,MAAM,OAAO,GAAG,SAAS,CAAC,IAAI,CAAC,CAAC,CAAM,KAAK,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC;AAC7D,YAAA,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE;gBACxB,MAAM,IAAI,2CAA2C;AACrD,gBAAA,KAAK,MAAM,IAAI,IAAI,SAAS,EAAE;AAC5B,oBAAA,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE;AACxB,wBAAA,MAAM,IAAI,CAAA,wBAAA,EAA2B,IAAI,CAAC,OAAO,EAAE;oBACrD;yBAAO;AACL,wBAAA,MAAM,IAAI,CAAA,QAAA,EAAW,IAAI,CAAC,IAAI,CAAA,EAAA,EAAK,IAAI,CAAC,IAAI,CAAA,GAAA,EAAM,IAAI,CAAC,OAAO,EAAE;oBAClE;gBACF;YACF;YACA,IAAI,OAAO,EAAE;gBACX,MAAM,IAAI,gOAAgO;YAC5O;iBAAO;gBACL,MAAM,IAAI,8IAA8I;YAC1J;QACF;QAAE,OAAO,KAAK,EAAE;AACd,YAAA,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,yCAAyC,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC;YACjF,MAAM,IAAI,+DAA+D;QAC3E;AACA,QAAA,OAAO,MAAM;IACf;AACD;;;;"}

package/dist/src/stage-executor.d.ts

@@ -13,6 +13,7 @@ export declare class StageExecutor {
     constructor(registry: AgentRegistry, logger: Logger, promptBuilder?: PromptBuilder, eventHandler?: (event: AgentEvent) => void, mcpServers?: Record<string, McpServerConfig>);
     setEventHandler(handler?: (event: AgentEvent) => void): void;
     execute(task: Task, stage: WorkflowStage, options: WorkflowExecutionOptions): Promise<WorkflowStageExecutionResult>;
+    private runResearch;
     private runPlanning;
     private runExecution;
 }

package/dist/src/stage-executor.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"stage-executor.d.ts","sourceRoot":"","sources":["../../src/stage-executor.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAE3C,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACpD,OAAO,KAAK,EAAE,UAAU,EAAE,IAAI,EAAE,eAAe,EAAE,MAAM,YAAY,CAAC;AACpE,OAAO,KAAK,EAAE,aAAa,EAAE,4BAA4B,EAAE,wBAAwB,EAAE,MAAM,qBAAqB,CAAC;AAGjH,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AAEpD,qBAAa,aAAa;IACxB,OAAO,CAAC,QAAQ,CAAgB;IAChC,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,OAAO,CAAgB;IAC/B,OAAO,CAAC,aAAa,CAAgB;IACrC,OAAO,CAAC,YAAY,CAAC,CAA8B;IACnD,OAAO,CAAC,UAAU,CAAC,CAAkC;gBAGnD,QAAQ,EAAE,aAAa,EACvB,MAAM,EAAE,MAAM,EACd,aAAa,CAAC,EAAE,aAAa,EAC7B,YAAY,CAAC,EAAE,CAAC,KAAK,EAAE,UAAU,KAAK,IAAI,EAC1C,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,eAAe,CAAC;IAc9C,eAAe,CAAC,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,UAAU,KAAK,IAAI,GAAG,IAAI;IAItD,OAAO,CAAC,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,aAAa,EAAE,OAAO,EAAE,wBAAwB,GAAG,OAAO,CAAC,4BAA4B,CAAC;YA+B3G,WAAW;YA+CX,YAAY;CAwC3B"}
+{"version":3,"file":"stage-executor.d.ts","sourceRoot":"","sources":["../../src/stage-executor.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAE3C,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACpD,OAAO,KAAK,EAAE,UAAU,EAAE,IAAI,EAAE,eAAe,EAAE,MAAM,YAAY,CAAC;AACpE,OAAO,KAAK,EAAE,aAAa,EAAE,4BAA4B,EAAE,wBAAwB,EAAE,MAAM,qBAAqB,CAAC;AAIjH,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AAEpD,qBAAa,aAAa;IACxB,OAAO,CAAC,QAAQ,CAAgB;IAChC,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,OAAO,CAAgB;IAC/B,OAAO,CAAC,aAAa,CAAgB;IACrC,OAAO,CAAC,YAAY,CAAC,CAA8B;IACnD,OAAO,CAAC,UAAU,CAAC,CAAkC;gBAGnD,QAAQ,EAAE,aAAa,EACvB,MAAM,EAAE,MAAM,EACd,aAAa,CAAC,EAAE,aAAa,EAC7B,YAAY,CAAC,EAAE,CAAC,KAAK,EAAE,UAAU,KAAK,IAAI,EAC1C,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,eAAe,CAAC;IAc9C,eAAe,CAAC,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,UAAU,KAAK,IAAI,GAAG,IAAI;IAItD,OAAO,CAAC,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,aAAa,EAAE,OAAO,EAAE,wBAAwB,GAAG,OAAO,CAAC,4BAA4B,CAAC;YAiC3G,WAAW;YA+CX,WAAW;YA+CX,YAAY;CAwC3B"}

package/dist/src/stage-executor.js

@@ -1,6 +1,7 @@
 import { query } from '@anthropic-ai/claude-agent-sdk';
 import './utils/logger.js';
 import { ClaudeAdapter } from './adapters/claude/claude-adapter.js';
+import { RESEARCH_SYSTEM_PROMPT } from './agents/research.js';
 import { PLANNING_SYSTEM_PROMPT } from './agents/planning.js';
 import { EXECUTION_SYSTEM_PROMPT } from './agents/execution.js';
 import { PromptBuilder } from './prompt-builder.js';
@@ -42,6 +43,8 @@ class StageExecutor {
         const permissionMode = options.permissionMode || 'acceptEdits';
         const cwd = options.repositoryPath || process.cwd();
         switch (agent.agent_type) {
+            case 'research':
+                return this.runResearch(task, cwd, options, stage.key);
             case 'planning':
                 return this.runPlanning(task, cwd, options, stage.key);
             case 'execution':
@@ -54,6 +57,46 @@ class StageExecutor {
                 return { results: [] };
         }
     }
+    async runResearch(task, cwd, options, stageKey) {
+        const contextPrompt = await this.promptBuilder.buildResearchPrompt(task, cwd);
+        let prompt = RESEARCH_SYSTEM_PROMPT + '\n\n' + contextPrompt;
+        const stageOverrides = options.stageOverrides?.[stageKey] || options.stageOverrides?.['research'];
+        const mergedOverrides = {
+            ...(options.queryOverrides || {}),
+            ...(stageOverrides?.queryOverrides || {}),
+        };
+        const baseOptions = {
+            model: 'claude-sonnet-4-5-20250929',
+            cwd,
+            permissionMode: 'plan',
+            settingSources: ['local'],
+            mcpServers: this.mcpServers
+        };
+        const response = query({
+            prompt,
+            options: { ...baseOptions, ...mergedOverrides },
+        });
+        let research = '';
+        for await (const message of response) {
+            // Emit raw SDK event first
+            this.eventHandler?.(this.adapter.createRawSDKEvent(message));
+            // Then emit transformed event
+            const transformed = this.adapter.transform(message);
+            if (transformed) {
+                if (transformed.type !== 'token') {
+                    this.logger.debug('Research event', { type: transformed.type });
+                }
+                this.eventHandler?.(transformed);
+            }
+            if (message.type === 'assistant' && message.message?.content) {
+                for (const c of message.message.content) {
+                    if (c.type === 'text' && c.text)
+                        research += c.text + '\n';
+                }
+            }
+        }
+        return { plan: research.trim() }; // Return as 'plan' field to match existing interface
+    }
     async runPlanning(task, cwd, options, stageKey) {
         const contextPrompt = await this.promptBuilder.buildPlanningPrompt(task, cwd);
         let prompt = PLANNING_SYSTEM_PROMPT + '\n\n' + contextPrompt;

package/dist/src/stage-executor.js.map

@@ -1 +1 @@
-{"version":3,"file":"stage-executor.js","sources":["../../src/stage-executor.ts"],"sourcesContent":["import { query } from '@anthropic-ai/claude-agent-sdk';\nimport { Logger } from './utils/logger.js';\nimport { ClaudeAdapter } from './adapters/claude/claude-adapter.js';\nimport { AgentRegistry } from './agent-registry.js';\nimport type { AgentEvent, Task, McpServerConfig } from './types.js';\nimport type { WorkflowStage, WorkflowStageExecutionResult, WorkflowExecutionOptions } from './workflow-types.js';\nimport { PLANNING_SYSTEM_PROMPT } from './agents/planning.js';\nimport { EXECUTION_SYSTEM_PROMPT } from './agents/execution.js';\nimport { PromptBuilder } from './prompt-builder.js';\n\nexport class StageExecutor {\n  private registry: AgentRegistry;\n  private logger: Logger;\n  private adapter: ClaudeAdapter;\n  private promptBuilder: PromptBuilder;\n  private eventHandler?: (event: AgentEvent) => void;\n  private mcpServers?: Record<string, McpServerConfig>;\n\n  constructor(\n    registry: AgentRegistry,\n    logger: Logger,\n    promptBuilder?: PromptBuilder,\n    eventHandler?: (event: AgentEvent) => void,\n    mcpServers?: Record<string, McpServerConfig>,\n  ) {\n    this.registry = registry;\n    this.logger = logger.child('StageExecutor');\n    this.adapter = new ClaudeAdapter();\n    this.promptBuilder = promptBuilder || new PromptBuilder({\n      getTaskFiles: async () => [],\n      generatePlanTemplate: async () => '',\n      logger,\n    });\n    this.eventHandler = eventHandler;\n    this.mcpServers = mcpServers;\n  }\n\n  setEventHandler(handler?: (event: AgentEvent) => void): void {\n    this.eventHandler = handler;\n  }\n\n  async execute(task: Task, stage: WorkflowStage, options: WorkflowExecutionOptions): Promise<WorkflowStageExecutionResult> {\n    const isManual = stage.is_manual_only === true;\n    if (isManual) {\n      this.logger.info('Manual stage detected; skipping agent execution', { stage: stage.key });\n      return { results: [] };\n    }\n\n    const inferredAgent = stage.key.toLowerCase().includes('plan') ? 'planning_basic' : 'code_generation';\n    const agentName = stage.agent_name || inferredAgent;\n    const agent = this.registry.getAgent(agentName);\n    if (!agent) {\n      throw new Error(`Unknown agent '${agentName}' for stage '${stage.key}'`);\n    }\n\n    const permissionMode = (options.permissionMode as any) || 'acceptEdits';\n    const cwd = options.repositoryPath || process.cwd();\n\n    switch (agent.agent_type) {\n      case 'planning':\n        return this.runPlanning(task, cwd, options, stage.key);\n      case 'execution':\n        return this.runExecution(task, cwd, permissionMode, options, stage.key);\n      case 'review': // TODO: Implement review\n      case 'testing': // TODO: Implement testing\n      default:\n        // throw new Error(`Unsupported agent type: ${agent.agent_type}`);\n        console.warn(`Unsupported agent type: ${agent.agent_type}`);\n        return { results: [] };\n    }\n  }\n\n  private async runPlanning(task: Task, cwd: string, options: WorkflowExecutionOptions, stageKey: string): Promise<WorkflowStageExecutionResult> {\n    const contextPrompt = await this.promptBuilder.buildPlanningPrompt(task, cwd);\n    let prompt = PLANNING_SYSTEM_PROMPT + '\\n\\n' + contextPrompt;\n\n    const stageOverrides = options.stageOverrides?.[stageKey] || options.stageOverrides?.['plan'];\n    const mergedOverrides = {\n      ...(options.queryOverrides || {}),\n      ...(stageOverrides?.queryOverrides || {}),\n    } as Record<string, any>;\n\n    const baseOptions: Record<string, any> = {\n      model: 'claude-sonnet-4-5-20250929',\n      cwd,\n      permissionMode: 'plan',\n      settingSources: ['local'],\n      mcpServers: this.mcpServers\n    };\n\n    const response = query({\n      prompt,\n      options: { ...baseOptions, ...mergedOverrides },\n    });\n\n    let plan = '';\n    for await (const message of response) {\n      // Emit raw SDK event first\n      this.eventHandler?.(this.adapter.createRawSDKEvent(message));\n\n      // Then emit transformed event\n      const transformed = this.adapter.transform(message);\n      if (transformed) {\n        if (transformed.type !== 'token') {\n          this.logger.debug('Planning event', { type: transformed.type });\n        }\n        this.eventHandler?.(transformed);\n      }\n\n      if (message.type === 'assistant' && message.message?.content) {\n        for (const c of message.message.content) {\n          if (c.type === 'text' && c.text) plan += c.text + '\\n';\n        }\n      }\n    }\n\n    return { plan: plan.trim() };\n  }\n\n  private async runExecution(task: Task, cwd: string, permissionMode: WorkflowExecutionOptions['permissionMode'], options: WorkflowExecutionOptions, stageKey: string): Promise<WorkflowStageExecutionResult> {\n    const contextPrompt = await this.promptBuilder.buildExecutionPrompt(task, cwd);\n    let prompt = EXECUTION_SYSTEM_PROMPT + '\\n\\n' + contextPrompt;\n\n    const stageOverrides = options.stageOverrides?.[stageKey];\n    const mergedOverrides = {\n      ...(options.queryOverrides || {}),\n      ...(stageOverrides?.queryOverrides || {}),\n    } as Record<string, any>;\n\n    const baseOptions: Record<string, any> = {\n      model: 'claude-sonnet-4-5-20250929',\n      cwd,\n      permissionMode,\n      settingSources: ['local'],\n      mcpServers: this.mcpServers\n    };\n\n    const response = query({\n      prompt,\n      options: { ...baseOptions, ...mergedOverrides },\n    });\n    const results: any[] = [];\n    for await (const message of response) {\n      // Emit raw SDK event first\n      this.eventHandler?.(this.adapter.createRawSDKEvent(message));\n\n      // Then emit transformed event\n      const transformed = this.adapter.transform(message);\n      if (transformed) {\n        if (transformed.type !== 'token') {\n          this.logger.debug('Execution event', { type: transformed.type });\n        }\n        this.eventHandler?.(transformed);\n      }\n\n      results.push(message);\n    }\n    return { results };\n  }\n}\n"],"names":[],"mappings":";;;;;;;MAUa,aAAa,CAAA;AAChB,IAAA,QAAQ;AACR,IAAA,MAAM;AACN,IAAA,OAAO;AACP,IAAA,aAAa;AACb,IAAA,YAAY;AACZ,IAAA,UAAU;IAElB,WAAA,CACE,QAAuB,EACvB,MAAc,EACd,aAA6B,EAC7B,YAA0C,EAC1C,UAA4C,EAAA;AAE5C,QAAA,IAAI,CAAC,QAAQ,GAAG,QAAQ;QACxB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC;AAC3C,QAAA,IAAI,CAAC,OAAO,GAAG,IAAI,aAAa,EAAE;AAClC,QAAA,IAAI,CAAC,aAAa,GAAG,aAAa,IAAI,IAAI,aAAa,CAAC;AACtD,YAAA,YAAY,EAAE,YAAY,EAAE;AAC5B,YAAA,oBAAoB,EAAE,YAAY,EAAE;YACpC,MAAM;AACP,SAAA,CAAC;AACF,QAAA,IAAI,CAAC,YAAY,GAAG,YAAY;AAChC,QAAA,IAAI,CAAC,UAAU,GAAG,UAAU;IAC9B;AAEA,IAAA,eAAe,CAAC,OAAqC,EAAA;AACnD,QAAA,IAAI,CAAC,YAAY,GAAG,OAAO;IAC7B;AAEA,IAAA,MAAM,OAAO,CAAC,IAAU,EAAE,KAAoB,EAAE,OAAiC,EAAA;AAC/E,QAAA,MAAM,QAAQ,GAAG,KAAK,CAAC,cAAc,KAAK,IAAI;QAC9C,IAAI,QAAQ,EAAE;AACZ,YAAA,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,iDAAiD,EAAE,EAAE,KAAK,EAAE,KAAK,CAAC,GAAG,EAAE,CAAC;AACzF,YAAA,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE;QACxB;QAEA,MAAM,aAAa,GAAG,KAAK,CAAC,GAAG,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,GAAG,gBAAgB,GAAG,iBAAiB;AACrG,QAAA,MAAM,SAAS,GAAG,KAAK,CAAC,UAAU,IAAI,aAAa;QACnD,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAAC;QAC/C,IAAI,CAAC,KAAK,EAAE;YACV,MAAM,IAAI,KAAK,CAAC,CAAA,eAAA,EAAkB,SAAS,CAAA,aAAA,EAAgB,KAAK,CAAC,GAAG,CAAA,CAAA,CAAG,CAAC;QAC1E;AAEA,QAAA,MAAM,cAAc,GAAI,OAAO,CAAC,cAAsB,IAAI,aAAa;QACvE,MAAM,GAAG,GAAG,OAAO,CAAC,cAAc,IAAI,OAAO,CAAC,GAAG,EAAE;AAEnD,QAAA,QAAQ,KAAK,CAAC,UAAU;AACtB,YAAA,KAAK,UAAU;AACb,gBAAA,OAAO,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,GAAG,EAAE,OAAO,EAAE,KAAK,CAAC,GAAG,CAAC;AACxD,YAAA,KAAK,WAAW;AACd,gBAAA,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,EAAE,GAAG,EAAE,cAAc,EAAE,OAAO,EAAE,KAAK,CAAC,GAAG,CAAC;YACzE,KAAK,QAAQ,CAAC;YACd,KAAK,SAAS,CAAC;AACf,YAAA;;gBAEE,OAAO,CAAC,IAAI,CAAC,CAAA,wBAAA,EAA2B,KAAK,CAAC,UAAU,CAAA,CAAE,CAAC;AAC3D,gBAAA,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE;;IAE5B;IAEQ,MAAM,WAAW,CAAC,IAAU,EAAE,GAAW,EAAE,OAAiC,EAAE,QAAgB,EAAA;AACpG,QAAA,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,mBAAmB,CAAC,IAAI,EAAE,GAAG,CAAC;AAC7E,QAAA,IAAI,MAAM,GAAG,sBAAsB,GAAG,MAAM,GAAG,aAAa;AAE5D,QAAA,MAAM,cAAc,GAAG,OAAO,CAAC,cAAc,GAAG,QAAQ,CAAC,IAAI,OAAO,CAAC,cAAc,GAAG,MAAM,CAAC;AAC7F,QAAA,MAAM,eAAe,GAAG;AACtB,YAAA,IAAI,OAAO,CAAC,cAAc,IAAI,EAAE,CAAC;AACjC,YAAA,IAAI,cAAc,EAAE,cAAc,IAAI,EAAE,CAAC;SACnB;AAExB,QAAA,MAAM,WAAW,GAAwB;AACvC,YAAA,KAAK,EAAE,4BAA4B;YACnC,GAAG;AACH,YAAA,cAAc,EAAE,MAAM;YACtB,cAAc,EAAE,CAAC,OAAO,CAAC;YACzB,UAAU,EAAE,IAAI,CAAC;SAClB;QAED,MAAM,QAAQ,GAAG,KAAK,CAAC;YACrB,MAAM;AACN,YAAA,OAAO,EAAE,EAAE,GAAG,WAAW,EAAE,GAAG,eAAe,EAAE;AAChD,SAAA,CAAC;QAEF,IAAI,IAAI,GAAG,EAAE;AACb,QAAA,WAAW,MAAM,OAAO,IAAI,QAAQ,EAAE;;AAEpC,YAAA,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,OAAO,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;;YAG5D,MAAM,WAAW,GAAG,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,OAAO,CAAC;YACnD,IAAI,WAAW,EAAE;AACf,gBAAA,IAAI,WAAW,CAAC,IAAI,KAAK,OAAO,EAAE;AAChC,oBAAA,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,gBAAgB,EAAE,EAAE,IAAI,EAAE,WAAW,CAAC,IAAI,EAAE,CAAC;gBACjE;AACA,gBAAA,IAAI,CAAC,YAAY,GAAG,WAAW,CAAC;YAClC;AAEA,YAAA,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,IAAI,OAAO,CAAC,OAAO,EAAE,OAAO,EAAE;gBAC5D,KAAK,MAAM,CAAC,IAAI,OAAO,CAAC,OAAO,CAAC,OAAO,EAAE;oBACvC,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM,IAAI,CAAC,CAAC,IAAI;AAAE,wBAAA,IAAI,IAAI,CAAC,CAAC,IAAI,GAAG,IAAI;gBACxD;YACF;QACF;QAEA,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,EAAE;IAC9B;IAEQ,MAAM,YAAY,CAAC,IAAU,EAAE,GAAW,EAAE,cAA0D,EAAE,OAAiC,EAAE,QAAgB,EAAA;AACjK,QAAA,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,oBAAoB,CAAC,IAAI,EAAE,GAAG,CAAC;AAC9E,QAAA,IAAI,MAAM,GAAG,uBAAuB,GAAG,MAAM,GAAG,aAAa;QAE7D,MAAM,cAAc,GAAG,OAAO,CAAC,cAAc,GAAG,QAAQ,CAAC;AACzD,QAAA,MAAM,eAAe,GAAG;AACtB,YAAA,IAAI,OAAO,CAAC,cAAc,IAAI,EAAE,CAAC;AACjC,YAAA,IAAI,cAAc,EAAE,cAAc,IAAI,EAAE,CAAC;SACnB;AAExB,QAAA,MAAM,WAAW,GAAwB;AACvC,YAAA,KAAK,EAAE,4BAA4B;YACnC,GAAG;YACH,cAAc;YACd,cAAc,EAAE,CAAC,OAAO,CAAC;YACzB,UAAU,EAAE,IAAI,CAAC;SAClB;QAED,MAAM,QAAQ,GAAG,KAAK,CAAC;YACrB,MAAM;AACN,YAAA,OAAO,EAAE,EAAE,GAAG,WAAW,EAAE,GAAG,eAAe,EAAE;AAChD,SAAA,CAAC;QACF,MAAM,OAAO,GAAU,EAAE;AACzB,QAAA,WAAW,MAAM,OAAO,IAAI,QAAQ,EAAE;;AAEpC,YAAA,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,OAAO,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;;YAG5D,MAAM,WAAW,GAAG,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,OAAO,CAAC;YACnD,IAAI,WAAW,EAAE;AACf,gBAAA,IAAI,WAAW,CAAC,IAAI,KAAK,OAAO,EAAE;AAChC,oBAAA,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,iBAAiB,EAAE,EAAE,IAAI,EAAE,WAAW,CAAC,IAAI,EAAE,CAAC;gBAClE;AACA,gBAAA,IAAI,CAAC,YAAY,GAAG,WAAW,CAAC;YAClC;AAEA,YAAA,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC;QACvB;QACA,OAAO,EAAE,OAAO,EAAE;IACpB;AACD;;;;"}
+{"version":3,"file":"stage-executor.js","sources":["../../src/stage-executor.ts"],"sourcesContent":["import { query } from '@anthropic-ai/claude-agent-sdk';\nimport { Logger } from './utils/logger.js';\nimport { ClaudeAdapter } from './adapters/claude/claude-adapter.js';\nimport { AgentRegistry } from './agent-registry.js';\nimport type { AgentEvent, Task, McpServerConfig } from './types.js';\nimport type { WorkflowStage, WorkflowStageExecutionResult, WorkflowExecutionOptions } from './workflow-types.js';\nimport { RESEARCH_SYSTEM_PROMPT } from './agents/research.js';\nimport { PLANNING_SYSTEM_PROMPT } from './agents/planning.js';\nimport { EXECUTION_SYSTEM_PROMPT } from './agents/execution.js';\nimport { PromptBuilder } from './prompt-builder.js';\n\nexport class StageExecutor {\n  private registry: AgentRegistry;\n  private logger: Logger;\n  private adapter: ClaudeAdapter;\n  private promptBuilder: PromptBuilder;\n  private eventHandler?: (event: AgentEvent) => void;\n  private mcpServers?: Record<string, McpServerConfig>;\n\n  constructor(\n    registry: AgentRegistry,\n    logger: Logger,\n    promptBuilder?: PromptBuilder,\n    eventHandler?: (event: AgentEvent) => void,\n    mcpServers?: Record<string, McpServerConfig>,\n  ) {\n    this.registry = registry;\n    this.logger = logger.child('StageExecutor');\n    this.adapter = new ClaudeAdapter();\n    this.promptBuilder = promptBuilder || new PromptBuilder({\n      getTaskFiles: async () => [],\n      generatePlanTemplate: async () => '',\n      logger,\n    });\n    this.eventHandler = eventHandler;\n    this.mcpServers = mcpServers;\n  }\n\n  setEventHandler(handler?: (event: AgentEvent) => void): void {\n    this.eventHandler = handler;\n  }\n\n  async execute(task: Task, stage: WorkflowStage, options: WorkflowExecutionOptions): Promise<WorkflowStageExecutionResult> {\n    const isManual = stage.is_manual_only === true;\n    if (isManual) {\n      this.logger.info('Manual stage detected; skipping agent execution', { stage: stage.key });\n      return { results: [] };\n    }\n\n    const inferredAgent = stage.key.toLowerCase().includes('plan') ? 'planning_basic' : 'code_generation';\n    const agentName = stage.agent_name || inferredAgent;\n    const agent = this.registry.getAgent(agentName);\n    if (!agent) {\n      throw new Error(`Unknown agent '${agentName}' for stage '${stage.key}'`);\n    }\n\n    const permissionMode = (options.permissionMode as any) || 'acceptEdits';\n    const cwd = options.repositoryPath || process.cwd();\n\n    switch (agent.agent_type) {\n      case 'research':\n        return this.runResearch(task, cwd, options, stage.key);\n      case 'planning':\n        return this.runPlanning(task, cwd, options, stage.key);\n      case 'execution':\n        return this.runExecution(task, cwd, permissionMode, options, stage.key);\n      case 'review': // TODO: Implement review\n      case 'testing': // TODO: Implement testing\n      default:\n        // throw new Error(`Unsupported agent type: ${agent.agent_type}`);\n        console.warn(`Unsupported agent type: ${agent.agent_type}`);\n        return { results: [] };\n    }\n  }\n\n  private async runResearch(task: Task, cwd: string, options: WorkflowExecutionOptions, stageKey: string): Promise<WorkflowStageExecutionResult> {\n    const contextPrompt = await this.promptBuilder.buildResearchPrompt(task, cwd);\n    let prompt = RESEARCH_SYSTEM_PROMPT + '\\n\\n' + contextPrompt;\n\n    const stageOverrides = options.stageOverrides?.[stageKey] || options.stageOverrides?.['research'];\n    const mergedOverrides = {\n      ...(options.queryOverrides || {}),\n      ...(stageOverrides?.queryOverrides || {}),\n    } as Record<string, any>;\n\n    const baseOptions: Record<string, any> = {\n      model: 'claude-sonnet-4-5-20250929',\n      cwd,\n      permissionMode: 'plan',\n      settingSources: ['local'],\n      mcpServers: this.mcpServers\n    };\n\n    const response = query({\n      prompt,\n      options: { ...baseOptions, ...mergedOverrides },\n    });\n\n    let research = '';\n    for await (const message of response) {\n      // Emit raw SDK event first\n      this.eventHandler?.(this.adapter.createRawSDKEvent(message));\n\n      // Then emit transformed event\n      const transformed = this.adapter.transform(message);\n      if (transformed) {\n        if (transformed.type !== 'token') {\n          this.logger.debug('Research event', { type: transformed.type });\n        }\n        this.eventHandler?.(transformed);\n      }\n\n      if (message.type === 'assistant' && message.message?.content) {\n        for (const c of message.message.content) {\n          if (c.type === 'text' && c.text) research += c.text + '\\n';\n        }\n      }\n    }\n\n    return { plan: research.trim() }; // Return as 'plan' field to match existing interface\n  }\n\n  private async runPlanning(task: Task, cwd: string, options: WorkflowExecutionOptions, stageKey: string): Promise<WorkflowStageExecutionResult> {\n    const contextPrompt = await this.promptBuilder.buildPlanningPrompt(task, cwd);\n    let prompt = PLANNING_SYSTEM_PROMPT + '\\n\\n' + contextPrompt;\n\n    const stageOverrides = options.stageOverrides?.[stageKey] || options.stageOverrides?.['plan'];\n    const mergedOverrides = {\n      ...(options.queryOverrides || {}),\n      ...(stageOverrides?.queryOverrides || {}),\n    } as Record<string, any>;\n\n    const baseOptions: Record<string, any> = {\n      model: 'claude-sonnet-4-5-20250929',\n      cwd,\n      permissionMode: 'plan',\n      settingSources: ['local'],\n      mcpServers: this.mcpServers\n    };\n\n    const response = query({\n      prompt,\n      options: { ...baseOptions, ...mergedOverrides },\n    });\n\n    let plan = '';\n    for await (const message of response) {\n      // Emit raw SDK event first\n      this.eventHandler?.(this.adapter.createRawSDKEvent(message));\n\n      // Then emit transformed event\n      const transformed = this.adapter.transform(message);\n      if (transformed) {\n        if (transformed.type !== 'token') {\n          this.logger.debug('Planning event', { type: transformed.type });\n        }\n        this.eventHandler?.(transformed);\n      }\n\n      if (message.type === 'assistant' && message.message?.content) {\n        for (const c of message.message.content) {\n          if (c.type === 'text' && c.text) plan += c.text + '\\n';\n        }\n      }\n    }\n\n    return { plan: plan.trim() };\n  }\n\n  private async runExecution(task: Task, cwd: string, permissionMode: WorkflowExecutionOptions['permissionMode'], options: WorkflowExecutionOptions, stageKey: string): Promise<WorkflowStageExecutionResult> {\n    const contextPrompt = await this.promptBuilder.buildExecutionPrompt(task, cwd);\n    let prompt = EXECUTION_SYSTEM_PROMPT + '\\n\\n' + contextPrompt;\n\n    const stageOverrides = options.stageOverrides?.[stageKey];\n    const mergedOverrides = {\n      ...(options.queryOverrides || {}),\n      ...(stageOverrides?.queryOverrides || {}),\n    } as Record<string, any>;\n\n    const baseOptions: Record<string, any> = {\n      model: 'claude-sonnet-4-5-20250929',\n      cwd,\n      permissionMode,\n      settingSources: ['local'],\n      mcpServers: this.mcpServers\n    };\n\n    const response = query({\n      prompt,\n      options: { ...baseOptions, ...mergedOverrides },\n    });\n    const results: any[] = [];\n    for await (const message of response) {\n      // Emit raw SDK event first\n      this.eventHandler?.(this.adapter.createRawSDKEvent(message));\n\n      // Then emit transformed event\n      const transformed = this.adapter.transform(message);\n      if (transformed) {\n        if (transformed.type !== 'token') {\n          this.logger.debug('Execution event', { type: transformed.type });\n        }\n        this.eventHandler?.(transformed);\n      }\n\n      results.push(message);\n    }\n    return { results };\n  }\n}\n"],"names":[],"mappings":";;;;;;;;MAWa,aAAa,CAAA;AAChB,IAAA,QAAQ;AACR,IAAA,MAAM;AACN,IAAA,OAAO;AACP,IAAA,aAAa;AACb,IAAA,YAAY;AACZ,IAAA,UAAU;IAElB,WAAA,CACE,QAAuB,EACvB,MAAc,EACd,aAA6B,EAC7B,YAA0C,EAC1C,UAA4C,EAAA;AAE5C,QAAA,IAAI,CAAC,QAAQ,GAAG,QAAQ;QACxB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC;AAC3C,QAAA,IAAI,CAAC,OAAO,GAAG,IAAI,aAAa,EAAE;AAClC,QAAA,IAAI,CAAC,aAAa,GAAG,aAAa,IAAI,IAAI,aAAa,CAAC;AACtD,YAAA,YAAY,EAAE,YAAY,EAAE;AAC5B,YAAA,oBAAoB,EAAE,YAAY,EAAE;YACpC,MAAM;AACP,SAAA,CAAC;AACF,QAAA,IAAI,CAAC,YAAY,GAAG,YAAY;AAChC,QAAA,IAAI,CAAC,UAAU,GAAG,UAAU;IAC9B;AAEA,IAAA,eAAe,CAAC,OAAqC,EAAA;AACnD,QAAA,IAAI,CAAC,YAAY,GAAG,OAAO;IAC7B;AAEA,IAAA,MAAM,OAAO,CAAC,IAAU,EAAE,KAAoB,EAAE,OAAiC,EAAA;AAC/E,QAAA,MAAM,QAAQ,GAAG,KAAK,CAAC,cAAc,KAAK,IAAI;QAC9C,IAAI,QAAQ,EAAE;AACZ,YAAA,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,iDAAiD,EAAE,EAAE,KAAK,EAAE,KAAK,CAAC,GAAG,EAAE,CAAC;AACzF,YAAA,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE;QACxB;QAEA,MAAM,aAAa,GAAG,KAAK,CAAC,GAAG,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,GAAG,gBAAgB,GAAG,iBAAiB;AACrG,QAAA,MAAM,SAAS,GAAG,KAAK,CAAC,UAAU,IAAI,aAAa;QACnD,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAAC;QAC/C,IAAI,CAAC,KAAK,EAAE;YACV,MAAM,IAAI,KAAK,CAAC,CAAA,eAAA,EAAkB,SAAS,CAAA,aAAA,EAAgB,KAAK,CAAC,GAAG,CAAA,CAAA,CAAG,CAAC;QAC1E;AAEA,QAAA,MAAM,cAAc,GAAI,OAAO,CAAC,cAAsB,IAAI,aAAa;QACvE,MAAM,GAAG,GAAG,OAAO,CAAC,cAAc,IAAI,OAAO,CAAC,GAAG,EAAE;AAEnD,QAAA,QAAQ,KAAK,CAAC,UAAU;AACtB,YAAA,KAAK,UAAU;AACb,gBAAA,OAAO,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,GAAG,EAAE,OAAO,EAAE,KAAK,CAAC,GAAG,CAAC;AACxD,YAAA,KAAK,UAAU;AACb,gBAAA,OAAO,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,GAAG,EAAE,OAAO,EAAE,KAAK,CAAC,GAAG,CAAC;AACxD,YAAA,KAAK,WAAW;AACd,gBAAA,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,EAAE,GAAG,EAAE,cAAc,EAAE,OAAO,EAAE,KAAK,CAAC,GAAG,CAAC;YACzE,KAAK,QAAQ,CAAC;YACd,KAAK,SAAS,CAAC;AACf,YAAA;;gBAEE,OAAO,CAAC,IAAI,CAAC,CAAA,wBAAA,EAA2B,KAAK,CAAC,UAAU,CAAA,CAAE,CAAC;AAC3D,gBAAA,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE;;IAE5B;IAEQ,MAAM,WAAW,CAAC,IAAU,EAAE,GAAW,EAAE,OAAiC,EAAE,QAAgB,EAAA;AACpG,QAAA,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,mBAAmB,CAAC,IAAI,EAAE,GAAG,CAAC;AAC7E,QAAA,IAAI,MAAM,GAAG,sBAAsB,GAAG,MAAM,GAAG,aAAa;AAE5D,QAAA,MAAM,cAAc,GAAG,OAAO,CAAC,cAAc,GAAG,QAAQ,CAAC,IAAI,OAAO,CAAC,cAAc,GAAG,UAAU,CAAC;AACjG,QAAA,MAAM,eAAe,GAAG;AACtB,YAAA,IAAI,OAAO,CAAC,cAAc,IAAI,EAAE,CAAC;AACjC,YAAA,IAAI,cAAc,EAAE,cAAc,IAAI,EAAE,CAAC;SACnB;AAExB,QAAA,MAAM,WAAW,GAAwB;AACvC,YAAA,KAAK,EAAE,4BAA4B;YACnC,GAAG;AACH,YAAA,cAAc,EAAE,MAAM;YACtB,cAAc,EAAE,CAAC,OAAO,CAAC;YACzB,UAAU,EAAE,IAAI,CAAC;SAClB;QAED,MAAM,QAAQ,GAAG,KAAK,CAAC;YACrB,MAAM;AACN,YAAA,OAAO,EAAE,EAAE,GAAG,WAAW,EAAE,GAAG,eAAe,EAAE;AAChD,SAAA,CAAC;QAEF,IAAI,QAAQ,GAAG,EAAE;AACjB,QAAA,WAAW,MAAM,OAAO,IAAI,QAAQ,EAAE;;AAEpC,YAAA,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,OAAO,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;;YAG5D,MAAM,WAAW,GAAG,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,OAAO,CAAC;YACnD,IAAI,WAAW,EAAE;AACf,gBAAA,IAAI,WAAW,CAAC,IAAI,KAAK,OAAO,EAAE;AAChC,oBAAA,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,gBAAgB,EAAE,EAAE,IAAI,EAAE,WAAW,CAAC,IAAI,EAAE,CAAC;gBACjE;AACA,gBAAA,IAAI,CAAC,YAAY,GAAG,WAAW,CAAC;YAClC;AAEA,YAAA,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,IAAI,OAAO,CAAC,OAAO,EAAE,OAAO,EAAE;gBAC5D,KAAK,MAAM,CAAC,IAAI,OAAO,CAAC,OAAO,CAAC,OAAO,EAAE;oBACvC,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM,IAAI,CAAC,CAAC,IAAI;AAAE,wBAAA,QAAQ,IAAI,CAAC,CAAC,IAAI,GAAG,IAAI;gBAC5D;YACF;QACF;QAEA,OAAO,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,EAAE,EAAE,CAAC;IACnC;IAEQ,MAAM,WAAW,CAAC,IAAU,EAAE,GAAW,EAAE,OAAiC,EAAE,QAAgB,EAAA;AACpG,QAAA,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,mBAAmB,CAAC,IAAI,EAAE,GAAG,CAAC;AAC7E,QAAA,IAAI,MAAM,GAAG,sBAAsB,GAAG,MAAM,GAAG,aAAa;AAE5D,QAAA,MAAM,cAAc,GAAG,OAAO,CAAC,cAAc,GAAG,QAAQ,CAAC,IAAI,OAAO,CAAC,cAAc,GAAG,MAAM,CAAC;AAC7F,QAAA,MAAM,eAAe,GAAG;AACtB,YAAA,IAAI,OAAO,CAAC,cAAc,IAAI,EAAE,CAAC;AACjC,YAAA,IAAI,cAAc,EAAE,cAAc,IAAI,EAAE,CAAC;SACnB;AAExB,QAAA,MAAM,WAAW,GAAwB;AACvC,YAAA,KAAK,EAAE,4BAA4B;YACnC,GAAG;AACH,YAAA,cAAc,EAAE,MAAM;YACtB,cAAc,EAAE,CAAC,OAAO,CAAC;YACzB,UAAU,EAAE,IAAI,CAAC;SAClB;QAED,MAAM,QAAQ,GAAG,KAAK,CAAC;YACrB,MAAM;AACN,YAAA,OAAO,EAAE,EAAE,GAAG,WAAW,EAAE,GAAG,eAAe,EAAE;AAChD,SAAA,CAAC;QAEF,IAAI,IAAI,GAAG,EAAE;AACb,QAAA,WAAW,MAAM,OAAO,IAAI,QAAQ,EAAE;;AAEpC,YAAA,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,OAAO,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;;YAG5D,MAAM,WAAW,GAAG,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,OAAO,CAAC;YACnD,IAAI,WAAW,EAAE;AACf,gBAAA,IAAI,WAAW,CAAC,IAAI,KAAK,OAAO,EAAE;AAChC,oBAAA,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,gBAAgB,EAAE,EAAE,IAAI,EAAE,WAAW,CAAC,IAAI,EAAE,CAAC;gBACjE;AACA,gBAAA,IAAI,CAAC,YAAY,GAAG,WAAW,CAAC;YAClC;AAEA,YAAA,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,IAAI,OAAO,CAAC,OAAO,EAAE,OAAO,EAAE;gBAC5D,KAAK,MAAM,CAAC,IAAI,OAAO,CAAC,OAAO,CAAC,OAAO,EAAE;oBACvC,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM,IAAI,CAAC,CAAC,IAAI;AAAE,wBAAA,IAAI,IAAI,CAAC,CAAC,IAAI,GAAG,IAAI;gBACxD;YACF;QACF;QAEA,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,EAAE;IAC9B;IAEQ,MAAM,YAAY,CAAC,IAAU,EAAE,GAAW,EAAE,cAA0D,EAAE,OAAiC,EAAE,QAAgB,EAAA;AACjK,QAAA,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,oBAAoB,CAAC,IAAI,EAAE,GAAG,CAAC;AAC9E,QAAA,IAAI,MAAM,GAAG,uBAAuB,GAAG,MAAM,GAAG,aAAa;QAE7D,MAAM,cAAc,GAAG,OAAO,CAAC,cAAc,GAAG,QAAQ,CAAC;AACzD,QAAA,MAAM,eAAe,GAAG;AACtB,YAAA,IAAI,OAAO,CAAC,cAAc,IAAI,EAAE,CAAC;AACjC,YAAA,IAAI,cAAc,EAAE,cAAc,IAAI,EAAE,CAAC;SACnB;AAExB,QAAA,MAAM,WAAW,GAAwB;AACvC,YAAA,KAAK,EAAE,4BAA4B;YACnC,GAAG;YACH,cAAc;YACd,cAAc,EAAE,CAAC,OAAO,CAAC;YACzB,UAAU,EAAE,IAAI,CAAC;SAClB;QAED,MAAM,QAAQ,GAAG,KAAK,CAAC;YACrB,MAAM;AACN,YAAA,OAAO,EAAE,EAAE,GAAG,WAAW,EAAE,GAAG,eAAe,EAAE;AAChD,SAAA,CAAC;QACF,MAAM,OAAO,GAAU,EAAE;AACzB,QAAA,WAAW,MAAM,OAAO,IAAI,QAAQ,EAAE;;AAEpC,YAAA,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,OAAO,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;;YAG5D,MAAM,WAAW,GAAG,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,OAAO,CAAC;YACnD,IAAI,WAAW,EAAE;AACf,gBAAA,IAAI,WAAW,CAAC,IAAI,KAAK,OAAO,EAAE;AAChC,oBAAA,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,iBAAiB,EAAE,EAAE,IAAI,EAAE,WAAW,CAAC,IAAI,EAAE,CAAC;gBAClE;AACA,gBAAA,IAAI,CAAC,YAAY,GAAG,WAAW,CAAC;YAClC;AAEA,YAAA,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC;QACvB;QACA,OAAO,EAAE,OAAO,EAAE;IACpB;AACD;;;;"}

package/dist/src/structured-extraction.d.ts

@@ -0,0 +1,22 @@
+import { Logger } from './utils/logger.js';
+export interface ExtractedQuestion {
+    id: string;
+    question: string;
+    options: string[];
+}
+export interface ExtractedQuestionWithAnswer extends ExtractedQuestion {
+    recommendedAnswer: string;
+    justification: string;
+}
+export interface StructuredExtractor {
+    extractQuestions(researchContent: string): Promise<ExtractedQuestion[]>;
+    extractQuestionsWithAnswers(researchContent: string): Promise<ExtractedQuestionWithAnswer[]>;
+}
+export declare class OpenAIExtractor implements StructuredExtractor {
+    private client;
+    private logger;
+    constructor(logger?: Logger);
+    extractQuestions(researchContent: string): Promise<ExtractedQuestion[]>;
+    extractQuestionsWithAnswers(researchContent: string): Promise<ExtractedQuestionWithAnswer[]>;
+}
+//# sourceMappingURL=structured-extraction.d.ts.map

package/dist/src/structured-extraction.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"structured-extraction.d.ts","sourceRoot":"","sources":["../../src/structured-extraction.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAE3C,MAAM,WAAW,iBAAiB;IAChC,EAAE,EAAE,MAAM,CAAC;IACX,QAAQ,EAAE,MAAM,CAAC;IACjB,OAAO,EAAE,MAAM,EAAE,CAAC;CACnB;AAED,MAAM,WAAW,2BAA4B,SAAQ,iBAAiB;IACpE,iBAAiB,EAAE,MAAM,CAAC;IAC1B,aAAa,EAAE,MAAM,CAAC;CACvB;AAoDD,MAAM,WAAW,mBAAmB;IAClC,gBAAgB,CAAC,eAAe,EAAE,MAAM,GAAG,OAAO,CAAC,iBAAiB,EAAE,CAAC,CAAC;IACxE,2BAA2B,CAAC,eAAe,EAAE,MAAM,GAAG,OAAO,CAAC,2BAA2B,EAAE,CAAC,CAAC;CAC9F;AAED,qBAAa,eAAgB,YAAW,mBAAmB;IACzD,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,MAAM,CAAS;gBAEX,MAAM,CAAC,EAAE,MAAM;IAUrB,gBAAgB,CAAC,eAAe,EAAE,MAAM,GAAG,OAAO,CAAC,iBAAiB,EAAE,CAAC;IAyCvE,2BAA2B,CAC/B,eAAe,EAAE,MAAM,GACtB,OAAO,CAAC,2BAA2B,EAAE,CAAC;CAwC1C"}

package/dist/src/structured-extraction.js

@@ -0,0 +1,136 @@
+import OpenAI from 'openai';
+import { Logger } from './utils/logger.js';
+
+const questionsOnlySchema = {
+    type: 'object',
+    properties: {
+        questions: {
+            type: 'array',
+            items: {
+                type: 'object',
+                properties: {
+                    id: { type: 'string' },
+                    question: { type: 'string' },
+                    options: {
+                        type: 'array',
+                        items: { type: 'string' }
+                    }
+                },
+                required: ['id', 'question', 'options'],
+                additionalProperties: false
+            }
+        }
+    },
+    required: ['questions'],
+    additionalProperties: false
+};
+const questionsWithAnswersSchema = {
+    type: 'object',
+    properties: {
+        questions: {
+            type: 'array',
+            items: {
+                type: 'object',
+                properties: {
+                    id: { type: 'string' },
+                    question: { type: 'string' },
+                    options: {
+                        type: 'array',
+                        items: { type: 'string' }
+                    },
+                    recommendedAnswer: { type: 'string' },
+                    justification: { type: 'string' }
+                },
+                required: ['id', 'question', 'options', 'recommendedAnswer', 'justification'],
+                additionalProperties: false
+            }
+        }
+    },
+    required: ['questions'],
+    additionalProperties: false
+};
+class OpenAIExtractor {
+    client;
+    logger;
+    constructor(logger) {
+        const apiKey = process.env.OPENAI_API_KEY;
+        if (!apiKey) {
+            throw new Error('OPENAI_API_KEY environment variable is required for structured extraction');
+        }
+        this.client = new OpenAI({ apiKey });
+        this.logger = logger || new Logger({ debug: false, prefix: '[OpenAIExtractor]' });
+    }
+    async extractQuestions(researchContent) {
+        this.logger.debug('Extracting questions from research content', {
+            contentLength: researchContent.length,
+        });
+        const completion = await this.client.chat.completions.create({
+            model: 'gpt-4o-mini',
+            messages: [
+                {
+                    role: 'system',
+                    content: 'Extract the research questions from the provided markdown. Return a JSON object matching the schema.',
+                },
+                {
+                    role: 'user',
+                    content: researchContent,
+                },
+            ],
+            response_format: {
+                type: 'json_schema',
+                json_schema: {
+                    name: 'questions',
+                    strict: true,
+                    schema: questionsOnlySchema,
+                },
+            },
+        });
+        const content = completion.choices[0].message.content;
+        if (!content) {
+            throw new Error('No content in OpenAI response');
+        }
+        const parsed = JSON.parse(content);
+        this.logger.info('Successfully extracted questions', {
+            questionCount: parsed.questions.length,
+        });
+        return parsed.questions;
+    }
+    async extractQuestionsWithAnswers(researchContent) {
+        this.logger.debug('Extracting questions with recommended answers', {
+            contentLength: researchContent.length,
+        });
+        const completion = await this.client.chat.completions.create({
+            model: 'gpt-4o-mini',
+            messages: [
+                {
+                    role: 'system',
+                    content: 'Extract the research questions from the markdown and provide recommended answers based on the analysis. For each question, include a recommendedAnswer (the letter: a, b, c, etc.) and a brief justification. Return a JSON object matching the schema.',
+                },
+                {
+                    role: 'user',
+                    content: researchContent,
+                },
+            ],
+            response_format: {
+                type: 'json_schema',
+                json_schema: {
+                    name: 'questions_with_answers',
+                    strict: true,
+                    schema: questionsWithAnswersSchema,
+                },
+            },
+        });
+        const content = completion.choices[0].message.content;
+        if (!content) {
+            throw new Error('No content in OpenAI response');
+        }
+        const parsed = JSON.parse(content);
+        this.logger.info('Successfully extracted questions with answers', {
+            questionCount: parsed.questions.length,
+        });
+        return parsed.questions;
+    }
+}
+
+export { OpenAIExtractor };
+//# sourceMappingURL=structured-extraction.js.map

package/dist/src/structured-extraction.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"structured-extraction.js","sources":["../../src/structured-extraction.ts"],"sourcesContent":["import OpenAI from 'openai';\nimport { Logger } from './utils/logger.js';\n\nexport interface ExtractedQuestion {\n  id: string;\n  question: string;\n  options: string[];\n}\n\nexport interface ExtractedQuestionWithAnswer extends ExtractedQuestion {\n  recommendedAnswer: string;\n  justification: string;\n}\n\nconst questionsOnlySchema = {\n  type: 'object',\n  properties: {\n    questions: {\n      type: 'array',\n      items: {\n        type: 'object',\n        properties: {\n          id: { type: 'string' },\n          question: { type: 'string' },\n          options: {\n            type: 'array',\n            items: { type: 'string' }\n          }\n        },\n        required: ['id', 'question', 'options'],\n        additionalProperties: false\n      }\n    }\n  },\n  required: ['questions'],\n  additionalProperties: false\n};\n\nconst questionsWithAnswersSchema = {\n  type: 'object',\n  properties: {\n    questions: {\n      type: 'array',\n      items: {\n        type: 'object',\n        properties: {\n          id: { type: 'string' },\n          question: { type: 'string' },\n          options: {\n            type: 'array',\n            items: { type: 'string' }\n          },\n          recommendedAnswer: { type: 'string' },\n          justification: { type: 'string' }\n        },\n        required: ['id', 'question', 'options', 'recommendedAnswer', 'justification'],\n        additionalProperties: false\n      }\n    }\n  },\n  required: ['questions'],\n  additionalProperties: false\n};\n\nexport interface StructuredExtractor {\n  extractQuestions(researchContent: string): Promise<ExtractedQuestion[]>;\n  extractQuestionsWithAnswers(researchContent: string): Promise<ExtractedQuestionWithAnswer[]>;\n}\n\nexport class OpenAIExtractor implements StructuredExtractor {\n  private client: OpenAI;\n  private logger: Logger;\n\n  constructor(logger?: Logger) {\n    const apiKey = process.env.OPENAI_API_KEY;\n    if (!apiKey) {\n      throw new Error('OPENAI_API_KEY environment variable is required for structured extraction');\n    }\n    \n    this.client = new OpenAI({ apiKey });\n    this.logger = logger || new Logger({ debug: false, prefix: '[OpenAIExtractor]' });\n  }\n\n  async extractQuestions(researchContent: string): Promise<ExtractedQuestion[]> {\n    this.logger.debug('Extracting questions from research content', {\n      contentLength: researchContent.length,\n    });\n\n    const completion = await this.client.chat.completions.create({\n      model: 'gpt-4o-mini',\n      messages: [\n        {\n          role: 'system',\n          content: 'Extract the research questions from the provided markdown. Return a JSON object matching the schema.',\n        },\n        {\n          role: 'user',\n          content: researchContent,\n        },\n      ],\n      response_format: {\n        type: 'json_schema',\n        json_schema: {\n          name: 'questions',\n          strict: true,\n          schema: questionsOnlySchema,\n        },\n      },\n    });\n\n    const content = completion.choices[0].message.content;\n    if (!content) {\n      throw new Error('No content in OpenAI response');\n    }\n\n    const parsed = JSON.parse(content) as { questions: ExtractedQuestion[] };\n    \n    this.logger.info('Successfully extracted questions', {\n      questionCount: parsed.questions.length,\n    });\n\n    return parsed.questions;\n  }\n\n  async extractQuestionsWithAnswers(\n    researchContent: string,\n  ): Promise<ExtractedQuestionWithAnswer[]> {\n    this.logger.debug('Extracting questions with recommended answers', {\n      contentLength: researchContent.length,\n    });\n\n    const completion = await this.client.chat.completions.create({\n      model: 'gpt-4o-mini',\n      messages: [\n        {\n          role: 'system',\n          content: 'Extract the research questions from the markdown and provide recommended answers based on the analysis. For each question, include a recommendedAnswer (the letter: a, b, c, etc.) and a brief justification. Return a JSON object matching the schema.',\n        },\n        {\n          role: 'user',\n          content: researchContent,\n        },\n      ],\n      response_format: {\n        type: 'json_schema',\n        json_schema: {\n          name: 'questions_with_answers',\n          strict: true,\n          schema: questionsWithAnswersSchema,\n        },\n      },\n    });\n\n    const content = completion.choices[0].message.content;\n    if (!content) {\n      throw new Error('No content in OpenAI response');\n    }\n\n    const parsed = JSON.parse(content) as { questions: ExtractedQuestionWithAnswer[] };\n    \n    this.logger.info('Successfully extracted questions with answers', {\n      questionCount: parsed.questions.length,\n    });\n\n    return parsed.questions;\n  }\n}\n"],"names":[],"mappings":";;;AAcA,MAAM,mBAAmB,GAAG;AAC1B,IAAA,IAAI,EAAE,QAAQ;AACd,IAAA,UAAU,EAAE;AACV,QAAA,SAAS,EAAE;AACT,YAAA,IAAI,EAAE,OAAO;AACb,YAAA,KAAK,EAAE;AACL,gBAAA,IAAI,EAAE,QAAQ;AACd,gBAAA,UAAU,EAAE;AACV,oBAAA,EAAE,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;AACtB,oBAAA,QAAQ,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;AAC5B,oBAAA,OAAO,EAAE;AACP,wBAAA,IAAI,EAAE,OAAO;AACb,wBAAA,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ;AACxB;AACF,iBAAA;AACD,gBAAA,QAAQ,EAAE,CAAC,IAAI,EAAE,UAAU,EAAE,SAAS,CAAC;AACvC,gBAAA,oBAAoB,EAAE;AACvB;AACF;AACF,KAAA;IACD,QAAQ,EAAE,CAAC,WAAW,CAAC;AACvB,IAAA,oBAAoB,EAAE;CACvB;AAED,MAAM,0BAA0B,GAAG;AACjC,IAAA,IAAI,EAAE,QAAQ;AACd,IAAA,UAAU,EAAE;AACV,QAAA,SAAS,EAAE;AACT,YAAA,IAAI,EAAE,OAAO;AACb,YAAA,KAAK,EAAE;AACL,gBAAA,IAAI,EAAE,QAAQ;AACd,gBAAA,UAAU,EAAE;AACV,oBAAA,EAAE,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;AACtB,oBAAA,QAAQ,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;AAC5B,oBAAA,OAAO,EAAE;AACP,wBAAA,IAAI,EAAE,OAAO;AACb,wBAAA,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ;AACxB,qBAAA;AACD,oBAAA,iBAAiB,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;AACrC,oBAAA,aAAa,EAAE,EAAE,IAAI,EAAE,QAAQ;AAChC,iBAAA;gBACD,QAAQ,EAAE,CAAC,IAAI,EAAE,UAAU,EAAE,SAAS,EAAE,mBAAmB,EAAE,eAAe,CAAC;AAC7E,gBAAA,oBAAoB,EAAE;AACvB;AACF;AACF,KAAA;IACD,QAAQ,EAAE,CAAC,WAAW,CAAC;AACvB,IAAA,oBAAoB,EAAE;CACvB;MAOY,eAAe,CAAA;AAClB,IAAA,MAAM;AACN,IAAA,MAAM;AAEd,IAAA,WAAA,CAAY,MAAe,EAAA;AACzB,QAAA,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,cAAc;QACzC,IAAI,CAAC,MAAM,EAAE;AACX,YAAA,MAAM,IAAI,KAAK,CAAC,2EAA2E,CAAC;QAC9F;QAEA,IAAI,CAAC,MAAM,GAAG,IAAI,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC;AACpC,QAAA,IAAI,CAAC,MAAM,GAAG,MAAM,IAAI,IAAI,MAAM,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,mBAAmB,EAAE,CAAC;IACnF;IAEA,MAAM,gBAAgB,CAAC,eAAuB,EAAA;AAC5C,QAAA,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,4CAA4C,EAAE;YAC9D,aAAa,EAAE,eAAe,CAAC,MAAM;AACtC,SAAA,CAAC;AAEF,QAAA,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;AAC3D,YAAA,KAAK,EAAE,aAAa;AACpB,YAAA,QAAQ,EAAE;AACR,gBAAA;AACE,oBAAA,IAAI,EAAE,QAAQ;AACd,oBAAA,OAAO,EAAE,sGAAsG;AAChH,iBAAA;AACD,gBAAA;AACE,oBAAA,IAAI,EAAE,MAAM;AACZ,oBAAA,OAAO,EAAE,eAAe;AACzB,iBAAA;AACF,aAAA;AACD,YAAA,eAAe,EAAE;AACf,gBAAA,IAAI,EAAE,aAAa;AACnB,gBAAA,WAAW,EAAE;AACX,oBAAA,IAAI,EAAE,WAAW;AACjB,oBAAA,MAAM,EAAE,IAAI;AACZ,oBAAA,MAAM,EAAE,mBAAmB;AAC5B,iBAAA;AACF,aAAA;AACF,SAAA,CAAC;AAEF,QAAA,MAAM,OAAO,GAAG,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO;QACrD,IAAI,CAAC,OAAO,EAAE;AACZ,YAAA,MAAM,IAAI,KAAK,CAAC,+BAA+B,CAAC;QAClD;QAEA,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAuC;AAExE,QAAA,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,kCAAkC,EAAE;AACnD,YAAA,aAAa,EAAE,MAAM,CAAC,SAAS,CAAC,MAAM;AACvC,SAAA,CAAC;QAEF,OAAO,MAAM,CAAC,SAAS;IACzB;IAEA,MAAM,2BAA2B,CAC/B,eAAuB,EAAA;AAEvB,QAAA,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,+CAA+C,EAAE;YACjE,aAAa,EAAE,eAAe,CAAC,MAAM;AACtC,SAAA,CAAC;AAEF,QAAA,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;AAC3D,YAAA,KAAK,EAAE,aAAa;AACpB,YAAA,QAAQ,EAAE;AACR,gBAAA;AACE,oBAAA,IAAI,EAAE,QAAQ;AACd,oBAAA,OAAO,EAAE,yPAAyP;AACnQ,iBAAA;AACD,gBAAA;AACE,oBAAA,IAAI,EAAE,MAAM;AACZ,oBAAA,OAAO,EAAE,eAAe;AACzB,iBAAA;AACF,aAAA;AACD,YAAA,eAAe,EAAE;AACf,gBAAA,IAAI,EAAE,aAAa;AACnB,gBAAA,WAAW,EAAE;AACX,oBAAA,IAAI,EAAE,wBAAwB;AAC9B,oBAAA,MAAM,EAAE,IAAI;AACZ,oBAAA,MAAM,EAAE,0BAA0B;AACnC,iBAAA;AACF,aAAA;AACF,SAAA,CAAC;AAEF,QAAA,MAAM,OAAO,GAAG,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO;QACrD,IAAI,CAAC,OAAO,EAAE;AACZ,YAAA,MAAM,IAAI,KAAK,CAAC,+BAA+B,CAAC;QAClD;QAEA,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAiD;AAElF,QAAA,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,+CAA+C,EAAE;AAChE,YAAA,aAAa,EAAE,MAAM,CAAC,SAAS,CAAC,MAAM;AACvC,SAAA,CAAC;QAEF,OAAO,MAAM,CAAC,SAAS;IACzB;AACD;;;;"}

package/dist/src/task-progress-reporter.d.ts

@@ -2,9 +2,6 @@ import type { Logger } from './utils/logger.js';
 import type { PostHogAPIClient } from './posthog-api.js';
 import type { AgentEvent } from './types.js';
 interface ProgressMetadata {
-    workflowId?: string;
-    workflowRunId?: string;
-    activityId?: string;
     totalSteps?: number;
 }
 /**
@@ -16,13 +13,13 @@ interface ProgressMetadata {
 export declare class TaskProgressReporter {
     private posthogAPI?;
     private logger;
-    private progressRecord?;
+    private taskRun?;
     private taskId?;
     private outputLog;
     private totalSteps?;
     private lastLogEntry?;
     constructor(posthogAPI: PostHogAPIClient | undefined, logger: Logger);
-    get progressId(): string | undefined;
+    get runId(): string | undefined;
     start(taskId: string, metadata?: ProgressMetadata): Promise<void>;
     stageStarted(stageKey: string, stageIndex: number): Promise<void>;
     stageCompleted(stageKey: string, completedStages: number): Promise<void>;

package/dist/src/task-progress-reporter.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"task-progress-reporter.d.ts","sourceRoot":"","sources":["../../src/task-progress-reporter.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,KAAK,EAAE,gBAAgB,EAA0C,MAAM,kBAAkB,CAAC;AACjG,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAE7C,UAAU,gBAAgB;IACxB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAED;;;;;GAKG;AACH,qBAAa,oBAAoB;IAC/B,OAAO,CAAC,UAAU,CAAC,CAAmB;IACtC,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,cAAc,CAAC,CAAqB;IAC5C,OAAO,CAAC,MAAM,CAAC,CAAS;IACxB,OAAO,CAAC,SAAS,CAAgB;IACjC,OAAO,CAAC,UAAU,CAAC,CAAS;IAC5B,OAAO,CAAC,YAAY,CAAC,CAAS;gBAElB,UAAU,EAAE,gBAAgB,GAAG,SAAS,EAAE,MAAM,EAAE,MAAM;IAKpE,IAAI,UAAU,IAAI,MAAM,GAAG,SAAS,CAEnC;IAEK,KAAK,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,GAAE,gBAAqB,GAAG,OAAO,CAAC,IAAI,CAAC;IA2BrE,YAAY,CAAC,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAQjE,cAAc,CAAC,QAAQ,EAAE,MAAM,EAAE,eAAe,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAQxE,aAAa,CAAC,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAIlE,UAAU,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,gBAAgB,GAAG,OAAO,CAAC,IAAI,CAAC;IAI5E,kBAAkB,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAIlE,WAAW,CAAC,QAAQ,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAQ7C,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;IAIzB,IAAI,CAAC,KAAK,EAAE,KAAK,GAAG,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAK1C,SAAS,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAItC,WAAW,CAAC,KAAK,EAAE,UAAU,GAAG,OAAO,CAAC,IAAI,CAAC;YAqErC,MAAM;IA6BpB,OAAO,CAAC,oBAAoB;IAkE5B,OAAO,CAAC,iBAAiB;IAQzB,OAAO,CAAC,mBAAmB;IAqB3B,OAAO,CAAC,qBAAqB;CAK9B"}
+{"version":3,"file":"task-progress-reporter.d.ts","sourceRoot":"","sources":["../../src/task-progress-reporter.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,KAAK,EAAE,gBAAgB,EAAiB,MAAM,kBAAkB,CAAC;AACxE,OAAO,KAAK,EAAE,UAAU,EAAW,MAAM,YAAY,CAAC;AAEtD,UAAU,gBAAgB;IACxB,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAED;;;;;GAKG;AACH,qBAAa,oBAAoB;IAC/B,OAAO,CAAC,UAAU,CAAC,CAAmB;IACtC,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,OAAO,CAAC,CAAU;IAC1B,OAAO,CAAC,MAAM,CAAC,CAAS;IACxB,OAAO,CAAC,SAAS,CAAgB;IACjC,OAAO,CAAC,UAAU,CAAC,CAAS;IAC5B,OAAO,CAAC,YAAY,CAAC,CAAS;gBAElB,UAAU,EAAE,gBAAgB,GAAG,SAAS,EAAE,MAAM,EAAE,MAAM;IAKpE,IAAI,KAAK,IAAI,MAAM,GAAG,SAAS,CAE9B;IAEK,KAAK,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,GAAE,gBAAqB,GAAG,OAAO,CAAC,IAAI,CAAC;IAqBrE,YAAY,CAAC,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAMjE,cAAc,CAAC,QAAQ,EAAE,MAAM,EAAE,eAAe,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAMxE,aAAa,CAAC,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAIlE,UAAU,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,gBAAgB,GAAG,OAAO,CAAC,IAAI,CAAC;IAI5E,kBAAkB,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAIlE,WAAW,CAAC,QAAQ,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAQ7C,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;IAIzB,IAAI,CAAC,KAAK,EAAE,KAAK,GAAG,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAK1C,SAAS,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAItC,WAAW,CAAC,KAAK,EAAE,UAAU,GAAG,OAAO,CAAC,IAAI,CAAC;YAqErC,MAAM;IAoCpB,OAAO,CAAC,oBAAoB;IAkE5B,OAAO,CAAC,iBAAiB;IAQzB,OAAO,CAAC,mBAAmB;IAqB3B,OAAO,CAAC,qBAAqB;CAK9B"}

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

@@ -7,7 +7,7 @@
 class TaskProgressReporter {
     posthogAPI;
     logger;
-    progressRecord;
+    taskRun;
     taskId;
     outputLog = [];
     totalSteps;
@@ -16,8 +16,8 @@ class TaskProgressReporter {
         this.posthogAPI = posthogAPI;
         this.logger = logger.child('TaskProgressReporter');
     }
-    get progressId() {
-        return this.progressRecord?.id;
+    get runId() {
+        return this.taskRun?.id;
     }
     async start(taskId, metadata = {}) {
         if (!this.posthogAPI) {
@@ -26,36 +26,26 @@ class TaskProgressReporter {
         this.taskId = taskId;
         this.totalSteps = metadata.totalSteps;
         try {
-            const record = await this.posthogAPI.createTaskProgress(taskId, {
+            const run = await this.posthogAPI.createTaskRun(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: '',
+                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 });
+            this.taskRun = run;
+            this.outputLog = [];
+            this.logger.debug('Created task run', { taskId, runId: run.id });
         }
         catch (error) {
-            this.logger.warn('Failed to create task progress record', { taskId, error: error.message });
+            this.logger.warn('Failed to create task run', { 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) {
@@ -73,7 +63,7 @@ class TaskProgressReporter {
             : 'No next stage available. Execution halted.');
     }
     async complete() {
-        await this.update({ status: 'completed', completed_steps: this.totalSteps }, 'Workflow execution completed');
+        await this.update({ status: 'completed' }, 'Workflow execution completed');
     }
     async fail(error) {
         const message = typeof error === 'string' ? error : error.message;
@@ -83,7 +73,7 @@ class TaskProgressReporter {
         await this.update({}, line);
     }
     async recordEvent(event) {
-        if (!this.posthogAPI || !this.progressId || !this.taskId) {
+        if (!this.posthogAPI || !this.runId || !this.taskId) {
             return;
         }
         switch (event.type) {
@@ -140,30 +130,38 @@ class TaskProgressReporter {
         }
     }
     async update(update, logLine) {
-        if (!this.posthogAPI || !this.progressId || !this.taskId) {
+        if (!this.posthogAPI || !this.runId || !this.taskId) {
             return;
         }
-        if (logLine) {
-            if (logLine !== this.lastLogEntry) {
-                this.outputLog.push(logLine);
+        // If there's a log line, append it separately using the append_log endpoint
+        if (logLine && logLine !== this.lastLogEntry) {
+            try {
+                await this.posthogAPI.appendTaskRunLog(this.taskId, this.runId, [
+                    { type: 'info', message: 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 append log entry', {
+                    taskId: this.taskId,
+                    runId: this.runId,
+                    error: error.message,
+                });
             }
         }
-        catch (error) {
-            this.logger.warn('Failed to update task progress record', {
-                taskId: this.taskId,
-                progressId: this.progressId,
-                error: error.message,
-            });
+        // Update other fields if provided
+        if (Object.keys(update).length > 0) {
+            try {
+                const run = await this.posthogAPI.updateTaskRun(this.taskId, this.runId, update);
+                this.taskRun = run;
+            }
+            catch (error) {
+                this.logger.warn('Failed to update task run', {
+                    taskId: this.taskId,
+                    runId: this.runId,
+                    error: error.message,
+                });
+            }
         }
     }
     summarizeUserMessage(content) {