@rudderhq/server 0.7.2 → 0.7.3

package/dist/services/messenger.js

@@ -1,4 +1,4 @@
-import { activityLog, agents, approvalComments, approvals, authUsers, chatConversations, heartbeatRuns, issueComments, issueFollows, issues, joinRequests, messengerCustomGroupEntries, messengerCustomGroups, messengerSavedViews, messengerThreadUserStates, projects } from "@rudderhq/db";
+import { activityLog, agentIssueCreationRequests, agents, approvalComments, approvals, authUsers, chatConversations, heartbeatRuns, issueComments, issueFollows, issues, joinRequests, messengerCustomGroupEntries, messengerCustomGroups, messengerSavedViews, messengerThreadUserStates, projects } from "@rudderhq/db";
 import { formatMessengerPreview, formatMessengerTitle, isUuidLike, issueUpdatedChangedKeys, messengerSavedViewIdSchema } from "@rudderhq/shared";
 import { and, asc, desc, eq, gt, inArray, isNull, ne, or, sql } from "drizzle-orm";
 import { badRequest, conflict, notFound } from "../errors.js";
@@ -16,6 +16,7 @@ const ISSUE_ACTIVITY_ACTIONS = [
     "issue.updated",
     "issue.followed",
     "automation.issue_created_notification",
+    "agent.issue_created_notification",
     "issue.approval_linked",
     "issue.work_product_created",
     "issue.work_product_updated",
@@ -48,6 +49,15 @@ function maxDate(...values) {
         return null;
     return dates.sort((a, b) => b.getTime() - a.getTime())[0] ?? null;
 }
+function failedRunSummary(run) {
+    const summary = failedRunUserSummary(run);
+    return run.agentIssueCreationRequestId
+        ? `Agent Issue creation failed. ${summary}`
+        : summary;
+}
+function failedAgentIssueRequestSummary(request) {
+    return `Agent Issue creation failed. ${request.error ?? "Agent run completed without creating an Issue"}`;
+}
 function compareLatestActivity(a, b) {
     const aTime = a.latestActivityAt?.getTime() ?? Number.NEGATIVE_INFINITY;
     const bTime = b.latestActivityAt?.getTime() ?? Number.NEGATIVE_INFINITY;
@@ -228,6 +238,8 @@ function summarizeIssueActivity(activity, issue) {
             return "Followed";
         case "automation.issue_created_notification":
             return "Automation created issue";
+        case "agent.issue_created_notification":
+            return "Agent created issue";
         case "issue.approval_linked":
             return "Approval linked";
         case "issue.work_product_created":
@@ -525,21 +537,31 @@ function approvalCard(approval, requesterAgent, latestComment, currentUserId, la
         requesterAgent,
     };
 }
-function failedRunCard(run, agentName, origin) {
-    const summary = failedRunUserSummary(run);
+function failedRunCard(run, agentName, origin, orgId, userId) {
+    const summary = failedRunSummary(run);
     const sourceAction = messengerFailedRunSourceAction(origin);
+    const agentIssueRequestId = run.agentIssueCreationRequestId;
+    const canRetryAgentIssue = Boolean(agentIssueRequestId
+        && run.agentIssueCreationRequestedByUserId === userId);
+    const retryAction = agentIssueRequestId
+        ? canRetryAgentIssue
+            ? buildAction("Retry Agent Issue", `/orgs/${orgId}/agent-issue-creation-requests/${agentIssueRequestId}/retry`, "POST")
+            : null
+        : buildAction("Retry", `/agent-runs/${run.id}/retry`, "POST");
     return {
         id: run.id,
         threadKey: "failed-runs",
         kind: "failed-runs",
-        title: agentName ? `${agentName} · Failed run` : "Failed run",
-        subtitle: run.status.replaceAll("_", " "),
+        title: agentIssueRequestId
+            ? (agentName ? `${agentName} · Agent Issue creation failed` : "Agent Issue creation failed")
+            : (agentName ? `${agentName} · Failed run` : "Failed run"),
+        subtitle: agentIssueRequestId ? "Agent Issue request failed" : run.status.replaceAll("_", " "),
         body: summary,
         preview: summary,
         href: `/agents/${run.agentId}/runs/${run.id}`,
         latestActivityAt: run.updatedAt ?? run.createdAt,
         actions: [
-            buildAction("Retry", `/agent-runs/${run.id}/retry`, "POST"),
+            ...(retryAction ? [retryAction] : []),
             ...(sourceAction ? [sourceAction] : []),
             buildAction("Open run", `/agents/${run.agentId}/runs/${run.id}`, "GET"),
         ],
@@ -547,10 +569,46 @@ function failedRunCard(run, agentName, origin) {
             runId: run.id,
             agentId: run.agentId,
             status: run.status,
+            ...(agentIssueRequestId
+                ? {
+                    agentIssueCreationRequestId: agentIssueRequestId,
+                    agentIssueCreationRetryable: canRetryAgentIssue,
+                    agentIssueCreationError: run.agentIssueCreationError,
+                }
+                : {}),
         },
         origin,
     };
 }
+function failedAgentIssueRequestCard(request, agentName, orgId) {
+    const summary = failedAgentIssueRequestSummary(request);
+    return {
+        id: request.id,
+        threadKey: "failed-runs",
+        kind: "failed-runs",
+        title: agentName ? `${agentName} · Agent Issue creation failed` : "Agent Issue creation failed",
+        subtitle: "Agent Issue request failed",
+        body: summary,
+        preview: summary,
+        href: request.runId ? `/agents/${request.agentId}/runs/${request.runId}` : null,
+        latestActivityAt: request.updatedAt ?? request.createdAt,
+        actions: [
+            buildAction("Retry Agent Issue", `/orgs/${orgId}/agent-issue-creation-requests/${request.id}/retry`, "POST"),
+            ...(request.runId
+                ? [buildAction("Open run", `/agents/${request.agentId}/runs/${request.runId}`, "GET")]
+                : []),
+        ],
+        metadata: {
+            requestId: request.id,
+            agentId: request.agentId,
+            status: request.status,
+            agentIssueCreationRequestId: request.id,
+            agentIssueCreationRetryable: true,
+            agentIssueCreationError: request.error,
+            ...(request.runId ? { runId: request.runId } : {}),
+        },
+    };
+}
 function budgetCard(incident) {
     return {
         id: incident.id,
@@ -1181,7 +1239,7 @@ export function messengerService(db) {
           and notification_issue.org_id = automation_notification_activity.org_id
         where automation_notification_activity.org_id = ${orgId}
           and automation_notification_activity.entity_type = 'issue'
-          and automation_notification_activity.action = 'automation.issue_created_notification'
+          and automation_notification_activity.action in ('automation.issue_created_notification', 'agent.issue_created_notification')
           and automation_notification_activity.details->>'userId' = ${userId}
           and notification_issue.hidden_at is null
       ),
@@ -1285,7 +1343,7 @@ export function messengerService(db) {
               where automation_notification_visibility.org_id = ${orgId}
                 and automation_notification_visibility.entity_type = 'issue'
                 and automation_notification_visibility.entity_id = issue_row.id::text
-                and automation_notification_visibility.action = 'automation.issue_created_notification'
+                and automation_notification_visibility.action in ('automation.issue_created_notification', 'agent.issue_created_notification')
                 and automation_notification_visibility.details->>'userId' = ${userId}
             )
           )
@@ -1327,6 +1385,10 @@ export function messengerService(db) {
             and activity_row.entity_id = issue_row.id::text
             and activity_row.action in (${issueActionSqlList})
             and not ${lowSignalContentOnlyActivity}
+            and (
+              activity_row.action not in ('automation.issue_created_notification', 'agent.issue_created_notification')
+              or activity_row.details->>'userId' = ${userId}
+            )
           order by activity_row.created_at desc, activity_row.id desc
           limit 1
         ) latest_activity on true
@@ -1346,6 +1408,10 @@ export function messengerService(db) {
             and external_activity_row.action in (${issueActionSqlList})
             and not ${externalLowSignalContentOnlyActivity}
             and (external_activity_row.actor_type <> 'user' or external_activity_row.actor_id <> ${userId})
+            and (
+              external_activity_row.action not in ('automation.issue_created_notification', 'agent.issue_created_notification')
+              or external_activity_row.details->>'userId' = ${userId}
+            )
           order by external_activity_row.created_at desc, external_activity_row.id desc
           limit 1
         ) latest_external_activity on true
@@ -1818,9 +1884,8 @@ export function messengerService(db) {
             summary: approvalSummary(itemCount, latestItem?.latestActivityAt ?? null, Number(unreadRows[0]?.unreadCount ?? 0), lastReadAt, latestItem?.preview ?? null),
         };
     }
-    async function loadFailedRunData(orgId, userId, threadStates) {
-        const lastReadAtPromise = lastReadAtForThread(db, orgId, userId, "failed-runs", threadStates);
-        const [runRows, agentRows] = await Promise.all([
+    async function loadFailedRunRows(orgId, userId) {
+        const [runRows, requestRows, agentRows] = await Promise.all([
             db
                 .select({
                 id: heartbeatRuns.id,
@@ -1835,10 +1900,28 @@ export function messengerService(db) {
                 contextSnapshot: heartbeatRuns.contextSnapshot,
                 createdAt: heartbeatRuns.createdAt,
                 updatedAt: heartbeatRuns.updatedAt,
+                agentIssueCreationRequestId: agentIssueCreationRequests.id,
+                agentIssueCreationRequestedByUserId: agentIssueCreationRequests.requestedByUserId,
+                agentIssueCreationError: agentIssueCreationRequests.error,
             })
                 .from(heartbeatRuns)
+                .leftJoin(agentIssueCreationRequests, and(eq(agentIssueCreationRequests.orgId, heartbeatRuns.orgId), eq(agentIssueCreationRequests.runId, heartbeatRuns.id)))
                 .where(and(eq(heartbeatRuns.orgId, orgId), eq(heartbeatRuns.status, "failed")))
                 .orderBy(desc(heartbeatRuns.updatedAt), desc(heartbeatRuns.createdAt)),
+            db
+                .select({
+                id: agentIssueCreationRequests.id,
+                orgId: agentIssueCreationRequests.orgId,
+                agentId: agentIssueCreationRequests.agentId,
+                runId: agentIssueCreationRequests.runId,
+                status: agentIssueCreationRequests.status,
+                error: agentIssueCreationRequests.error,
+                createdAt: agentIssueCreationRequests.createdAt,
+                updatedAt: agentIssueCreationRequests.updatedAt,
+            })
+                .from(agentIssueCreationRequests)
+                .where(and(eq(agentIssueCreationRequests.orgId, orgId), eq(agentIssueCreationRequests.requestedByUserId, userId), inArray(agentIssueCreationRequests.status, ["failed", "cancelled"]), isNull(agentIssueCreationRequests.createdIssueId)))
+                .orderBy(desc(agentIssueCreationRequests.updatedAt), desc(agentIssueCreationRequests.createdAt)),
             db
                 .select({
                 id: agents.id,
@@ -1847,14 +1930,28 @@ export function messengerService(db) {
                 .from(agents)
                 .where(eq(agents.orgId, orgId)),
         ]);
+        const linkedRequestIds = new Set(runRows
+            .map((run) => run.agentIssueCreationRequestId)
+            .filter((requestId) => Boolean(requestId)));
+        return {
+            runRows,
+            requestRows: requestRows.filter((request) => !linkedRequestIds.has(request.id)),
+            agentRows,
+        };
+    }
+    async function loadFailedRunData(orgId, userId, threadStates) {
+        const lastReadAtPromise = lastReadAtForThread(db, orgId, userId, "failed-runs", threadStates);
+        const { runRows, requestRows, agentRows } = await loadFailedRunRows(orgId, userId);
         const lastReadAt = await lastReadAtPromise;
         const agentNames = new Map(agentRows.map((row) => [row.id, row.name]));
         const origins = await hydrateMessengerFailedRunOrigins(db, orgId, runRows);
-        const items = runRows.map((run) => failedRunCard(run, agentNames.get(run.agentId) ?? null, origins.get(run.id)));
+        const runItems = runRows.map((run) => failedRunCard(run, agentNames.get(run.agentId) ?? null, origins.get(run.id), orgId, userId));
+        const requestItems = requestRows.map((request) => failedAgentIssueRequestCard(request, agentNames.get(request.agentId) ?? null, orgId));
+        const items = [...runItems, ...requestItems];
         const latestFirstItems = [...items].sort(compareLatestActivity);
         const chronologicalItems = [...items].sort(compareChronologicalActivity);
         const latestActivityAt = latestFirstItems[0]?.latestActivityAt ?? null;
-        const unreadCount = systemUnreadCountSince(runRows, lastReadAt);
+        const unreadCount = systemUnreadCountSince([...runRows, ...requestRows], lastReadAt);
         return {
             summary: systemSummary("failed-runs", "Failed runs", runRows.length, latestActivityAt, unreadCount, lastReadAt, "No failed runs yet", latestFirstItems[0]?.preview ?? null),
             detail: {
@@ -1876,43 +1973,23 @@ export function messengerService(db) {
     }
     async function loadFailedRunSummaryData(orgId, userId, threadStates) {
         const lastReadAt = await lastReadAtForThread(db, orgId, userId, "failed-runs", threadStates);
-        const latestActivitySql = sql `max(coalesce(${heartbeatRuns.updatedAt}, ${heartbeatRuns.createdAt}))`;
-        const failedRunPredicate = and(eq(heartbeatRuns.orgId, orgId), eq(heartbeatRuns.status, "failed"));
-        const [summaryRows, latestRows, unreadRows] = await Promise.all([
-            db
-                .select({
-                itemCount: sql `count(*)::int`,
-                latestActivityAt: latestActivitySql,
-            })
-                .from(heartbeatRuns)
-                .where(failedRunPredicate),
-            db
-                .select({
-                error: heartbeatRuns.error,
-                errorCode: heartbeatRuns.errorCode,
-                resultJson: heartbeatRuns.resultJson,
-                stderrExcerpt: heartbeatRuns.stderrExcerpt,
-            })
-                .from(heartbeatRuns)
-                .where(failedRunPredicate)
-                .orderBy(desc(heartbeatRuns.updatedAt), desc(heartbeatRuns.createdAt))
-                .limit(1),
-            lastReadAt
-                ? db
-                    .select({
-                    unreadCount: sql `count(*)::int`,
-                })
-                    .from(heartbeatRuns)
-                    .where(and(failedRunPredicate, gt(heartbeatRuns.updatedAt, lastReadAt)))
-                : Promise.resolve([]),
-        ]);
-        const summaryRow = summaryRows[0];
-        const latestRun = latestRows[0] ?? null;
-        const itemCount = Number(summaryRow?.itemCount ?? 0);
-        const unreadCount = lastReadAt ? Number(unreadRows[0]?.unreadCount ?? 0) : itemCount;
+        const { runRows, requestRows } = await loadFailedRunRows(orgId, userId);
+        const failureRows = [...runRows, ...requestRows];
+        const itemCount = failureRows.length;
+        const latestRow = [...failureRows].sort((a, b) => {
+            const aTime = normalizeDate(a.updatedAt ?? a.createdAt)?.getTime() ?? Number.NEGATIVE_INFINITY;
+            const bTime = normalizeDate(b.updatedAt ?? b.createdAt)?.getTime() ?? Number.NEGATIVE_INFINITY;
+            return bTime - aTime;
+        })[0] ?? null;
+        const unreadCount = systemUnreadCountSince(failureRows, lastReadAt);
+        const latestPreview = latestRow
+            ? "agentIssueCreationRequestId" in latestRow
+                ? failedRunSummary(latestRow)
+                : failedAgentIssueRequestSummary(latestRow)
+            : null;
         return {
             itemCount,
-            summary: systemSummary("failed-runs", "Failed runs", itemCount, normalizeDate(summaryRow?.latestActivityAt ?? null), unreadCount, lastReadAt, "No failed runs yet", latestRun ? failedRunUserSummary(latestRun) : null),
+            summary: systemSummary("failed-runs", "Failed runs", itemCount, normalizeDate(latestRow?.updatedAt ?? latestRow?.createdAt ?? null), unreadCount, lastReadAt, "No failed runs yet", latestPreview),
         };
     }
     async function loadBudgetAlertData(orgId, userId, threadStates) {
@@ -2238,7 +2315,7 @@ export function messengerService(db) {
             where automation_notification_activity.org_id = ${orgId}
               and automation_notification_activity.entity_type = 'issue'
               and automation_notification_activity.entity_id = ${issues.id}::text
-              and automation_notification_activity.action = 'automation.issue_created_notification'
+              and automation_notification_activity.action in ('automation.issue_created_notification', 'agent.issue_created_notification')
               and automation_notification_activity.details->>'userId' = ${userId}
           )`)))
             .limit(1);