chrome-devtools-frontend 1.0.1577886 → 1.0.1578729

package/front_end/core/common/Revealer.ts

@@ -34,10 +34,6 @@ const UIStrings = {
    * @description The UI destination when right clicking an item that can be revealed
    */
   applicationPanel: 'Application panel',
-  /**
-   * @description The UI destination when right clicking an item that can be revealed
-   */
-  securityPanel: 'Security panel',
   /**
    * @description The UI destination when right clicking an item that can be revealed
    */
@@ -189,7 +185,6 @@ export const RevealerDestination = {
   TIMELINE_PANEL: i18nLazyString(UIStrings.timelinePanel),
   APPLICATION_PANEL: i18nLazyString(UIStrings.applicationPanel),
   SOURCES_PANEL: i18nLazyString(UIStrings.sourcesPanel),
-  SECURITY_PANEL: i18nLazyString(UIStrings.securityPanel),
   MEMORY_INSPECTOR_PANEL: i18nLazyString(UIStrings.memoryInspectorPanel),
   ANIMATIONS_PANEL: i18nLazyString(UIStrings.animationsPanel),
 };

package/front_end/core/sdk/ConsoleModel.ts

@@ -545,7 +545,6 @@ export interface ConsoleMessageDetails {
   context?: string;
   affectedResources?: AffectedResources;
   category?: Protocol.Log.LogEntryCategory;
-  isCookieReportIssue?: boolean;
 }
 
 export class ConsoleMessage {
@@ -569,7 +568,6 @@ export class ConsoleMessage {
   #exceptionId?: number = undefined;
   #affectedResources?: AffectedResources;
   category?: Protocol.Log.LogEntryCategory;
-  isCookieReportIssue = false;
 
   /**
    * The parent frame of the `console.log` call of logpoints or conditional breakpoints
@@ -600,7 +598,6 @@ export class ConsoleMessage {
     this.workerId = details?.workerId;
     this.#affectedResources = details?.affectedResources;
     this.category = details?.category;
-    this.isCookieReportIssue = Boolean(details?.isCookieReportIssue);
 
     if (!this.#executionContextId && this.#runtimeModel) {
       if (this.scriptId) {

package/front_end/core/sdk/HeapProfilerModel.ts

@@ -11,6 +11,7 @@ import type {RemoteObject} from './RemoteObject.js';
 import {RuntimeModel} from './RuntimeModel.js';
 import {SDKModel} from './SDKModel.js';
 import {Capability, type Target} from './Target.js';
+import {TargetManager} from './TargetManager.js';
 
 export class HeapProfilerModel extends SDKModel<EventTypes> {
   #enabled: boolean;
@@ -106,7 +107,12 @@ export class HeapProfilerModel extends SDKModel<EventTypes> {
   }
 
   async takeHeapSnapshot(heapSnapshotOptions: Protocol.HeapProfiler.TakeHeapSnapshotRequest): Promise<void> {
-    await this.#heapProfilerAgent.invoke_takeHeapSnapshot(heapSnapshotOptions);
+    await TargetManager.instance().suspendAllTargets('heap-snapshot');
+    try {
+      await this.#heapProfilerAgent.invoke_takeHeapSnapshot(heapSnapshotOptions);
+    } finally {
+      await TargetManager.instance().resumeAllTargets();
+    }
   }
 
   async startTrackingHeapObjects(recordAllocationStacks: boolean): Promise<boolean> {

package/front_end/generated/InspectorBackendCommands.ts

@@ -783,7 +783,7 @@ inspectorBackend.registerEnum("Network.InterceptionStage", {Request: "Request",
 inspectorBackend.registerEnum("Network.SignedExchangeErrorField", {SignatureSig: "signatureSig", SignatureIntegrity: "signatureIntegrity", SignatureCertUrl: "signatureCertUrl", SignatureCertSha256: "signatureCertSha256", SignatureValidityUrl: "signatureValidityUrl", SignatureTimestamps: "signatureTimestamps"});
 inspectorBackend.registerEnum("Network.ContentEncoding", {Deflate: "deflate", Gzip: "gzip", Br: "br", Zstd: "zstd"});
 inspectorBackend.registerEnum("Network.DirectSocketDnsQueryType", {Ipv4: "ipv4", Ipv6: "ipv6"});
-inspectorBackend.registerEnum("Network.PrivateNetworkRequestPolicy", {Allow: "Allow", BlockFromInsecureToMorePrivate: "BlockFromInsecureToMorePrivate", WarnFromInsecureToMorePrivate: "WarnFromInsecureToMorePrivate", PermissionBlock: "PermissionBlock", PermissionWarn: "PermissionWarn"});
+inspectorBackend.registerEnum("Network.LocalNetworkAccessRequestPolicy", {Allow: "Allow", BlockFromInsecureToMorePrivate: "BlockFromInsecureToMorePrivate", WarnFromInsecureToMorePrivate: "WarnFromInsecureToMorePrivate", PermissionBlock: "PermissionBlock", PermissionWarn: "PermissionWarn"});
 inspectorBackend.registerEnum("Network.IPAddressSpace", {Loopback: "Loopback", Local: "Local", Public: "Public", Unknown: "Unknown"});
 inspectorBackend.registerEnum("Network.CrossOriginOpenerPolicyValue", {SameOrigin: "SameOrigin", SameOriginAllowPopups: "SameOriginAllowPopups", RestrictProperties: "RestrictProperties", UnsafeNone: "UnsafeNone", SameOriginPlusCoep: "SameOriginPlusCoep", RestrictPropertiesPlusCoep: "RestrictPropertiesPlusCoep", NoopenerAllowPopups: "NoopenerAllowPopups"});
 inspectorBackend.registerEnum("Network.CrossOriginEmbedderPolicyValue", {None: "None", Credentialless: "Credentialless", RequireCorp: "RequireCorp"});
@@ -913,7 +913,7 @@ inspectorBackend.registerType("Network.DirectTCPSocketOptions", [{"name": "noDel
 inspectorBackend.registerType("Network.DirectUDPSocketOptions", [{"name": "remoteAddr", "type": "string", "optional": true, "description": "", "typeRef": null}, {"name": "remotePort", "type": "number", "optional": true, "description": "Unsigned int 16.", "typeRef": null}, {"name": "localAddr", "type": "string", "optional": true, "description": "", "typeRef": null}, {"name": "localPort", "type": "number", "optional": true, "description": "Unsigned int 16.", "typeRef": null}, {"name": "dnsQueryType", "type": "string", "optional": true, "description": "", "typeRef": "Network.DirectSocketDnsQueryType"}, {"name": "sendBufferSize", "type": "number", "optional": true, "description": "Expected to be unsigned integer.", "typeRef": null}, {"name": "receiveBufferSize", "type": "number", "optional": true, "description": "Expected to be unsigned integer.", "typeRef": null}, {"name": "multicastLoopback", "type": "boolean", "optional": true, "description": "", "typeRef": null}, {"name": "multicastTimeToLive", "type": "number", "optional": true, "description": "Unsigned int 8.", "typeRef": null}, {"name": "multicastAllowAddressSharing", "type": "boolean", "optional": true, "description": "", "typeRef": null}]);
 inspectorBackend.registerType("Network.DirectUDPMessage", [{"name": "data", "type": "string", "optional": false, "description": "", "typeRef": null}, {"name": "remoteAddr", "type": "string", "optional": true, "description": "Null for connected mode.", "typeRef": null}, {"name": "remotePort", "type": "number", "optional": true, "description": "Null for connected mode. Expected to be unsigned integer.", "typeRef": null}]);
 inspectorBackend.registerType("Network.ConnectTiming", [{"name": "requestTime", "type": "number", "optional": false, "description": "Timing's requestTime is a baseline in seconds, while the other numbers are ticks in milliseconds relatively to this requestTime. Matches ResourceTiming's requestTime for the same request (but not for redirected requests).", "typeRef": null}]);
-inspectorBackend.registerType("Network.ClientSecurityState", [{"name": "initiatorIsSecureContext", "type": "boolean", "optional": false, "description": "", "typeRef": null}, {"name": "initiatorIPAddressSpace", "type": "string", "optional": false, "description": "", "typeRef": "Network.IPAddressSpace"}, {"name": "privateNetworkRequestPolicy", "type": "string", "optional": false, "description": "", "typeRef": "Network.PrivateNetworkRequestPolicy"}]);
+inspectorBackend.registerType("Network.ClientSecurityState", [{"name": "initiatorIsSecureContext", "type": "boolean", "optional": false, "description": "", "typeRef": null}, {"name": "initiatorIPAddressSpace", "type": "string", "optional": false, "description": "", "typeRef": "Network.IPAddressSpace"}, {"name": "localNetworkAccessRequestPolicy", "type": "string", "optional": false, "description": "", "typeRef": "Network.LocalNetworkAccessRequestPolicy"}]);
 inspectorBackend.registerType("Network.CrossOriginOpenerPolicyStatus", [{"name": "value", "type": "string", "optional": false, "description": "", "typeRef": "Network.CrossOriginOpenerPolicyValue"}, {"name": "reportOnlyValue", "type": "string", "optional": false, "description": "", "typeRef": "Network.CrossOriginOpenerPolicyValue"}, {"name": "reportingEndpoint", "type": "string", "optional": true, "description": "", "typeRef": null}, {"name": "reportOnlyReportingEndpoint", "type": "string", "optional": true, "description": "", "typeRef": null}]);
 inspectorBackend.registerType("Network.CrossOriginEmbedderPolicyStatus", [{"name": "value", "type": "string", "optional": false, "description": "", "typeRef": "Network.CrossOriginEmbedderPolicyValue"}, {"name": "reportOnlyValue", "type": "string", "optional": false, "description": "", "typeRef": "Network.CrossOriginEmbedderPolicyValue"}, {"name": "reportingEndpoint", "type": "string", "optional": true, "description": "", "typeRef": null}, {"name": "reportOnlyReportingEndpoint", "type": "string", "optional": true, "description": "", "typeRef": null}]);
 inspectorBackend.registerType("Network.ContentSecurityPolicyStatus", [{"name": "effectiveDirectives", "type": "string", "optional": false, "description": "", "typeRef": null}, {"name": "isEnforced", "type": "boolean", "optional": false, "description": "", "typeRef": null}, {"name": "source", "type": "string", "optional": false, "description": "", "typeRef": "Network.ContentSecurityPolicySource"}]);

package/front_end/generated/protocol.ts

@@ -11223,7 +11223,7 @@ export namespace Network {
     remotePort?: integer;
   }
 
-  export const enum PrivateNetworkRequestPolicy {
+  export const enum LocalNetworkAccessRequestPolicy {
     Allow = 'Allow',
     BlockFromInsecureToMorePrivate = 'BlockFromInsecureToMorePrivate',
     WarnFromInsecureToMorePrivate = 'WarnFromInsecureToMorePrivate',
@@ -11250,7 +11250,7 @@ export namespace Network {
   export interface ClientSecurityState {
     initiatorIsSecureContext: boolean;
     initiatorIPAddressSpace: IPAddressSpace;
-    privateNetworkRequestPolicy: PrivateNetworkRequestPolicy;
+    localNetworkAccessRequestPolicy: LocalNetworkAccessRequestPolicy;
   }
 
   export const enum CrossOriginOpenerPolicyValue {

package/front_end/models/ai_assistance/AiConversation.ts

@@ -17,10 +17,10 @@ import {
   ResponseType
 } from './agents/AiAgent.js';
 import {ContextSelectionAgent} from './agents/ContextSelectionAgent.js';
-import {FileAgent} from './agents/FileAgent.js';
-import {NetworkAgent} from './agents/NetworkAgent.js';
-import {PerformanceAgent} from './agents/PerformanceAgent.js';
-import {StylingAgent} from './agents/StylingAgent.js';
+import {FileAgent, FileContext} from './agents/FileAgent.js';
+import {NetworkAgent, RequestContext} from './agents/NetworkAgent.js';
+import {PerformanceAgent, PerformanceTraceContext} from './agents/PerformanceAgent.js';
+import {NodeContext, StylingAgent} from './agents/StylingAgent.js';
 import {AiHistoryStorage, ConversationType, type SerializedConversation} from './AiHistoryStorage.js';
 import type {ChangeManager} from './ChangeManager.js';
 import {NetworkRequestFormatter} from './data_formatters/NetworkRequestFormatter.js';
@@ -59,14 +59,17 @@ export class AiConversation {
   }
 
   readonly id: string;
-  #type: ConversationType;
+  // Handled in #updateAgent
+  #type!: ConversationType;
+  // Handled in #updateAgent
+  #agent!: AiAgent<unknown>;
+
   #isReadOnly: boolean;
   readonly history: ResponseData[];
   #isExternal: boolean;
 
   #aidaClient: Host.AidaClient.AidaClient;
   #changeManager: ChangeManager|undefined;
-  #agent: AiAgent<unknown>;
   #origin?: string;
 
   #contexts: Array<ConversationContext<unknown>> = [];
@@ -82,14 +85,13 @@ export class AiConversation {
   ) {
     this.#changeManager = changeManager;
     this.#aidaClient = aidaClient;
-    this.#type = type;
 
     this.id = id;
     this.#isReadOnly = isReadOnly;
     this.#isExternal = isExternal;
     this.history = this.#reconstructHistory(data);
     // Needs to be last
-    this.#agent = this.#createAgent();
+    this.#updateAgent(type);
   }
 
   get isReadOnly(): boolean {
@@ -123,10 +125,26 @@ export class AiConversation {
   setContext(updateContext: ConversationContext<unknown>|null): void {
     if (!updateContext) {
       this.#contexts = [];
+      if (isAiAssistanceContextSelectionAgentEnabled()) {
+        this.#updateAgent(ConversationType.NONE);
+      }
+
       return;
     }
 
     this.#contexts = [updateContext];
+
+    if (isAiAssistanceContextSelectionAgentEnabled()) {
+      if (updateContext instanceof FileContext) {
+        this.#updateAgent(ConversationType.FILE);
+      } else if (updateContext instanceof NodeContext) {
+        this.#updateAgent(ConversationType.STYLING);
+      } else if (updateContext instanceof RequestContext) {
+        this.#updateAgent(ConversationType.NETWORK);
+      } else if (updateContext instanceof PerformanceTraceContext) {
+        this.#updateAgent(ConversationType.PERFORMANCE);
+      }
+    }
   }
 
   get selectedContext(): ConversationContext<unknown>|undefined {
@@ -249,37 +267,40 @@ export class AiConversation {
     };
   }
 
-  #createAgent(): AiAgent<unknown> {
+  #updateAgent(type: ConversationType): void {
+    if (this.#type === type) {
+      return;
+    }
+
+    this.#type = type;
     const options = {
       aidaClient: this.#aidaClient,
       serverSideLoggingEnabled: isAiAssistanceServerSideLoggingEnabled(),
       sessionId: this.id,
       changeManager: this.#changeManager,
     };
-    let agent: AiAgent<unknown>;
-    switch (this.#type) {
+    switch (type) {
       case ConversationType.STYLING: {
-        agent = new StylingAgent(options);
+        this.#agent = new StylingAgent(options);
         break;
       }
       case ConversationType.NETWORK: {
-        agent = new NetworkAgent(options);
+        this.#agent = new NetworkAgent(options);
         break;
       }
       case ConversationType.FILE: {
-        agent = new FileAgent(options);
+        this.#agent = new FileAgent(options);
         break;
       }
       case ConversationType.PERFORMANCE: {
-        agent = new PerformanceAgent(options);
+        this.#agent = new PerformanceAgent(options);
         break;
       }
       case ConversationType.NONE: {
-        agent = new ContextSelectionAgent(options);
+        this.#agent = new ContextSelectionAgent(options);
         break;
       }
     }
-    return agent;
   }
 
   #factsCache = new Map<ExtraContext, Host.AidaClient.RequestFact>();
@@ -399,11 +420,6 @@ Time: ${micros(time)}`;
         },
         options.multimodalInput,
         )) {
-      if (data.type === ResponseType.CONTEXT_CHANGE) {
-        this.#type = ConversationType.NETWORK;
-        this.#agent = this.#createAgent();
-      }
-
       if (shouldAddToHistory(data)) {
         void this.addHistoryItem(data);
       }
@@ -439,3 +455,7 @@ function isAiAssistanceServerSideLoggingEnabled(): boolean {
 type ExtraContext = SDK.DOMModel.DOMNode|SDK.NetworkRequest.NetworkRequest|
                     {event: Trace.Types.Events.Event, traceStartTime: Trace.Types.Timing.Micro}|
                     {insight: Trace.Insights.Types.InsightModel, trace: Trace.TraceModel.ParsedTrace};
+
+function isAiAssistanceContextSelectionAgentEnabled(): boolean {
+  return Boolean(Root.Runtime.hostConfig.devToolsAiAssistanceContextSelectionAgent?.enabled);
+}

package/front_end/models/issues_manager/CookieIssue.ts

@@ -327,7 +327,6 @@ export class CookieIssue extends Issue<Protocol.Audits.CookieIssueDetails> {
           Protocol.Log.LogEntryLevel.Warning, UIStrings.consoleTpcdErrorMessage, {
             url: this.details().request?.url as Platform.DevToolsPath.UrlString | undefined,
             affectedResources: {requestId: this.details().request?.requestId, issueId: this.issueId},
-            isCookieReportIssue: true
           });
     }
     return;

package/front_end/models/issues_manager/IssuesManager.ts

@@ -3,7 +3,6 @@
 // found in the LICENSE file.
 
 import * as Common from '../../core/common/common.js';
-import * as Root from '../../core/root/root.js';
 import * as SDK from '../../core/sdk/sdk.js';
 import * as Protocol from '../../generated/protocol.js';
 
@@ -12,7 +11,7 @@ import {BounceTrackingIssue} from './BounceTrackingIssue.js';
 import {ClientHintIssue} from './ClientHintIssue.js';
 import {ContentSecurityPolicyIssue} from './ContentSecurityPolicyIssue.js';
 import {CookieDeprecationMetadataIssue} from './CookieDeprecationMetadataIssue.js';
-import {CookieIssue, CookieIssueSubCategory} from './CookieIssue.js';
+import {CookieIssue} from './CookieIssue.js';
 import {CorsIssue} from './CorsIssue.js';
 import {CrossOriginEmbedderPolicyIssue, isCrossOriginEmbedderPolicyIssue} from './CrossOriginEmbedderPolicyIssue.js';
 import {DeprecationIssue} from './DeprecationIssue.js';
@@ -211,7 +210,6 @@ export class IssuesManager extends Common.ObjectWrapper.ObjectWrapper<EventTypes
   #thirdPartyCookiePhaseoutIssueCount = new Map<IssueKind, number>();
   #issuesById = new Map<string, Issue>();
   #issuesByOutermostTarget: WeakMap<SDK.Target.Target, Set<Issue>> = new Map();
-  #thirdPartyCookiePhaseoutIssueMessageSent = false;
 
   constructor(
       private readonly showThirdPartyIssuesSetting?: Common.Settings.Setting<boolean>,
@@ -314,7 +312,6 @@ export class IssuesManager extends Common.ObjectWrapper.ObjectWrapper<EventTypes
 
   #onIssueAddedEvent(event: Common.EventTarget.EventTargetEvent<SDK.IssuesModel.IssueAddedEvent>): void {
     const {issuesModel, inspectorIssue} = event.data;
-    const isPrivacyUiEnabled = Root.Runtime.hostConfig.devToolsPrivacyUI?.enabled;
 
     const issues = createIssuesFromProtocolIssue(issuesModel, inspectorIssue);
     for (const issue of issues) {
@@ -323,16 +320,7 @@ export class IssuesManager extends Common.ObjectWrapper.ObjectWrapper<EventTypes
       if (!message) {
         continue;
       }
-
-      // Only show one message for third-party cookie phaseout issues if the new privacy ui is enabled
-      const is3rdPartyCookiePhaseoutIssue =
-          CookieIssue.getSubCategory(issue.code()) === CookieIssueSubCategory.THIRD_PARTY_PHASEOUT_COOKIE;
-      if (!is3rdPartyCookiePhaseoutIssue || !isPrivacyUiEnabled || !this.#thirdPartyCookiePhaseoutIssueMessageSent) {
-        issuesModel.target().model(SDK.ConsoleModel.ConsoleModel)?.addMessage(message);
-      }
-      if (is3rdPartyCookiePhaseoutIssue && isPrivacyUiEnabled) {
-        this.#thirdPartyCookiePhaseoutIssueMessageSent = true;
-      }
+      issuesModel.target().model(SDK.ConsoleModel.ConsoleModel)?.addMessage(message);
     }
   }
 
@@ -453,7 +441,6 @@ export class IssuesManager extends Common.ObjectWrapper.ObjectWrapper<EventTypes
     this.#issuesById.clear();
     this.#hiddenIssueCount.clear();
     this.#thirdPartyCookiePhaseoutIssueCount.clear();
-    this.#thirdPartyCookiePhaseoutIssueMessageSent = false;
     const values = this.hideIssueSetting?.get();
     for (const [key, issue] of this.#allIssues) {
       if (this.#issueFilter(issue)) {

package/front_end/panels/ai_assistance/AiAssistancePanel.ts

@@ -729,7 +729,9 @@ export class AiAssistancePanel extends UI.Panel.Panel {
       this.#conversation = conversation;
     }
 
-    this.#conversation?.setContext(this.#getConversationContext(this.#conversation));
+    this.#conversation?.setContext(this.#getConversationContext(
+        isAiAssistanceContextSelectionAgentEnabled() ? this.#getDefaultConversationType() :
+                                                       (this.#conversation?.type ?? null)));
 
     this.requestUpdate();
   }
@@ -1269,12 +1271,9 @@ export class AiAssistancePanel extends UI.Panel.Panel {
     this.#runAbortController = new AbortController();
   }
 
-  #getConversationContext(conversation?: AiAssistanceModel.AiConversation.AiConversation):
+  #getConversationContext(type?: AiAssistanceModel.AiHistoryStorage.ConversationType):
       AiAssistanceModel.AiAgent.ConversationContext<unknown>|null {
-    if (!conversation) {
-      return null;
-    }
-    switch (conversation.type) {
+    switch (type) {
       case AiAssistanceModel.AiHistoryStorage.ConversationType.STYLING:
         return this.#selectedElement;
       case AiAssistanceModel.AiHistoryStorage.ConversationType.FILE:
@@ -1284,35 +1283,28 @@ export class AiAssistancePanel extends UI.Panel.Panel {
       case AiAssistanceModel.AiHistoryStorage.ConversationType.PERFORMANCE:
         return this.#selectedPerformanceTrace;
       case AiAssistanceModel.AiHistoryStorage.ConversationType.NONE:
+      case undefined:
         return null;
     }
   }
 
   #handleConversationContextChange = (data: unknown): void => {
     if (data instanceof Workspace.UISourceCode.UISourceCode) {
-      if (this.#selectedFile?.getItem() === data) {
-        return;
-      }
-      this.#selectedFile = new AiAssistanceModel.FileAgent.FileContext(data);
+      const context = new AiAssistanceModel.FileAgent.FileContext(data);
+      this.#selectedFile = context;
 
     } else if (data instanceof SDK.DOMModel.DOMNode) {
-      if (this.#selectedElement?.getItem() === data ||
-          // Ignore non node type like comments or html tags
-          data.nodeType() === Node.ELEMENT_NODE) {
-        return;
-      }
-      this.#selectedElement = new AiAssistanceModel.StylingAgent.NodeContext(data);
+      const context = new AiAssistanceModel.StylingAgent.NodeContext(data);
+      this.#selectedElement = context;
+
     } else if (data instanceof SDK.NetworkRequest.NetworkRequest) {
-      if (this.#selectedRequest?.getItem() === data) {
-        return;
-      }
       const calculator = NetworkPanel.NetworkPanel.NetworkPanel.instance().networkLogView.timeCalculator();
-      this.#selectedRequest = new AiAssistanceModel.NetworkAgent.RequestContext(data, calculator);
+
+      const context = new AiAssistanceModel.NetworkAgent.RequestContext(data, calculator);
+      this.#selectedRequest = context;
     } else if (data instanceof AiAssistanceModel.AIContext.AgentFocus) {
-      if (this.#selectedPerformanceTrace?.getItem() === data) {
-        return;
-      }
-      this.#selectedPerformanceTrace = new AiAssistanceModel.PerformanceAgent.PerformanceTraceContext(data);
+      const context = new AiAssistanceModel.PerformanceAgent.PerformanceTraceContext(data);
+      this.#selectedPerformanceTrace = context;
     }
 
     this.#updateConversationState(this.#conversation);
@@ -1329,7 +1321,7 @@ export class AiAssistancePanel extends UI.Panel.Panel {
     // Cancel any previous in-flight conversation.
     this.#cancel();
     const signal = this.#runAbortController.signal;
-    const context = this.#getConversationContext(this.#conversation);
+    const context = this.#getConversationContext(this.#conversation.type);
     this.#conversation.setContext(context);
 
     // If a different context is provided, it must be from the same origin.

package/front_end/panels/ai_assistance/ai_assistance-meta.ts

@@ -174,7 +174,6 @@ UI.ActionRegistration.registerActionExtension({
     return new AiAssistance.ActionDelegate();
   },
   condition: config => isAnyFeatureAvailable(config) && !isPolicyRestricted(config) && !isGeoRestricted(config),
-  featurePromotionId: 'debug-with-ai',
 });
 
 UI.ActionRegistration.registerActionExtension({

package/front_end/panels/ai_assistance/components/ChatMessage.ts

@@ -323,7 +323,6 @@ export const DEFAULT_VIEW = (input: ChatMessageViewInput, output: ViewOutput, ta
           }
           return renderStep({
             step: part.step,
-            isLoading: input.isLoading,
             markdownRenderer: input.markdownRenderer,
             isLast: isLastPart,
           });
@@ -451,12 +450,11 @@ function renderStepDetails({
   // clang-format on
 }
 
-function renderStepBadge({step, isLoading, isLast}: {
+function renderStepBadge({step, isLast}: {
   step: Step,
-  isLoading: boolean,
   isLast: boolean,
 }): Lit.LitTemplate {
-  if (isLoading && isLast && !step.sideEffect) {
+  if (step.isLoading && isLast && !step.sideEffect) {
     return html`<devtools-spinner></devtools-spinner>`;
   }
 
@@ -480,9 +478,8 @@ function renderStepBadge({step, isLoading, isLast}: {
     ></devtools-icon>`;
 }
 
-function renderStep({step, isLoading, markdownRenderer, isLast}: {
+function renderStep({step, markdownRenderer, isLast}: {
   step: Step,
-  isLoading: boolean,
   markdownRenderer: MarkdownLitRenderer,
   isLast: boolean,
 }): Lit.LitTemplate {
@@ -499,7 +496,7 @@ function renderStep({step, isLoading, markdownRenderer, isLast}: {
       .open=${Boolean(step.sideEffect)}>
       <summary>
         <div class="summary">
-          ${renderStepBadge({ step, isLoading, isLast })}
+          ${renderStepBadge({ step, isLast })}
           ${renderTitle(step)}
           <devtools-icon
             class="arrow"

package/front_end/panels/ai_assistance/components/ChatView.ts

@@ -105,7 +105,7 @@ const DEFAULT_VIEW: View = (input, output, target) => {
               ${repeat(input.messages, message =>
                 html`<devtools-widget .widgetConfig=${UI.Widget.widgetConfig(ChatMessage, {
                   message,
-                  isLoading: input.isLoading,
+                  isLoading: input.isLoading && input.messages.at(-1) === message,
                   isReadOnly: input.isReadOnly,
                   canShowFeedbackForm: input.canShowFeedbackForm,
                   userInfo: input.userInfo,