1
>
/*---------------------------------------------------------------------------------------------
agentService.ts
2
>
* Copyright (c) Microsoft Corporation. All rights reserved.
3
>
* Licensed under the MIT License. See License.txt in the project root for license information.
4
>
*--------------------------------------------------------------------------------------------*/
5
>
6
>
import { open, unlink, type FileHandle } from 'fs/promises';
7
>
import { decodeBase64, VSBuffer } from '../../../base/common/buffer.js';
8
>
import { DeferredPromise, disposableTimeout, ResourceQueue } from '../../../base/common/async.js';
9
>
import { toErrorMessage } from '../../../base/common/errorMessage.js';
10
>
import { Emitter } from '../../../base/common/event.js';
11
>
import { Disposable, DisposableMap, DisposableResourceMap, DisposableStore, IDisposable, MutableDisposable } from '../../../base/common/lifecycle.js';
12
>
import { ResourceMap } from '../../../base/common/map.js';
13
>
import { getExtensionForMimeType, getMediaMime } from '../../../base/common/mime.js';
14
>
import { Schemas } from '../../../base/common/network.js';
15
>
import { IObservable, observableValue } from '../../../base/common/observable.js';
16
>
import { dirname as resourcesDirname, extname as resourcesExtname, extUriBiasedIgnorePathCase, isEqual, isEqualOrParent, joinPath } from '../../../base/common/resources.js';
17
>
import { URI } from '../../../base/common/uri.js';
18
>
import { generateUuid } from '../../../base/common/uuid.js';
19
>
import { hasKey } from '../../../base/common/types.js';
20
>
import { localize } from '../../../nls.js';
21
>
import { FileChangeType, FileOperationResult, IFileChange, IFileService, toFileOperationResult, type FileChangesEvent } from '../../files/common/files.js';
22
>
import { InstantiationService } from '../../instantiation/common/instantiationService.js';
23
>
import { ServiceCollection } from '../../instantiation/common/serviceCollection.js';
24
>
import { ILogService } from '../../log/common/log.js';
25
>
import { AgentProvider, AgentSession, AgentSignal, AgentHostSessionReleaseGraceMsEnvVar, IAgent, IAgentChatDataChange, IAgentCreateChatOptions, IAgentCreateChatResult, IAgentCreateChatSideChatSelection, IAgentCreateChatSideChatSource, IAgentCreateSessionConfig, IAgentCreateSessionResult, IAgentHostAuthTokenRequest, IAgentHostManagedSettingsDiagnostics, IAgentHostNetworkDiagnosticsInfo, IAgentHostNetworkEndpoint, IAgentHostNetworkFetchResult, IAgentMaterializeSessionEvent, IAgentModelInfo, IAgentResolveSessionConfigParams, IAgentService, IAgentSessionConfigCompletionsParams, IAgentSessionMetadata, IAgentSpawnChatEvent, AuthenticateParams, AuthenticateResult, IMcpNotification, IRestoredSubagentSession, SubagentChatSignal } from '../common/agentService.js';
26
>
import { ISessionDataService, SESSION_ATTACHMENTS_DIRNAME } from '../common/sessionDataService.js';
27
>
import { SessionConfigKey } from '../common/sessionConfigKeys.js';
28
>
import type { IAgentCustomizationSettingsRegistration } from '../common/agentCustomizationSettings.js';
29
>
import { parseChangesetUri } from '../common/changesetUri.js';
30
>
import { ActionType, ActionEnvelope, AuthRequiredReason, INotification, type ChatAction, type IRootConfigChangedAction, type SessionAction, type TerminalAction, type ClientAnnotationsAction, type ClientChangesetAction } from '../common/state/sessionActions.js';
31
>
import type { CompletionsParams, CompletionsResult, CreateTerminalParams, ResolveSessionConfigResult, SessionConfigCompletionsResult, SessionConfigPropertySchema } from '../common/state/protocol/commands.js';
32
>
import type { InvokeChangesetOperationParams, InvokeChangesetOperationResult } from '../common/state/protocol/channels-changeset/commands.js';
33
>
import { AhpErrorCodes, AHP_SESSION_NOT_FOUND, ContentEncoding, JSON_RPC_INTERNAL_ERROR, ProtocolError, ResourceChangeType, ResourceType, ResourceWriteMode, type CreateResourceWatchParams, type CreateResourceWatchResult, type DirectoryEntry, type ResourceCopyParams, type ResourceCopyResult, type ResourceDeleteParams, type ResourceDeleteResult, type ResourceListResult, type ResourceMkdirParams, type ResourceMkdirResult, type ResourceMoveParams, type ResourceMoveResult, type ResourceReadResult, type ResourceResolveParams, type ResourceResolveResult, type ResourceWatchState, type ResourceWriteParams, type ResourceWriteResult, type IStateSnapshot } from '../common/state/sessionProtocol.js';
34
>
import { ChangesSummary, ChatInteractivity, ChatOriginKind, MessageAttachmentKind, type ChatOrigin, type Message, type MessageAttachment, type MessageResourceAttachment } from '../common/state/protocol/state.js';
35
>
import type { ChatPendingMessageSetAction, ChatTurnStartedAction } from '../common/state/protocol/actions.js';
36
>
import { ISessionGitHubState, ISessionGitState, MessageKind, ResponsePartKind, SESSION_META_GITHUB_KEY, SESSION_META_GIT_KEY, readSessionSpawnDepth, withSessionSpawnDepth, SessionStatus, ToolCallStatus, ToolResultContentType, AH_META_WORKSPACELESS_DB_KEY, AH_META_IS_ARCHIVED_DB_KEY, AH_META_IS_DONE_DB_KEY, buildChatUri, buildDefaultChatUri, buildResourceWatchChannelUri, buildSubagentChatUri, buildSubagentSessionUriPrefix, hostBuildInfoFromProduct, isAhpChatChannel, isDefaultChatUri, isSubagentChatUri, isSubagentSession, parseDefaultChatUri, parseRequiredSessionUriFromChatUri, parseResourceWatchChannelUri, parseSubagentSessionUri, readSessionGitState, readSessionWorkspaceless, withSessionGitHubState, withSessionGitState, withSessionWorkspaceless, type SessionConfigState, type SessionSummary, type ToolResultSubagentContent, type Turn } from '../common/state/sessionState.js';
37
>
import { readToolCallMeta } from '../common/meta/agentToolCallMeta.js';
38
>
import { IProductService } from '../../product/common/productService.js';
39
>
import { buildBoundedSideChatSourceContext, getSideChatPartialResponse } from './agentPeerChats.js';
40
>
import { AgentConfigurationService, IAgentConfigurationService } from './agentConfigurationService.js';
41
>
import { AgentHostTerminalManager, IAgentHostTerminalManager } from './agentHostTerminalManager.js';
42
>
import { ISessionDbUriFields, parseSessionDbUri } from './shared/fileEditTracker.js';
43
>
import { IGitBlobUriFields, parseGitBlobUri } from './gitDiffContent.js';
44
>
import { AgentHostStateManager, IAgentHostStateManager } from './agentHostStateManager.js';
45
>
import { IAgentHostGitService } from '../common/agentHostGitService.js';
46
>
import { AgentSideEffects } from './agentSideEffects.js';
47
>
import { AgentHostLocalTurns } from './agentHostLocalTurns.js';
48
>
import { AgentServerToolHost } from './shared/agentServerToolHost.js';
49
>
import { buildServerToolGroups } from './shared/serverToolGroups.js';
50
>
import { type IChatContextSnapshot, type ISessionServerToolAccessor } from './shared/sessionServerTools.js';
51
>
52
>
import { WorktreeIsolation, WORKTREE_META_REPOSITORY_ROOT, worktreeProjectFromRepositoryRoot } from './shared/worktreeIsolation.js';
53
>
import { AgentHostChangesetService } from './agentHostChangesetService.js';
54
>
import { AgentHostFileMonitorService, IAgentHostFileMonitorService } from './agentHostFileMonitorService.js';
55
>
import { IAgentHostCheckpointService, NULL_CHECKPOINT_SERVICE } from '../common/agentHostCheckpointService.js';
56
>
import { IAgentHostReviewService } from '../common/agentHostReviewService.js';
57
>
import { AgentHostChangesetCoordinator } from './agentHostChangesetCoordinator.js';
58
>
import { AgentHostCompletions, IAgentHostCompletions } from './agentHostCompletions.js';
59
>
import { AgentHostFileCompletionProvider } from './agentHostFileCompletionProvider.js';
60
>
import { AgentHostRenameCompletionProvider } from './agentHostRenameCommand.js';
61
>
import { AgentHostSkillCompletionProvider } from './agentHostSkillCompletionProvider.js';
62
>
import { AgentHostWorkspaceFiles } from './agentHostWorkspaceFiles.js';
63
>
import { CopilotApiService, ICopilotApiService } from './shared/copilotApiService.js';
64
>
import { INetworkDiagnosticsService } from './networkDiagnosticsService.js';
65
>
import { parseMcpChannelUri } from './shared/mcpCustomizationController.js';
66
>
import { toAgentClientUri } from '../common/agentClientUri.js';
67
>
import { AgentHostChangesetOperationService } from './agentHostChangesetOperationService.js';
68
>
import { AgentHostGitStateService } from './agentHostGitStateService.js';
69
>
import { AgentHostGitHubEndpointService, IAgentHostGitHubEndpointService } from './agentHostGitHubEndpointService.js';
70
>
import { ITelemetryService } from '../../telemetry/common/telemetry.js';
71
>
import { NullTelemetryService } from '../../telemetry/common/telemetryUtils.js';
72
>
import { AgentHostAuthenticationService } from './agentHostAuthenticationService.js';
73
>
import { updateAgentHostTelemetryLevelFromConfig } from './agentHostTelemetryService.js';
74
>
import { AgentHostOctoKitService, IAgentHostOctoKitService } from './shared/agentHostOctoKitService.js';
75
>
import { IAgentHostChangesetService, CHANGESET_DB_METADATA_KEYS, META_CHANGES_SUMMARY } from '../common/agentHostChangesetService.js';
76
>
import { IAgentHostChangesetSubscriptionService } from '../common/agentHostChangesetSubscriptionService.js';
77
>
import { AgentHostChangesetSubscriptionService } from './agentHostChangesetSubscriptionService.js';
78
>
import { GIT_DB_METADATA_KEYS, IAgentHostGitStateService, META_GIT_STATE, META_GITHUB_STATE } from '../common/agentHostGitStateService.js';
79
>
import { IAgentHostChangesetOperationService } from '../common/agentHostChangesetOperationService.js';
80
>
import { AgentHostCommitOperationContribution } from './agentHostCommitOperationProvider.js';
81
>
import { AgentHostDiscardChangesOperationContribution } from './agentHostDiscardChangesOperationProvider.js';
82
>
import { AgentHostPullRequestOperationContribution } from './agentHostPullRequestOperationProvider.js';
83
>
import { AgentHostSyncOperationContribution } from './agentHostSyncOperationProvider.js';
84
>
import { AgentHostReviewService } from './agentHostReviewService.js';
85
>
86
>
/**
87
>
* Grace period before an empty, unsubscribed session is garbage-collected
88
>
* via {@link AgentService._runSessionGc}. Gives a disconnected client time
89
>
* to reconnect (or a workspace switch to settle) before we tear down the
90
>
* provider-side session, worktree, and on-disk state.
91
>
*/
92
>
const SESSION_GC_GRACE_MS = 30_000;
93
>
94
>
const HOST_OWNED_SESSION_CONFIG_KEYS = [
95
>
SessionConfigKey.Isolation,
96
>
SessionConfigKey.Branch,
97
>
SessionConfigKey.WorktreeBranchPrefix,
98
>
SessionConfigKey.WorktreeIncludeFiles,
99
>
] as const;
100
>
101
function omitHostOwnedSessionConfig<T>(config: Record<string, T>): Record<string, T> {
102
const result = { ...config };