feat: consume chat participant detection provider

src/vs/workbench/contrib/chat/common/chatAgents.ts

@@ -188,8 +188,9 @@ export interface IChatAgentService {
  registerAgentImplementation(id: string, agent: IChatAgentImplementation): IDisposable;
  registerDynamicAgent(data: IChatAgentData, agentImpl: IChatAgentImplementation): IDisposable;
  registerAgentCompletionProvider(id: string, provider: (query: string, token: CancellationToken) => Promise<IChatAgentCompletionItem[]>): IDisposable;
- registerChatParticipantDetectionProvider(handle: number, provider: IChatParticipantDetectionProvider): IDisposable;
  getAgentCompletionItems(id: string, query: string, token: CancellationToken): Promise<IChatAgentCompletionItem[]>;
+ registerChatParticipantDetectionProvider(handle: number, provider: IChatParticipantDetectionProvider): IDisposable;
+ detectAgentOrCommand(request: IChatAgentRequest, history: IChatAgentHistoryEntry[], options: { location: ChatAgentLocation }, token: CancellationToken): Promise<{ agent: IChatAgentData; command?: IChatAgentCommand } | undefined>;
  invokeAgent(agent: string, request: IChatAgentRequest, progress: (part: IChatProgress) => void, history: IChatAgentHistoryEntry[], token: CancellationToken): Promise<IChatAgentResult>;
  getFollowups(id: string, request: IChatAgentRequest, result: IChatAgentResult, history: IChatAgentHistoryEntry[], token: CancellationToken): Promise<IChatFollowup[]>;
  getAgent(id: string): IChatAgentData | undefined;

src/vs/workbench/contrib/chat/common/chatServiceImpl.ts

@@ -15,14 +15,15 @@ import { revive } from 'vs/base/common/marshalling';
 import { StopWatch } from 'vs/base/common/stopwatch';
 import { URI, UriComponents } from 'vs/base/common/uri';
 import { localize } from 'vs/nls';
+import { IConfigurationService } from 'vs/platform/configuration/common/configuration';
 import { IContextKeyService } from 'vs/platform/contextkey/common/contextkey';
 import { IInstantiationService } from 'vs/platform/instantiation/common/instantiation';
 import { ILogService } from 'vs/platform/log/common/log';
 import { Progress } from 'vs/platform/progress/common/progress';
 import { IStorageService, StorageScope, StorageTarget } from 'vs/platform/storage/common/storage';
 import { ITelemetryService } from 'vs/platform/telemetry/common/telemetry';
 import { IWorkspaceContextService } from 'vs/platform/workspace/common/workspace';
-import { ChatAgentLocation, IChatAgent, IChatAgentRequest, IChatAgentResult, IChatAgentService } from 'vs/workbench/contrib/chat/common/chatAgents';
+import { ChatAgentLocation, IChatAgent, IChatAgentCommand, IChatAgentData, IChatAgentRequest, IChatAgentResult, IChatAgentService } from 'vs/workbench/contrib/chat/common/chatAgents';
 import { CONTEXT_VOTE_UP_ENABLED } from 'vs/workbench/contrib/chat/common/chatContextKeys';
 import { ChatModel, ChatRequestModel, ChatRequestRemovalReason, ChatWelcomeMessageModel, IChatModel, IChatRequestModel, IChatRequestVariableData, IChatResponseModel, IExportableChatData, ISerializableChatData, ISerializableChatsData, getHistoryEntriesFromModel, updateRanges } from 'vs/workbench/contrib/chat/common/chatModel';
 import { ChatRequestAgentPart, ChatRequestAgentSubcommandPart, ChatRequestSlashCommandPart, IParsedChatRequest, chatAgentLeader, chatSubcommandLeader, getPromptText } from 'vs/workbench/contrib/chat/common/chatParserTypes';
@@ -111,7 +112,8 @@ export class ChatService extends Disposable implements IChatService {
  @IChatVariablesService private readonly chatVariablesService: IChatVariablesService,
  @IChatAgentService private readonly chatAgentService: IChatAgentService,
  @IWorkbenchAssignmentService workbenchAssignmentService: IWorkbenchAssignmentService,
- @IContextKeyService contextKeyService: IContextKeyService
+ @IContextKeyService contextKeyService: IContextKeyService,
+ @IConfigurationService private readonly configurationService: IConfigurationService
  ) {
  super();
 
@@ -549,35 +551,60 @@ export class ChatService extends Disposable implements IChatService {
  let rawResult: IChatAgentResult | null | undefined;
  let agentOrCommandFollowups: Promise<IChatFollowup[] | undefined> | undefined = undefined;
 
+
  if (agentPart || (defaultAgent && !commandPart)) {
- const agent = (agentPart?.agent ?? defaultAgent)!;
+ const prepareChatAgentRequest = async (agent: IChatAgentData, command?: IChatAgentCommand, chatRequest?: ChatRequestModel) => {
+ const initVariableData: IChatRequestVariableData = { variables: [] };
+ request = chatRequest ?? model.addRequest(parsedRequest, initVariableData, attempt, agent, command, options?.confirmation);
+
+ // Variables may have changed if the agent and slash command changed, so resolve them again even if we already had a chatRequest
+ const variableData = await this.chatVariablesService.resolveVariables(parsedRequest, options?.attachedContext, model, progressCallback, token);
+ model.updateRequest(request, variableData);
+ const promptTextResult = getPromptText(request.message);
+ const updatedVariableData = updateRanges(variableData, promptTextResult.diff); // TODO bit of a hack
+
+ return {
+ sessionId,
+ requestId: request.id,
+ agentId: agent.id,
+ message: promptTextResult.message,
+ command: command?.name,
+ variables: updatedVariableData,
+ enableCommandDetection,
+ attempt,
+ location,
+ locationData: options?.locationData,
+ acceptedConfirmationData: options?.acceptedConfirmationData,
+ rejectedConfirmationData: options?.rejectedConfirmationData,
+ } satisfies IChatAgentRequest;
+ };
+
+ let detectedAgent: IChatAgentData | undefined;
+ let detectedCommand: IChatAgentCommand | undefined;
+ if (this.configurationService.getValue('chat.experimental.detectParticipant.enabled') && !agentPart && !commandPart) {
+ // We have no agent or command to scope history with, pass the full history to the participant detection provider
+ const defaultAgentHistory = getHistoryEntriesFromModel(model, defaultAgent.id);
+
+ // Prepare the request object that we will send to the participant detection provider
+ const chatAgentRequest = await prepareChatAgentRequest(defaultAgent, agentSlashCommandPart?.command);
+
+ const result = await this.chatAgentService.detectAgentOrCommand(chatAgentRequest, defaultAgentHistory, { location }, token);
+ if (result) {
+ // Update the response in the ChatModel to reflect the detected agent and command
+ request.response?.setAgent(result.agent, result.command);
+ detectedAgent = result.agent;
+ detectedCommand = result.command;
+ }
+ }
+
+ const agent = (detectedAgent ?? agentPart?.agent ?? defaultAgent)!;
+ const command = detectedCommand ?? agentSlashCommandPart?.command;
  await this.extensionService.activateByEvent(`onChatParticipant:${agent.id}`);
- const history = getHistoryEntriesFromModel(model, agentPart?.agent.id);
 
- const initVariableData: IChatRequestVariableData = { variables: [] };
- request = model.addRequest(parsedRequest, initVariableData, attempt, agent, agentSlashCommandPart?.command, options?.confirmation);
+ // Recompute history in case the agent or command changed
+ const history = getHistoryEntriesFromModel(model, agent.id);
+ const requestProps = await prepareChatAgentRequest(agent, command, request /* Reuse the request object if we already created it for participant detection */);
  completeResponseCreated();
- const variableData = await this.chatVariablesService.resolveVariables(parsedRequest, options?.attachedContext, model, progressCallback, token);
- model.updateRequest(request, variableData);
-
- const promptTextResult = getPromptText(request.message);
- const updatedVariableData = updateRanges(variableData, promptTextResult.diff); // TODO bit of a hack
-
- const requestProps: IChatAgentRequest = {
- sessionId,
- requestId: request.id,
- agentId: agent.id,
- message: promptTextResult.message,
- command: agentSlashCommandPart?.command.name,
- variables: updatedVariableData,
- enableCommandDetection,
- attempt,
- location,
- locationData: options?.locationData,
- acceptedConfirmationData: options?.acceptedConfirmationData,
- rejectedConfirmationData: options?.rejectedConfirmationData,
- };
-
  const agentResult = await this.chatAgentService.invokeAgent(agent.id, requestProps, progressCallback, history, token);
  rawResult = agentResult;
  agentOrCommandFollowups = this.chatAgentService.getFollowups(agent.id, requestProps, agentResult, history, followupsCancelToken);