chrome-devtools-frontend 1.0.1362775 → 1.0.1367881

package/front_end/panels/freestyler/DrJonesFileAgent.ts

@@ -0,0 +1,196 @@
+// Copyright 2024 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+import * as Common from '../../core/common/common.js';
+import * as Host from '../../core/host/host.js';
+import * as i18n from '../../core/i18n/i18n.js';
+import type * as Workspace from '../../models/workspace/workspace.js';
+
+import {
+  AiAgent,
+  type AidaRequestOptions,
+  type ContextDetail,
+  debugLog,
+  ErrorType,
+  isDebugMode,
+  type ResponseData,
+  ResponseType,
+  type ThoughtResponse,
+  type TitleResponse,
+} from './AiAgent.js';
+
+const preamble =
+    `You are a highly skilled software engineer with expertise in various programming languages and frameworks.
+You are provided with the content of a file from the Chrome DevTools Sources panel.
+
+**Important Note:** The provided code may represent an incomplete fragment of a larger file.
+
+Analyze the code and provide the following information:
+* Describe the primary functionality of the code. What does it do? Be specific and concise.
+* If possible, identify the framework or library the code is associated with (e.g., React, Angular, jQuery). List any key technologies, APIs, or patterns used in the code (e.g., Fetch API, WebSockets, object-oriented programming).
+* (Only provide if available and accessible externally) External Resources: Suggest relevant documentation that could help a developer understand the code better. Prioritize official documentation if available. Do not provide any internal resources.
+
+# Considerations
+* Keep your analysis concise and focused, highlighting only the most critical aspects for a software engineer.
+* **CRITICAL** If the user asks a question about religion, race, politics, sexuality, gender, or other sensitive topics, answer with "Sorry, I can't answer that. I'm best at questions about files."
+
+## Example session
+
+**User:** (Selects a file containing the following JavaScript code)
+
+function calculateTotal(price, quantity) {
+  const total = price * quantity;
+  return total;
+}
+Explain this file.
+
+
+This code defines a function called calculateTotal that calculates the total cost by multiplying the price and quantity arguments.
+This code is written in JavaScript and doesn't seem to be associated with a specific framework. It's likely a utility function.
+Relevant Technologies: JavaScript, functions, arithmetic operations.
+External Resources:
+MDN Web Docs: JavaScript Functions: https://developer.mozilla.org/en-US/docs/Web/JavaScript/Guide/Functions
+`;
+
+/*
+* Strings that don't need to be translated at this time.
+*/
+const UIStringsNotTranslate = {
+  /**
+   *@description Title for thinking step of DrJones File agent.
+   */
+  analyzingFile: 'Analyzing file',
+  /**
+   *@description Thought text for thinking step of DrJones File agent.
+   */
+  dataUsedToGenerateThisResponse: 'Data used to generate this response',
+};
+
+const lockedString = i18n.i18n.lockedString;
+
+const MAX_FILE_SIZE = 50000;
+
+/**
+ * One agent instance handles one conversation. Create a new agent
+ * instance for a new conversation.
+ */
+export class DrJonesFileAgent extends AiAgent {
+  readonly preamble = preamble;
+  readonly clientFeature = Host.AidaClient.ClientFeature.CHROME_DRJONES_FILE_AGENT;
+  // TODO(b/369822364): use a feature param instead.
+  readonly userTier = 'BETA';
+  get options(): AidaRequestOptions {
+    const config = Common.Settings.Settings.instance().getHostConfig();
+    const temperature = AiAgent.validTemperature(config.devToolsAiAssistanceFileAgentDogfood?.temperature);
+    const modelId = config.devToolsAiAssistanceFileAgentDogfood?.modelId;
+
+    return {
+      temperature,
+      model_id: modelId,
+    };
+  }
+
+  *
+      handleContextDetails(selectedFile: Workspace.UISourceCode.UISourceCode|null):
+          Generator<ThoughtResponse|TitleResponse, void, void> {
+    if (!selectedFile) {
+      return;
+    }
+
+    yield {
+      type: ResponseType.TITLE,
+      title: lockedString(UIStringsNotTranslate.analyzingFile),
+    };
+    yield {
+      type: ResponseType.THOUGHT,
+      thought: lockedString(UIStringsNotTranslate.dataUsedToGenerateThisResponse),
+      contextDetails: createContextDetailsForDrJonesFileAgent(selectedFile),
+    };
+  }
+
+  async enhanceQuery(query: string, selectedFile: Workspace.UISourceCode.UISourceCode|null): Promise<string> {
+    const fileEnchantmentQuery =
+        selectedFile ? `# Selected file\n${formatFile(selectedFile)}\n\n# User request\n\n` : '';
+    return `${fileEnchantmentQuery}${query}`;
+  }
+
+  #runId = 0;
+  async * run(query: string, options: {
+    signal?: AbortSignal, selectedFile: Workspace.UISourceCode.UISourceCode|null,
+  }): AsyncGenerator<ResponseData, void, void> {
+    yield* this.handleContextDetails(options.selectedFile);
+
+    query = await this.enhanceQuery(query, options.selectedFile);
+    const currentRunId = ++this.#runId;
+
+    let response: string;
+    let rpcId: number|undefined;
+    try {
+      const fetchResult = await this.aidaFetch(query, {signal: options.signal});
+      response = fetchResult.response;
+      rpcId = fetchResult.rpcId;
+    } catch (err) {
+      debugLog('Error calling the AIDA API', err);
+      if (err instanceof Host.AidaClient.AidaAbortError) {
+        this.removeHistoryRun(currentRunId);
+        yield {
+          type: ResponseType.ERROR,
+          error: ErrorType.ABORT,
+          rpcId,
+        };
+        return;
+      }
+
+      yield {
+        type: ResponseType.ERROR,
+        error: ErrorType.UNKNOWN,
+        rpcId,
+      };
+      return;
+    }
+
+    this.addToHistory({
+      id: currentRunId,
+      query,
+      output: response,
+    });
+
+    yield {
+      type: ResponseType.ANSWER,
+      text: response,
+      rpcId,
+    };
+    if (isDebugMode()) {
+      window.dispatchEvent(new CustomEvent('freestylerdone'));
+    }
+  }
+}
+
+function createContextDetailsForDrJonesFileAgent(selectedFile: Workspace.UISourceCode.UISourceCode):
+    [ContextDetail, ...ContextDetail[]] {
+  return [
+    {
+      title: 'Selected file',
+      text: formatFile(selectedFile),
+    },
+  ];
+}
+
+function formatFile(selectedFile: Workspace.UISourceCode.UISourceCode): string {
+  return `File Name: ${selectedFile.displayName()}
+URL: ${selectedFile.url()}
+File Content:
+${selectedFile.content().slice(0, MAX_FILE_SIZE)}`;
+}
+
+function setDebugFreestylerEnabled(enabled: boolean): void {
+  if (enabled) {
+    localStorage.setItem('debugFreestylerEnabled', 'true');
+  } else {
+    localStorage.removeItem('debugFreestylerEnabled');
+  }
+}
+
+// @ts-ignore
+globalThis.setDebugFreestylerEnabled = setDebugFreestylerEnabled;

package/front_end/panels/freestyler/DrJonesNetworkAgent.test.ts

@@ -107,7 +107,7 @@ describeWithEnvironment('DrJonesNetworkAgent', () => {
               model_id: 'test model',
               temperature: undefined,
             },
-            client_feature: 2,
+            client_feature: 7,
             functionality_type: 1,
           },
       );

package/front_end/panels/freestyler/DrJonesNetworkAgent.ts

@@ -59,6 +59,9 @@ const MAX_HEADERS_SIZE = 1000;
 * Strings that don't need to be translated at this time.
 */
 const UIStringsNotTranslate = {
+  /**
+   *@description Title for thinking step of DrJones Network agent.
+   */
   inspectingNetworkData: 'Inspecting network data',
   /**
    *@description Thought text for thinking step of DrJones Network agent.
@@ -106,7 +109,7 @@ const lockedString = i18n.i18n.lockedString;
  */
 export class DrJonesNetworkAgent extends AiAgent {
   readonly preamble = preamble;
-  readonly clientFeature = Host.AidaClient.ClientFeature.CHROME_FREESTYLER;
+  readonly clientFeature = Host.AidaClient.ClientFeature.CHROME_DRJONES_NETWORK_AGENT;
   // TODO(b/369822364): use a feature param instead.
   readonly userTier = 'BETA';
   get options(): AidaRequestOptions {
@@ -123,6 +126,10 @@ export class DrJonesNetworkAgent extends AiAgent {
   *
       handleContextDetails(selectedNetworkRequest: SDK.NetworkRequest.NetworkRequest|null):
           Generator<ThoughtResponse|TitleResponse, void, void> {
+    if (!selectedNetworkRequest) {
+      return;
+    }
+
     yield {
       type: ResponseType.TITLE,
       title: lockedString(UIStringsNotTranslate.inspectingNetworkData),
@@ -276,8 +283,6 @@ function formatRequestInitiatorChain(request: SDK.NetworkRequest.NetworkRequest)
 }
 
 export function formatNetworkRequest(request: SDK.NetworkRequest.NetworkRequest): string {
-  const formatHeaders = (title: string, headers: SDK.NetworkRequest.NameValue[]): string => formatLines(
-      title, headers.filter(allowHeader).map(header => header.name + ': ' + header.value + '\n'), MAX_HEADERS_SIZE);
   // TODO: anything else that might be relavant?
   // TODO: handle missing headers
   return `Request: ${request.url()}
@@ -293,32 +298,30 @@ Request Timing:\n${formatNetworkRequestTiming(request)}
 Request Initiator Chain:\n${formatRequestInitiatorChain(request)}`;
 }
 
-function createContextDetailsForDrJonesNetworkAgent(request: SDK.NetworkRequest.NetworkRequest|null): ContextDetail[] {
-  if (request) {
-    const requestContextDetail: ContextDetail = {
-      title: lockedString(UIStringsNotTranslate.request),
-      text: lockedString(UIStringsNotTranslate.requestUrl) + ': ' + request.url() + '\n\n' +
-          formatHeaders(lockedString(UIStringsNotTranslate.requestHeaders), request.requestHeaders()),
-    };
-    const responseContextDetail: ContextDetail = {
-      title: lockedString(UIStringsNotTranslate.response),
-      text: lockedString(UIStringsNotTranslate.responseStatus) + ': ' + request.statusCode + ' ' + request.statusText +
-          '\n\n' + formatHeaders(lockedString(UIStringsNotTranslate.responseHeaders), request.responseHeaders),
-    };
-    const timingContextDetail: ContextDetail = {
-      title: lockedString(UIStringsNotTranslate.timing),
-      text: formatNetworkRequestTiming(request),
-    };
-    const initiatorChainContextDetail: ContextDetail = {
-      title: lockedString(UIStringsNotTranslate.requestInitiatorChain),
-      text: formatRequestInitiatorChain(request),
-    };
-    return [
-      requestContextDetail,
-      responseContextDetail,
-      timingContextDetail,
-      initiatorChainContextDetail,
-    ];
-  }
-  return [];
+function createContextDetailsForDrJonesNetworkAgent(request: SDK.NetworkRequest.NetworkRequest):
+    [ContextDetail, ...ContextDetail[]] {
+  const requestContextDetail: ContextDetail = {
+    title: lockedString(UIStringsNotTranslate.request),
+    text: lockedString(UIStringsNotTranslate.requestUrl) + ': ' + request.url() + '\n\n' +
+        formatHeaders(lockedString(UIStringsNotTranslate.requestHeaders), request.requestHeaders()),
+  };
+  const responseContextDetail: ContextDetail = {
+    title: lockedString(UIStringsNotTranslate.response),
+    text: lockedString(UIStringsNotTranslate.responseStatus) + ': ' + request.statusCode + ' ' + request.statusText +
+        '\n\n' + formatHeaders(lockedString(UIStringsNotTranslate.responseHeaders), request.responseHeaders),
+  };
+  const timingContextDetail: ContextDetail = {
+    title: lockedString(UIStringsNotTranslate.timing),
+    text: formatNetworkRequestTiming(request),
+  };
+  const initiatorChainContextDetail: ContextDetail = {
+    title: lockedString(UIStringsNotTranslate.requestInitiatorChain),
+    text: formatRequestInitiatorChain(request),
+  };
+  return [
+    requestContextDetail,
+    responseContextDetail,
+    timingContextDetail,
+    initiatorChainContextDetail,
+  ];
 }

package/front_end/panels/freestyler/DrJonesPerformanceAgent.test.ts

@@ -0,0 +1,195 @@
+// Copyright 2024 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+import * as Host from '../../core/host/host.js';
+import * as Trace from '../../models/trace/trace.js';
+import {
+  describeWithEnvironment,
+  getGetHostConfigStub,
+} from '../../testing/EnvironmentHelpers.js';
+
+import {DrJonesPerformanceAgent, ResponseType} from './freestyler.js';
+
+describeWithEnvironment('DrJonesPerformanceAgent', () => {
+  function mockHostConfig(modelId?: string, temperature?: number) {
+    getGetHostConfigStub({
+      devToolsAiAssistancePerformanceAgentDogfood: {
+        modelId,
+        temperature,
+      },
+    });
+  }
+
+  function mockAidaClient(
+      fetch: () => AsyncGenerator<Host.AidaClient.AidaResponse, void, void>,
+      ): Host.AidaClient.AidaClient {
+    return {
+      fetch,
+      registerClientEvent: () => Promise.resolve({}),
+    };
+  }
+
+  describe('buildRequest', () => {
+    beforeEach(() => {
+      sinon.restore();
+    });
+
+    it('builds a request with a model id', async () => {
+      mockHostConfig('test model');
+      const agent = new DrJonesPerformanceAgent({
+        aidaClient: {} as Host.AidaClient.AidaClient,
+      });
+      assert.strictEqual(
+          agent.buildRequest({input: 'test input'}).options?.model_id,
+          'test model',
+      );
+    });
+
+    it('builds a request with a temperature', async () => {
+      mockHostConfig('test model', 1);
+      const agent = new DrJonesPerformanceAgent({
+        aidaClient: {} as Host.AidaClient.AidaClient,
+      });
+      assert.strictEqual(
+          agent.buildRequest({input: 'test input'}).options?.temperature,
+          1,
+      );
+    });
+
+    it('structure matches the snapshot', () => {
+      mockHostConfig('test model');
+      sinon.stub(crypto, 'randomUUID').returns('sessionId' as `${string}-${string}-${string}-${string}-${string}`);
+      const agent = new DrJonesPerformanceAgent({
+        aidaClient: {} as Host.AidaClient.AidaClient,
+        serverSideLoggingEnabled: true,
+      });
+      sinon.stub(agent, 'preamble').value('preamble');
+      agent.chatHistoryForTesting = new Map([[
+        0,
+        [
+          {
+            text: 'first',
+            entity: Host.AidaClient.Entity.UNKNOWN,
+          },
+          {
+            text: 'second',
+            entity: Host.AidaClient.Entity.SYSTEM,
+          },
+          {
+            text: 'third',
+            entity: Host.AidaClient.Entity.USER,
+          },
+        ],
+      ]]);
+      assert.deepStrictEqual(
+          agent.buildRequest({
+            input: 'test input',
+          }),
+          {
+            input: 'test input',
+            client: 'CHROME_DEVTOOLS',
+            preamble: 'preamble',
+            chat_history: [
+              {
+                entity: 0,
+                text: 'first',
+              },
+              {
+                entity: 2,
+                text: 'second',
+              },
+              {
+                entity: 1,
+                text: 'third',
+              },
+            ],
+            metadata: {
+              disable_user_content_logging: false,
+              string_session_id: 'sessionId',
+              user_tier: 2,
+            },
+            options: {
+              model_id: 'test model',
+              temperature: undefined,
+            },
+            client_feature: 2,
+            functionality_type: 1,
+          },
+      );
+    });
+  });
+  describe('run', () => {
+    const rootNodeEntry =
+        new Trace.Helpers.TreeHelpers.TraceEntryNodeForAI('RunTask', Trace.Types.Timing.MilliSeconds(0));
+    const node1 = new Trace.Helpers.TreeHelpers.TraceEntryNodeForAI('ProfileCall', Trace.Types.Timing.MilliSeconds(1));
+    const node2 = new Trace.Helpers.TreeHelpers.TraceEntryNodeForAI('ProfileCall', Trace.Types.Timing.MilliSeconds(2));
+    const node3 = new Trace.Helpers.TreeHelpers.TraceEntryNodeForAI('ProfileCall', Trace.Types.Timing.MilliSeconds(10));
+    const node4 = new Trace.Helpers.TreeHelpers.TraceEntryNodeForAI('ProfileCall', Trace.Types.Timing.MilliSeconds(11));
+    const node5 = new Trace.Helpers.TreeHelpers.TraceEntryNodeForAI('ProfileCall', Trace.Types.Timing.MilliSeconds(15));
+
+    beforeEach(() => {
+      rootNodeEntry.children = [node1, node3];
+      node1.function = 'foo';
+      node1.children = [node2];
+      node2.function = 'foofoo';
+      node3.function = 'foo2';
+      node3.children = [node4, node5];
+      node4.function = 'foo2foo';
+      node5.function = 'foo2foo2';
+    });
+
+    it('generates an answer', async () => {
+      async function* generateAnswer() {
+        yield {
+          explanation: 'This is the answer',
+          metadata: {
+            rpcGlobalId: 123,
+          },
+          completed: true,
+        };
+      }
+
+      const agent = new DrJonesPerformanceAgent({
+        aidaClient: mockAidaClient(generateAnswer),
+      });
+
+      // Select node3
+      node3.selected = true;
+      const responses = await Array.fromAsync(agent.run('test', {selectedStackTrace: rootNodeEntry}));
+
+      assert.deepStrictEqual(responses, [
+        {
+          type: ResponseType.TITLE,
+          title: 'Analyzing stack trace',
+        },
+        {
+          type: ResponseType.THOUGHT,
+          thought: 'Data used to generate this response',
+          contextDetails: [
+            {
+              title: 'Selected stack trace',
+              text: JSON.stringify(rootNodeEntry),
+            },
+          ],
+        },
+        {
+          type: ResponseType.ANSWER,
+          text: 'This is the answer',
+          rpcId: 123,
+        },
+      ]);
+
+      assert.deepStrictEqual(agent.chatHistoryForTesting, [
+        {
+          entity: 1,
+          text: `# Selected stack trace\n${JSON.stringify(rootNodeEntry)}\n\n# User request\n\ntest`,
+        },
+        {
+          entity: 2,
+          text: 'This is the answer',
+        },
+      ]);
+    });
+  });
+});

package/front_end/panels/freestyler/DrJonesPerformanceAgent.ts

@@ -0,0 +1,193 @@
+// Copyright 2024 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+import * as Common from '../../core/common/common.js';
+import * as Host from '../../core/host/host.js';
+import * as i18n from '../../core/i18n/i18n.js';
+import type * as Trace from '../../models/trace/trace.js';
+
+import {
+  AiAgent,
+  type AidaRequestOptions,
+  type ContextDetail,
+  debugLog,
+  ErrorType,
+  isDebugMode,
+  type ResponseData,
+  ResponseType,
+  type ThoughtResponse,
+  type TitleResponse,
+} from './AiAgent.js';
+
+const preamble = `You are a performance expert deeply integrated with Chrome DevTools.
+You specialize in analyzing web application behavior captured by Chrome DevTools Performance Panel.
+You will be provided with a string containing the JSON.stringify representation of a tree of events captured in a Chrome DevTools performance recording.
+This tree originates from the root task of a specific event that was selected by a user in the Performance panel's flame graph.
+Each node in this tree represents an event and contains the following information:
+* id: A unique identifier for the event.
+* url:  The URL of the JavaScript file where the event originated (if applicable).
+* line: The line number in the JavaScript file where the event originated (if applicable).
+* column: The column number in the JavaScript file where the event originated (if applicable).
+* function: The name of the function associated with this event.
+* start: The start time of the event in milliseconds.
+* totalTime: The total duration of the event, including the time spent in its children, in milliseconds.
+* selfTime:  The duration of the event itself, excluding the time spent in its children, in milliseconds.
+* selected: A boolean value indicating whether this is the event the user selected in the Performance panel.
+* children: An array of child events, each represented as another node with the same structure.
+
+Your task is to analyze this event and its surrounding context within the performance recording. Your analysis may include:
+* Clearly state the name and purpose of the selected event based on its properties (e.g., function name, URL, line number). Explain what the task is broadly doing. You can also mention the function
+* Describe its execution context:
+  * Ancestors: Trace back through the tree to identify the chain of parent events that led to the execution of the selected event. Describe this execution path.
+  * Descendants:  Analyze the children of the selected event. What tasks did it initiate? Did it spawn any long-running or resource-intensive sub-tasks?
+* Quantify performance:
+    * Duration
+    * Relative Cost:  How much did this event contribute to the overall duration of its parent tasks and the entire recorded trace?
+    * Potential Bottlenecks: Analyze the totalTime and selfTime of the selected event and its children to identify any potential performance bottlenecks. Are there any excessively long tasks or periods of idle time?
+4. (Only provide if you have specific suggestions) Based on your analysis, provide specific and actionable suggestions for improving the performance of the selected event and its related tasks.  Are there any resources being acquired or held for longer than necessary?
+
+# Considerations
+* Keep your analysis concise and focused, highlighting only the most critical aspects for a software engineer.
+* Do not mention id of the event in your response.
+
+## Example session
+
+Selected call tree
+'{"id":"1","function":"main","start":0,"totalTime":500,"selfTime":100,"children":[{"id":"2","function":"update","start":100,"totalTime":200,"selfTime":50,"children":[{"id":"3","function":"animate","url":"[invalid URL removed]","line":120,"column":10,"start":150,"totalTime":150,"selfTime":20,"selected":true,"children":[{"id":"4","function":"calculatePosition","start":160,"totalTime":80,"selfTime":80,"children":[]},{"id":"5","function":"applyStyles","start":240,"totalTime":50,"selfTime":50,"children":[]}]}]}]}'
+Explain the selected task.
+
+
+It looks like you've selected the animate function, which is responsible for animating elements on the page.
+This function took a total of 150ms to execute, but only 20ms of that time was spent within the animate function itself.
+The remaining 130ms were spent in its child functions, calculatePosition and applyStyles.
+It seems like a significant portion of the animation time is spent calculating the position of the elements.
+Perhaps there's room for optimization there. You could investigate whether the calculatePosition function can be made more efficient or if the number of calculations can be reduced.
+`;
+
+/*
+* Strings that don't need to be translated at this time.
+*/
+const UIStringsNotTranslate = {
+  analyzingStackTrace: 'Analyzing stack trace',
+  /**
+   *@description Thought text for thinking step of DrJones Performance agent.
+   */
+  dataUsedToGenerateThisResponse: 'Data used to generate this response',
+};
+
+const lockedString = i18n.i18n.lockedString;
+
+/**
+ * One agent instance handles one conversation. Create a new agent
+ * instance for a new conversation.
+ */
+export class DrJonesPerformanceAgent extends AiAgent {
+  readonly preamble = preamble;
+  readonly clientFeature = Host.AidaClient.ClientFeature.CHROME_FREESTYLER;
+  // TODO(b/369822364): use a feature param instead.
+  readonly userTier = 'BETA';
+  get options(): AidaRequestOptions {
+    const config = Common.Settings.Settings.instance().getHostConfig();
+    const temperature = AiAgent.validTemperature(config.devToolsAiAssistancePerformanceAgentDogfood?.temperature);
+    const modelId = config.devToolsAiAssistancePerformanceAgentDogfood?.modelId;
+
+    return {
+      temperature,
+      model_id: modelId,
+    };
+  }
+
+  *
+      handleContextDetails(selectedStackTrace: Trace.Helpers.TreeHelpers.TraceEntryNodeForAI|null):
+          Generator<ThoughtResponse|TitleResponse, void, void> {
+    yield {
+      type: ResponseType.TITLE,
+      title: lockedString(UIStringsNotTranslate.analyzingStackTrace),
+    };
+    yield {
+      type: ResponseType.THOUGHT,
+      thought: lockedString(UIStringsNotTranslate.dataUsedToGenerateThisResponse),
+      contextDetails: createContextDetailsForDrJonesPerformanceAgent(selectedStackTrace),
+    };
+  }
+
+  async enhanceQuery(query: string, selectedStackTrace: Trace.Helpers.TreeHelpers.TraceEntryNodeForAI|null):
+      Promise<string> {
+    const networkEnchantmentQuery =
+        selectedStackTrace ? `# Selected stack trace\n${JSON.stringify(selectedStackTrace)}\n\n# User request\n\n` : '';
+    return `${networkEnchantmentQuery}${query}`;
+  }
+
+  #runId = 0;
+  async * run(query: string, options: {
+    signal?: AbortSignal, selectedStackTrace: Trace.Helpers.TreeHelpers.TraceEntryNodeForAI|null,
+  }): AsyncGenerator<ResponseData, void, void> {
+    yield* this.handleContextDetails(options.selectedStackTrace);
+
+    query = await this.enhanceQuery(query, options.selectedStackTrace);
+    const currentRunId = ++this.#runId;
+
+    let response: string;
+    let rpcId: number|undefined;
+    try {
+      const fetchResult = await this.aidaFetch(query, {signal: options.signal});
+      response = fetchResult.response;
+      rpcId = fetchResult.rpcId;
+    } catch (err) {
+      debugLog('Error calling the AIDA API', err);
+      if (err instanceof Host.AidaClient.AidaAbortError) {
+        this.removeHistoryRun(currentRunId);
+        yield {
+          type: ResponseType.ERROR,
+          error: ErrorType.ABORT,
+          rpcId,
+        };
+        return;
+      }
+
+      yield {
+        type: ResponseType.ERROR,
+        error: ErrorType.UNKNOWN,
+        rpcId,
+      };
+      return;
+    }
+
+    this.addToHistory({
+      id: currentRunId,
+      query,
+      output: response,
+    });
+
+    yield {
+      type: ResponseType.ANSWER,
+      text: response,
+      rpcId,
+    };
+    if (isDebugMode()) {
+      window.dispatchEvent(new CustomEvent('freestylerdone'));
+    }
+  }
+}
+
+function createContextDetailsForDrJonesPerformanceAgent(
+    selectedStackTrace: Trace.Helpers.TreeHelpers.TraceEntryNodeForAI|null): [ContextDetail, ...ContextDetail[]] {
+  return [
+    {
+      title: 'Selected stack trace',
+      text: JSON.stringify(selectedStackTrace),
+    },
+  ];
+}
+
+function setDebugFreestylerEnabled(enabled: boolean): void {
+  if (enabled) {
+    localStorage.setItem('debugFreestylerEnabled', 'true');
+  } else {
+    localStorage.removeItem('debugFreestylerEnabled');
+  }
+}
+
+// @ts-ignore
+globalThis.setDebugFreestylerEnabled = setDebugFreestylerEnabled;