feat: support configuring chat context in AI assistant

This is important for reports with many apps, or many checks. The model
limits can be easily hit.

Author: devversion

report-app/report-server.ts

@@ -10,7 +10,12 @@ import {fileURLToPath} from 'node:url';
 import {chatWithReportAI} from '../runner/reporting/report-ai-chat';
 import {convertV2ReportToV3Report} from '../runner/reporting/migrations/v2_to_v3';
 import {FetchedLocalReports, fetchReportsFromDisk} from '../runner/reporting/report-local-disk';
-import {AiChatRequest, AIConfigState, RunInfo} from '../runner/shared-interfaces';
+import {
+  AiChatRequest,
+  AIConfigState,
+  IndividualAssessmentState,
+  RunInfo,
+} from '../runner/shared-interfaces';
 
 // This will result in a lot of loading and would slow down the serving,
 // so it's loaded lazily below.
@@ -55,8 +60,19 @@ async function fetchAndMigrateReports(id: string): Promise<RunInfo[] | null> {
     return null;
   }
 
+  let checkID = 0;
   // Convert potential older v2 reports.
-  return result.map(r => convertV2ReportToV3Report(r));
+  return result.map(run => {
+    run = convertV2ReportToV3Report(run);
+    run = {
+      ...run,
+      results: run.results.map(check => ({
+        id: `${id}-${checkID++}`,
+        ...check,
+      })),
+    };
+    return run;
+  });
 }
 
 // Endpoint for fetching a specific report group.
@@ -89,16 +105,19 @@ app.post('/api/reports/:id/chat', async (req, res) => {
   }
 
   try {
-    const {prompt, pastMessages, model} = req.body as AiChatRequest;
-    const assessments = reports.flatMap(run => run.results);
+    const {prompt, pastMessages, model, context, openAppIDs} = req.body as AiChatRequest;
+    const allAssessments = reports.flatMap(run => run.results);
+
     const abortController = new AbortController();
     const summary = await chatWithReportAI(
       await (llm ?? getOrCreateGenkitLlmRunner()),
       prompt,
       abortController.signal,
-      assessments,
+      allAssessments,
       pastMessages,
       model,
+      context,
+      openAppIDs,
     );
     res.json(summary);
   } catch (e) {