Set up maxRequests setting to be controlled by separate experiments f…

…or free vs pro users (#239170)

src/vs/workbench/contrib/chat/browser/chatSetup.ts

@@ -68,6 +68,7 @@ import { IConfigurationRegistry, Extensions as ConfigurationExtensions, IConfigu
 import { IQuickInputService } from '../../../../platform/quickinput/common/quickInput.js';
 import { ILifecycleService } from '../../../services/lifecycle/common/lifecycle.js';
 import { equalsIgnoreCase } from '../../../../base/common/strings.js';
+import { IWorkbenchAssignmentService } from '../../../services/assignment/common/assignmentService.js';
 
 const defaultChat = {
 	extensionId: product.defaultChatAgent?.extensionId ?? '',
@@ -120,6 +121,7 @@ export class ChatSetupContribution extends Disposable implements IWorkbenchContr
 		@IWorkbenchEnvironmentService private readonly environmentService: IWorkbenchEnvironmentService,
 		@ICommandService private readonly commandService: ICommandService,
 		@ITelemetryService private readonly telemetryService: ITelemetryService,
+		@IWorkbenchAssignmentService private readonly experimentService: IWorkbenchAssignmentService,
 	) {
 		super();
 
@@ -324,23 +326,29 @@ export class ChatSetupContribution extends Disposable implements IWorkbenchContr
 
 		let lastNode: IConfigurationNode | undefined;
 		const registerSetting = () => {
-			const node: IConfigurationNode = {
-				id: 'chatSidebar',
-				title: localize('interactiveSessionConfigurationTitle', "Chat"),
-				type: 'object',
-				properties: {
-					'chat.agent.maxRequests': {
-						type: 'number',
-						description: localize('chat.agent.maxRequests', "The maximum number of requests to allow Copilot Edits to use in agent mode."),
-						default: context.state.entitlement === ChatEntitlement.Limited ? 5 : 15,
-						tags: ['experimental', 'onExp']
-					},
-				}
-			};
-			configurationRegistry.updateConfigurations({ remove: lastNode ? [lastNode] : [], add: [node] });
-			lastNode = node;
+			const treatmentId = context.state.entitlement === ChatEntitlement.Limited ?
+				'chatAgentMaxRequestsFree' :
+				'chatAgentMaxRequestsPro';
+			this.experimentService.getTreatment<number>(treatmentId).then(value => {
+				const defaultValue = value ?? (context.state.entitlement === ChatEntitlement.Limited ? 5 : 15);
+				const node: IConfigurationNode = {
+					id: 'chatSidebar',
+					title: localize('interactiveSessionConfigurationTitle', "Chat"),
+					type: 'object',
+					properties: {
+						'chat.agent.maxRequests': {
+							type: 'number',
+							description: localize('chat.agent.maxRequests', "The maximum number of requests to allow Copilot Edits to use in agent mode."),
+							default: defaultValue,
+							tags: ['experimental']
+						},
+					}
+				};
+				configurationRegistry.updateConfigurations({ remove: lastNode ? [lastNode] : [], add: [node] });
+				lastNode = node;
+			});
 		};
-		this._register(Event.runAndSubscribe(context.onDidChange, () => registerSetting()));
+		this._register(Event.runAndSubscribe(Event.debounce(context.onDidChange, () => { }, 1000), () => registerSetting()));
 	}
 }