[AI Assistance] Configure userTier

Bug: 393064551
Change-Id: Ic0da9c66c80a4a87f9f49efd25055b4622e0270b
Reviewed-on: https://chromium-review.googlesource.com/c/devtools/devtools-frontend/+/6349178
Auto-Submit: Alex Rudenko <[email protected]>
Reviewed-by: Nikolay Vitkov <[email protected]>
Commit-Queue: Nikolay Vitkov <[email protected]>

Author: OrKoN

front_end/panels/ai_assistance/agents/PatchAgent.test.ts

@@ -2,9 +2,10 @@
 // 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 Platform from '../../../core/platform/platform.js';
 import {mockAidaClient, type MockAidaResponse} from '../../../testing/AiAssistanceHelpers.js';
-import {describeWithEnvironment} from '../../../testing/EnvironmentHelpers.js';
+import {describeWithEnvironment, updateHostConfig} from '../../../testing/EnvironmentHelpers.js';
 import {createFileSystemUISourceCode} from '../../../testing/UISourceCodeHelpers.js';
 import {type ActionResponse, FileUpdateAgent, PatchAgent, type ResponseData, ResponseType} from '../ai_assistance.js';
 
@@ -99,4 +100,29 @@ describeWithEnvironment('PatchAgent', () => {
         action,
         {type: 'action' as ActionResponse['type'], output: '{"success":true}', code: undefined, canceled: false});
   });
+
+  it('builds a request with a user tier', async () => {
+    updateHostConfig({
+      devToolsFreestyler: {
+        userTier: 'PUBLIC',
+      },
+    });
+    const {project} = createFileSystemUISourceCode({
+      url: Platform.DevToolsPath.urlString`file:///path/to/overrides/example.html`,
+      fileSystemPath: Platform.DevToolsPath.urlString`file:///path/to/overrides`,
+      mimeType: 'text/html',
+      content: 'content',
+    });
+    const agent = new PatchAgent({
+      aidaClient: mockAidaClient(),
+      project,
+      fileUpdateAgent: new FileUpdateAgent({
+        aidaClient: mockAidaClient(),
+      })
+    });
+    assert.strictEqual(
+        agent.buildRequest({text: 'test input'}, Host.AidaClient.Role.USER).metadata?.user_tier,
+        3,
+    );
+  });
 });

front_end/panels/ai_assistance/agents/PatchAgent.ts

@@ -3,6 +3,7 @@
 // found in the LICENSE file.
 
 import * as Host from '../../../core/host/host.js';
+import * as Root from '../../../core/root/root.js';
 import type * as Workspace from '../../../models/workspace/workspace.js';
 import {AgentProject} from '../AgentProject.js';
 import {debugLog} from '../debug.js';
@@ -46,7 +47,7 @@ export class PatchAgent extends AiAgent<Workspace.Workspace.Project> {
   readonly clientFeature = Host.AidaClient.ClientFeature.CHROME_PATCH_AGENT;
 
   get userTier(): string|undefined {
-    return 'TESTERS';
+    return Root.Runtime.hostConfig.devToolsFreestyler?.userTier;
   }
 
   get options(): RequestOptions {
@@ -222,7 +223,7 @@ export class FileUpdateAgent extends AiAgent<Workspace.Workspace.Project> {
   readonly clientFeature = Host.AidaClient.ClientFeature.CHROME_PATCH_AGENT;
 
   get userTier(): string|undefined {
-    return 'TESTERS';
+    return Root.Runtime.hostConfig.devToolsFreestyler?.userTier;
   }
 
   get options(): RequestOptions {