Implement copilot token flag for agent mode (microsoft#257892)

* Implement copilot token flag for agent mode

* include account tag

Author: roblourens

src/vs/base/common/policy.ts

@@ -8,7 +8,8 @@ export type PolicyName = string;
 export enum PolicyTag {
 	Account = 'ACCOUNT',
 	MCP = 'MCP',
-	Preview = 'PREVIEW'
+	Preview = 'PREVIEW',
+	Agent = 'AGENT',
 }
 
 export interface IPolicy {

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

@@ -385,6 +385,7 @@ configurationRegistry.registerConfiguration({
 			policy: {
 				name: 'ChatAgentMode',
 				minimumVersion: '1.99',
+				tags: [PolicyTag.Account, PolicyTag.Agent]
 			}
 		},
 		[ChatConfiguration.EnableMath]: {

src/vs/workbench/services/accounts/common/defaultAccount.ts

@@ -49,6 +49,7 @@ export interface IDefaultAccount {
 		readonly completions: number;
 	};
 	readonly limited_user_reset_date?: string;
+	readonly chat_agent_enabled?: boolean;
 }
 
 interface IChatEntitlementsResponse {
@@ -250,6 +251,7 @@ export class DefaultAccountManagementContribution extends Disposable implements
 				return {
 					// Editor preview features are disabled if the flag is present and set to 0
 					chat_preview_features_enabled: tokenMap.get('editor_preview_features') !== '0',
+					chat_agent_enabled: tokenMap.get('chat_agent_enabled') !== '0',
 					// MCP is disabled if the flag is present and set to 0
 					mcp: tokenMap.get('mcp') !== '0',
 				};

src/vs/workbench/services/policies/common/accountPolicyService.ts

@@ -13,12 +13,14 @@ import { IDefaultAccountService } from '../../accounts/common/defaultAccount.js'
 interface IAccountPolicy {
 	readonly chatPreviewFeaturesEnabled: boolean;
 	readonly mcpEnabled: boolean;
+	readonly chatAgentEnabled: boolean;
 }
 
 export class AccountPolicyService extends AbstractPolicyService implements IPolicyService {
 	private accountPolicy: IAccountPolicy = {
 		chatPreviewFeaturesEnabled: true,
-		mcpEnabled: true
+		mcpEnabled: true,
+		chatAgentEnabled: true
 	};
 	constructor(
 		@ILogService private readonly logService: ILogService,
@@ -30,12 +32,14 @@ export class AccountPolicyService extends AbstractPolicyService implements IPoli
 			.then(account => {
 				this._update({
 					chatPreviewFeaturesEnabled: account?.chat_preview_features_enabled ?? true,
-					mcpEnabled: account?.mcp ?? true
+					mcpEnabled: account?.mcp ?? true,
+					chatAgentEnabled: account?.chat_agent_enabled ?? true
 				});
 				this._register(this.defaultAccountService.onDidChangeDefaultAccount(
 					account => this._update({
 						chatPreviewFeaturesEnabled: account?.chat_preview_features_enabled ?? true,
-						mcpEnabled: account?.mcp ?? true
+						mcpEnabled: account?.mcp ?? true,
+						chatAgentEnabled: account?.chat_agent_enabled ?? true
 					})
 				));
 			});
@@ -84,6 +88,9 @@ export class AccountPolicyService extends AbstractPolicyService implements IPoli
 			else if (hasAllTags(policy, [PolicyTag.Account, PolicyTag.MCP])) {
 				updateIfNeeded(key, policy, this.accountPolicy?.mcpEnabled);
 			}
+			else if (hasAllTags(policy, [PolicyTag.Account, PolicyTag.Agent])) {
+				updateIfNeeded(key, policy, this.accountPolicy?.chatAgentEnabled);
+			}
 		}
 
 		if (updated.length) {