chrome-devtools-frontend 1.0.1380117 → 1.0.1402249

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

@@ -17,8 +17,7 @@ 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 {
@@ -39,8 +38,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,7 +56,7 @@ describeWithEnvironment('FreestylerPanel', () => {
 
   describe('consent view', () => {
     it('should render consent view when the consent is not given before', async () => {
-      panel = new Freestyler.FreestylerPanel(mockView, {
+      panel = new AiAssistance.AiAssistancePanel(mockView, {
         aidaClient: getTestAidaClient(),
         aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
         syncInfo: getTestSyncInfo(),
@@ -65,28 +64,28 @@ describeWithEnvironment('FreestylerPanel', () => {
       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, {
+      panel = new AiAssistance.AiAssistancePanel(mockView, {
         aidaClient: getTestAidaClient(),
         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, {
+      panel = new AiAssistance.AiAssistancePanel(mockView, {
         aidaClient: getTestAidaClient(),
         aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
         syncInfo: getTestSyncInfo(),
@@ -94,18 +93,18 @@ describeWithEnvironment('FreestylerPanel', () => {
       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, {
+      panel = new AiAssistance.AiAssistancePanel(viewStub, {
         aidaClient: getTestAidaClient(),
         aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
         syncInfo: getTestSyncInfo(),
@@ -114,8 +113,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,7 +129,7 @@ describeWithEnvironment('FreestylerPanel', () => {
         },
       });
 
-      panel = new Freestyler.FreestylerPanel(mockView, {
+      panel = new AiAssistance.AiAssistancePanel(mockView, {
         aidaClient: getTestAidaClient(),
         aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
         syncInfo: getTestSyncInfo(),
@@ -138,14 +137,14 @@ describeWithEnvironment('FreestylerPanel', () => {
       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, {
+      panel = new AiAssistance.AiAssistancePanel(mockView, {
         aidaClient: getTestAidaClient(),
         aidaAvailability: Host.AidaClient.AidaAccessPreconditions.NO_ACCOUNT_EMAIL,
         syncInfo: getTestSyncInfo(),
@@ -154,7 +153,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 +164,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,
       }));
 
@@ -176,13 +175,13 @@ describeWithEnvironment('FreestylerPanel', () => {
   describe('on rate click', () => {
     afterEach(() => {
       // @ts-expect-error global test variable
-      setFreestylerServerSideLoggingEnabled(false);
+      setAiAssistanceServerSideLoggingEnabled(false);
     });
 
     it('renders a button linking to settings', () => {
       const stub = sinon.stub(UI.ViewManager.ViewManager.instance(), 'showView');
 
-      panel = new Freestyler.FreestylerPanel(mockView, {
+      panel = new AiAssistance.AiAssistancePanel(mockView, {
         aidaClient: getTestAidaClient(),
         aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
         syncInfo: getTestSyncInfo(),
@@ -197,7 +196,7 @@ describeWithEnvironment('FreestylerPanel', () => {
 
     it('should allow logging if configured', () => {
       // @ts-expect-error global test variable
-      setFreestylerServerSideLoggingEnabled(true);
+      setAiAssistanceServerSideLoggingEnabled(true);
       const stub = getGetHostConfigStub({
         aidaAvailability: {
           disallowLogging: false,
@@ -205,7 +204,7 @@ describeWithEnvironment('FreestylerPanel', () => {
       });
 
       const aidaClient = getTestAidaClient();
-      panel = new Freestyler.FreestylerPanel(mockView, {
+      panel = new AiAssistance.AiAssistancePanel(mockView, {
         aidaClient,
         aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
         syncInfo: getTestSyncInfo(),
@@ -226,7 +225,7 @@ describeWithEnvironment('FreestylerPanel', () => {
       const RPC_ID = 0;
 
       const aidaClient = getTestAidaClient();
-      panel = new Freestyler.FreestylerPanel(mockView, {
+      panel = new AiAssistance.AiAssistancePanel(mockView, {
         aidaClient,
         aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
         syncInfo: getTestSyncInfo(),
@@ -251,7 +250,7 @@ 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, {
+      panel = new AiAssistance.AiAssistancePanel(mockView, {
         aidaClient,
         aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
         syncInfo: getTestSyncInfo(),
@@ -277,7 +276,7 @@ describeWithEnvironment('FreestylerPanel', () => {
       const RPC_ID = 0;
       const feedback = 'This helped me a ton.';
       const aidaClient = getTestAidaClient();
-      panel = new Freestyler.FreestylerPanel(mockView, {
+      panel = new AiAssistance.AiAssistancePanel(mockView, {
         aidaClient,
         aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
         syncInfo: getTestSyncInfo(),
@@ -307,7 +306,7 @@ 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, {
+        panel = new AiAssistance.AiAssistancePanel(mockView, {
           aidaClient: getTestAidaClient(),
           aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
           syncInfo: getTestSyncInfo(),
@@ -322,13 +321,13 @@ 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, {
+        panel = new AiAssistance.AiAssistancePanel(mockView, {
           aidaClient: getTestAidaClient(),
           aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
           syncInfo: getTestSyncInfo(),
@@ -346,13 +345,13 @@ 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, {
+        panel = new AiAssistance.AiAssistancePanel(mockView, {
           aidaClient: getTestAidaClient(),
           aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
           syncInfo: getTestSyncInfo(),
@@ -376,7 +375,7 @@ 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, {
+        panel = new AiAssistance.AiAssistancePanel(mockView, {
           aidaClient: getTestAidaClient(),
           aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
           syncInfo: getTestSyncInfo(),
@@ -394,7 +393,7 @@ 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, {
+        panel = new AiAssistance.AiAssistancePanel(mockView, {
           aidaClient: getTestAidaClient(),
           aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
           syncInfo: getTestSyncInfo(),
@@ -407,13 +406,13 @@ 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, {
+        panel = new AiAssistance.AiAssistancePanel(mockView, {
           aidaClient: getTestAidaClient(),
           aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
           syncInfo: getTestSyncInfo(),
@@ -429,13 +428,13 @@ 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, {
+        panel = new AiAssistance.AiAssistancePanel(mockView, {
           aidaClient: getTestAidaClient(),
           aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
           syncInfo: getTestSyncInfo(),
@@ -451,7 +450,7 @@ 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, {
+        panel = new AiAssistance.AiAssistancePanel(mockView, {
           aidaClient: getTestAidaClient(),
           aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
           syncInfo: getTestSyncInfo(),
@@ -464,13 +463,13 @@ 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, {
+        panel = new AiAssistance.AiAssistancePanel(mockView, {
           aidaClient: getTestAidaClient(),
           aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
           syncInfo: getTestSyncInfo(),
@@ -486,13 +485,13 @@ 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, {
+        panel = new AiAssistance.AiAssistancePanel(mockView, {
           aidaClient: getTestAidaClient(),
           aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
           syncInfo: getTestSyncInfo(),
@@ -508,7 +507,7 @@ 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, {
+        panel = new AiAssistance.AiAssistancePanel(mockView, {
           aidaClient: getTestAidaClient(),
           aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
           syncInfo: getTestSyncInfo(),
@@ -521,13 +520,13 @@ 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, {
+        panel = new AiAssistance.AiAssistancePanel(mockView, {
           aidaClient: getTestAidaClient(),
           aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
           syncInfo: getTestSyncInfo(),
@@ -543,13 +542,13 @@ 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, {
+        panel = new AiAssistance.AiAssistancePanel(mockView, {
           aidaClient: getTestAidaClient(),
           aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
           syncInfo: getTestSyncInfo(),
@@ -572,7 +571,7 @@ describeWithEnvironment('FreestylerPanel', () => {
     });
 
     it('should set inspectElementToggled when the widget is shown', () => {
-      panel = new Freestyler.FreestylerPanel(mockView, {
+      panel = new AiAssistance.AiAssistancePanel(mockView, {
         aidaClient: getTestAidaClient(),
         aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
         syncInfo: getTestSyncInfo(),
@@ -589,7 +588,7 @@ describeWithEnvironment('FreestylerPanel', () => {
 
     it('should update inspectElementToggled when the action is toggled', () => {
       toggleSearchElementAction.setToggled(false);
-      panel = new Freestyler.FreestylerPanel(mockView, {
+      panel = new AiAssistance.AiAssistancePanel(mockView, {
         aidaClient: getTestAidaClient(),
         aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
         syncInfo: getTestSyncInfo(),
@@ -610,7 +609,7 @@ 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, {
+         panel = new AiAssistance.AiAssistancePanel(mockView, {
            aidaClient: getTestAidaClient(),
            aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
            syncInfo: getTestSyncInfo(),
@@ -624,7 +623,7 @@ describeWithEnvironment('FreestylerPanel', () => {
 
   describe('history interactions', () => {
     it('should have empty messages after new chat', async () => {
-      panel = new Freestyler.FreestylerPanel(mockView, {
+      panel = new AiAssistance.AiAssistancePanel(mockView, {
         aidaClient: getTestAidaClient(),
         aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
         syncInfo: getTestSyncInfo(),
@@ -635,12 +634,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: [],
@@ -659,7 +658,7 @@ describeWithEnvironment('FreestylerPanel', () => {
           enabled: true,
         },
       });
-      panel = new Freestyler.FreestylerPanel(mockView, {
+      panel = new AiAssistance.AiAssistancePanel(mockView, {
         aidaClient: getTestAidaClient(),
         aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
         syncInfo: getTestSyncInfo(),
@@ -673,12 +672,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: [],
@@ -689,12 +688,12 @@ describeWithEnvironment('FreestylerPanel', () => {
       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, {
+      panel = new AiAssistance.AiAssistancePanel(mockView, {
         aidaClient: getTestAidaClient(),
         aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
         syncInfo: getTestSyncInfo(),
@@ -704,12 +703,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,12 +720,12 @@ 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,
+          entity: AiAssistance.ChatMessageEntity.MODEL,
           rpcId: undefined,
           suggestions: undefined,
           steps: [],
@@ -746,12 +745,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,7 +760,7 @@ describeWithEnvironment('FreestylerPanel', () => {
   });
 
   it('should have empty state after clear chat', async () => {
-    panel = new Freestyler.FreestylerPanel(mockView, {
+    panel = new AiAssistance.AiAssistancePanel(mockView, {
       aidaClient: getTestAidaClient(),
       aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
       syncInfo: getTestSyncInfo(),
@@ -772,23 +771,23 @@ 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 = toolbar!.shadowRoot!.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,7 +796,7 @@ describeWithEnvironment('FreestylerPanel', () => {
         enabled: true,
       },
     });
-    panel = new Freestyler.FreestylerPanel(mockView, {
+    panel = new AiAssistance.AiAssistancePanel(mockView, {
       aidaClient: getTestAidaClient(),
       aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
       syncInfo: getTestSyncInfo(),
@@ -810,28 +809,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 = toolbar!.shadowRoot!.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, {
+    panel = new AiAssistance.AiAssistancePanel(mockView, {
       aidaClient: getTestAidaClient(),
       aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
       syncInfo: getTestSyncInfo(),
@@ -841,12 +840,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,12 +857,12 @@ 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,
+        entity: AiAssistance.ChatMessageEntity.MODEL,
         rpcId: undefined,
         suggestions: undefined,
         steps: [],
@@ -876,12 +875,12 @@ describeWithEnvironment('FreestylerPanel', () => {
     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();
@@ -896,13 +895,14 @@ describeWithEnvironment('FreestylerPanel', () => {
     const menuItem = findMenuItemWithLabel(contextMenu.defaultSection(), 'No past conversations');
     assert(menuItem);
   });
+
   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,
       });
       UI.Context.Context.instance().setFlavor(SDK.NetworkRequest.NetworkRequest, networkRequest);
-      panel = new Freestyler.FreestylerPanel(mockView, {
+      panel = new AiAssistance.AiAssistancePanel(mockView, {
         aidaClient: getTestAidaClient(),
         aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
         syncInfo: getTestSyncInfo(),
@@ -912,7 +912,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,
       }));
 
@@ -930,10 +930,80 @@ describeWithEnvironment('FreestylerPanel', () => {
       await drainMicroTasks();
 
       sinon.assert.calledWith(mockView, sinon.match({
-        selectedContext: new Freestyler.RequestContext(networkRequest2),
+        selectedContext: new AiAssistance.RequestContext(networkRequest2),
         blockedByCrossOrigin: true,
       }));
     });
+
+    it('should be able to continue same-origin requests', async () => {
+      const stub = getGetHostConfigStub({
+        devToolsFreestyler: {
+          enabled: true,
+        },
+      });
+      panel = new AiAssistance.AiAssistancePanel(mockView, {
+        aidaClient: getTestAidaClient(),
+        aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
+        syncInfo: getTestSyncInfo(),
+      });
+      UI.Context.Context.instance().setFlavor(
+          ElementsPanel.ElementsPanel.ElementsPanel,
+          sinon.createStubInstance(ElementsPanel.ElementsPanel.ElementsPanel));
+
+      panel.handleAction('freestyler.elements-floating-button');
+      mockView.lastCall.args[0].onTextSubmit('test');
+      await drainMicroTasks();
+
+      assert.deepEqual(mockView.lastCall.args[0].messages, [
+        {
+          entity: AiAssistance.ChatMessageEntity.USER,
+          text: 'test',
+        },
+        {
+          answer: 'test',
+          entity: AiAssistance.ChatMessageEntity.MODEL,
+          rpcId: undefined,
+          suggestions: undefined,
+          steps: [],
+        },
+      ]);
+
+      UI.Context.Context.instance().setFlavor(
+          ElementsPanel.ElementsPanel.ElementsPanel,
+          sinon.createStubInstance(ElementsPanel.ElementsPanel.ElementsPanel));
+
+      panel.handleAction('freestyler.elements-floating-button');
+      mockView.lastCall.args[0].onTextSubmit('test2');
+      await drainMicroTasks();
+
+      assert.isFalse(mockView.lastCall.args[0].isReadOnly);
+      assert.deepEqual(mockView.lastCall.args[0].messages, [
+        {
+          entity: AiAssistance.ChatMessageEntity.USER,
+          text: 'test',
+        },
+        {
+          answer: 'test',
+          entity: AiAssistance.ChatMessageEntity.MODEL,
+          rpcId: undefined,
+          suggestions: undefined,
+          steps: [],
+        },
+        {
+          entity: AiAssistance.ChatMessageEntity.USER,
+          text: 'test2',
+        },
+        {
+          answer: 'test',
+          entity: AiAssistance.ChatMessageEntity.MODEL,
+          rpcId: undefined,
+          suggestions: undefined,
+          steps: [],
+        },
+      ]);
+
+      stub.restore();
+    });
   });
 
   describe('auto agent selection for panels', () => {
@@ -947,7 +1017,7 @@ describeWithEnvironment('FreestylerPanel', () => {
         UI.Context.Context.instance().setFlavor(
             ElementsPanel.ElementsPanel.ElementsPanel,
             sinon.createStubInstance(ElementsPanel.ElementsPanel.ElementsPanel));
-        panel = new Freestyler.FreestylerPanel(mockView, {
+        panel = new AiAssistance.AiAssistancePanel(mockView, {
           aidaClient: getTestAidaClient(),
           aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
           syncInfo: getTestSyncInfo(),
@@ -957,7 +1027,7 @@ describeWithEnvironment('FreestylerPanel', () => {
         panel.show(document.body);
 
         sinon.assert.calledWith(mockView, sinon.match({
-          agentType: AiAgent.AgentType.FREESTYLER,
+          agentType: AiAssistance.AgentType.STYLING,
         }));
         stub.restore();
       });
@@ -971,7 +1041,7 @@ describeWithEnvironment('FreestylerPanel', () => {
         UI.Context.Context.instance().setFlavor(
             ElementsPanel.ElementsPanel.ElementsPanel,
             sinon.createStubInstance(ElementsPanel.ElementsPanel.ElementsPanel));
-        panel = new Freestyler.FreestylerPanel(mockView, {
+        panel = new AiAssistance.AiAssistancePanel(mockView, {
           aidaClient: getTestAidaClient(),
           aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
           syncInfo: getTestSyncInfo(),
@@ -980,11 +1050,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();
       });
 
@@ -997,7 +1067,7 @@ describeWithEnvironment('FreestylerPanel', () => {
         UI.Context.Context.instance().setFlavor(
             ElementsPanel.ElementsPanel.ElementsPanel,
             sinon.createStubInstance(ElementsPanel.ElementsPanel.ElementsPanel));
-        panel = new Freestyler.FreestylerPanel(mockView, {
+        panel = new AiAssistance.AiAssistancePanel(mockView, {
           aidaClient: getTestAidaClient(),
           aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
           syncInfo: getTestSyncInfo(),
@@ -1006,7 +1076,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();
       });
     });
@@ -1020,7 +1090,7 @@ describeWithEnvironment('FreestylerPanel', () => {
         });
         UI.Context.Context.instance().setFlavor(
             NetworkPanel.NetworkPanel.NetworkPanel, sinon.createStubInstance(NetworkPanel.NetworkPanel.NetworkPanel));
-        panel = new Freestyler.FreestylerPanel(mockView, {
+        panel = new AiAssistance.AiAssistancePanel(mockView, {
           aidaClient: getTestAidaClient(),
           aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
           syncInfo: getTestSyncInfo(),
@@ -1030,7 +1100,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();
       });
@@ -1043,7 +1113,7 @@ describeWithEnvironment('FreestylerPanel', () => {
         });
         UI.Context.Context.instance().setFlavor(
             NetworkPanel.NetworkPanel.NetworkPanel, sinon.createStubInstance(NetworkPanel.NetworkPanel.NetworkPanel));
-        panel = new Freestyler.FreestylerPanel(mockView, {
+        panel = new AiAssistance.AiAssistancePanel(mockView, {
           aidaClient: getTestAidaClient(),
           aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
           syncInfo: getTestSyncInfo(),
@@ -1052,11 +1122,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();
       });
 
@@ -1070,7 +1140,7 @@ describeWithEnvironment('FreestylerPanel', () => {
            UI.Context.Context.instance().setFlavor(
                NetworkPanel.NetworkPanel.NetworkPanel,
                sinon.createStubInstance(NetworkPanel.NetworkPanel.NetworkPanel));
-           panel = new Freestyler.FreestylerPanel(mockView, {
+           panel = new AiAssistance.AiAssistancePanel(mockView, {
              aidaClient: getTestAidaClient(),
              aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
              syncInfo: getTestSyncInfo(),
@@ -1079,7 +1149,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();
          });
     });
@@ -1093,7 +1163,7 @@ describeWithEnvironment('FreestylerPanel', () => {
         });
         UI.Context.Context.instance().setFlavor(
             SourcesPanel.SourcesPanel.SourcesPanel, sinon.createStubInstance(SourcesPanel.SourcesPanel.SourcesPanel));
-        panel = new Freestyler.FreestylerPanel(mockView, {
+        panel = new AiAssistance.AiAssistancePanel(mockView, {
           aidaClient: getTestAidaClient(),
           aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
           syncInfo: getTestSyncInfo(),
@@ -1103,7 +1173,7 @@ describeWithEnvironment('FreestylerPanel', () => {
         panel.show(document.body);
 
         sinon.assert.calledWith(mockView, sinon.match({
-          agentType: AiAgent.AgentType.DRJONES_FILE,
+          agentType: AiAssistance.AgentType.FILE,
         }));
         stub.restore();
       });
@@ -1116,7 +1186,7 @@ describeWithEnvironment('FreestylerPanel', () => {
         });
         UI.Context.Context.instance().setFlavor(
             SourcesPanel.SourcesPanel.SourcesPanel, sinon.createStubInstance(SourcesPanel.SourcesPanel.SourcesPanel));
-        panel = new Freestyler.FreestylerPanel(mockView, {
+        panel = new AiAssistance.AiAssistancePanel(mockView, {
           aidaClient: getTestAidaClient(),
           aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
           syncInfo: getTestSyncInfo(),
@@ -1125,11 +1195,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();
       });
 
@@ -1143,7 +1213,7 @@ describeWithEnvironment('FreestylerPanel', () => {
            UI.Context.Context.instance().setFlavor(
                SourcesPanel.SourcesPanel.SourcesPanel,
                sinon.createStubInstance(SourcesPanel.SourcesPanel.SourcesPanel));
-           panel = new Freestyler.FreestylerPanel(mockView, {
+           panel = new AiAssistance.AiAssistancePanel(mockView, {
              aidaClient: getTestAidaClient(),
              aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
              syncInfo: getTestSyncInfo(),
@@ -1152,7 +1222,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();
          });
     });
@@ -1167,7 +1237,7 @@ describeWithEnvironment('FreestylerPanel', () => {
         UI.Context.Context.instance().setFlavor(
             TimelinePanel.TimelinePanel.TimelinePanel,
             sinon.createStubInstance(TimelinePanel.TimelinePanel.TimelinePanel));
-        panel = new Freestyler.FreestylerPanel(mockView, {
+        panel = new AiAssistance.AiAssistancePanel(mockView, {
           aidaClient: getTestAidaClient(),
           aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
           syncInfo: getTestSyncInfo(),
@@ -1177,7 +1247,7 @@ describeWithEnvironment('FreestylerPanel', () => {
         panel.show(document.body);
 
         sinon.assert.calledWith(mockView, sinon.match({
-          agentType: AiAgent.AgentType.DRJONES_PERFORMANCE,
+          agentType: AiAssistance.AgentType.PERFORMANCE,
         }));
         stub.restore();
       });
@@ -1191,7 +1261,7 @@ describeWithEnvironment('FreestylerPanel', () => {
         UI.Context.Context.instance().setFlavor(
             TimelinePanel.TimelinePanel.TimelinePanel,
             sinon.createStubInstance(TimelinePanel.TimelinePanel.TimelinePanel));
-        panel = new Freestyler.FreestylerPanel(mockView, {
+        panel = new AiAssistance.AiAssistancePanel(mockView, {
           aidaClient: getTestAidaClient(),
           aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
           syncInfo: getTestSyncInfo(),
@@ -1200,11 +1270,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();
       });
 
@@ -1218,7 +1288,7 @@ describeWithEnvironment('FreestylerPanel', () => {
            UI.Context.Context.instance().setFlavor(
                TimelinePanel.TimelinePanel.TimelinePanel,
                sinon.createStubInstance(TimelinePanel.TimelinePanel.TimelinePanel));
-           panel = new Freestyler.FreestylerPanel(mockView, {
+           panel = new AiAssistance.AiAssistancePanel(mockView, {
              aidaClient: getTestAidaClient(),
              aidaAvailability: Host.AidaClient.AidaAccessPreconditions.AVAILABLE,
              syncInfo: getTestSyncInfo(),
@@ -1227,9 +1297,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');
+  });
 });