chrome-devtools-frontend 1.0.1642899 → 1.0.1643855

package/front_end/models/ai_assistance/{agents/ConversationSummaryAgent.ts → ConversationSummary.ts}

@@ -2,10 +2,10 @@
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
-import * as Host from '../../../core/host/host.js';
-import * as Root from '../../../core/root/root.js';
+import * as Host from '../../core/host/host.js';
+import * as Root from '../../core/root/root.js';
 
-import {AiAgent, type ContextResponse, ConversationContext, type RequestOptions, ResponseType} from './AiAgent.js';
+import {runOneShotPrompt} from './AiUtils.js';
 
 const preamble = `### Role
 You are a Conversation Summarizer. Your task is to take a transcript of a conversation between a user and a DevTools AI agent and produce a succinct, actionable Markdown summary. This summary will be used to help apply fixes in an IDE, so it must capture all relevant technical details, findings, and proposed code changes without any conversational fluff.
@@ -98,90 +98,53 @@ color: red;
 - Professional, objective, and dense.
 - Past tense for actions; Present tense for technical facts.`;
 
-export class ConversationSummaryContext extends ConversationContext<string> {
-  #conversation: string;
-  constructor(conversation: string) {
-    super();
-    this.#conversation = conversation;
-  }
-
-  override getURL(): string {
-    return 'devtools://ai-assistance';
-  }
-
-  getItem(): string {
-    return this.#conversation;
-  }
-
-  override getTitle(): string {
-    return 'Conversation';
-  }
+export interface ConversationSummaryOptions {
+  aidaClient: Host.AidaClient.AidaClient;
+  serverSideLoggingEnabled?: boolean;
 }
 
 /**
- * An agent that takes a full conversation between a user and an agent in markdown
+ * A class that takes a full conversation between a user and an agent in markdown
  * format and produces a succinct summary of the conversation.
  *
  * This summary is designed to be read by a local agent in the user's IDE and it
  * will be used to help apply fixes to the user's local codebase based on the
  * debugging information the devtools agent found.
- *
- * This agent is not intended to be used directly by users in the AI Assistance
- * panel when chatting with DevTools AI.
  */
-export class ConversationSummaryAgent extends AiAgent<string> {
-  override preamble = preamble;
+export class ConversationSummary {
+  readonly #aidaClient: Host.AidaClient.AidaClient;
+  readonly #serverSideLoggingEnabled: boolean;
 
-  get clientFeature(): Host.AidaClient.ClientFeature {
-    return Host.AidaClient.ClientFeature.CHROME_CONVERSATION_SUMMARY_AGENT;
+  constructor(options: ConversationSummaryOptions) {
+    this.#aidaClient = options.aidaClient;
+    this.#serverSideLoggingEnabled = options.serverSideLoggingEnabled ?? false;
   }
 
-  get userTier(): string|undefined {
-    // TODO(b/491772868): tidy up userTier & feature flags in the backend.
-    return Root.Runtime.hostConfig.devToolsFreestyler?.userTier;
-  }
+  async summarizeConversation(conversation: string): Promise<string> {
+    const enhancedQuery = `Summarize the following conversation:\n\n${conversation}`;
 
-  get options(): RequestOptions {
     // TODO(b/491772868): tidy up userTier & feature flags in the backend.
     const temperature = Root.Runtime.hostConfig.devToolsFreestyler?.temperature;
     const modelId = Root.Runtime.hostConfig.devToolsFreestyler?.modelId;
+    const userTier = Root.Runtime.hostConfig.devToolsFreestyler?.userTier;
 
-    return {
+    const resultText = await runOneShotPrompt({
+      aidaClient: this.#aidaClient,
+      preamble,
+      query: enhancedQuery,
+      clientFeature: Host.AidaClient.ClientFeature.CHROME_CONVERSATION_SUMMARY_AGENT,
       temperature,
       modelId,
-    };
-  }
+      userTier,
+      serverSideLoggingEnabled: this.#serverSideLoggingEnabled,
+    });
 
-  async * handleContextDetails(context: ConversationContext<string>|null): AsyncGenerator<ContextResponse, void, void> {
-    if (!context) {
-      return;
+    if (!resultText) {
+      throw new Error('Failed to summarize conversation');
     }
 
-    yield {
-      type: ResponseType.CONTEXT,
-      details: [
-        {
-          title: 'Conversation transcript',
-          text: context.getItem(),
-        },
-      ],
-    };
-  }
-
-  override async enhanceQuery(query: string, context: ConversationContext<string>|null): Promise<string> {
-    const conversation = context ? context.getItem() : query;
-    return `Summarize the following conversation:\n\n${conversation}`;
-  }
-
-  async summarizeConversation(conversation: string): Promise<string> {
-    const context = new ConversationSummaryContext(conversation);
-    const response = await Array.fromAsync(this.run('', {selected: context}));
-    const lastResponse = response.at(-1);
-    if (lastResponse && lastResponse.type === ResponseType.ANSWER && lastResponse.complete === true) {
-      const disclaimer =
-          '*Note: The code fixes and findings above were identified on a live page in DevTools. When applying them to your codebase, please adapt them to your project\'s specific technical stack (e.g., Tailwind CSS classes, CSS modules, framework components) rather than applying them as literal CSS overrides.*';
-      return `${lastResponse.text.trim()}\n\n${disclaimer}`;
-    }
-    throw new Error('Failed to summarize conversation');
+    const disclaimer =
+        '*Note: The code fixes and findings above were identified on a live page in DevTools. When applying them to your codebase, please adapt them to your project\'s specific technical stack (e.g., Tailwind CSS classes, CSS modules, framework components) rather than applying them as literal CSS overrides.*';
+    return `${resultText.trim()}\n\n${disclaimer}`;
   }
 }

package/front_end/models/ai_assistance/ExtensionScope.ts

@@ -34,7 +34,6 @@ export class ExtensionScope {
                     }) => Promise<void>> = [];
   #changeManager: ChangeManager;
   #agentId: string;
-  #turnId?: number;
   /** Don't use directly use the getter */
   #frameId?: Protocol.Page.FrameId|null;
   /** Don't use directly use the getter */
@@ -42,12 +41,11 @@ export class ExtensionScope {
 
   readonly #bindingMutex = new Common.Mutex.Mutex();
 
-  constructor(changes: ChangeManager, agentId: string, selectedNode: SDK.DOMModel.DOMNode|null, turnId?: number) {
+  constructor(changes: ChangeManager, agentId: string, selectedNode: SDK.DOMModel.DOMNode|null) {
     this.#changeManager = changes;
     const frameId = selectedNode?.frameId();
     const target = selectedNode?.domModel().target();
     this.#agentId = agentId;
-    this.#turnId = turnId;
     this.#target = target;
     this.#frameId = frameId;
   }
@@ -357,7 +355,6 @@ export class ExtensionScope {
         const sanitizedStyles = await this.sanitizedStyleChanges(context.selector, arg.styles);
         const styleChanges = await this.#changeManager.addChange(cssModel, this.frameId, {
           groupId: this.#agentId,
-          turnId: this.#turnId,
           sourceLocation: context.sourceLocation,
           selector: context.selector,
           simpleSelector: context.simpleSelector,

package/front_end/models/ai_assistance/{agents/PerformanceAnnotationsAgent.ts → PerformanceAnnotations.ts}

@@ -2,21 +2,12 @@
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
-import * as Host from '../../../core/host/host.js';
-import * as Root from '../../../core/root/root.js';
-import type {AICallTree} from '../performance/AICallTree.js';
-import type {AgentFocus} from '../performance/AIContext.js';
-
-import {AiAgent, type ContextResponse, type ConversationContext, type RequestOptions, ResponseType} from './AiAgent.js';
-import {PerformanceTraceContext} from './PerformanceAgent.js';
-
-/**
- * Preamble clocks in at ~970 tokens.
- *   The prose is around 4.5 chars per token.
- * The data can be as bad as 1.8 chars per token
- *
- * Check token length in https://aistudio.google.com/
- */
+import * as Host from '../../core/host/host.js';
+import * as Root from '../../core/root/root.js';
+
+import {runOneShotPrompt} from './AiUtils.js';
+import type {AICallTree} from './performance/AICallTree.js';
+
 const callTreePreamble = `You are an expert performance analyst embedded within Chrome DevTools.
 You meticulously examine web application behavior captured by the Chrome DevTools Performance Panel and Chrome tracing.
 You will receive a structured text representation of a call tree, derived from a user-selected call frame within a performance trace's flame chart.
@@ -79,80 +70,6 @@ The 'calculatePosition' function, taking 80ms, is a potential bottleneck.
 Consider optimizing the position calculation logic or reducing the frequency of calls to improve animation performance.
 `;
 
-export class PerformanceAnnotationsAgent extends AiAgent<AgentFocus> {
-  override preamble = callTreePreamble;
-
-  get clientFeature(): Host.AidaClient.ClientFeature {
-    return Host.AidaClient.ClientFeature.CHROME_PERFORMANCE_ANNOTATIONS_AGENT;
-  }
-
-  get userTier(): string|undefined {
-    return Root.Runtime.hostConfig.devToolsAiAssistancePerformanceAgent?.userTier;
-  }
-
-  get options(): RequestOptions {
-    const temperature = Root.Runtime.hostConfig.devToolsAiAssistancePerformanceAgent?.temperature;
-    const modelId = Root.Runtime.hostConfig.devToolsAiAssistancePerformanceAgent?.modelId;
-
-    return {
-      temperature,
-      modelId,
-    };
-  }
-
-  async *
-      handleContextDetails(context: ConversationContext<AgentFocus>|null): AsyncGenerator<ContextResponse, void, void> {
-    if (!context) {
-      return;
-    }
-
-    const focus = context.getItem();
-    if (!focus.callTree) {
-      throw new Error('unexpected context');
-    }
-
-    const callTree = focus.callTree;
-
-    yield {
-      type: ResponseType.CONTEXT,
-      details: [
-        {
-          title: 'Selected call tree',
-          text: callTree.serialize(),
-        },
-      ],
-    };
-  }
-
-  override async enhanceQuery(query: string, context: ConversationContext<AgentFocus>|null): Promise<string> {
-    if (!context) {
-      return query;
-    }
-
-    const focus = context.getItem();
-    if (!focus.callTree) {
-      throw new Error('unexpected context');
-    }
-
-    const callTree = focus.callTree;
-    const contextString = callTree.serialize();
-    return `${contextString}\n\n# User request\n\n${query}`;
-  }
-
-  /**
-   * Used in the Performance panel to automatically generate a label for a selected entry.
-   */
-  async generateAIEntryLabel(callTree: AICallTree): Promise<string> {
-    const context = PerformanceTraceContext.fromCallTree(callTree);
-    const response = await Array.fromAsync(this.run(AI_LABEL_GENERATION_PROMPT, {selected: context}));
-    const lastResponse = response.at(-1);
-    if (lastResponse && lastResponse.type === ResponseType.ANSWER && lastResponse.complete === true) {
-      return lastResponse.text.trim();
-    }
-    throw new Error('Failed to generate AI entry label');
-  }
-}
-
 const AI_LABEL_GENERATION_PROMPT = `## Instruction:
 Generate a concise label (max 60 chars, single line) describing the *user-visible effect* of the selected call tree's activity, based solely on the provided call tree data.
 
@@ -168,3 +85,44 @@ Generate a concise label (max 60 chars, single line) describing the *user-visibl
 - Only include third-party script names if their identification is highly confident.
 - Very important: Only output the 60 character label text, your response will be used in full to show to the user as an annotation in the timeline.
 `;
+
+export interface PerformanceAnnotationsOptions {
+  aidaClient: Host.AidaClient.AidaClient;
+  serverSideLoggingEnabled?: boolean;
+}
+
+export class PerformanceAnnotations {
+  readonly #aidaClient: Host.AidaClient.AidaClient;
+  readonly #serverSideLoggingEnabled: boolean;
+
+  constructor(options: PerformanceAnnotationsOptions) {
+    this.#aidaClient = options.aidaClient;
+    this.#serverSideLoggingEnabled = options.serverSideLoggingEnabled ?? false;
+  }
+
+  async generateAIEntryLabel(callTree: AICallTree): Promise<string> {
+    const contextString = callTree.serialize();
+    const query = `${contextString}\n\n# User request\n\n${AI_LABEL_GENERATION_PROMPT}`;
+
+    const temperature = Root.Runtime.hostConfig.devToolsAiAssistancePerformanceAgent?.temperature;
+    const modelId = Root.Runtime.hostConfig.devToolsAiAssistancePerformanceAgent?.modelId;
+    const userTier = Root.Runtime.hostConfig.devToolsAiAssistancePerformanceAgent?.userTier;
+
+    const resultText = await runOneShotPrompt({
+      aidaClient: this.#aidaClient,
+      preamble: callTreePreamble,
+      query,
+      clientFeature: Host.AidaClient.ClientFeature.CHROME_PERFORMANCE_ANNOTATIONS_AGENT,
+      temperature,
+      modelId,
+      userTier,
+      serverSideLoggingEnabled: this.#serverSideLoggingEnabled,
+    });
+
+    if (!resultText) {
+      throw new Error('Failed to generate AI entry label');
+    }
+
+    return resultText.trim();
+  }
+}

package/front_end/models/ai_assistance/agents/AccessibilityAgent.ts

@@ -13,6 +13,8 @@ import {ChangeManager} from '../ChangeManager.js';
 import {LighthouseFormatter} from '../data_formatters/LighthouseFormatter.js';
 import {debugLog} from '../debug.js';
 import {ExtensionScope} from '../ExtensionScope.js';
+import {ToolName} from '../tools/Tool.js';
+import {ToolRegistry} from '../tools/ToolRegistry.js';
 
 import {
   AiAgent,
@@ -25,10 +27,8 @@ import {
 } from './AiAgent.js';
 import {
   type CreateExtensionScopeFunction,
-  executeJavaScriptFunction,
   type ExecuteJsAgentOptions,
   executeJsCode,
-  JavascriptExecutor
 } from './ExecuteJavascript.js';
 
 /**
@@ -81,12 +81,6 @@ If the user asks a question that requires an investigation of a problem, use thi
     - [Suggestion 2]
 `;
 
-const SECURITY_WARNING = `**CRITICAL CONSTRAINT**: This Lighthouse report was imported from a file and is static.
-You do NOT have access to the inspected page.
-Tools like \`executeJavaScript\`, \`getStyles\`, or \`runAccessibilityAudits\` are disabled.
-Do NOT attempt to use them or instruct the user that you will use them.
-Rely ONLY on the static report data below.`;
-
 export class AccessibilityContext extends ConversationContext<LHModel.ReporterTypes.ReportJSON> {
   #lh: LHModel.ReporterTypes.ReportJSON;
 
@@ -123,28 +117,17 @@ export class AccessibilityAgent extends AiAgent<LHModel.ReporterTypes.ReportJSON
       (overrides?: LHModel.RunTypes.RunOverrides) => Promise<LHModel.ReporterTypes.ReportJSON|null>;
 
   #execJs: typeof executeJsCode;
-  #javascriptExecutor: JavascriptExecutor;
   #changes: ChangeManager;
   #createExtensionScope: CreateExtensionScopeFunction;
-  #currentTurnId = 0;
 
   constructor(opts: ExecuteJsAgentOptions) {
     super(opts);
     this.#lighthouseRecording = opts.lighthouseRecording;
     this.#changes = opts.changeManager || new ChangeManager();
     this.#execJs = opts.execJs ?? executeJsCode;
-    this.#createExtensionScope =
-        opts.createExtensionScope ?? ((changes: ChangeManager) => {
-          return new ExtensionScope(changes, this.sessionId, this.#getDocumentBodyNode(), this.#currentTurnId);
-        });
-    this.#javascriptExecutor = new JavascriptExecutor(
-        {
-          executionMode: this.executionMode,
-          getContextNode: () => this.#getDocumentBodyNode(),
-          createExtensionScope: this.#createExtensionScope.bind(this),
-          changes: this.#changes,
-        },
-        this.#execJs);
+    this.#createExtensionScope = opts.createExtensionScope ?? ((changes: ChangeManager) => {
+                                   return new ExtensionScope(changes, this.sessionId, this.#getDocumentBodyNode());
+                                 });
   }
 
   get userTier(): string|undefined {
@@ -168,7 +151,6 @@ export class AccessibilityAgent extends AiAgent<LHModel.ReporterTypes.ReportJSON
   }
 
   protected override async preRun(): Promise<void> {
-    this.#currentTurnId++;
     const target = SDK.TargetManager.TargetManager.instance().primaryPageTarget();
     const domModel = target?.model(SDK.DOMModel.DOMModel);
     // We need to ensure the document is requested so that #getDocumentBodyNode()
@@ -280,11 +262,33 @@ export class AccessibilityAgent extends AiAgent<LHModel.ReporterTypes.ReportJSON
       }
     });
 
-    if (isImported) {
-      return;
+    const executeJsTool = ToolRegistry.get(ToolName.EXECUTE_JAVASCRIPT);
+    if (!executeJsTool) {
+      throw new Error('Required tool "executeJavaScript" not found');
     }
-
-    this.declareFunction('executeJavaScript', executeJavaScriptFunction(this.#javascriptExecutor));
+    this.declareFunction(executeJsTool.name, {
+      description: executeJsTool.description,
+      parameters: executeJsTool.parameters,
+      displayInfoFromArgs: executeJsTool.displayInfoFromArgs,
+      handler: async (args, options) => {
+        if (isImported) {
+          return {
+            error: 'Cannot use this tool on an imported file.',
+          };
+        }
+        return await executeJsTool.handler(
+            args,
+            {
+              conversationContext: this.context ?? null,
+              changeManager: this.#changes,
+              createExtensionScope: this.#createExtensionScope.bind(this),
+              execJs: this.#execJs,
+              getExecutionContextNode: () => this.#getDocumentBodyNode(),
+            },
+            options,
+        );
+      },
+    });
 
     this.declareFunction<{explanation: string}, {audits: string}>('runAccessibilityAudits', {
       description:
@@ -311,6 +315,11 @@ export class AccessibilityAgent extends AiAgent<LHModel.ReporterTypes.ReportJSON
       },
       handler: async params => {
         debugLog('Function call: runAccessibilityAudits', params);
+        if (isImported) {
+          return {
+            error: 'Cannot use this tool on an imported file.',
+          };
+        }
         if (!this.#lighthouseRecording) {
           return {error: 'Lighthouse recording is not available.'};
         }
@@ -375,6 +384,11 @@ export class AccessibilityAgent extends AiAgent<LHModel.ReporterTypes.ReportJSON
       },
       handler: async params => {
         debugLog('Function call: getStyles', params);
+        if (isImported) {
+          return {
+            error: 'Cannot use this tool on an imported file.',
+          };
+        }
         const node = await this.#resolvePathToNode(params.path);
         if (!node) {
           return {error: `Could not find the element with path: ${params.path}`};
@@ -445,6 +459,11 @@ export class AccessibilityAgent extends AiAgent<LHModel.ReporterTypes.ReportJSON
       },
       handler: async params => {
         debugLog('Function call: getElementAccessibilityDetails', params);
+        if (isImported) {
+          return {
+            error: 'Cannot use this tool on an imported file.',
+          };
+        }
         const node = await this.#resolvePathToNode(params.path);
         if (!node) {
           return {error: `Could not find the element with path: ${params.path}`};
@@ -522,10 +541,7 @@ export class AccessibilityAgent extends AiAgent<LHModel.ReporterTypes.ReportJSON
     if (lhr) {
       this.#declareFunctions();
     }
-    let enhancedQuery = lhr ? `${this.#getInitialPayload(lhr)}\n# User request:\n\n` : '';
-    if (lhr?.getItem().isImported) {
-      enhancedQuery = `${SECURITY_WARNING}\n\n${enhancedQuery}`;
-    }
+    const enhancedQuery = lhr ? `${this.#getInitialPayload(lhr)}\n# User request:\n\n` : '';
     return `${enhancedQuery}${query}`;
   }
 

package/front_end/models/ai_assistance/agents/AiAgent.ts

@@ -332,6 +332,13 @@ export interface SourceFilesListAiWidget {
   };
 }
 
+export interface NetworkRequestsListAiWidget {
+  name: 'NETWORK_REQUESTS_LIST';
+  data: {
+    requests: SDK.NetworkRequest.NetworkRequest[],
+  };
+}
+
 export interface LighthouseReportAiWidget {
   name: 'LIGHTHOUSE_REPORT';
   data: {
@@ -366,10 +373,10 @@ export interface SourceCodeAiWidget {
   };
 }
 
-export type AiWidget =
-    ComputedStyleAiWidget|CoreVitalsAiWidget|StylePropertiesAiWidget|DomTreeAiWidget|PerformanceTraceAiWidget|
-    PerfInsightAiWidget|TimelineRangeSummaryAiWidget|BottomUpTreeAiWidget|SourceFileAiWidget|LighthouseReportAiWidget|
-    TimelineEventSummaryAiWidget|NetworkRequestGeneralHeadersAiWidget|SourceCodeAiWidget|SourceFilesListAiWidget;
+export type AiWidget = ComputedStyleAiWidget|CoreVitalsAiWidget|StylePropertiesAiWidget|DomTreeAiWidget|
+    PerformanceTraceAiWidget|PerfInsightAiWidget|TimelineRangeSummaryAiWidget|BottomUpTreeAiWidget|SourceFileAiWidget|
+    LighthouseReportAiWidget|TimelineEventSummaryAiWidget|NetworkRequestGeneralHeadersAiWidget|SourceCodeAiWidget|
+    SourceFilesListAiWidget|NetworkRequestsListAiWidget;
 
 export type FunctionCallHandlerResult<Result> = {
   requiresApproval: true,
@@ -463,7 +470,7 @@ export abstract class AiAgent<T> {
 
   readonly #sessionId: string;
   readonly #aidaClient: Host.AidaClient.AidaClient;
-  readonly #serverSideLoggingEnabled: boolean;
+  #serverSideLoggingEnabled: boolean;
   readonly confirmSideEffect: typeof Promise.withResolvers;
   readonly #functionDeclarations = new Map<string, FunctionDeclaration<Record<string, unknown>, unknown>>();
   readonly #allowedOrigin?: () => AllowedOriginResult;
@@ -502,8 +509,8 @@ export abstract class AiAgent<T> {
     this.#allowedOrigin = opts.allowedOrigin;
   }
 
-  async enhanceQuery(query: string, selected: ConversationContext<T>|null, multimodalInputType?: MultimodalInputType):
-      Promise<string>;
+  async enhanceQuery(query: string, selected: ConversationContext<T>|null,
+                     multimodalInputType?: MultimodalInputType): Promise<string>;
   async enhanceQuery(query: string): Promise<string> {
     return query;
   }
@@ -542,13 +549,29 @@ export abstract class AiAgent<T> {
   clearCache(): void {
   }
 
+  protected disableServerSideLogging(): void {
+    this.#serverSideLoggingEnabled = false;
+  }
+
   popPendingMultimodalInput(): MultimodalInput|undefined {
     return undefined;
   }
 
-  buildRequest(
-      part: Host.AidaClient.Part|Host.AidaClient.Part[],
-      role: Host.AidaClient.Role.USER|Host.AidaClient.Role.ROLE_UNSPECIFIED): Host.AidaClient.DoConversationRequest {
+  /**
+   * Preamble features appended to the `client_version` in metadata.
+   * This is required ONLY for the Styling Agent for legacy reasons to serve
+   * different server-side preambles based on the Chrome version.
+   * Other agents should NOT set or override this.
+   * If you are curious about this, look for `do_conversation_handler.cc` in
+   * Google3 or chat to @jacktfranklin.
+   */
+  preambleFeatures(): string[] {
+    return [];
+  }
+
+  buildRequest(part: Host.AidaClient.Part|Host.AidaClient.Part[],
+               role: Host.AidaClient.Role.USER|
+               Host.AidaClient.Role.ROLE_UNSPECIFIED): Host.AidaClient.DoConversationRequest {
     const parts = Array.isArray(part) ? part : [part];
     const currentMessage: Host.AidaClient.Content = {
       parts,
@@ -589,7 +612,8 @@ export abstract class AiAgent<T> {
         disable_user_content_logging: !(this.#serverSideLoggingEnabled ?? false),
         string_session_id: this.#sessionId,
         user_tier: userTier,
-        client_version: Root.Runtime.getChromeVersion(),
+        client_version:
+            Root.Runtime.getChromeVersion() + this.preambleFeatures().map(feature => `+${feature}`).join(''),
       },
 
       functionality_type: enableAidaFunctionCalling ? Host.AidaClient.FunctionalityType.AGENTIC_CHAT :
@@ -693,6 +717,10 @@ export abstract class AiAgent<T> {
     this.#functionDeclarations.clear();
   }
 
+  /**
+   * Executed immediately after the current context is populated with the selected
+   * context and before the request is built.
+   */
   protected async preRun(): Promise<void> {
   }
 
@@ -705,11 +733,11 @@ export abstract class AiAgent<T> {
           },
           multimodalInput?: MultimodalInput,
           ): AsyncGenerator<ResponseData, void, void> {
-    await this.preRun();
     await options.selected?.refresh();
     if (options.selected) {
       this.context = options.selected;
     }
+    await this.preRun();
 
     const enhancedQuery = await this.enhanceQuery(initialQuery, options.selected, multimodalInput?.type);
     Host.userMetrics.freestylerQueryLength(enhancedQuery.length);

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

@@ -129,6 +129,17 @@ Content:
         "required": [],
         "properties": {}
       }
+    },
+    {
+      "name": "analyzeStorage",
+      "description": "Selects the page storage. Use this when asked about browser storage (localStorage, sessionStorage, cookies) and issues related to these.",
+      "parameters": {
+        "type": 6,
+        "description": "",
+        "nullable": true,
+        "required": [],
+        "properties": {}
+      }
     }
   ],
   "options": {},