chrome-devtools-frontend 1.0.1386602 → 1.0.1418433

package/front_end/panels/{freestyler/FreestylerPanel.test.ts → ai_assistance/AiAssistancePanel.test.ts}

@@ -4,9 +4,10 @@
 
 import * as Common from '../../core/common/common.js';
 import * as Host from '../../core/host/host.js';
-import type * as Platform from '../../core/platform/platform.js';
+import * as Platform from '../../core/platform/platform.js';
 import * as SDK from '../../core/sdk/sdk.js';
 import * as Workspace from '../../models/workspace/workspace.js';
+import {mockAidaClient} from '../../testing/AiAssistanceHelpers.js';
 import {findMenuItemWithLabel, getMenu} from '../../testing/ContextMenuHelpers.js';
 import {dispatchClickEvent} from '../../testing/DOMHelpers.js';
 import {describeWithEnvironment, getGetHostConfigStub, registerNoopActions} from '../../testing/EnvironmentHelpers.js';
@@ -17,18 +18,9 @@ import * as SourcesPanel from '../sources/sources.js';
 import * as TimelinePanel from '../timeline/timeline.js';
 import * as TimelineUtils from '../timeline/utils/utils.js';
 
-import * as AiAgent from './AiAgent.js';
-import * as Freestyler from './freestyler.js';
+import * as AiAssistance from './ai_assistance.js';
 
-function getTestAidaClient() {
-  return {
-    async *
-        fetch() {
-          yield {explanation: 'test', metadata: {}, completed: true};
-        },
-    registerClientEvent: sinon.spy(),
-  };
-}
+const {urlString} = Platform.DevToolsPath;
 
 function getTestSyncInfo(): Host.InspectorFrontendHostAPI.SyncInformation {
   return {isSyncActive: true};
@@ -39,8 +31,8 @@ async function drainMicroTasks() {
 }
 
 describeWithEnvironment('FreestylerPanel', () => {
-  let mockView: sinon.SinonStub<[Freestyler.Props, unknown, HTMLElement]>;
-  let panel: Freestyler.FreestylerPanel;
+  let mockView: sinon.SinonStub<[AiAssistance.Props, unknown, HTMLElement]>;
+  let panel: AiAssistance.AiAssistancePanel;
 
   beforeEach(() => {
     mockView = sinon.stub();
@@ -57,56 +49,56 @@ describeWithEnvironment('FreestylerPanel', () => {
 
   describe('consent view', () => {
     it('should render consent view when the consent is not given before', async () => {
-      panel = new Freestyler.FreestylerPanel(mockView, {
-        aidaClient: getTestAidaClient(),
+      panel = new AiAssistance.AiAssistancePanel(mockView, {
+        aidaClient: mockAidaClient([[{explanation: 'test'}]]),
         aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
         syncInfo: getTestSyncInfo(),
       });
       panel.markAsRoot();
       panel.show(document.body);
 
-      sinon.assert.calledWith(mockView, sinon.match({state: Freestyler.State.CONSENT_VIEW}));
+      sinon.assert.calledWith(mockView, sinon.match({state: AiAssistance.State.CONSENT_VIEW}));
     });
 
     it('should switch from consent view to chat view when enabling setting', async () => {
-      panel = new Freestyler.FreestylerPanel(mockView, {
-        aidaClient: getTestAidaClient(),
+      panel = new AiAssistance.AiAssistancePanel(mockView, {
+        aidaClient: mockAidaClient([[{explanation: 'test'}]]),
         aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
         syncInfo: getTestSyncInfo(),
       });
       panel.markAsRoot();
       panel.show(document.body);
-      sinon.assert.calledWith(mockView, sinon.match({state: Freestyler.State.CONSENT_VIEW}));
+      sinon.assert.calledWith(mockView, sinon.match({state: AiAssistance.State.CONSENT_VIEW}));
 
       Common.Settings.moduleSetting('ai-assistance-enabled').set(true);
-      sinon.assert.calledWith(mockView, sinon.match({state: Freestyler.State.CHAT_VIEW}));
+      sinon.assert.calledWith(mockView, sinon.match({state: AiAssistance.State.CHAT_VIEW}));
       await drainMicroTasks();
     });
 
     it('should render chat view when the consent is given before', async () => {
       Common.Settings.moduleSetting('ai-assistance-enabled').set(true);
 
-      panel = new Freestyler.FreestylerPanel(mockView, {
-        aidaClient: getTestAidaClient(),
+      panel = new AiAssistance.AiAssistancePanel(mockView, {
+        aidaClient: mockAidaClient([[{explanation: 'test'}]]),
         aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
         syncInfo: getTestSyncInfo(),
       });
       panel.markAsRoot();
       panel.show(document.body);
 
-      sinon.assert.calledWith(mockView, sinon.match({state: Freestyler.State.CHAT_VIEW}));
+      sinon.assert.calledWith(mockView, sinon.match({state: AiAssistance.State.CHAT_VIEW}));
     });
 
     it('should render the consent view when the setting is disabled', async () => {
       Common.Settings.moduleSetting('ai-assistance-enabled').set(true);
       Common.Settings.moduleSetting('ai-assistance-enabled').setDisabled(true);
-      const chatUiStates: Freestyler.State[] = [];
+      const chatUiStates: AiAssistance.State[] = [];
       const viewStub = sinon.stub().callsFake(props => {
         chatUiStates.push(props.state);
       });
 
-      panel = new Freestyler.FreestylerPanel(viewStub, {
-        aidaClient: getTestAidaClient(),
+      panel = new AiAssistance.AiAssistancePanel(viewStub, {
+        aidaClient: mockAidaClient([[{explanation: 'test'}]]),
         aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
         syncInfo: getTestSyncInfo(),
       });
@@ -114,8 +106,8 @@ describeWithEnvironment('FreestylerPanel', () => {
       panel.show(document.body);
       await drainMicroTasks();
 
-      sinon.assert.calledWith(viewStub, sinon.match({state: Freestyler.State.CONSENT_VIEW}));
-      assert.isFalse(chatUiStates.includes(Freestyler.State.CHAT_VIEW));
+      sinon.assert.calledWith(viewStub, sinon.match({state: AiAssistance.State.CONSENT_VIEW}));
+      assert.isFalse(chatUiStates.includes(AiAssistance.State.CHAT_VIEW));
       Common.Settings.moduleSetting('ai-assistance-enabled').setDisabled(false);
     });
 
@@ -130,23 +122,23 @@ describeWithEnvironment('FreestylerPanel', () => {
         },
       });
 
-      panel = new Freestyler.FreestylerPanel(mockView, {
-        aidaClient: getTestAidaClient(),
+      panel = new AiAssistance.AiAssistancePanel(mockView, {
+        aidaClient: mockAidaClient([[{explanation: 'test'}]]),
         aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
         syncInfo: getTestSyncInfo(),
       });
       panel.markAsRoot();
       panel.show(document.body);
 
-      sinon.assert.calledWith(mockView, sinon.match({state: Freestyler.State.CONSENT_VIEW}));
+      sinon.assert.calledWith(mockView, sinon.match({state: AiAssistance.State.CONSENT_VIEW}));
       stub.restore();
     });
 
     it('updates when the user logs in', async () => {
       Common.Settings.moduleSetting('ai-assistance-enabled').set(true);
 
-      panel = new Freestyler.FreestylerPanel(mockView, {
-        aidaClient: getTestAidaClient(),
+      panel = new AiAssistance.AiAssistancePanel(mockView, {
+        aidaClient: mockAidaClient([[{explanation: 'test'}]]),
         aidaAvailability: Host.AidaClient.AidaAccessPreconditions.NO_ACCOUNT_EMAIL,
         syncInfo: getTestSyncInfo(),
       });
@@ -154,7 +146,7 @@ describeWithEnvironment('FreestylerPanel', () => {
       panel.show(document.body);
       await drainMicroTasks();
       sinon.assert.calledWith(mockView, sinon.match({
-        state: Freestyler.State.CHAT_VIEW,
+        state: AiAssistance.State.CHAT_VIEW,
         aidaAvailability: Host.AidaClient.AidaAccessPreconditions.NO_ACCOUNT_EMAIL,
       }));
       mockView.reset();
@@ -165,7 +157,7 @@ describeWithEnvironment('FreestylerPanel', () => {
           Host.AidaClient.Events.AIDA_AVAILABILITY_CHANGED);
       await drainMicroTasks();
       sinon.assert.calledWith(mockView, sinon.match({
-        state: Freestyler.State.CHAT_VIEW,
+        state: AiAssistance.State.CHAT_VIEW,
         aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
       }));
 
@@ -174,21 +166,15 @@ describeWithEnvironment('FreestylerPanel', () => {
   });
 
   describe('on rate click', () => {
-    afterEach(() => {
-      // @ts-expect-error global test variable
-      setFreestylerServerSideLoggingEnabled(false);
-    });
-
     it('renders a button linking to settings', () => {
       const stub = sinon.stub(UI.ViewManager.ViewManager.instance(), 'showView');
 
-      panel = new Freestyler.FreestylerPanel(mockView, {
-        aidaClient: getTestAidaClient(),
+      panel = new AiAssistance.AiAssistancePanel(mockView, {
+        aidaClient: mockAidaClient([[{explanation: 'test'}]]),
         aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
         syncInfo: getTestSyncInfo(),
       });
-      const toolbar = panel.contentElement.querySelector('.freestyler-right-toolbar');
-      const button = toolbar!.shadowRoot!.querySelector('devtools-button[aria-label=\'Settings\']');
+      const button = panel.contentElement.querySelector('devtools-button[aria-label=\'Settings\']');
       assert.instanceOf(button, HTMLElement);
       button.click();
       assert.isTrue(stub.calledWith('chrome-ai'));
@@ -196,16 +182,14 @@ describeWithEnvironment('FreestylerPanel', () => {
     });
 
     it('should allow logging if configured', () => {
-      // @ts-expect-error global test variable
-      setFreestylerServerSideLoggingEnabled(true);
       const stub = getGetHostConfigStub({
         aidaAvailability: {
           disallowLogging: false,
         },
       });
 
-      const aidaClient = getTestAidaClient();
-      panel = new Freestyler.FreestylerPanel(mockView, {
+      const aidaClient = mockAidaClient([[{explanation: 'test'}]]);
+      panel = new AiAssistance.AiAssistancePanel(mockView, {
         aidaClient,
         aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
         syncInfo: getTestSyncInfo(),
@@ -216,7 +200,7 @@ describeWithEnvironment('FreestylerPanel', () => {
       mockView.reset();
       callArgs.onFeedbackSubmit(0, Host.AidaClient.Rating.POSITIVE);
 
-      sinon.assert.match(aidaClient.registerClientEvent.firstCall.firstArg, sinon.match({
+      sinon.assert.match((aidaClient.registerClientEvent as sinon.SinonStub).firstCall.firstArg, sinon.match({
         disable_user_content_logging: false,
       }));
       stub.restore();
@@ -225,8 +209,8 @@ describeWithEnvironment('FreestylerPanel', () => {
     it('should send POSITIVE rating to aida client when the user clicks on positive rating', () => {
       const RPC_ID = 0;
 
-      const aidaClient = getTestAidaClient();
-      panel = new Freestyler.FreestylerPanel(mockView, {
+      const aidaClient = mockAidaClient([[{explanation: 'test'}]]);
+      panel = new AiAssistance.AiAssistancePanel(mockView, {
         aidaClient,
         aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
         syncInfo: getTestSyncInfo(),
@@ -237,7 +221,7 @@ describeWithEnvironment('FreestylerPanel', () => {
       mockView.reset();
       callArgs.onFeedbackSubmit(RPC_ID, Host.AidaClient.Rating.POSITIVE);
 
-      sinon.assert.match(aidaClient.registerClientEvent.firstCall.firstArg, sinon.match({
+      sinon.assert.match((aidaClient.registerClientEvent as sinon.SinonStub).firstCall.firstArg, sinon.match({
         corresponding_aida_rpc_global_id: RPC_ID,
         do_conversation_client_event: {
           user_feedback: {
@@ -250,8 +234,8 @@ describeWithEnvironment('FreestylerPanel', () => {
 
     it('should send NEGATIVE rating to aida client when the user clicks on positive rating', () => {
       const RPC_ID = 0;
-      const aidaClient = getTestAidaClient();
-      panel = new Freestyler.FreestylerPanel(mockView, {
+      const aidaClient = mockAidaClient([[{explanation: 'test'}]]);
+      panel = new AiAssistance.AiAssistancePanel(mockView, {
         aidaClient,
         aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
         syncInfo: getTestSyncInfo(),
@@ -262,7 +246,7 @@ describeWithEnvironment('FreestylerPanel', () => {
       mockView.reset();
       callArgs.onFeedbackSubmit(RPC_ID, Host.AidaClient.Rating.NEGATIVE);
 
-      sinon.assert.match(aidaClient.registerClientEvent.firstCall.firstArg, sinon.match({
+      sinon.assert.match((aidaClient.registerClientEvent as sinon.SinonStub).firstCall.firstArg, sinon.match({
         corresponding_aida_rpc_global_id: RPC_ID,
         do_conversation_client_event: {
           user_feedback: {
@@ -276,8 +260,8 @@ describeWithEnvironment('FreestylerPanel', () => {
     it('should send feedback text with data', () => {
       const RPC_ID = 0;
       const feedback = 'This helped me a ton.';
-      const aidaClient = getTestAidaClient();
-      panel = new Freestyler.FreestylerPanel(mockView, {
+      const aidaClient = mockAidaClient([[{explanation: 'test'}]]);
+      panel = new AiAssistance.AiAssistancePanel(mockView, {
         aidaClient,
         aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
         syncInfo: getTestSyncInfo(),
@@ -288,7 +272,7 @@ describeWithEnvironment('FreestylerPanel', () => {
       mockView.reset();
       callArgs.onFeedbackSubmit(RPC_ID, Host.AidaClient.Rating.POSITIVE, feedback);
 
-      sinon.assert.match(aidaClient.registerClientEvent.firstCall.firstArg, sinon.match({
+      sinon.assert.match((aidaClient.registerClientEvent as sinon.SinonStub).firstCall.firstArg, sinon.match({
         corresponding_aida_rpc_global_id: RPC_ID,
         do_conversation_client_event: {
           user_feedback: {
@@ -307,8 +291,8 @@ describeWithEnvironment('FreestylerPanel', () => {
     describe('SDK.DOMModel.DOMNode flavor changes for selected element', () => {
       it('should set the selected element when the widget is shown', () => {
         UI.Context.Context.instance().setFlavor(SDK.DOMModel.DOMNode, null);
-        panel = new Freestyler.FreestylerPanel(mockView, {
-          aidaClient: getTestAidaClient(),
+        panel = new AiAssistance.AiAssistancePanel(mockView, {
+          aidaClient: mockAidaClient([[{explanation: 'test'}]]),
           aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
           syncInfo: getTestSyncInfo(),
         });
@@ -322,14 +306,14 @@ describeWithEnvironment('FreestylerPanel', () => {
         panel.handleAction('freestyler.elements-floating-button');
 
         sinon.assert.calledWith(mockView, sinon.match({
-          selectedContext: new Freestyler.NodeContext(node),
+          selectedContext: new AiAssistance.NodeContext(node),
         }));
       });
 
       it('should update the selected element when the changed DOMNode flavor is an ELEMENT_NODE', () => {
         UI.Context.Context.instance().setFlavor(SDK.DOMModel.DOMNode, null);
-        panel = new Freestyler.FreestylerPanel(mockView, {
-          aidaClient: getTestAidaClient(),
+        panel = new AiAssistance.AiAssistancePanel(mockView, {
+          aidaClient: mockAidaClient([[{explanation: 'test'}]]),
           aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
           syncInfo: getTestSyncInfo(),
         });
@@ -346,14 +330,14 @@ describeWithEnvironment('FreestylerPanel', () => {
         UI.Context.Context.instance().setFlavor(SDK.DOMModel.DOMNode, node);
 
         sinon.assert.calledWith(mockView, sinon.match({
-          selectedContext: new Freestyler.NodeContext(node),
+          selectedContext: new AiAssistance.NodeContext(node),
         }));
       });
 
       it('should set selected element to null when the change DOMNode flavor is not an ELEMENT_NODE', () => {
         UI.Context.Context.instance().setFlavor(SDK.DOMModel.DOMNode, null);
-        panel = new Freestyler.FreestylerPanel(mockView, {
-          aidaClient: getTestAidaClient(),
+        panel = new AiAssistance.AiAssistancePanel(mockView, {
+          aidaClient: mockAidaClient([[{explanation: 'test'}]]),
           aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
           syncInfo: getTestSyncInfo(),
         });
@@ -376,8 +360,8 @@ describeWithEnvironment('FreestylerPanel', () => {
 
       it('should not handle DOMNode flavor changes if the widget is not shown', () => {
         UI.Context.Context.instance().setFlavor(SDK.DOMModel.DOMNode, null);
-        panel = new Freestyler.FreestylerPanel(mockView, {
-          aidaClient: getTestAidaClient(),
+        panel = new AiAssistance.AiAssistancePanel(mockView, {
+          aidaClient: mockAidaClient([[{explanation: 'test'}]]),
           aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
           syncInfo: getTestSyncInfo(),
         });
@@ -394,8 +378,8 @@ describeWithEnvironment('FreestylerPanel', () => {
     describe('SDK.NetworkRequest.NetworkRequest flavor changes for selected network request', () => {
       it('should set the selected network request when the widget is shown', () => {
         UI.Context.Context.instance().setFlavor(SDK.NetworkRequest.NetworkRequest, null);
-        panel = new Freestyler.FreestylerPanel(mockView, {
-          aidaClient: getTestAidaClient(),
+        panel = new AiAssistance.AiAssistancePanel(mockView, {
+          aidaClient: mockAidaClient([[{explanation: 'test'}]]),
           aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
           syncInfo: getTestSyncInfo(),
         });
@@ -407,14 +391,14 @@ describeWithEnvironment('FreestylerPanel', () => {
         panel.handleAction('drjones.network-floating-button');
 
         sinon.assert.calledWith(mockView, sinon.match({
-          selectedContext: new Freestyler.RequestContext(networkRequest),
+          selectedContext: new AiAssistance.RequestContext(networkRequest),
         }));
       });
 
       it('should set selected network request when the NetworkRequest flavor changes', () => {
         UI.Context.Context.instance().setFlavor(SDK.NetworkRequest.NetworkRequest, null);
-        panel = new Freestyler.FreestylerPanel(mockView, {
-          aidaClient: getTestAidaClient(),
+        panel = new AiAssistance.AiAssistancePanel(mockView, {
+          aidaClient: mockAidaClient([[{explanation: 'test'}]]),
           aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
           syncInfo: getTestSyncInfo(),
         });
@@ -429,14 +413,14 @@ describeWithEnvironment('FreestylerPanel', () => {
         UI.Context.Context.instance().setFlavor(SDK.NetworkRequest.NetworkRequest, networkRequest);
 
         sinon.assert.calledWith(mockView, sinon.match({
-          selectedContext: new Freestyler.RequestContext(networkRequest),
+          selectedContext: new AiAssistance.RequestContext(networkRequest),
         }));
       });
 
       it('should not handle NetworkRequest flavor changes if the widget is not shown', () => {
         UI.Context.Context.instance().setFlavor(SDK.NetworkRequest.NetworkRequest, null);
-        panel = new Freestyler.FreestylerPanel(mockView, {
-          aidaClient: getTestAidaClient(),
+        panel = new AiAssistance.AiAssistancePanel(mockView, {
+          aidaClient: mockAidaClient([[{explanation: 'test'}]]),
           aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
           syncInfo: getTestSyncInfo(),
         });
@@ -451,8 +435,8 @@ describeWithEnvironment('FreestylerPanel', () => {
     describe('TimelineUtils.AICallTree.AICallTree flavor changes for selected call tree', () => {
       it('should set the selected call tree when the widget is shown', () => {
         UI.Context.Context.instance().setFlavor(TimelineUtils.AICallTree.AICallTree, null);
-        panel = new Freestyler.FreestylerPanel(mockView, {
-          aidaClient: getTestAidaClient(),
+        panel = new AiAssistance.AiAssistancePanel(mockView, {
+          aidaClient: mockAidaClient([[{explanation: 'test'}]]),
           aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
           syncInfo: getTestSyncInfo(),
         });
@@ -464,14 +448,14 @@ describeWithEnvironment('FreestylerPanel', () => {
         panel.handleAction('drjones.performance-panel-context');
 
         sinon.assert.calledWith(mockView, sinon.match({
-          selectedContext: new Freestyler.CallTreeContext(selectedAiCallTree as TimelineUtils.AICallTree.AICallTree),
+          selectedContext: new AiAssistance.CallTreeContext(selectedAiCallTree as TimelineUtils.AICallTree.AICallTree),
         }));
       });
 
       it('should set selected call tree when the AICallTree flavor changes', () => {
         UI.Context.Context.instance().setFlavor(TimelineUtils.AICallTree.AICallTree, null);
-        panel = new Freestyler.FreestylerPanel(mockView, {
-          aidaClient: getTestAidaClient(),
+        panel = new AiAssistance.AiAssistancePanel(mockView, {
+          aidaClient: mockAidaClient([[{explanation: 'test'}]]),
           aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
           syncInfo: getTestSyncInfo(),
         });
@@ -486,14 +470,14 @@ describeWithEnvironment('FreestylerPanel', () => {
         UI.Context.Context.instance().setFlavor(TimelineUtils.AICallTree.AICallTree, selectedAiCallTree);
 
         sinon.assert.calledWith(mockView, sinon.match({
-          selectedContext: new Freestyler.CallTreeContext(selectedAiCallTree as TimelineUtils.AICallTree.AICallTree),
+          selectedContext: new AiAssistance.CallTreeContext(selectedAiCallTree as TimelineUtils.AICallTree.AICallTree),
         }));
       });
 
       it('should not handle AICallTree flavor changes if the widget is not shown', () => {
         UI.Context.Context.instance().setFlavor(TimelineUtils.AICallTree.AICallTree, null);
-        panel = new Freestyler.FreestylerPanel(mockView, {
-          aidaClient: getTestAidaClient(),
+        panel = new AiAssistance.AiAssistancePanel(mockView, {
+          aidaClient: mockAidaClient([[{explanation: 'test'}]]),
           aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
           syncInfo: getTestSyncInfo(),
         });
@@ -508,8 +492,8 @@ describeWithEnvironment('FreestylerPanel', () => {
     describe('Workspace.UISourceCode.UISourceCode flavor changes for selected network request', () => {
       it('should set selected file when the widget is shown', () => {
         UI.Context.Context.instance().setFlavor(Workspace.UISourceCode.UISourceCode, null);
-        panel = new Freestyler.FreestylerPanel(mockView, {
-          aidaClient: getTestAidaClient(),
+        panel = new AiAssistance.AiAssistancePanel(mockView, {
+          aidaClient: mockAidaClient([[{explanation: 'test'}]]),
           aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
           syncInfo: getTestSyncInfo(),
         });
@@ -521,14 +505,14 @@ describeWithEnvironment('FreestylerPanel', () => {
         panel.handleAction('drjones.sources-panel-context');
 
         sinon.assert.calledWith(mockView, sinon.match({
-          selectedContext: new Freestyler.FileContext(uiSourceCode),
+          selectedContext: new AiAssistance.FileContext(uiSourceCode),
         }));
       });
 
       it('should set selected file when the UISourceCode flavor changes', () => {
         UI.Context.Context.instance().setFlavor(Workspace.UISourceCode.UISourceCode, null);
-        panel = new Freestyler.FreestylerPanel(mockView, {
-          aidaClient: getTestAidaClient(),
+        panel = new AiAssistance.AiAssistancePanel(mockView, {
+          aidaClient: mockAidaClient([[{explanation: 'test'}]]),
           aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
           syncInfo: getTestSyncInfo(),
         });
@@ -543,14 +527,14 @@ describeWithEnvironment('FreestylerPanel', () => {
         UI.Context.Context.instance().setFlavor(Workspace.UISourceCode.UISourceCode, uiSourceCode);
 
         sinon.assert.calledWith(mockView, sinon.match({
-          selectedContext: new Freestyler.FileContext(uiSourceCode),
+          selectedContext: new AiAssistance.FileContext(uiSourceCode),
         }));
       });
 
       it('should not handle NetworkRequest flavor changes if the widget is not shown', () => {
         UI.Context.Context.instance().setFlavor(Workspace.UISourceCode.UISourceCode, null);
-        panel = new Freestyler.FreestylerPanel(mockView, {
-          aidaClient: getTestAidaClient(),
+        panel = new AiAssistance.AiAssistancePanel(mockView, {
+          aidaClient: mockAidaClient([[{explanation: 'test'}]]),
           aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
           syncInfo: getTestSyncInfo(),
         });
@@ -572,8 +556,8 @@ describeWithEnvironment('FreestylerPanel', () => {
     });
 
     it('should set inspectElementToggled when the widget is shown', () => {
-      panel = new Freestyler.FreestylerPanel(mockView, {
-        aidaClient: getTestAidaClient(),
+      panel = new AiAssistance.AiAssistancePanel(mockView, {
+        aidaClient: mockAidaClient([[{explanation: 'test'}]]),
         aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
         syncInfo: getTestSyncInfo(),
       });
@@ -589,8 +573,8 @@ describeWithEnvironment('FreestylerPanel', () => {
 
     it('should update inspectElementToggled when the action is toggled', () => {
       toggleSearchElementAction.setToggled(false);
-      panel = new Freestyler.FreestylerPanel(mockView, {
-        aidaClient: getTestAidaClient(),
+      panel = new AiAssistance.AiAssistancePanel(mockView, {
+        aidaClient: mockAidaClient([[{explanation: 'test'}]]),
         aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
         syncInfo: getTestSyncInfo(),
       });
@@ -610,8 +594,8 @@ describeWithEnvironment('FreestylerPanel', () => {
     it('should not update toggleSearchElementAction even after the action is toggled when the widget is not shown',
        () => {
          toggleSearchElementAction.setToggled(false);
-         panel = new Freestyler.FreestylerPanel(mockView, {
-           aidaClient: getTestAidaClient(),
+         panel = new AiAssistance.AiAssistancePanel(mockView, {
+           aidaClient: mockAidaClient([[{explanation: 'test'}]]),
            aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
            syncInfo: getTestSyncInfo(),
          });
@@ -624,8 +608,8 @@ describeWithEnvironment('FreestylerPanel', () => {
 
   describe('history interactions', () => {
     it('should have empty messages after new chat', async () => {
-      panel = new Freestyler.FreestylerPanel(mockView, {
-        aidaClient: getTestAidaClient(),
+      panel = new AiAssistance.AiAssistancePanel(mockView, {
+        aidaClient: mockAidaClient([[{explanation: 'test'}]]),
         aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
         syncInfo: getTestSyncInfo(),
       });
@@ -635,19 +619,18 @@ describeWithEnvironment('FreestylerPanel', () => {
 
       assert.deepEqual(mockView.lastCall.args[0].messages, [
         {
-          entity: Freestyler.ChatMessageEntity.USER,
+          entity: AiAssistance.ChatMessageEntity.USER,
           text: 'test',
         },
         {
           answer: 'test',
-          entity: Freestyler.ChatMessageEntity.MODEL,
+          entity: AiAssistance.ChatMessageEntity.MODEL,
           rpcId: undefined,
           suggestions: undefined,
           steps: [],
         },
       ]);
-      const toolbar = panel.contentElement.querySelector('.freestyler-left-toolbar');
-      const button = toolbar!.shadowRoot!.querySelector('devtools-button[aria-label=\'New chat\']');
+      const button = panel.contentElement.querySelector('devtools-button[aria-label=\'New chat\']');
       assert.instanceOf(button, HTMLElement);
       dispatchClickEvent(button);
       assert.deepEqual(mockView.lastCall.args[0].messages, []);
@@ -659,8 +642,8 @@ describeWithEnvironment('FreestylerPanel', () => {
           enabled: true,
         },
       });
-      panel = new Freestyler.FreestylerPanel(mockView, {
-        aidaClient: getTestAidaClient(),
+      panel = new AiAssistance.AiAssistancePanel(mockView, {
+        aidaClient: mockAidaClient([[{explanation: 'test'}]]),
         aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
         syncInfo: getTestSyncInfo(),
       });
@@ -673,29 +656,28 @@ describeWithEnvironment('FreestylerPanel', () => {
 
       assert.deepEqual(mockView.lastCall.args[0].messages, [
         {
-          entity: Freestyler.ChatMessageEntity.USER,
+          entity: AiAssistance.ChatMessageEntity.USER,
           text: 'test',
         },
         {
           answer: 'test',
-          entity: Freestyler.ChatMessageEntity.MODEL,
+          entity: AiAssistance.ChatMessageEntity.MODEL,
           rpcId: undefined,
           suggestions: undefined,
           steps: [],
         },
       ]);
-      const toolbar = panel.contentElement.querySelector('.freestyler-left-toolbar');
-      const button = toolbar!.shadowRoot!.querySelector('devtools-button[aria-label=\'New chat\']');
+      const button = panel.contentElement.querySelector('devtools-button[aria-label=\'New chat\']');
       assert.instanceOf(button, HTMLElement);
       dispatchClickEvent(button);
       assert.deepEqual(mockView.lastCall.args[0].messages, []);
-      assert.deepEqual(mockView.lastCall.args[0].agentType, AiAgent.AgentType.FREESTYLER);
+      assert.deepEqual(mockView.lastCall.args[0].agentType, AiAssistance.AgentType.STYLING);
       stub.restore();
     });
 
     it('should switch agents and restore history', async () => {
-      panel = new Freestyler.FreestylerPanel(mockView, {
-        aidaClient: getTestAidaClient(),
+      panel = new AiAssistance.AiAssistancePanel(mockView, {
+        aidaClient: mockAidaClient([[{explanation: 'test'}], [{explanation: 'test2'}]]),
         aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
         syncInfo: getTestSyncInfo(),
       });
@@ -704,12 +686,12 @@ describeWithEnvironment('FreestylerPanel', () => {
       await drainMicroTasks();
       assert.deepEqual(mockView.lastCall.args[0].messages, [
         {
-          entity: Freestyler.ChatMessageEntity.USER,
+          entity: AiAssistance.ChatMessageEntity.USER,
           text: 'User question to Freestyler?',
         },
         {
           answer: 'test',
-          entity: Freestyler.ChatMessageEntity.MODEL,
+          entity: AiAssistance.ChatMessageEntity.MODEL,
           rpcId: undefined,
           suggestions: undefined,
           steps: [],
@@ -721,20 +703,19 @@ describeWithEnvironment('FreestylerPanel', () => {
       await drainMicroTasks();
       assert.deepEqual(mockView.lastCall.args[0].messages, [
         {
-          entity: Freestyler.ChatMessageEntity.USER,
+          entity: AiAssistance.ChatMessageEntity.USER,
           text: 'User question to DrJones?',
         },
         {
-          answer: 'test',
-          entity: Freestyler.ChatMessageEntity.MODEL,
+          answer: 'test2',
+          entity: AiAssistance.ChatMessageEntity.MODEL,
           rpcId: undefined,
           suggestions: undefined,
           steps: [],
         },
       ]);
 
-      const toolbar = panel.contentElement.querySelector('.freestyler-left-toolbar');
-      const button = toolbar!.shadowRoot!.querySelector('devtools-button[aria-label=\'History\']');
+      const button = panel.contentElement.querySelector('devtools-button[aria-label=\'History\']');
       assert.instanceOf(button, HTMLElement);
       const contextMenu = getMenu(() => {
         dispatchClickEvent(button);
@@ -746,12 +727,12 @@ describeWithEnvironment('FreestylerPanel', () => {
       await drainMicroTasks();
       assert.deepEqual(mockView.lastCall.args[0].messages, [
         {
-          entity: Freestyler.ChatMessageEntity.USER,
+          entity: AiAssistance.ChatMessageEntity.USER,
           text: 'User question to Freestyler?',
         },
         {
           answer: 'test',
-          entity: Freestyler.ChatMessageEntity.MODEL,
+          entity: AiAssistance.ChatMessageEntity.MODEL,
           rpcId: undefined,
           suggestions: undefined,
           steps: [],
@@ -761,8 +742,8 @@ describeWithEnvironment('FreestylerPanel', () => {
   });
 
   it('should have empty state after clear chat', async () => {
-    panel = new Freestyler.FreestylerPanel(mockView, {
-      aidaClient: getTestAidaClient(),
+    panel = new AiAssistance.AiAssistancePanel(mockView, {
+      aidaClient: mockAidaClient([[{explanation: 'test'}]]),
       aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
       syncInfo: getTestSyncInfo(),
     });
@@ -772,23 +753,22 @@ describeWithEnvironment('FreestylerPanel', () => {
 
     assert.deepEqual(mockView.lastCall.args[0].messages, [
       {
-        entity: Freestyler.ChatMessageEntity.USER,
+        entity: AiAssistance.ChatMessageEntity.USER,
         text: 'test',
       },
       {
         answer: 'test',
-        entity: Freestyler.ChatMessageEntity.MODEL,
+        entity: AiAssistance.ChatMessageEntity.MODEL,
         rpcId: undefined,
         suggestions: undefined,
         steps: [],
       },
     ]);
-    const toolbar = panel.contentElement.querySelector('.freestyler-left-toolbar');
-    const button = toolbar!.shadowRoot!.querySelector('devtools-button[aria-label=\'Delete chat\']');
+    const button = panel.contentElement.querySelector('devtools-button[aria-label=\'Delete local chat\']');
     assert.instanceOf(button, HTMLElement);
     dispatchClickEvent(button);
     assert.deepEqual(mockView.lastCall.args[0].messages, []);
-    assert.deepEqual(mockView.lastCall.args[0].agentType, undefined);
+    assert.isUndefined(mockView.lastCall.args[0].agentType);
   });
 
   it('should select default agent based on open panel after clearing the chat', async () => {
@@ -797,8 +777,8 @@ describeWithEnvironment('FreestylerPanel', () => {
         enabled: true,
       },
     });
-    panel = new Freestyler.FreestylerPanel(mockView, {
-      aidaClient: getTestAidaClient(),
+    panel = new AiAssistance.AiAssistancePanel(mockView, {
+      aidaClient: mockAidaClient([[{explanation: 'test'}]]),
       aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
       syncInfo: getTestSyncInfo(),
     });
@@ -810,29 +790,28 @@ describeWithEnvironment('FreestylerPanel', () => {
         ElementsPanel.ElementsPanel.ElementsPanel, sinon.createStubInstance(ElementsPanel.ElementsPanel.ElementsPanel));
     assert.deepEqual(mockView.lastCall.args[0].messages, [
       {
-        entity: Freestyler.ChatMessageEntity.USER,
+        entity: AiAssistance.ChatMessageEntity.USER,
         text: 'test',
       },
       {
         answer: 'test',
-        entity: Freestyler.ChatMessageEntity.MODEL,
+        entity: AiAssistance.ChatMessageEntity.MODEL,
         rpcId: undefined,
         suggestions: undefined,
         steps: [],
       },
     ]);
-    const toolbar = panel.contentElement.querySelector('.freestyler-left-toolbar');
-    const button = toolbar!.shadowRoot!.querySelector('devtools-button[aria-label=\'Delete chat\']');
+    const button = panel.contentElement.querySelector('devtools-button[aria-label=\'Delete local chat\']');
     assert.instanceOf(button, HTMLElement);
     dispatchClickEvent(button);
     assert.deepEqual(mockView.lastCall.args[0].messages, []);
-    assert.deepEqual(mockView.lastCall.args[0].agentType, AiAgent.AgentType.FREESTYLER);
+    assert.deepEqual(mockView.lastCall.args[0].agentType, AiAssistance.AgentType.STYLING);
     stub.restore();
   });
 
   it('should have empty state after clear chat history', async () => {
-    panel = new Freestyler.FreestylerPanel(mockView, {
-      aidaClient: getTestAidaClient(),
+    panel = new AiAssistance.AiAssistancePanel(mockView, {
+      aidaClient: mockAidaClient([[{explanation: 'test'}], [{explanation: 'test2'}]]),
       aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
       syncInfo: getTestSyncInfo(),
     });
@@ -841,12 +820,12 @@ describeWithEnvironment('FreestylerPanel', () => {
     await drainMicroTasks();
     assert.deepEqual(mockView.lastCall.args[0].messages, [
       {
-        entity: Freestyler.ChatMessageEntity.USER,
+        entity: AiAssistance.ChatMessageEntity.USER,
         text: 'User question to Freestyler?',
       },
       {
         answer: 'test',
-        entity: Freestyler.ChatMessageEntity.MODEL,
+        entity: AiAssistance.ChatMessageEntity.MODEL,
         rpcId: undefined,
         suggestions: undefined,
         steps: [],
@@ -858,37 +837,35 @@ describeWithEnvironment('FreestylerPanel', () => {
     await drainMicroTasks();
     assert.deepEqual(mockView.lastCall.args[0].messages, [
       {
-        entity: Freestyler.ChatMessageEntity.USER,
+        entity: AiAssistance.ChatMessageEntity.USER,
         text: 'User question to DrJones?',
       },
       {
-        answer: 'test',
-        entity: Freestyler.ChatMessageEntity.MODEL,
+        answer: 'test2',
+        entity: AiAssistance.ChatMessageEntity.MODEL,
         rpcId: undefined,
         suggestions: undefined,
         steps: [],
       },
     ]);
 
-    let toolbar = panel.contentElement.querySelector('.freestyler-left-toolbar');
-    let button = toolbar!.shadowRoot!.querySelector('devtools-button[aria-label=\'History\']');
+    let button = panel.contentElement.querySelector('devtools-button[aria-label=\'History\']');
     assert.instanceOf(button, HTMLElement);
     let contextMenu = getMenu(() => {
       dispatchClickEvent(button!);
     });
-    const clearAll = findMenuItemWithLabel(contextMenu.footerSection(), 'Clear chat history')!;
+    const clearAll = findMenuItemWithLabel(contextMenu.footerSection(), 'Clear local chats')!;
     assert.isDefined(clearAll);
     contextMenu.invokeHandler(clearAll.id());
     await drainMicroTasks();
     assert.deepEqual(mockView.lastCall.args[0].messages, []);
-    assert.deepEqual(mockView.lastCall.args[0].agentType, undefined);
+    assert.isUndefined(mockView.lastCall.args[0].agentType);
 
     await drainMicroTasks();
     contextMenu.discard();
     await drainMicroTasks();
 
-    toolbar = panel.contentElement.querySelector('.freestyler-left-toolbar');
-    button = toolbar!.shadowRoot!.querySelector('devtools-button[aria-label=\'History\']');
+    button = panel.contentElement.querySelector('devtools-button[aria-label=\'History\']');
     assert.instanceOf(button, HTMLElement);
     contextMenu = getMenu(() => {
       dispatchClickEvent(button);
@@ -900,11 +877,11 @@ describeWithEnvironment('FreestylerPanel', () => {
   describe('cross-origin', () => {
     it('blocks input on cross origin requests', async () => {
       const networkRequest = sinon.createStubInstance(SDK.NetworkRequest.NetworkRequest, {
-        url: 'https://a.test' as Platform.DevToolsPath.UrlString,
+        url: urlString`https://a.test`,
       });
       UI.Context.Context.instance().setFlavor(SDK.NetworkRequest.NetworkRequest, networkRequest);
-      panel = new Freestyler.FreestylerPanel(mockView, {
-        aidaClient: getTestAidaClient(),
+      panel = new AiAssistance.AiAssistancePanel(mockView, {
+        aidaClient: mockAidaClient([[{explanation: 'test'}]]),
         aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
         syncInfo: getTestSyncInfo(),
       });
@@ -913,7 +890,7 @@ describeWithEnvironment('FreestylerPanel', () => {
       panel.handleAction('drjones.network-floating-button');
 
       sinon.assert.calledWith(mockView, sinon.match({
-        selectedContext: new Freestyler.RequestContext(networkRequest),
+        selectedContext: new AiAssistance.RequestContext(networkRequest),
         blockedByCrossOrigin: false,
       }));
 
@@ -924,14 +901,14 @@ describeWithEnvironment('FreestylerPanel', () => {
 
       // Change context to https://b.test.
       const networkRequest2 = sinon.createStubInstance(SDK.NetworkRequest.NetworkRequest, {
-        url: 'https://b.test' as Platform.DevToolsPath.UrlString,
+        url: urlString`https://b.test`,
       });
       UI.Context.Context.instance().setFlavor(SDK.NetworkRequest.NetworkRequest, networkRequest2);
       panel.handleAction('drjones.network-floating-button');
       await drainMicroTasks();
 
       sinon.assert.calledWith(mockView, sinon.match({
-        selectedContext: new Freestyler.RequestContext(networkRequest2),
+        selectedContext: new AiAssistance.RequestContext(networkRequest2),
         blockedByCrossOrigin: true,
       }));
     });
@@ -942,8 +919,8 @@ describeWithEnvironment('FreestylerPanel', () => {
           enabled: true,
         },
       });
-      panel = new Freestyler.FreestylerPanel(mockView, {
-        aidaClient: getTestAidaClient(),
+      panel = new AiAssistance.AiAssistancePanel(mockView, {
+        aidaClient: mockAidaClient([[{explanation: 'test'}], [{explanation: 'test2'}]]),
         aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
         syncInfo: getTestSyncInfo(),
       });
@@ -957,12 +934,12 @@ describeWithEnvironment('FreestylerPanel', () => {
 
       assert.deepEqual(mockView.lastCall.args[0].messages, [
         {
-          entity: Freestyler.ChatMessageEntity.USER,
+          entity: AiAssistance.ChatMessageEntity.USER,
           text: 'test',
         },
         {
           answer: 'test',
-          entity: Freestyler.ChatMessageEntity.MODEL,
+          entity: AiAssistance.ChatMessageEntity.MODEL,
           rpcId: undefined,
           suggestions: undefined,
           steps: [],
@@ -977,26 +954,26 @@ describeWithEnvironment('FreestylerPanel', () => {
       mockView.lastCall.args[0].onTextSubmit('test2');
       await drainMicroTasks();
 
-      assert.strictEqual(mockView.lastCall.args[0].isReadOnly, false);
+      assert.isFalse(mockView.lastCall.args[0].isReadOnly);
       assert.deepEqual(mockView.lastCall.args[0].messages, [
         {
-          entity: Freestyler.ChatMessageEntity.USER,
+          entity: AiAssistance.ChatMessageEntity.USER,
           text: 'test',
         },
         {
           answer: 'test',
-          entity: Freestyler.ChatMessageEntity.MODEL,
+          entity: AiAssistance.ChatMessageEntity.MODEL,
           rpcId: undefined,
           suggestions: undefined,
           steps: [],
         },
         {
-          entity: Freestyler.ChatMessageEntity.USER,
+          entity: AiAssistance.ChatMessageEntity.USER,
           text: 'test2',
         },
         {
-          answer: 'test',
-          entity: Freestyler.ChatMessageEntity.MODEL,
+          answer: 'test2',
+          entity: AiAssistance.ChatMessageEntity.MODEL,
           rpcId: undefined,
           suggestions: undefined,
           steps: [],
@@ -1018,8 +995,8 @@ describeWithEnvironment('FreestylerPanel', () => {
         UI.Context.Context.instance().setFlavor(
             ElementsPanel.ElementsPanel.ElementsPanel,
             sinon.createStubInstance(ElementsPanel.ElementsPanel.ElementsPanel));
-        panel = new Freestyler.FreestylerPanel(mockView, {
-          aidaClient: getTestAidaClient(),
+        panel = new AiAssistance.AiAssistancePanel(mockView, {
+          aidaClient: mockAidaClient([[{explanation: 'test'}]]),
           aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
           syncInfo: getTestSyncInfo(),
         });
@@ -1028,7 +1005,7 @@ describeWithEnvironment('FreestylerPanel', () => {
         panel.show(document.body);
 
         sinon.assert.calledWith(mockView, sinon.match({
-          agentType: AiAgent.AgentType.FREESTYLER,
+          agentType: AiAssistance.AgentType.STYLING,
         }));
         stub.restore();
       });
@@ -1042,8 +1019,8 @@ describeWithEnvironment('FreestylerPanel', () => {
         UI.Context.Context.instance().setFlavor(
             ElementsPanel.ElementsPanel.ElementsPanel,
             sinon.createStubInstance(ElementsPanel.ElementsPanel.ElementsPanel));
-        panel = new Freestyler.FreestylerPanel(mockView, {
-          aidaClient: getTestAidaClient(),
+        panel = new AiAssistance.AiAssistancePanel(mockView, {
+          aidaClient: mockAidaClient([[{explanation: 'test'}]]),
           aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
           syncInfo: getTestSyncInfo(),
         });
@@ -1051,11 +1028,11 @@ describeWithEnvironment('FreestylerPanel', () => {
         panel.markAsRoot();
         panel.show(document.body);
         sinon.assert.calledWith(mockView, sinon.match({
-          agentType: AiAgent.AgentType.FREESTYLER,
+          agentType: AiAssistance.AgentType.STYLING,
         }));
 
         UI.Context.Context.instance().setFlavor(ElementsPanel.ElementsPanel.ElementsPanel, null);
-        assert.deepStrictEqual(mockView.lastCall.args[0].agentType, undefined);
+        assert.isUndefined(mockView.lastCall.args[0].agentType);
         stub.restore();
       });
 
@@ -1068,8 +1045,8 @@ describeWithEnvironment('FreestylerPanel', () => {
         UI.Context.Context.instance().setFlavor(
             ElementsPanel.ElementsPanel.ElementsPanel,
             sinon.createStubInstance(ElementsPanel.ElementsPanel.ElementsPanel));
-        panel = new Freestyler.FreestylerPanel(mockView, {
-          aidaClient: getTestAidaClient(),
+        panel = new AiAssistance.AiAssistancePanel(mockView, {
+          aidaClient: mockAidaClient([[{explanation: 'test'}]]),
           aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
           syncInfo: getTestSyncInfo(),
         });
@@ -1077,7 +1054,7 @@ describeWithEnvironment('FreestylerPanel', () => {
         panel.markAsRoot();
         panel.show(document.body);
 
-        assert.deepStrictEqual(mockView.lastCall.args[0].agentType, undefined);
+        assert.isUndefined(mockView.lastCall.args[0].agentType);
         stub.restore();
       });
     });
@@ -1091,8 +1068,8 @@ describeWithEnvironment('FreestylerPanel', () => {
         });
         UI.Context.Context.instance().setFlavor(
             NetworkPanel.NetworkPanel.NetworkPanel, sinon.createStubInstance(NetworkPanel.NetworkPanel.NetworkPanel));
-        panel = new Freestyler.FreestylerPanel(mockView, {
-          aidaClient: getTestAidaClient(),
+        panel = new AiAssistance.AiAssistancePanel(mockView, {
+          aidaClient: mockAidaClient([[{explanation: 'test'}]]),
           aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
           syncInfo: getTestSyncInfo(),
         });
@@ -1101,7 +1078,7 @@ describeWithEnvironment('FreestylerPanel', () => {
         panel.show(document.body);
 
         sinon.assert.calledWith(mockView, sinon.match({
-          agentType: AiAgent.AgentType.DRJONES_NETWORK_REQUEST,
+          agentType: AiAssistance.AgentType.NETWORK,
         }));
         stub.restore();
       });
@@ -1114,8 +1091,8 @@ describeWithEnvironment('FreestylerPanel', () => {
         });
         UI.Context.Context.instance().setFlavor(
             NetworkPanel.NetworkPanel.NetworkPanel, sinon.createStubInstance(NetworkPanel.NetworkPanel.NetworkPanel));
-        panel = new Freestyler.FreestylerPanel(mockView, {
-          aidaClient: getTestAidaClient(),
+        panel = new AiAssistance.AiAssistancePanel(mockView, {
+          aidaClient: mockAidaClient([[{explanation: 'test'}]]),
           aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
           syncInfo: getTestSyncInfo(),
         });
@@ -1123,11 +1100,11 @@ describeWithEnvironment('FreestylerPanel', () => {
         panel.markAsRoot();
         panel.show(document.body);
         sinon.assert.calledWith(mockView, sinon.match({
-          agentType: AiAgent.AgentType.DRJONES_NETWORK_REQUEST,
+          agentType: AiAssistance.AgentType.NETWORK,
         }));
 
         UI.Context.Context.instance().setFlavor(NetworkPanel.NetworkPanel.NetworkPanel, null);
-        assert.deepStrictEqual(mockView.lastCall.args[0].agentType, undefined);
+        assert.isUndefined(mockView.lastCall.args[0].agentType);
         stub.restore();
       });
 
@@ -1141,8 +1118,8 @@ describeWithEnvironment('FreestylerPanel', () => {
            UI.Context.Context.instance().setFlavor(
                NetworkPanel.NetworkPanel.NetworkPanel,
                sinon.createStubInstance(NetworkPanel.NetworkPanel.NetworkPanel));
-           panel = new Freestyler.FreestylerPanel(mockView, {
-             aidaClient: getTestAidaClient(),
+           panel = new AiAssistance.AiAssistancePanel(mockView, {
+             aidaClient: mockAidaClient([[{explanation: 'test'}]]),
              aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
              syncInfo: getTestSyncInfo(),
            });
@@ -1150,7 +1127,7 @@ describeWithEnvironment('FreestylerPanel', () => {
            panel.markAsRoot();
            panel.show(document.body);
 
-           assert.deepStrictEqual(mockView.lastCall.args[0].agentType, undefined);
+           assert.isUndefined(mockView.lastCall.args[0].agentType);
            stub.restore();
          });
     });
@@ -1164,8 +1141,8 @@ describeWithEnvironment('FreestylerPanel', () => {
         });
         UI.Context.Context.instance().setFlavor(
             SourcesPanel.SourcesPanel.SourcesPanel, sinon.createStubInstance(SourcesPanel.SourcesPanel.SourcesPanel));
-        panel = new Freestyler.FreestylerPanel(mockView, {
-          aidaClient: getTestAidaClient(),
+        panel = new AiAssistance.AiAssistancePanel(mockView, {
+          aidaClient: mockAidaClient([[{explanation: 'test'}]]),
           aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
           syncInfo: getTestSyncInfo(),
         });
@@ -1174,7 +1151,7 @@ describeWithEnvironment('FreestylerPanel', () => {
         panel.show(document.body);
 
         sinon.assert.calledWith(mockView, sinon.match({
-          agentType: AiAgent.AgentType.DRJONES_FILE,
+          agentType: AiAssistance.AgentType.FILE,
         }));
         stub.restore();
       });
@@ -1187,8 +1164,8 @@ describeWithEnvironment('FreestylerPanel', () => {
         });
         UI.Context.Context.instance().setFlavor(
             SourcesPanel.SourcesPanel.SourcesPanel, sinon.createStubInstance(SourcesPanel.SourcesPanel.SourcesPanel));
-        panel = new Freestyler.FreestylerPanel(mockView, {
-          aidaClient: getTestAidaClient(),
+        panel = new AiAssistance.AiAssistancePanel(mockView, {
+          aidaClient: mockAidaClient([[{explanation: 'test'}]]),
           aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
           syncInfo: getTestSyncInfo(),
         });
@@ -1196,11 +1173,11 @@ describeWithEnvironment('FreestylerPanel', () => {
         panel.markAsRoot();
         panel.show(document.body);
         sinon.assert.calledWith(mockView, sinon.match({
-          agentType: AiAgent.AgentType.DRJONES_FILE,
+          agentType: AiAssistance.AgentType.FILE,
         }));
 
         UI.Context.Context.instance().setFlavor(SourcesPanel.SourcesPanel.SourcesPanel, null);
-        assert.deepStrictEqual(mockView.lastCall.args[0].agentType, undefined);
+        assert.isUndefined(mockView.lastCall.args[0].agentType);
         stub.restore();
       });
 
@@ -1214,8 +1191,8 @@ describeWithEnvironment('FreestylerPanel', () => {
            UI.Context.Context.instance().setFlavor(
                SourcesPanel.SourcesPanel.SourcesPanel,
                sinon.createStubInstance(SourcesPanel.SourcesPanel.SourcesPanel));
-           panel = new Freestyler.FreestylerPanel(mockView, {
-             aidaClient: getTestAidaClient(),
+           panel = new AiAssistance.AiAssistancePanel(mockView, {
+             aidaClient: mockAidaClient([[{explanation: 'test'}]]),
              aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
              syncInfo: getTestSyncInfo(),
            });
@@ -1223,7 +1200,7 @@ describeWithEnvironment('FreestylerPanel', () => {
            panel.markAsRoot();
            panel.show(document.body);
 
-           assert.deepStrictEqual(mockView.lastCall.args[0].agentType, undefined);
+           assert.isUndefined(mockView.lastCall.args[0].agentType);
            stub.restore();
          });
     });
@@ -1238,8 +1215,8 @@ describeWithEnvironment('FreestylerPanel', () => {
         UI.Context.Context.instance().setFlavor(
             TimelinePanel.TimelinePanel.TimelinePanel,
             sinon.createStubInstance(TimelinePanel.TimelinePanel.TimelinePanel));
-        panel = new Freestyler.FreestylerPanel(mockView, {
-          aidaClient: getTestAidaClient(),
+        panel = new AiAssistance.AiAssistancePanel(mockView, {
+          aidaClient: mockAidaClient([[{explanation: 'test'}]]),
           aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
           syncInfo: getTestSyncInfo(),
         });
@@ -1248,7 +1225,7 @@ describeWithEnvironment('FreestylerPanel', () => {
         panel.show(document.body);
 
         sinon.assert.calledWith(mockView, sinon.match({
-          agentType: AiAgent.AgentType.DRJONES_PERFORMANCE,
+          agentType: AiAssistance.AgentType.PERFORMANCE,
         }));
         stub.restore();
       });
@@ -1262,8 +1239,8 @@ describeWithEnvironment('FreestylerPanel', () => {
         UI.Context.Context.instance().setFlavor(
             TimelinePanel.TimelinePanel.TimelinePanel,
             sinon.createStubInstance(TimelinePanel.TimelinePanel.TimelinePanel));
-        panel = new Freestyler.FreestylerPanel(mockView, {
-          aidaClient: getTestAidaClient(),
+        panel = new AiAssistance.AiAssistancePanel(mockView, {
+          aidaClient: mockAidaClient([[{explanation: 'test'}]]),
           aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
           syncInfo: getTestSyncInfo(),
         });
@@ -1271,11 +1248,11 @@ describeWithEnvironment('FreestylerPanel', () => {
         panel.markAsRoot();
         panel.show(document.body);
         sinon.assert.calledWith(mockView, sinon.match({
-          agentType: AiAgent.AgentType.DRJONES_PERFORMANCE,
+          agentType: AiAssistance.AgentType.PERFORMANCE,
         }));
 
         UI.Context.Context.instance().setFlavor(TimelinePanel.TimelinePanel.TimelinePanel, null);
-        assert.deepStrictEqual(mockView.lastCall.args[0].agentType, undefined);
+        assert.isUndefined(mockView.lastCall.args[0].agentType);
         stub.restore();
       });
 
@@ -1289,8 +1266,8 @@ describeWithEnvironment('FreestylerPanel', () => {
            UI.Context.Context.instance().setFlavor(
                TimelinePanel.TimelinePanel.TimelinePanel,
                sinon.createStubInstance(TimelinePanel.TimelinePanel.TimelinePanel));
-           panel = new Freestyler.FreestylerPanel(mockView, {
-             aidaClient: getTestAidaClient(),
+           panel = new AiAssistance.AiAssistancePanel(mockView, {
+             aidaClient: mockAidaClient([[{explanation: 'test'}]]),
              aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
              syncInfo: getTestSyncInfo(),
            });
@@ -1298,9 +1275,44 @@ describeWithEnvironment('FreestylerPanel', () => {
            panel.markAsRoot();
            panel.show(document.body);
 
-           assert.deepStrictEqual(mockView.lastCall.args[0].agentType, undefined);
+           assert.isUndefined(mockView.lastCall.args[0].agentType);
            stub.restore();
          });
     });
   });
+
+  it('erases previous partial response on blocked error', async () => {
+    async function* generateAnswerAndError() {
+      yield {
+        explanation: 'ANSWER: This is the first part of the answer.',
+        metadata: {},
+        completed: false,
+      };
+      throw new Host.AidaClient.AidaBlockError();
+    }
+
+    const mockAidaClient = {
+      fetch: generateAnswerAndError,
+      registerClientEvent: sinon.spy(),
+    };
+
+    const messages: AiAssistance.ModelChatMessage[] = [];
+    const viewMock = sinon.stub().callsFake((props: AiAssistance.Props) => {
+      messages.push(JSON.parse(JSON.stringify(props.messages[1])));
+    });
+
+    panel = new AiAssistance.AiAssistancePanel(viewMock, {
+      aidaClient: mockAidaClient,
+      aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
+      syncInfo: getTestSyncInfo(),
+    });
+    panel.handleAction('freestyler.elements-floating-button');
+    viewMock.lastCall.args[0].onTextSubmit('test');
+    await drainMicroTasks();
+
+    assert.strictEqual(messages[2].answer, 'This is the first part of the answer.');
+    assert.isUndefined(messages[2].error);
+    assert.isUndefined(messages[3].answer);
+    assert.strictEqual(messages[3].error, 'block');
+  });
 });