@posthog/agent 1.24.2 → 1.25.0

package/src/prompt-builder.ts

@@ -1,27 +1,36 @@
-import type { Task, UrlMention, PostHogResource } from './types.js';
-import type { TemplateVariables } from './template-manager.js';
-import { Logger } from './utils/logger.js';
-import { promises as fs } from 'fs';
-import { join } from 'path';
+import { promises as fs } from "node:fs";
+import { join } from "node:path";
+import type { TemplateVariables } from "./template-manager.js";
+import type {
+  PostHogResource,
+  ResourceType,
+  SupportingFile,
+  Task,
+  UrlMention,
+} from "./types.js";
+import { Logger } from "./utils/logger.js";
 
 export interface PromptBuilderDeps {
-  getTaskFiles: (taskId: string) => Promise<any[]>;
+  getTaskFiles: (taskId: string) => Promise<SupportingFile[]>;
   generatePlanTemplate: (vars: TemplateVariables) => Promise<string>;
-  posthogClient?: { fetchResourceByUrl: (mention: UrlMention) => Promise<PostHogResource> };
+  posthogClient?: {
+    fetchResourceByUrl: (mention: UrlMention) => Promise<PostHogResource>;
+  };
   logger?: Logger;
 }
 
 export class PromptBuilder {
-  private getTaskFiles: PromptBuilderDeps['getTaskFiles'];
-  private generatePlanTemplate: PromptBuilderDeps['generatePlanTemplate'];
-  private posthogClient?: PromptBuilderDeps['posthogClient'];
+  private getTaskFiles: PromptBuilderDeps["getTaskFiles"];
+  private generatePlanTemplate: PromptBuilderDeps["generatePlanTemplate"];
+  private posthogClient?: PromptBuilderDeps["posthogClient"];
   private logger: Logger;
 
   constructor(deps: PromptBuilderDeps) {
     this.getTaskFiles = deps.getTaskFiles;
     this.generatePlanTemplate = deps.generatePlanTemplate;
     this.posthogClient = deps.posthogClient;
-    this.logger = deps.logger || new Logger({ debug: false, prefix: '[PromptBuilder]' });
+    this.logger =
+      deps.logger || new Logger({ debug: false, prefix: "[PromptBuilder]" });
   }
 
   /**
@@ -33,8 +42,10 @@ export class PromptBuilder {
     const paths: string[] = [];
     let match: RegExpExecArray | null;
 
-    while ((match = fileTagRegex.exec(description)) !== null) {
+    match = fileTagRegex.exec(description);
+    while (match !== null) {
       paths.push(match[1]);
+      match = fileTagRegex.exec(description);
     }
 
     return paths;
@@ -43,13 +54,18 @@ export class PromptBuilder {
   /**
    * Read file contents from repository
    */
-  private async readFileContent(repositoryPath: string, filePath: string): Promise<string | null> {
+  private async readFileContent(
+    repositoryPath: string,
+    filePath: string,
+  ): Promise<string | null> {
     try {
       const fullPath = join(repositoryPath, filePath);
-      const content = await fs.readFile(fullPath, 'utf8');
+      const content = await fs.readFile(fullPath, "utf8");
       return content;
     } catch (error) {
-      this.logger.warn(`Failed to read referenced file: ${filePath}`, { error });
+      this.logger.warn(`Failed to read referenced file: ${filePath}`, {
+        error,
+      });
       return null;
     }
   }
@@ -60,30 +76,35 @@ export class PromptBuilder {
    */
   private extractUrlMentions(description: string): UrlMention[] {
     const mentions: UrlMention[] = [];
-    
+
     // PostHog resource mentions: <error id="..." />, <experiment id="..." />, etc.
-    const resourceRegex = /<(error|experiment|insight|feature_flag)\s+id="([^"]+)"\s*\/>/g;
+    const resourceRegex =
+      /<(error|experiment|insight|feature_flag)\s+id="([^"]+)"\s*\/>/g;
     let match: RegExpExecArray | null;
 
-    while ((match = resourceRegex.exec(description)) !== null) {
+    match = resourceRegex.exec(description);
+    while (match !== null) {
       const [, type, id] = match;
       mentions.push({
-        url: '', // Will be reconstructed if needed
-        type: type as any,
+        url: "", // Will be reconstructed if needed
+        type: type as ResourceType,
         id,
-        label: this.generateUrlLabel('', type as any),
+        label: this.generateUrlLabel("", type as ResourceType),
       });
+      match = resourceRegex.exec(description);
     }
 
     // Generic URL mentions: <url href="..." />
     const urlRegex = /<url\s+href="([^"]+)"\s*\/>/g;
-    while ((match = urlRegex.exec(description)) !== null) {
+    match = urlRegex.exec(description);
+    while (match !== null) {
       const [, url] = match;
       mentions.push({
         url,
-        type: 'generic',
-        label: this.generateUrlLabel(url, 'generic'),
+        type: "generic",
+        label: this.generateUrlLabel(url, "generic"),
       });
+      match = urlRegex.exec(description);
     }
 
     return mentions;
@@ -96,21 +117,23 @@ export class PromptBuilder {
     try {
       const urlObj = new URL(url);
       switch (type) {
-        case 'error':
+        case "error": {
           const errorMatch = url.match(/error_tracking\/([a-f0-9-]+)/);
-          return errorMatch ? `Error ${errorMatch[1].slice(0, 8)}...` : 'Error';
-        case 'experiment':
+          return errorMatch ? `Error ${errorMatch[1].slice(0, 8)}...` : "Error";
+        }
+        case "experiment": {
           const expMatch = url.match(/experiments\/(\d+)/);
-          return expMatch ? `Experiment #${expMatch[1]}` : 'Experiment';
-        case 'insight':
-          return 'Insight';
-        case 'feature_flag':
-          return 'Feature Flag';
+          return expMatch ? `Experiment #${expMatch[1]}` : "Experiment";
+        }
+        case "insight":
+          return "Insight";
+        case "feature_flag":
+          return "Feature Flag";
         default:
           return urlObj.hostname;
       }
     } catch {
-      return 'URL';
+      return "URL";
     }
   }
 
@@ -118,7 +141,7 @@ export class PromptBuilder {
    * Process URL references and fetch their content
    */
   private async processUrlReferences(
-    description: string
+    description: string,
   ): Promise<{ description: string; referencedResources: PostHogResource[] }> {
     const urlMentions = this.extractUrlMentions(description);
     const referencedResources: PostHogResource[] = [];
@@ -133,13 +156,15 @@ export class PromptBuilder {
         const resource = await this.posthogClient.fetchResourceByUrl(mention);
         referencedResources.push(resource);
       } catch (error) {
-        this.logger.warn(`Failed to fetch resource from URL: ${mention.url}`, { error });
+        this.logger.warn(`Failed to fetch resource from URL: ${mention.url}`, {
+          error,
+        });
         // Add a placeholder resource for failed fetches
         referencedResources.push({
           type: mention.type,
-          id: mention.id || '',
+          id: mention.id || "",
           url: mention.url,
-          title: mention.label || 'Unknown Resource',
+          title: mention.label || "Unknown Resource",
           content: `Failed to fetch resource from ${mention.url}: ${error}`,
           metadata: {},
         });
@@ -149,20 +174,22 @@ export class PromptBuilder {
     // Replace URL tags with just the label for readability
     let processedDescription = description;
     for (const mention of urlMentions) {
-      if (mention.type === 'generic') {
+      if (mention.type === "generic") {
         // Generic URLs: <url href="..." />
-        const escapedUrl = mention.url.replace(/[.*+?^${}()|[\]\\]/g, '\\$&');
+        const escapedUrl = mention.url.replace(/[.*+?^${}()|[\]\\]/g, "\\$&");
         processedDescription = processedDescription.replace(
-          new RegExp(`<url\\s+href="${escapedUrl}"\\s*/>`, 'g'),
-          `@${mention.label}`
+          new RegExp(`<url\\s+href="${escapedUrl}"\\s*/>`, "g"),
+          `@${mention.label}`,
         );
       } else {
         // PostHog resources: <error id="..." />, <experiment id="..." />, etc.
-        const escapedType = mention.type.replace(/[.*+?^${}()|[\]\\]/g, '\\$&');
-        const escapedId = mention.id ? mention.id.replace(/[.*+?^${}()|[\]\\]/g, '\\$&') : '';
+        const escapedType = mention.type.replace(/[.*+?^${}()|[\]\\]/g, "\\$&");
+        const escapedId = mention.id
+          ? mention.id.replace(/[.*+?^${}()|[\]\\]/g, "\\$&")
+          : "";
         processedDescription = processedDescription.replace(
-          new RegExp(`<${escapedType}\\s+id="${escapedId}"\\s*/>`, 'g'),
-          `@${mention.label}`
+          new RegExp(`<${escapedType}\\s+id="${escapedId}"\\s*/>`, "g"),
+          `@${mention.label}`,
         );
       }
     }
@@ -176,8 +203,11 @@ export class PromptBuilder {
    */
   private async processFileReferences(
     description: string,
-    repositoryPath?: string
-  ): Promise<{ description: string; referencedFiles: Array<{ path: string; content: string }> }> {
+    repositoryPath?: string,
+  ): Promise<{
+    description: string;
+    referencedFiles: Array<{ path: string; content: string }>;
+  }> {
     const filePaths = this.extractFilePaths(description);
     const referencedFiles: Array<{ path: string; content: string }> = [];
 
@@ -196,211 +226,223 @@ export class PromptBuilder {
     // Replace file tags with just the filename for readability
     let processedDescription = description;
     for (const filePath of filePaths) {
-      const fileName = filePath.split('/').pop() || filePath;
+      const fileName = filePath.split("/").pop() || filePath;
       processedDescription = processedDescription.replace(
-        new RegExp(`<file\\s+path="${filePath.replace(/[.*+?^${}()|[\]\\]/g, '\\$&')}"\\s*/>`, 'g'),
-        `@${fileName}`
+        new RegExp(
+          `<file\\s+path="${filePath.replace(/[.*+?^${}()|[\]\\]/g, "\\$&")}"\\s*/>`,
+          "g",
+        ),
+        `@${fileName}`,
       );
     }
 
     return { description: processedDescription, referencedFiles };
   }
 
-  async buildResearchPrompt(task: Task, repositoryPath?: string): Promise<string> {
+  async buildResearchPrompt(
+    task: Task,
+    repositoryPath?: string,
+  ): Promise<string> {
     // Process file references in description
-    const { description: descriptionAfterFiles, referencedFiles } = await this.processFileReferences(
-      task.description,
-      repositoryPath
-    );
+    const { description: descriptionAfterFiles, referencedFiles } =
+      await this.processFileReferences(task.description, repositoryPath);
 
     // Process URL references in description
-    const { description: processedDescription, referencedResources } = await this.processUrlReferences(
-      descriptionAfterFiles
-    );
+    const { description: processedDescription, referencedResources } =
+      await this.processUrlReferences(descriptionAfterFiles);
 
-    let prompt = '<task>\n';
+    let prompt = "<task>\n";
     prompt += `<title>${task.title}</title>\n`;
     prompt += `<description>${processedDescription}</description>\n`;
 
-    if ((task as any).primary_repository) {
-      prompt += `<repository>${(task as any).primary_repository}</repository>\n`;
+    if (task.repository) {
+      prompt += `<repository>${task.repository}</repository>\n`;
     }
-    prompt += '</task>\n';
+    prompt += "</task>\n";
 
     // Add referenced files from @ mentions
     if (referencedFiles.length > 0) {
-      prompt += '\n<referenced_files>\n';
+      prompt += "\n<referenced_files>\n";
       for (const file of referencedFiles) {
         prompt += `<file path="${file.path}">\n\`\`\`\n${file.content}\n\`\`\`\n</file>\n`;
       }
-      prompt += '</referenced_files>\n';
+      prompt += "</referenced_files>\n";
     }
 
     // Add referenced resources from URL mentions
     if (referencedResources.length > 0) {
-      prompt += '\n<referenced_resources>\n';
+      prompt += "\n<referenced_resources>\n";
       for (const resource of referencedResources) {
         prompt += `<resource type="${resource.type}" url="${resource.url}">\n`;
         prompt += `<title>${resource.title}</title>\n`;
         prompt += `<content>${resource.content}</content>\n`;
-        prompt += '</resource>\n';
+        prompt += "</resource>\n";
       }
-      prompt += '</referenced_resources>\n';
+      prompt += "</referenced_resources>\n";
     }
 
     try {
       const taskFiles = await this.getTaskFiles(task.id);
-      const contextFiles = taskFiles.filter((f: any) => f.type === 'context' || f.type === 'reference');
+      const contextFiles = taskFiles.filter(
+        (f: SupportingFile) => f.type === "context" || f.type === "reference",
+      );
       if (contextFiles.length > 0) {
-        prompt += '\n<supporting_files>\n';
+        prompt += "\n<supporting_files>\n";
         for (const file of contextFiles) {
           prompt += `<file name="${file.name}" type="${file.type}">\n${file.content}\n</file>\n`;
         }
-        prompt += '</supporting_files>\n';
+        prompt += "</supporting_files>\n";
       }
-    } catch (error) {
-      this.logger.debug('No existing task files found for research', { taskId: task.id });
+    } catch (_error) {
+      this.logger.debug("No existing task files found for research", {
+        taskId: task.id,
+      });
     }
 
     return prompt;
   }
 
-  async buildPlanningPrompt(task: Task, repositoryPath?: string): Promise<string> {
+  async buildPlanningPrompt(
+    task: Task,
+    repositoryPath?: string,
+  ): Promise<string> {
     // Process file references in description
-    const { description: descriptionAfterFiles, referencedFiles } = await this.processFileReferences(
-      task.description,
-      repositoryPath
-    );
+    const { description: descriptionAfterFiles, referencedFiles } =
+      await this.processFileReferences(task.description, repositoryPath);
 
     // Process URL references in description
-    const { description: processedDescription, referencedResources } = await this.processUrlReferences(
-      descriptionAfterFiles
-    );
+    const { description: processedDescription, referencedResources } =
+      await this.processUrlReferences(descriptionAfterFiles);
 
-    let prompt = '<task>\n';
+    let prompt = "<task>\n";
     prompt += `<title>${task.title}</title>\n`;
     prompt += `<description>${processedDescription}</description>\n`;
 
-    if ((task as any).primary_repository) {
-      prompt += `<repository>${(task as any).primary_repository}</repository>\n`;
+    if (task.repository) {
+      prompt += `<repository>${task.repository}</repository>\n`;
     }
-    prompt += '</task>\n';
+    prompt += "</task>\n";
 
     // Add referenced files from @ mentions
     if (referencedFiles.length > 0) {
-      prompt += '\n<referenced_files>\n';
+      prompt += "\n<referenced_files>\n";
       for (const file of referencedFiles) {
         prompt += `<file path="${file.path}">\n\`\`\`\n${file.content}\n\`\`\`\n</file>\n`;
       }
-      prompt += '</referenced_files>\n';
+      prompt += "</referenced_files>\n";
     }
 
     // Add referenced resources from URL mentions
     if (referencedResources.length > 0) {
-      prompt += '\n<referenced_resources>\n';
+      prompt += "\n<referenced_resources>\n";
       for (const resource of referencedResources) {
         prompt += `<resource type="${resource.type}" url="${resource.url}">\n`;
         prompt += `<title>${resource.title}</title>\n`;
         prompt += `<content>${resource.content}</content>\n`;
-        prompt += '</resource>\n';
+        prompt += "</resource>\n";
       }
-      prompt += '</referenced_resources>\n';
+      prompt += "</referenced_resources>\n";
     }
 
     try {
       const taskFiles = await this.getTaskFiles(task.id);
-      const contextFiles = taskFiles.filter((f: any) => f.type === 'context' || f.type === 'reference');
+      const contextFiles = taskFiles.filter(
+        (f: SupportingFile) => f.type === "context" || f.type === "reference",
+      );
       if (contextFiles.length > 0) {
-        prompt += '\n<supporting_files>\n';
+        prompt += "\n<supporting_files>\n";
         for (const file of contextFiles) {
           prompt += `<file name="${file.name}" type="${file.type}">\n${file.content}\n</file>\n`;
         }
-        prompt += '</supporting_files>\n';
+        prompt += "</supporting_files>\n";
       }
-    } catch (error) {
-      this.logger.debug('No existing task files found for planning', { taskId: task.id });
+    } catch (_error) {
+      this.logger.debug("No existing task files found for planning", {
+        taskId: task.id,
+      });
     }
 
     const templateVariables = {
       task_id: task.id,
       task_title: task.title,
       task_description: processedDescription,
-      date: new Date().toISOString().split('T')[0],
-      repository: ((task as any).primary_repository || '') as string,
+      date: new Date().toISOString().split("T")[0],
+      repository: task.repository || "",
     };
 
     const planTemplate = await this.generatePlanTemplate(templateVariables);
 
-    prompt += '\n<instructions>\n';
-    prompt += 'Analyze the codebase and create a detailed implementation plan. Use the template structure below, filling each section with specific, actionable information.\n';
-    prompt += '</instructions>\n\n';
-    prompt += '<plan_template>\n';
+    prompt += "\n<instructions>\n";
+    prompt +=
+      "Analyze the codebase and create a detailed implementation plan. Use the template structure below, filling each section with specific, actionable information.\n";
+    prompt += "</instructions>\n\n";
+    prompt += "<plan_template>\n";
     prompt += planTemplate;
-    prompt += '\n</plan_template>';
+    prompt += "\n</plan_template>";
 
     return prompt;
   }
 
-  async buildExecutionPrompt(task: Task, repositoryPath?: string): Promise<string> {
+  async buildExecutionPrompt(
+    task: Task,
+    repositoryPath?: string,
+  ): Promise<string> {
     // Process file references in description
-    const { description: descriptionAfterFiles, referencedFiles } = await this.processFileReferences(
-      task.description,
-      repositoryPath
-    );
+    const { description: descriptionAfterFiles, referencedFiles } =
+      await this.processFileReferences(task.description, repositoryPath);
 
     // Process URL references in description
-    const { description: processedDescription, referencedResources } = await this.processUrlReferences(
-      descriptionAfterFiles
-    );
+    const { description: processedDescription, referencedResources } =
+      await this.processUrlReferences(descriptionAfterFiles);
 
-    let prompt = '<task>\n';
+    let prompt = "<task>\n";
     prompt += `<title>${task.title}</title>\n`;
     prompt += `<description>${processedDescription}</description>\n`;
 
-    if ((task as any).primary_repository) {
-      prompt += `<repository>${(task as any).primary_repository}</repository>\n`;
+    if (task.repository) {
+      prompt += `<repository>${task.repository}</repository>\n`;
     }
-    prompt += '</task>\n';
+    prompt += "</task>\n";
 
     // Add referenced files from @ mentions
     if (referencedFiles.length > 0) {
-      prompt += '\n<referenced_files>\n';
+      prompt += "\n<referenced_files>\n";
       for (const file of referencedFiles) {
         prompt += `<file path="${file.path}">\n\`\`\`\n${file.content}\n\`\`\`\n</file>\n`;
       }
-      prompt += '</referenced_files>\n';
+      prompt += "</referenced_files>\n";
     }
 
     // Add referenced resources from URL mentions
     if (referencedResources.length > 0) {
-      prompt += '\n<referenced_resources>\n';
+      prompt += "\n<referenced_resources>\n";
       for (const resource of referencedResources) {
         prompt += `<resource type="${resource.type}" url="${resource.url}">\n`;
         prompt += `<title>${resource.title}</title>\n`;
         prompt += `<content>${resource.content}</content>\n`;
-        prompt += '</resource>\n';
+        prompt += "</resource>\n";
       }
-      prompt += '</referenced_resources>\n';
+      prompt += "</referenced_resources>\n";
     }
 
     try {
       const taskFiles = await this.getTaskFiles(task.id);
-      const hasPlan = taskFiles.some((f: any) => f.type === 'plan');
-      const todosFile = taskFiles.find((f: any) => f.name === 'todos.json');
+      const hasPlan = taskFiles.some((f: SupportingFile) => f.type === "plan");
+      const todosFile = taskFiles.find(
+        (f: SupportingFile) => f.name === "todos.json",
+      );
 
       if (taskFiles.length > 0) {
-        prompt += '\n<context>\n';
+        prompt += "\n<context>\n";
         for (const file of taskFiles) {
-          if (file.type === 'plan') {
+          if (file.type === "plan") {
             prompt += `<plan>\n${file.content}\n</plan>\n`;
-          } else if (file.name === 'todos.json') {
-            // skip - we do this below
-            continue;
+          } else if (file.name === "todos.json") {
           } else {
             prompt += `<file name="${file.name}" type="${file.type}">\n${file.content}\n</file>\n`;
           }
         }
-        prompt += '</context>\n';
+        prompt += "</context>\n";
       }
 
       // Add todos context if resuming work
@@ -408,37 +450,48 @@ export class PromptBuilder {
         try {
           const todos = JSON.parse(todosFile.content);
           if (todos.items && todos.items.length > 0) {
-            prompt += '\n<previous_todos>\n';
-            prompt += 'You previously created the following todo list for this task:\n\n';
+            prompt += "\n<previous_todos>\n";
+            prompt +=
+              "You previously created the following todo list for this task:\n\n";
             for (const item of todos.items) {
-              const statusIcon = item.status === 'completed' ? '✓' : item.status === 'in_progress' ? '▶' : '○';
+              const statusIcon =
+                item.status === "completed"
+                  ? "✓"
+                  : item.status === "in_progress"
+                    ? "▶"
+                    : "○";
               prompt += `${statusIcon} [${item.status}] ${item.content}\n`;
             }
             prompt += `\nProgress: ${todos.metadata.completed}/${todos.metadata.total} completed\n`;
-            prompt += '\nYou can reference this list when resuming work or create an updated list as needed.\n';
-            prompt += '</previous_todos>\n';
+            prompt +=
+              "\nYou can reference this list when resuming work or create an updated list as needed.\n";
+            prompt += "</previous_todos>\n";
           }
         } catch (error) {
-          this.logger.debug('Failed to parse todos.json for context', { error });
+          this.logger.debug("Failed to parse todos.json for context", {
+            error,
+          });
         }
       }
 
-      prompt += '\n<instructions>\n';
+      prompt += "\n<instructions>\n";
       if (hasPlan) {
-        prompt += 'Implement the changes described in the execution plan. Follow the plan step-by-step and make the necessary file modifications.\n';
+        prompt +=
+          "Implement the changes described in the execution plan. Follow the plan step-by-step and make the necessary file modifications.\n";
       } else {
-        prompt += 'Implement the changes described in the task. Make the necessary file modifications to complete the task.\n';
+        prompt +=
+          "Implement the changes described in the task. Make the necessary file modifications to complete the task.\n";
       }
-      prompt += '</instructions>';
-    } catch (error) {
-      this.logger.debug('No supporting files found for execution', { taskId: task.id });
-      prompt += '\n<instructions>\n';
-      prompt += 'Implement the changes described in the task.\n';
-      prompt += '</instructions>';
+      prompt += "</instructions>";
+    } catch (_error) {
+      this.logger.debug("No supporting files found for execution", {
+        taskId: task.id,
+      });
+      prompt += "\n<instructions>\n";
+      prompt += "Implement the changes described in the task.\n";
+      prompt += "</instructions>";
     }
 
     return prompt;
   }
 }
-
-