@oneuptime/common 9.2.17 → 9.2.18

package/Server/Utils/AI/AlertAIContextBuilder.ts

@@ -0,0 +1,264 @@
+import ObjectID from "../../../Types/ObjectID";
+import Alert from "../../../Models/DatabaseModels/Alert";
+import AlertStateTimeline from "../../../Models/DatabaseModels/AlertStateTimeline";
+import AlertInternalNote from "../../../Models/DatabaseModels/AlertInternalNote";
+import AlertService from "../../Services/AlertService";
+import AlertStateTimelineService from "../../Services/AlertStateTimelineService";
+import AlertInternalNoteService from "../../Services/AlertInternalNoteService";
+import CaptureSpan from "../Telemetry/CaptureSpan";
+import OneUptimeDate from "../../../Types/Date";
+import SortOrder from "../../../Types/BaseDatabase/SortOrder";
+import { LLMMessage } from "../LLM/LLMService";
+
+export interface AlertContextData {
+  alert: Alert;
+  stateTimeline: Array<AlertStateTimeline>;
+  internalNotes: Array<AlertInternalNote>;
+}
+
+export interface AIGenerationContext {
+  contextText: string;
+  systemPrompt: string;
+  messages: Array<LLMMessage>;
+}
+
+export default class AlertAIContextBuilder {
+  @CaptureSpan()
+  public static async buildAlertContext(data: {
+    alertId: ObjectID;
+  }): Promise<AlertContextData> {
+    const alert: Alert | null = await AlertService.findOneById({
+      id: data.alertId,
+      select: {
+        _id: true,
+        title: true,
+        description: true,
+        createdAt: true,
+        customFields: true,
+        projectId: true,
+        alertSeverity: {
+          name: true,
+          color: true,
+        },
+        currentAlertState: {
+          name: true,
+          color: true,
+        },
+        monitor: {
+          name: true,
+        },
+        labels: {
+          name: true,
+          color: true,
+        },
+        rootCause: true,
+        remediationNotes: true,
+      },
+      props: {
+        isRoot: true,
+      },
+    });
+
+    if (!alert) {
+      throw new Error("Alert not found");
+    }
+
+    // Fetch state timeline
+    const stateTimeline: Array<AlertStateTimeline> =
+      await AlertStateTimelineService.findBy({
+        query: {
+          alertId: data.alertId,
+        },
+        select: {
+          _id: true,
+          createdAt: true,
+          startsAt: true,
+          endsAt: true,
+          rootCause: true,
+          alertState: {
+            name: true,
+            color: true,
+          },
+          createdByUser: {
+            name: true,
+            email: true,
+          },
+        },
+        sort: {
+          startsAt: SortOrder.Ascending,
+        },
+        limit: 100,
+        skip: 0,
+        props: {
+          isRoot: true,
+        },
+      });
+
+    // Fetch internal notes
+    const internalNotes: Array<AlertInternalNote> =
+      await AlertInternalNoteService.findBy({
+        query: {
+          alertId: data.alertId,
+        },
+        select: {
+          _id: true,
+          note: true,
+          createdAt: true,
+          createdByUser: {
+            name: true,
+            email: true,
+          },
+        },
+        sort: {
+          createdAt: SortOrder.Ascending,
+        },
+        limit: 100,
+        skip: 0,
+        props: {
+          isRoot: true,
+        },
+      });
+
+    return {
+      alert,
+      stateTimeline,
+      internalNotes,
+    };
+  }
+
+  @CaptureSpan()
+  public static formatAlertContextForNote(
+    contextData: AlertContextData,
+    template?: string,
+  ): AIGenerationContext {
+    const { alert, stateTimeline, internalNotes } = contextData;
+
+    let contextText: string = "";
+
+    // Basic alert information
+    contextText += "# Alert Information\n\n";
+    contextText += `**Title:** ${alert.title || "N/A"}\n\n`;
+    contextText += `**Description:** ${alert.description || "N/A"}\n\n`;
+    contextText += `**Severity:** ${alert.alertSeverity?.name || "N/A"}\n\n`;
+    contextText += `**Current State:** ${alert.currentAlertState?.name || "N/A"}\n\n`;
+    contextText += `**Created At:** ${alert.createdAt ? OneUptimeDate.getDateAsFormattedString(alert.createdAt) : "N/A"}\n\n`;
+
+    // Affected monitor
+    if (alert.monitor) {
+      contextText += `**Monitor:** ${alert.monitor.name || "N/A"}\n\n`;
+    }
+
+    // Labels
+    if (alert.labels && alert.labels.length > 0) {
+      contextText += "**Labels:** ";
+      contextText += alert.labels
+        .map((l: { name?: string }) => {
+          return l.name;
+        })
+        .join(", ");
+      contextText += "\n\n";
+    }
+
+    // Root cause if available
+    if (alert.rootCause) {
+      contextText += `**Root Cause:** ${alert.rootCause}\n\n`;
+    }
+
+    // Remediation notes if available
+    if (alert.remediationNotes) {
+      contextText += `**Remediation Notes:** ${alert.remediationNotes}\n\n`;
+    }
+
+    // State timeline
+    if (stateTimeline.length > 0) {
+      contextText += "# State Timeline\n\n";
+      for (const timeline of stateTimeline) {
+        const startTime: string = timeline.startsAt
+          ? OneUptimeDate.getDateAsFormattedString(timeline.startsAt)
+          : "N/A";
+        const stateName: string =
+          timeline.alertState?.name?.toString() || "Unknown";
+        const createdBy: string =
+          timeline.createdByUser?.name?.toString() ||
+          timeline.createdByUser?.email?.toString() ||
+          "System";
+
+        contextText += `- **${startTime}**: State changed to **${stateName}** by ${createdBy}\n`;
+        if (timeline.rootCause) {
+          contextText += `  - Root cause noted: ${timeline.rootCause}\n`;
+        }
+      }
+      contextText += "\n";
+    }
+
+    // Internal notes
+    if (internalNotes.length > 0) {
+      contextText += "# Internal Notes (Private)\n\n";
+      for (const note of internalNotes) {
+        const noteTime: string = note.createdAt
+          ? OneUptimeDate.getDateAsFormattedString(note.createdAt)
+          : "N/A";
+        const createdBy: string =
+          note.createdByUser?.name?.toString() ||
+          note.createdByUser?.email?.toString() ||
+          "Unknown";
+
+        contextText += `**[${noteTime}] ${createdBy}:**\n`;
+        contextText += `${note.note || "N/A"}\n\n`;
+      }
+    }
+
+    // System prompt for note generation (alerts only have internal notes)
+    let systemPrompt: string;
+
+    if (template) {
+      systemPrompt = `You are an expert Site Reliability Engineer (SRE). Your task is to fill in an internal alert note template based on the provided alert data.
+
+CRITICAL INSTRUCTIONS:
+- You MUST use ONLY the exact template structure provided below
+- Fill in each section of the template with relevant information from the alert data
+- Do NOT add any new sections, headers, or content that is not part of the template
+- Do NOT add introductions, conclusions, or any text outside the template structure
+- Be technical and detailed - this is for the internal team
+- Include relevant technical details, observations, and analysis
+
+TEMPLATE TO FILL (use this exact structure):
+
+${template}`;
+    } else {
+      systemPrompt = `You are an expert Site Reliability Engineer (SRE). Your task is to generate an internal alert note for the team.
+
+The note should:
+1. Provide technical details about the alert and its current status
+2. Document observations, findings, or actions taken
+3. Include relevant metrics or error messages if mentioned in the context
+4. Be detailed enough to help team members understand the situation
+5. Use technical language appropriate for the engineering team
+
+Write in markdown format for better readability. Be thorough and technical.`;
+    }
+
+    // Build user message
+    const userMessage: string = template
+      ? `Fill in the template above using ONLY the following alert data. Output only the filled template, nothing else:\n\n${contextText}`
+      : `Based on the following alert data, please generate an internal technical alert note:\n\n${contextText}`;
+
+    // Build messages array
+    const messages: Array<LLMMessage> = [
+      {
+        role: "system",
+        content: systemPrompt,
+      },
+      {
+        role: "user",
+        content: userMessage,
+      },
+    ];
+
+    return {
+      contextText,
+      systemPrompt,
+      messages,
+    };
+  }
+}

package/Server/Utils/AI/IncidentAIContextBuilder.ts

@@ -392,6 +392,218 @@ Write in a professional, clear, and concise manner. Use markdown formatting for
     };
   }
 
+  @CaptureSpan()
+  public static formatIncidentContextForNote(
+    contextData: IncidentContextData,
+    noteType: "public" | "internal",
+    template?: string,
+  ): AIGenerationContext {
+    const {
+      incident,
+      stateTimeline,
+      internalNotes,
+      publicNotes,
+      workspaceMessages,
+    } = contextData;
+
+    let contextText: string = "";
+
+    // Basic incident information
+    contextText += "# Incident Information\n\n";
+    contextText += `**Title:** ${incident.title || "N/A"}\n\n`;
+    contextText += `**Description:** ${incident.description || "N/A"}\n\n`;
+    contextText += `**Severity:** ${incident.incidentSeverity?.name || "N/A"}\n\n`;
+    contextText += `**Current State:** ${incident.currentIncidentState?.name || "N/A"}\n\n`;
+    contextText += `**Created At:** ${incident.createdAt ? OneUptimeDate.getDateAsFormattedString(incident.createdAt) : "N/A"}\n\n`;
+
+    // Affected monitors
+    if (incident.monitors && incident.monitors.length > 0) {
+      contextText += "**Affected Monitors:** ";
+      contextText += incident.monitors
+        .map((m: { name?: string }) => {
+          return m.name;
+        })
+        .join(", ");
+      contextText += "\n\n";
+    }
+
+    // Labels
+    if (incident.labels && incident.labels.length > 0) {
+      contextText += "**Labels:** ";
+      contextText += incident.labels
+        .map((l: { name?: string }) => {
+          return l.name;
+        })
+        .join(", ");
+      contextText += "\n\n";
+    }
+
+    // Root cause if available
+    if (incident.rootCause) {
+      contextText += `**Root Cause:** ${incident.rootCause}\n\n`;
+    }
+
+    // State timeline
+    if (stateTimeline.length > 0) {
+      contextText += "# State Timeline\n\n";
+      for (const timeline of stateTimeline) {
+        const startTime: string = timeline.startsAt
+          ? OneUptimeDate.getDateAsFormattedString(timeline.startsAt)
+          : "N/A";
+        const stateName: string =
+          timeline.incidentState?.name?.toString() || "Unknown";
+        const createdBy: string =
+          timeline.createdByUser?.name?.toString() ||
+          timeline.createdByUser?.email?.toString() ||
+          "System";
+
+        contextText += `- **${startTime}**: State changed to **${stateName}** by ${createdBy}\n`;
+        if (timeline.rootCause) {
+          contextText += `  - Root cause noted: ${timeline.rootCause}\n`;
+        }
+      }
+      contextText += "\n";
+    }
+
+    // Include internal notes for context (for both note types)
+    if (internalNotes.length > 0) {
+      contextText += "# Internal Notes (Private)\n\n";
+      for (const note of internalNotes) {
+        const noteTime: string = note.createdAt
+          ? OneUptimeDate.getDateAsFormattedString(note.createdAt)
+          : "N/A";
+        const createdBy: string =
+          note.createdByUser?.name?.toString() ||
+          note.createdByUser?.email?.toString() ||
+          "Unknown";
+
+        contextText += `**[${noteTime}] ${createdBy}:**\n`;
+        contextText += `${note.note || "N/A"}\n\n`;
+      }
+    }
+
+    // Public notes
+    if (publicNotes.length > 0) {
+      contextText += "# Public Notes\n\n";
+      for (const note of publicNotes) {
+        const noteTime: string = note.postedAt
+          ? OneUptimeDate.getDateAsFormattedString(note.postedAt)
+          : note.createdAt
+            ? OneUptimeDate.getDateAsFormattedString(note.createdAt)
+            : "N/A";
+        const createdBy: string =
+          note.createdByUser?.name?.toString() ||
+          note.createdByUser?.email?.toString() ||
+          "Unknown";
+
+        contextText += `**[${noteTime}] ${createdBy}:**\n`;
+        contextText += `${note.note || "N/A"}\n\n`;
+      }
+    }
+
+    // Workspace messages (Slack/Teams)
+    if (workspaceMessages.length > 0) {
+      contextText += "# Discussion from Incident Channel\n\n";
+      contextText += WorkspaceUtil.formatMessagesAsContext(workspaceMessages, {
+        includeTimestamp: true,
+        includeUsername: true,
+        maxLength: 20000,
+      });
+      contextText += "\n\n";
+    }
+
+    // System prompt for note generation
+    let systemPrompt: string;
+
+    if (noteType === "public") {
+      if (template) {
+        systemPrompt = `You are an expert incident communicator. Your task is to fill in a public incident note template based on the provided incident data.
+
+CRITICAL INSTRUCTIONS:
+- You MUST use ONLY the exact template structure provided below
+- Fill in each section of the template with relevant information from the incident data
+- Do NOT add any new sections, headers, or content that is not part of the template
+- Do NOT add introductions, conclusions, or any text outside the template structure
+- Write in a professional, clear, and customer-friendly manner
+- Focus on what customers need to know: impact, current status, and next steps
+- Avoid technical jargon - keep it understandable for non-technical readers
+- Be concise but informative
+
+TEMPLATE TO FILL (use this exact structure):
+
+${template}`;
+      } else {
+        systemPrompt = `You are an expert incident communicator. Your task is to generate a public incident note that will be visible to customers on the status page.
+
+The note should:
+1. Be written in a professional, customer-friendly tone
+2. Clearly communicate the current status of the incident
+3. Explain the impact on users without excessive technical details
+4. Provide information about what is being done to resolve the issue
+5. Set appropriate expectations about resolution timing if known
+6. Be concise but informative
+
+DO NOT include:
+- Internal technical details that customers don't need
+- Blame or finger-pointing
+- Confidential information
+- Excessive jargon
+
+Write in markdown format for better readability.`;
+      }
+    } else if (template) {
+      // Internal note with template
+      systemPrompt = `You are an expert Site Reliability Engineer (SRE). Your task is to fill in an internal incident note template based on the provided incident data.
+
+CRITICAL INSTRUCTIONS:
+- You MUST use ONLY the exact template structure provided below
+- Fill in each section of the template with relevant information from the incident data
+- Do NOT add any new sections, headers, or content that is not part of the template
+- Do NOT add introductions, conclusions, or any text outside the template structure
+- Be technical and detailed - this is for the internal team
+- Include relevant technical details, metrics, and observations
+
+TEMPLATE TO FILL (use this exact structure):
+
+${template}`;
+    } else {
+      // Internal note without template
+      systemPrompt = `You are an expert Site Reliability Engineer (SRE). Your task is to generate an internal incident note for the response team.
+
+The note should:
+1. Provide technical details about the current situation
+2. Document observations, findings, or actions taken
+3. Include relevant metrics or error messages if mentioned in the context
+4. Be detailed enough to help team members understand the situation
+5. Use technical language appropriate for the engineering team
+
+Write in markdown format for better readability. Be thorough and technical.`;
+    }
+
+    // Build user message
+    const userMessage: string = template
+      ? `Fill in the template above using ONLY the following incident data. Output only the filled template, nothing else:\n\n${contextText}`
+      : `Based on the following incident data, please generate ${noteType === "public" ? "a customer-facing public" : "an internal technical"} incident note:\n\n${contextText}`;
+
+    // Build messages array
+    const messages: Array<LLMMessage> = [
+      {
+        role: "system",
+        content: systemPrompt,
+      },
+      {
+        role: "user",
+        content: userMessage,
+      },
+    ];
+
+    return {
+      contextText,
+      systemPrompt,
+      messages,
+    };
+  }
+
   @CaptureSpan()
   public static buildGenericAIContext(data: {
     systemPrompt: string;