Add global debugProblem method for calling AI assistance

Bug: 416138589
Change-Id: I2426a50569117f994e4bf8725b6f8a54bbdbcd89
Reviewed-on: https://chromium-review.googlesource.com/c/devtools/devtools-frontend/+/6520239
Commit-Queue: Alex Rudenko <alexrudenko@chromium.org>
Reviewed-by: Alex Rudenko <alexrudenko@chromium.org>

Author: wolfib

front_end/panels/ai_assistance/AiAssistancePanel.test.ts

@@ -1393,6 +1393,93 @@ describeWithMockConnection('AI Assistance Panel', () => {
       assert.isUndefined((await view.nextInput).imageInput);
     });
   });
+
+  describe('debugProblem', () => {
+    const explanation = 'I need more information';
+    const notAvailable =
+        'For AI features to be available, you need to log into Chrome and enable AI assistance in DevTools settings';
+
+    beforeEach(() => {
+      Common.Settings.moduleSetting('ai-assistance-enabled').set(true);
+      updateHostConfig({
+        devToolsFreestyler: {
+          enabled: true,
+        },
+      });
+    });
+
+    it('can be blocked by a setting', async () => {
+      Common.Settings.moduleSetting('ai-assistance-enabled').set(false);
+      const {panel} = await createAiAssistancePanel({
+        aidaClient: mockAidaClient([[{explanation}]]),
+      });
+      try {
+        await panel.debugProblem('Please help me debug this problem');
+        assert.fail('Expected `debugProblem` to throw');
+      } catch (err) {
+        assert.strictEqual(err.message, notAvailable);
+      }
+    });
+
+    it('can be blocked by feature availability', async () => {
+      const {panel} = await createAiAssistancePanel({
+        aidaClient: mockAidaClient([[{explanation}]]),
+        aidaAvailability: Host.AidaClient.AidaAccessPreconditions.SYNC_IS_PAUSED,
+      });
+      try {
+        await panel.debugProblem('Please help me debug this problem');
+        assert.fail('Expected `debugProblem` to throw');
+      } catch (err) {
+        assert.strictEqual(err.message, notAvailable);
+      }
+    });
+
+    it('can be blocked by user age', async () => {
+      updateHostConfig({
+        aidaAvailability: {
+          blockedByAge: true,
+        },
+        devToolsFreestyler: {
+          enabled: true,
+        },
+      });
+      const {panel} = await createAiAssistancePanel({
+        aidaClient: mockAidaClient([[{explanation}]]),
+      });
+      try {
+        await panel.debugProblem('Please help me debug this problem');
+        assert.fail('Expected `debugProblem` to throw');
+      } catch (err) {
+        assert.strictEqual(err.message, notAvailable);
+      }
+    });
+
+    it('returns an explanation', async () => {
+      const {panel} = await createAiAssistancePanel({
+        aidaClient: mockAidaClient([[{explanation}]]),
+      });
+      const response = await panel.debugProblem('Please help me debug this problem');
+      assert.strictEqual(response, explanation);
+    });
+
+    it('throws an error if no answer could be generated', async () => {
+      const {panel} = await createAiAssistancePanel({
+        aidaClient: mockAidaClient([
+          [{
+            explanation: `ACTION
+$0.style.backgroundColor = 'red'
+STOP`,
+          }],
+        ])
+      });
+      try {
+        await panel.debugProblem('Please help me debug this problem');
+        assert.fail('Expected `debugProblem` to throw');
+      } catch (err) {
+        assert.strictEqual(err.message, 'Something went wrong. No answer was generated.');
+      }
+    });
+  });
 });
 
 describeWithEnvironment('AiAssistancePanel.ActionDelegate', () => {

front_end/panels/ai_assistance/AiAssistancePanel.ts

@@ -1310,6 +1310,43 @@ export class AiAssistancePanel extends UI.Panel.Panel {
     return context;
   }
 
+  // Called by MCP server via Puppeteer
+  // TODO(http://b/416460908) Make this generic and select the agent based on an input parameter
+  async debugProblem(prompt: string): Promise<string> {
+    const aiAssistanceSetting = this.#aiAssistanceEnabledSetting?.getIfNotDisabled();
+    const isBlockedByAge = Root.Runtime.hostConfig.aidaAvailability?.blockedByAge === true;
+    const isAidaAvailable = this.#aidaAvailability === Host.AidaClient.AidaAccessPreconditions.AVAILABLE;
+    if (!aiAssistanceSetting || isBlockedByAge || !isAidaAvailable) {
+      throw new Error(
+          'For AI features to be available, you need to log into Chrome and enable AI assistance in DevTools settings');
+    }
+
+    const stylingAgent = this.#createAgent(AiAssistanceModel.ConversationType.STYLING);
+    // Cancel any previous in-flight conversation.
+    this.#cancel();
+    const runner = stylingAgent.run(
+        prompt,
+        {
+          signal: this.#runAbortController.signal,
+          // TODO(crbug.com/416134018) provide context via MCP instead of using the currently selected element
+          selected: this.#getConversationContext(),
+        },
+    );
+    for await (const data of runner) {
+      if (data.type === AiAssistanceModel.ResponseType.SIDE_EFFECT) {
+        data.confirm(true);
+      }
+      if (data.type === AiAssistanceModel.ResponseType.ANSWER && data.complete) {
+        await this.#changeManager.stashChanges();
+        this.#changeManager.dropStashedChanges();
+        return data.text;
+      }
+    }
+    await this.#changeManager.stashChanges();
+    this.#changeManager.dropStashedChanges();
+    throw new Error('Something went wrong. No answer was generated.');
+  }
+
   async #startConversation(
       text: string, imageInput?: Host.AidaClient.Part,
       multimodalInputType?: AiAssistanceModel.MultimodalInputType): Promise<void> {

front_end/panels/ai_assistance/ai_assistance-meta.ts

@@ -263,3 +263,12 @@ UI.ActionRegistration.registerActionExtension({
   },
   condition: config => isFileAgentFeatureAvailable(config) && !isPolicyRestricted(config) && !isGeoRestricted(config),
 });
+
+// Called by MCP server via Puppeteer
+// TODO(http://b/416460908) Make this generic and select the agent based on an input parameter
+// @ts-expect-error
+globalThis.debugProblem = async(prompt: string): Promise<string> => {
+  const AiAssistance = await loadAiAssistanceModule();
+  const panelInstance = await AiAssistance.AiAssistancePanel.instance();
+  return await panelInstance.debugProblem(prompt);
+};