chrome-devtools-frontend 1.0.1642899 → 1.0.1643099

package/front_end/core/protocol_client/InspectorBackend.ts

@@ -339,6 +339,10 @@ export class TargetBase {
     return this.getAgent('Autofill');
   }
 
+  adsAgent(): ProtocolProxyApi.AdsApi {
+    return this.getAgent('Ads');
+  }
+
   browserAgent(): ProtocolProxyApi.BrowserApi {
     return this.getAgent('Browser');
   }

package/front_end/entrypoints/heap_snapshot_worker/HeapSnapshot.ts

@@ -876,7 +876,7 @@ export abstract class HeapSnapshot {
   readonly #progress: HeapSnapshotProgress;
   readonly #noDistance = -5;
   rootNodeIndexInternal = 0;
-  #snapshotDiffs: Record<string, Record<string, HeapSnapshotModel.HeapSnapshotModel.Diff>> = {};
+  #snapshotDiffs: Record<number, Record<string, HeapSnapshotModel.HeapSnapshotModel.Diff>> = {};
   #aggregatesForDiff?: {
     interfaceDefinitions: string,
     aggregates: Record<string, HeapSnapshotModel.HeapSnapshotModel.AggregateForDiff>,
@@ -2688,9 +2688,8 @@ export abstract class HeapSnapshot {
     throw new Error('Not implemented');
   }
 
-  calculateSnapshotDiff(
-      baseSnapshotId: string,
-      baseSnapshotAggregates: Record<string, HeapSnapshotModel.HeapSnapshotModel.AggregateForDiff>):
+  calculateSnapshotDiff(baseSnapshotId: number,
+                        baseSnapshotAggregates: Record<string, HeapSnapshotModel.HeapSnapshotModel.AggregateForDiff>):
       Record<string, HeapSnapshotModel.HeapSnapshotModel.Diff> {
     let snapshotDiff: Record<string, HeapSnapshotModel.HeapSnapshotModel.Diff> = this.#snapshotDiffs[baseSnapshotId];
     if (snapshotDiff) {
@@ -2976,7 +2975,7 @@ export abstract class HeapSnapshot {
     return {paths, limitsReached};
   }
 
-  createAddedNodesProvider(baseSnapshotId: string, classKey: string): HeapSnapshotNodesProvider {
+  createAddedNodesProvider(baseSnapshotId: number, classKey: string): HeapSnapshotNodesProvider {
     const snapshotDiff = this.#snapshotDiffs[baseSnapshotId];
     const diffForClass = snapshotDiff[classKey];
     return new HeapSnapshotNodesProvider(this, diffForClass.addedIndexes);

package/front_end/generated/InspectorBackendCommands.ts

@@ -435,7 +435,7 @@ inspectorBackend.registerCommand("DOM.getFrameOwner", [{"name": "frameId", "type
 inspectorBackend.registerCommand("DOM.getContainerForNode", [{"name": "nodeId", "type": "number", "optional": false, "description": "", "typeRef": "DOM.NodeId"}, {"name": "containerName", "type": "string", "optional": true, "description": "", "typeRef": null}, {"name": "physicalAxes", "type": "string", "optional": true, "description": "", "typeRef": "DOM.PhysicalAxes"}, {"name": "logicalAxes", "type": "string", "optional": true, "description": "", "typeRef": "DOM.LogicalAxes"}, {"name": "queriesScrollState", "type": "boolean", "optional": true, "description": "", "typeRef": null}, {"name": "queriesAnchored", "type": "boolean", "optional": true, "description": "", "typeRef": null}], ["nodeId"], "Returns the query container of the given node based on container query conditions: containerName, physical and logical axes, and whether it queries scroll-state or anchored elements. If no axes are provided and queriesScrollState is false, the style container is returned, which is the direct parent or the closest element with a matching container-name.");
 inspectorBackend.registerCommand("DOM.getQueryingDescendantsForContainer", [{"name": "nodeId", "type": "number", "optional": false, "description": "Id of the container node to find querying descendants from.", "typeRef": "DOM.NodeId"}], ["nodeIds"], "Returns the descendants of a container query container that have container queries against this container.");
 inspectorBackend.registerCommand("DOM.getAnchorElement", [{"name": "nodeId", "type": "number", "optional": false, "description": "Id of the positioned element from which to find the anchor.", "typeRef": "DOM.NodeId"}, {"name": "anchorSpecifier", "type": "string", "optional": true, "description": "An optional anchor specifier, as defined in https://www.w3.org/TR/css-anchor-position-1/#anchor-specifier. If not provided, it will return the implicit anchor element for the given positioned element.", "typeRef": null}], ["nodeId"], "Returns the target anchor element of the given anchor query according to https://www.w3.org/TR/css-anchor-position-1/#target.");
-inspectorBackend.registerCommand("DOM.forceShowPopover", [{"name": "nodeId", "type": "number", "optional": false, "description": "Id of the popover HTMLElement", "typeRef": "DOM.NodeId"}, {"name": "enable", "type": "boolean", "optional": false, "description": "If true, opens the popover and keeps it open. If false, closes the popover if it was previously force-opened.", "typeRef": null}], ["nodeIds"], "When enabling, this API force-opens the popover identified by nodeId and keeps it open until disabled.");
+inspectorBackend.registerCommand("DOM.forceShowPopover", [{"name": "nodeId", "type": "number", "optional": false, "description": "Id of the popover HTMLElement", "typeRef": "DOM.NodeId"}, {"name": "enable", "type": "boolean", "optional": false, "description": "If true, opens the popover and keeps it open. If false, closes the popover if it was previously force-opened.", "typeRef": null}, {"name": "invokerNodeId", "type": "number", "optional": true, "description": "Optional ID of the element invoking this popover, used to establish the implicit anchor. If not provided, it will fall back to the first invoker in the document, preferring elements with a popovertarget attribute over those with a commandfor attribute. Note that if there are multiple invokers, this is just an estimate.", "typeRef": "DOM.BackendNodeId"}], ["nodeIds"], "When enabling, this API force-opens the popover identified by nodeId and keeps it open until disabled.");
 inspectorBackend.registerType("DOM.BackendNode", [{"name": "nodeType", "type": "number", "optional": false, "description": "`Node`'s nodeType.", "typeRef": null}, {"name": "nodeName", "type": "string", "optional": false, "description": "`Node`'s nodeName.", "typeRef": null}, {"name": "backendNodeId", "type": "number", "optional": false, "description": "", "typeRef": "DOM.BackendNodeId"}]);
 inspectorBackend.registerType("DOM.Node", [{"name": "nodeId", "type": "number", "optional": false, "description": "Node identifier that is passed into the rest of the DOM messages as the `nodeId`. Backend will only push node with given `id` once. It is aware of all requested nodes and will only fire DOM events for nodes known to the client.", "typeRef": "DOM.NodeId"}, {"name": "parentId", "type": "number", "optional": true, "description": "The id of the parent node if any.", "typeRef": "DOM.NodeId"}, {"name": "backendNodeId", "type": "number", "optional": false, "description": "The BackendNodeId for this node.", "typeRef": "DOM.BackendNodeId"}, {"name": "nodeType", "type": "number", "optional": false, "description": "`Node`'s nodeType.", "typeRef": null}, {"name": "nodeName", "type": "string", "optional": false, "description": "`Node`'s nodeName.", "typeRef": null}, {"name": "localName", "type": "string", "optional": false, "description": "`Node`'s localName.", "typeRef": null}, {"name": "nodeValue", "type": "string", "optional": false, "description": "`Node`'s nodeValue.", "typeRef": null}, {"name": "childNodeCount", "type": "number", "optional": true, "description": "Child count for `Container` nodes.", "typeRef": null}, {"name": "children", "type": "array", "optional": true, "description": "Child nodes of this node when requested with children.", "typeRef": "DOM.Node"}, {"name": "attributes", "type": "array", "optional": true, "description": "Attributes of the `Element` node in the form of flat array `[name1, value1, name2, value2]`.", "typeRef": "string"}, {"name": "documentURL", "type": "string", "optional": true, "description": "Document URL that `Document` or `FrameOwner` node points to.", "typeRef": null}, {"name": "baseURL", "type": "string", "optional": true, "description": "Base URL that `Document` or `FrameOwner` node uses for URL completion.", "typeRef": null}, {"name": "publicId", "type": "string", "optional": true, "description": "`DocumentType`'s publicId.", "typeRef": null}, {"name": "systemId", "type": "string", "optional": true, "description": "`DocumentType`'s systemId.", "typeRef": null}, {"name": "internalSubset", "type": "string", "optional": true, "description": "`DocumentType`'s internalSubset.", "typeRef": null}, {"name": "xmlVersion", "type": "string", "optional": true, "description": "`Document`'s XML version in case of XML documents.", "typeRef": null}, {"name": "name", "type": "string", "optional": true, "description": "`Attr`'s name.", "typeRef": null}, {"name": "value", "type": "string", "optional": true, "description": "`Attr`'s value.", "typeRef": null}, {"name": "pseudoType", "type": "string", "optional": true, "description": "Pseudo element type for this node.", "typeRef": "DOM.PseudoType"}, {"name": "pseudoIdentifier", "type": "string", "optional": true, "description": "Pseudo element identifier for this node. Only present if there is a valid pseudoType.", "typeRef": null}, {"name": "shadowRootType", "type": "string", "optional": true, "description": "Shadow root type.", "typeRef": "DOM.ShadowRootType"}, {"name": "frameId", "type": "string", "optional": true, "description": "Frame ID for frame owner elements.", "typeRef": "Page.FrameId"}, {"name": "contentDocument", "type": "object", "optional": true, "description": "Content document for frame owner elements.", "typeRef": "DOM.Node"}, {"name": "shadowRoots", "type": "array", "optional": true, "description": "Shadow root list for given element host.", "typeRef": "DOM.Node"}, {"name": "templateContent", "type": "object", "optional": true, "description": "Content document fragment for template elements.", "typeRef": "DOM.Node"}, {"name": "pseudoElements", "type": "array", "optional": true, "description": "Pseudo elements associated with this node.", "typeRef": "DOM.Node"}, {"name": "importedDocument", "type": "object", "optional": true, "description": "Deprecated, as the HTML Imports API has been removed (crbug.com/937746). This property used to return the imported document for the HTMLImport links. The property is always undefined now.", "typeRef": "DOM.Node"}, {"name": "distributedNodes", "type": "array", "optional": true, "description": "Distributed nodes for given insertion point.", "typeRef": "DOM.BackendNode"}, {"name": "isSVG", "type": "boolean", "optional": true, "description": "Whether the node is SVG.", "typeRef": null}, {"name": "compatibilityMode", "type": "string", "optional": true, "description": "", "typeRef": "DOM.CompatibilityMode"}, {"name": "assignedSlot", "type": "object", "optional": true, "description": "", "typeRef": "DOM.BackendNode"}, {"name": "isScrollable", "type": "boolean", "optional": true, "description": "", "typeRef": null}, {"name": "affectedByStartingStyles", "type": "boolean", "optional": true, "description": "", "typeRef": null}, {"name": "adoptedStyleSheets", "type": "array", "optional": true, "description": "", "typeRef": "DOM.StyleSheetId"}, {"name": "adProvenance", "type": "object", "optional": true, "description": "", "typeRef": "Network.AdProvenance"}]);
 inspectorBackend.registerType("DOM.DetachedElementInfo", [{"name": "treeNode", "type": "object", "optional": false, "description": "", "typeRef": "DOM.Node"}, {"name": "retainedNodeIds", "type": "array", "optional": false, "description": "", "typeRef": "DOM.NodeId"}]);

package/front_end/generated/protocol.ts

@@ -5849,6 +5849,13 @@ export namespace DOM {
      * popover if it was previously force-opened.
      */
     enable: boolean;
+    /**
+     * Optional ID of the element invoking this popover, used to establish the implicit anchor.
+     * If not provided, it will fall back to the first invoker in the document, preferring
+     * elements with a popovertarget attribute over those with a commandfor attribute. Note that
+     * if there are multiple invokers, this is just an estimate.
+     */
+    invokerNodeId?: BackendNodeId;
   }
 
   export interface ForceShowPopoverResponse extends ProtocolResponseWithError {

package/front_end/models/ai_assistance/AiAgent2.ts

@@ -5,7 +5,6 @@
 import * as Host from '../../core/host/host.js';
 
 import {
-  type AgentOptions,
   AiAgent,
   type ContextResponse,
   type ConversationContext,
@@ -13,7 +12,11 @@ import {
   type RequestOptions,
   ResponseType
 } from './agents/AiAgent.js';
+import {type ExecuteJsAgentOptions, executeJsCode} from './agents/ExecuteJavascript.js';
+import {ChangeManager} from './ChangeManager.js';
+import {DOMNodeContext} from './contexts/DOMNodeContext.js';
 import {debugLog} from './debug.js';
+import {ExtensionScope} from './ExtensionScope.js';
 import type {Skill, SkillName} from './skills/Skill.js';
 import {SKILLS} from './skills/SkillRegistry.js';
 import type {Tool} from './tools/Tool.js';
@@ -30,6 +33,8 @@ export class AiAgent2 extends AiAgent<unknown> {
   readonly userTier = 'TESTERS';
 
   #skillsInjected = false;
+  #changes = new ChangeManager();
+  #execJs: typeof executeJsCode;
 
   get options(): RequestOptions {
     return {};
@@ -38,8 +43,9 @@ export class AiAgent2 extends AiAgent<unknown> {
   readonly #activeSkills = new Set<SkillName>();
   readonly #declaredTools = new Set<string>();
 
-  constructor(opts: AgentOptions) {
+  constructor(opts: ExecuteJsAgentOptions) {
     super(opts);
+    this.#execJs = opts.execJs ?? executeJsCode;
     this.#declaredTools.add('learnSkills');
     const skillsList = Object.keys(SKILLS).join(', ');
     this.declareFunction<{skills: SkillName[]}>('learnSkills', {
@@ -154,6 +160,11 @@ User query: ${enhancedQuery}`;
     return response.trim();
   }
 
+  #createExtensionScope(changes: ChangeManager): {install(): Promise<void>, uninstall(): Promise<void>} {
+    const selectedNode = this.context && this.context instanceof DOMNodeContext ? this.context.getItem() : null;
+    return new ExtensionScope(changes, this.sessionId, selectedNode);
+  }
+
   /**
    * Declares a tool to be available to the agent model, verifying first that
    * it hasn't already been declared to prevent duplicate declaration errors.
@@ -168,9 +179,16 @@ User query: ${enhancedQuery}`;
       description: tool.description,
       parameters: tool.parameters,
       displayInfoFromArgs: tool.displayInfoFromArgs,
-      handler: args => tool.handler(args, {
-        conversationContext: this.context ?? null,
-      }),
+      handler: (args, options) => tool.handler(
+          args,
+          {
+            conversationContext: this.context ?? null,
+            changeManager: this.#changes,
+            createExtensionScope: this.#createExtensionScope.bind(this),
+            execJs: this.#execJs,
+          },
+          options,
+          ),
     });
   }
 

package/front_end/models/ai_assistance/AiConversation.ts

@@ -328,24 +328,27 @@ export class AiConversation {
     };
   }
 
+  #filterHistoryForNewAgent(): Host.AidaClient.Content[] {
+    return this.#agent?.history
+               .map(content => {
+                 return {
+                   ...content,
+                   parts: content.parts.filter(part => !('functionCall' in part) && !('functionResponse' in part)),
+                 };
+               })
+               .filter(content => content.parts.length > 0) ??
+        [];
+  }
+
   #updateAgent(type: ConversationType): void {
     if (this.#type === type) {
       return;
     }
 
-    this.#type = type;
+    const isTransitioningFromStorage = this.#type === ConversationType.STORAGE && type !== ConversationType.STORAGE;
+    const history = isTransitioningFromStorage ? [] : this.#filterHistoryForNewAgent();
 
-    // We need to filter out the function calls
-    // as the LLM tries to call the existing ones.
-    const history =
-        this.#agent?.history
-            .map(content => {
-              return {
-                ...content,
-                parts: content.parts.filter(part => !('functionCall' in part) && !('functionResponse' in part)),
-              };
-            })
-            .filter(content => content.parts.length > 0);
+    this.#type = type;
 
     const options = {
       aidaClient: this.#aidaClient,

package/front_end/models/ai_assistance/AiUtils.ts

@@ -8,6 +8,8 @@ import * as i18n from '../../core/i18n/i18n.js';
 import type * as Platform from '../../core/platform/platform.js';
 import * as Root from '../../core/root/root.js';
 
+import {debugLog} from './debug.js';
+
 const UIStrings = {
   /**
    * @description Message shown to the user if the age check is not successful.
@@ -71,3 +73,72 @@ export function isSameOrigin(url1: Platform.DevToolsPath.UrlString, url2: Platfo
   const origin2 = Common.ParsedURL.ParsedURL.extractOrigin(url2);
   return origin1 !== '' && origin1 === origin2;
 }
+
+export interface OneShotPromptRequest {
+  aidaClient: Host.AidaClient.AidaClient;
+  preamble: string;
+  query: string;
+  clientFeature: Host.AidaClient.ClientFeature;
+  temperature?: number;
+  modelId?: string;
+  userTier?: string;
+  serverSideLoggingEnabled?: boolean;
+  signal?: AbortSignal;
+}
+
+export async function runOneShotPrompt({
+  aidaClient,
+  preamble,
+  query,
+  clientFeature,
+  temperature,
+  modelId,
+  userTier,
+  serverSideLoggingEnabled,
+  signal,
+}: OneShotPromptRequest): Promise<string> {
+  const chromeVersion = Root.Runtime.getChromeVersion();
+  if (!chromeVersion) {
+    throw new Error('Cannot determine Chrome version');
+  }
+  const disallowLogging = !serverSideLoggingEnabled;
+  const sessionId = crypto.randomUUID();
+
+  const userTierEnum = Host.AidaClient.convertToUserTierEnum(userTier);
+  const finalPreamble = userTierEnum === Host.AidaClient.UserTier.TESTERS ? preamble : undefined;
+
+  const request: Host.AidaClient.DoConversationRequest = {
+    client: Host.AidaClient.CLIENT_NAME,
+    current_message: {
+      parts: [{text: query}],
+      role: Host.AidaClient.Role.USER,
+    },
+    preamble: finalPreamble,
+    options: {
+      temperature: typeof temperature === 'number' && temperature >= 0 ? temperature : undefined,
+      model_id: modelId || undefined,
+    },
+    metadata: {
+      disable_user_content_logging: disallowLogging,
+      string_session_id: sessionId,
+      user_tier: userTierEnum,
+      client_version: chromeVersion,
+    },
+    functionality_type: Host.AidaClient.FunctionalityType.CHAT,
+    client_feature: clientFeature,
+  };
+
+  let textResponse = '';
+  try {
+    for await (const response of aidaClient.doConversation(request, {signal})) {
+      if (response.explanation) {
+        textResponse = response.explanation;
+      }
+    }
+  } catch (err) {
+    debugLog('Error calling AIDA for one-shot prompt', err);
+    throw err;
+  }
+
+  return textResponse;
+}

package/front_end/models/ai_assistance/ChangeManager.ts

@@ -9,8 +9,6 @@ import type * as Protocol from '../../generated/protocol.js';
 
 export interface Change {
   groupId: string;
-  // Optional turn ID to group changes from the same turn.
-  turnId?: number;
   // Optional about where in the source the selector was defined.
   sourceLocation?: string;
   // Selector used by the page or a simple selector as the fallback.
@@ -104,7 +102,6 @@ export class ChangeManager {
       // it currently causes crashes in the Styles tab when duplicate selectors exist (crbug.com/393515428).
       // This workaround avoids that crash.
       existingChange.groupId = change.groupId;
-      existingChange.turnId = change.turnId;
     } else {
       changes.push({
         ...change,
@@ -126,11 +123,11 @@ export class ChangeManager {
         .join('\n\n');
   }
 
-  getChangedNodesForGroupId(groupId: string, turnId?: number): Protocol.DOM.BackendNodeId[] {
+  getChangedNodesForGroupId(groupId: string): Protocol.DOM.BackendNodeId[] {
     const nodes = new Set<Protocol.DOM.BackendNodeId>();
     for (const changes of this.#stylesheetChanges.values()) {
       for (const change of changes) {
-        if (change.groupId === groupId && change.backendNodeId && (turnId === undefined || change.turnId === turnId)) {
+        if (change.groupId === groupId && change.backendNodeId) {
           nodes.add(change.backendNodeId);
         }
       }

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();
+  }
+}