AI: let the AI choose which mode to run Lighthouse in

Fixed: 498942434
Change-Id: Ieea5870da1fa211901df4ecd40943a0bd1344fa9
Reviewed-on: https://chromium-review.googlesource.com/c/devtools/devtools-frontend/+/7761792
Commit-Queue: Alina Varkki <alinavarkki@chromium.org>
Reviewed-by: Alina Varkki <alinavarkki@chromium.org>
Auto-Submit: Jack Franklin <jacktfranklin@chromium.org>

Author: jackfranklin

front_end/models/ai_assistance/agents/ContextSelectionAgent.snapshot.txt

@@ -107,8 +107,16 @@ Content:
         "type": 6,
         "description": "",
         "nullable": true,
-        "required": [],
-        "properties": {}
+        "required": [
+          "mode"
+        ],
+        "properties": {
+          "mode": {
+            "type": 1,
+            "description": "The mode to run Lighthouse in. Your ONLY options are \"navigation\" or \"snapshot\". You should determine this based on the user's question. If the user is asking specifically about accessibility, you can run in \"snapshot\" mode which avoids reloading the page. If the user asks for a full Lighthouse report, you should run in \"navigation\" mode which is the default. These are the only options you can pass.",
+            "nullable": false
+          }
+        }
       }
     },
     {

front_end/models/ai_assistance/agents/ContextSelectionAgent.ts

@@ -12,6 +12,7 @@ import * as Logs from '../../logs/logs.js';
 import * as NetworkTimeCalculator from '../../network_time_calculator/network_time_calculator.js';
 import type * as Trace from '../../trace/trace.js';
 import * as Workspace from '../../workspace/workspace.js';
+import {debugLog} from '../debug.js';
 
 import {AccessibilityContext} from './AccessibilityAgent.js';
 import {
@@ -306,29 +307,44 @@ export class ContextSelectionAgent extends AiAgent<never> {
       }
     });
 
-    this.declareFunction('runLighthouseAudits', {
+    type LHSupportedRunMode = Extract<LHModel.RunTypes.RunMode, 'navigation'|'snapshot'>;
+    const parseLighthouseMode = (mode?: string): LHSupportedRunMode => {
+      return mode === 'snapshot' ? 'snapshot' : 'navigation';
+    };
+
+    this.declareFunction<{mode: LHSupportedRunMode}>('runLighthouseAudits', {
       description:
           'Records a Lighthouse audit on the current page. Use this to debug accessibility, SEO, and best practices. (For performance metrics like LCP, use performanceRecordAndReload instead).',
       parameters: {
         type: Host.AidaClient.ParametersTypes.OBJECT,
         description: '',
         nullable: true,
-        required: [],
-        properties: {},
+        required: ['mode'],
+        properties: {
+          mode: {
+            type: Host.AidaClient.ParametersTypes.STRING,
+            description:
+                'The mode to run Lighthouse in. Your ONLY options are "navigation" or "snapshot". You should determine this based on the user\'s question. If the user is asking specifically about accessibility, you can run in "snapshot" mode which avoids reloading the page. If the user asks for a full Lighthouse report, you should run in "navigation" mode which is the default. These are the only options you can pass.',
+            nullable: false,
+          }
+        },
       },
-      displayInfoFromArgs: () => {
+      displayInfoFromArgs: args => {
+        const mode = parseLighthouseMode(args.mode);
         return {
           title: 'Auditing your page with Lighthouse',
-          action: 'runLighthouseAudits()',
+          action: `runLighthouseAudits(${mode})`,
         };
       },
-      handler: async () => {
+      handler: async params => {
         if (!this.#lighthouseRecording) {
           return {
             error: 'Lighthouse report is not available.',
           };
         }
-        const result = await this.#lighthouseRecording();
+        const mode = parseLighthouseMode(params.mode);
+        debugLog(`Recording with Lighthouse; runMode=${mode}`);
+        const result = await this.#lighthouseRecording({mode});
         if (!result) {
           return {error: 'Failed to generate Lighthouse report.'};
         }