diff --git a/src/vs/platform/agentHost/node/agentService.ts b/src/vs/platform/agentHost/node/agentService.ts index 1408cd840549d..eefdf2a2cac0c 100644 --- a/src/vs/platform/agentHost/node/agentService.ts +++ b/src/vs/platform/agentHost/node/agentService.ts @@ -4406,7 +4406,7 @@ export class AgentService extends Disposable implements IAgentService { const sessionArchived = ((sessionState?.status ?? 0) & SessionStatus.IsArchived) === SessionStatus.IsArchived; const turn = chatState?.turns.at(-1); const errorPart = getErrorResponsePart(turn); - const provider = this._findProviderForSession(sessionChannel); + const provider = this._providerService.getProviderForSession(sessionChannel); if (chatState?.activeTurn) { this._stateManager.rejectClientAction(channel, action, origin, 'Cannot resume while a turn is active.'); return; diff --git a/src/vs/platform/agentHost/test/node/agentService.test.ts b/src/vs/platform/agentHost/test/node/agentService.test.ts index 13a6169ed1ec6..10df6315329b8 100644 --- a/src/vs/platform/agentHost/test/node/agentService.test.ts +++ b/src/vs/platform/agentHost/test/node/agentService.test.ts @@ -554,7 +554,7 @@ suite('AgentService (node dispatcher)', () => { suite('failed turn resume', () => { async function createErroredTurn(): Promise<{ session: URI; chat: string }> { - service.registerProvider(copilotAgent); + registerTestAgentProvider(service, copilotAgent); const session = await service.createSession({ provider: 'copilot' }); const chat = buildDefaultChatUri(session.toString()); const stateManager = getStateManager(service);