1
>
/*---------------------------------------------------------------------------------------------
copilotAgentSession.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 type { CopilotSession, CurrentToolMetadata, ExitPlanModeRequest, McpServersLoadedServer, MessageOptions, PermissionAllowAllMode, PermissionAutoApproval, PermissionRequestResult, SessionConfig, Tool, ToolResultObject, McpServerStatus as SdkMcpServerStatus } from '@github/copilot-sdk';
7
>
import { DeferredPromise, Sequencer } from '../../../../base/common/async.js';
8
>
import { encodeBase64, VSBuffer } from '../../../../base/common/buffer.js';
9
>
import { Emitter } from '../../../../base/common/event.js';
10
>
import { CancellationError, getErrorMessage } from '../../../../base/common/errors.js';
11
>
import { escapeMarkdownSyntaxTokens } from '../../../../base/common/htmlContent.js';
12
>
import { Disposable, IReference, toDisposable } from '../../../../base/common/lifecycle.js';
13
>
import { Schemas } from '../../../../base/common/network.js';
14
>
import { isAuthorizationProtectedResourceMetadata } from '../../../../base/common/oauth.js';
15
>
import { safeStringify } from '../../../../base/common/objects.js';
16
>
import { isAbsolute, join } from '../../../../base/common/path.js';
17
>
import { extUriBiasedIgnorePathCase, normalizePath } from '../../../../base/common/resources.js';
18
>
import { StopWatch } from '../../../../base/common/stopwatch.js';
19
>
import { splitLinesIncludeSeparators } from '../../../../base/common/strings.js';
20
>
import { hasKey, isDefined, isObject, isString, type Mutable } from '../../../../base/common/types.js';
21
>
import { URI } from '../../../../base/common/uri.js';
22
>
import { generateUuid } from '../../../../base/common/uuid.js';
23
>
import { localize } from '../../../../nls.js';
24
>
import { INativeEnvironmentService } from '../../../environment/common/environment.js';
25
>
import { IFileService } from '../../../files/common/files.js';
26
>
import { IInstantiationService } from '../../../instantiation/common/instantiation.js';
27
>
import { ILogService, LogLevel } from '../../../log/common/log.js';
28
>
import { ITelemetryService } from '../../../telemetry/common/telemetry.js';
29
>
import { CopilotCliConfigKey, applyModelFamilyAlias, copilotCliConfigSchema } from '../../common/copilotCliConfig.js';
30
>
import type { ChatInputRequestWithPlanReview, IAgentHostPlanReviewAction } from '../../common/agentHostPlanReview.js';
31
>
import { gitHubMcpServerUrl } from '../../common/githubEndpoints.js';
32
>
import { AgentHostSandboxConfigKey, sandboxConfigSchema } from '../../common/sandboxConfigSchema.js';
33
>
import { AgentHostGlobalAutoApproveEnabledConfigKey, AgentHostAutoReplyEnabledConfigKey, AgentHostDisableRepoInfoTelemetryConfigKey, platformRootSchema, platformSessionSchema } from '../../common/agentHostSchema.js';
34
>
import { AgentSession, AgentSignal, AuthenticateParams, IMcpNotification, IRestoredSubagentSession, subagentChatTitle } from '../../common/agentService.js';
35
>
import { META_DIFF_BASE_BRANCH } from '../../common/agentHostGitService.js';
36
>
import { stripRedundantCdPrefix } from '../../common/commandLineHelpers.js';
37
>
import { readToolCallMeta, toToolCallMeta, type IToolCallMeta, type IToolCallUiMeta, type IToolSearchCandidate } from '../../common/meta/agentToolCallMeta.js';
38
>
import { OtelData, type OtelAttributeValue } from '../../common/otlp/otlpLogEmitter.js';
39
>
import { SessionConfigKey } from '../../common/sessionConfigKeys.js';
40
>
import { resolveCopilotConfigSlashCommandOnSend } from '../../common/copilotConfigSlashCommands.js';
41
>
import { isAgentFeedbackAnnotationsAttachment, renderAgentFeedbackAnnotationsAttachment } from '../../common/meta/agentFeedbackAttachments.js';
42
>
import { ISessionDatabase, ISessionDataService, SESSION_ATTACHMENTS_DIRNAME } from '../../common/sessionDataService.js';
43
>
import { MessageAttachmentKind, ToolCallContributorKind, type FileEdit, type MessageAttachment } from '../../common/state/protocol/state.js';
44
>
import { ActionType, isChatAction, type ChatAction, type SessionAction } from '../../common/state/sessionActions.js';
45
>
import { MessageKind, ResponsePartKind, ChatInputAnswerState, ChatInputAnswerValueKind, ChatInputQuestionKind, ChatInputResponseKind, ToolCallConfirmationReason, ToolCallRiskAssessmentKind, ToolCallRiskAssessmentStatus, ToolCallStatus, ToolResultContentType, buildSubagentSessionUri, getToolSubagentContent, isDefaultChatUri, isSubagentSession, withSessionPromptCacheState, type PendingMessage, type ChatInputAnswer, type ChatInputOption, type ChatInputQuestion, type ChatInputRequest, type ToolCallResult, type ToolResultContent, type ToolResultTerminalContent, type Turn, type UsageInfo, type UsageInfoMeta } from '../../common/state/sessionState.js';
46
>
import { IAgentConfigurationService } from '../agentConfigurationService.js';
47
>
import type { IExitPlanModeResponse } from './copilotAgent.js';
48
>
import { CopilotSessionWrapper } from './copilotSessionWrapper.js';
49
>
import { clientToolNamesFromSnapshot, type CopilotSessionLaunchPlan, type IActiveClientSnapshot, type ICopilotSessionLauncher, type ICopilotSessionRuntime } from './copilotSessionLauncher.js';
50
>
import { agentHostModelSupportsToolSearch, CLIENT_TOOL_SEARCH_REFERENCE_NAME, NON_DEFERRED_CLIENT_TOOL_NAMES, RUNTIME_TOOL_SEARCH_TOOL_NAME } from './toolSearchDeferral.js';
51
>
import { ActiveClientToolSet } from '../activeClientState.js';
52
>
import { AgentHostTelemetryReporter } from '../agentHostTelemetryReporter.js';
53
>
import { AgentHostRepoInfoTelemetry } from '../agentHostRepoInfoTelemetry.js';
54
>
import { PendingRequestRegistry } from '../../common/pendingRequestRegistry.js';
55
>
import { buildCopilotSystemNotification } from './copilotSystemNotification.js';
56
>
import { parseLeadingSlashCommand } from '../../common/agentHostSlashCommand.js';
57
>
import type { IUnsandboxedCommandConfirmationRequest, ShellManager } from './copilotShellTools.js';
58
>
import { NonPtyShellTerminalStreams } from './copilotNonPtyShellTerminals.js';
59
>
import { buildSandboxConfigForSdk, type ISdkSandboxConfig } from './sandboxConfigForSdk.js';
60
>
import type { IAgentServerToolHost } from '../../common/agentServerTools.js';
61
>
import { getEditFilePaths, getInvocationMessage, getPastTenseMessage, getPermissionDisplay, getShellIntention, getShellLanguage, getSubagentMetadata, getTaskCompleteMarkdown, getToolDisplayName, getToolInputString, getToolKind, isAgentCoordinationTool, isEditTool, isHiddenTool, isShellTool, isTaskCompleteTool, synthesizeSkillToolCall, tryStringify, type ITypedPermissionRequest } from './copilotToolDisplay.js';
62
>
import { FileEditTracker } from '../shared/fileEditTracker.js';
63
>
import { ICopilotApiService, type IRestrictedTelemetryContext } from '../shared/copilotApiService.js';
64
>
import type { IAgentHostRestrictedTelemetryContext } from '../agentHostRestrictedTelemetry.js';
65
>
import { stripProxyErrorMarker, tryBuildChatErrorMeta, tryBuildChatErrorMetaFromFields } from '../shared/forwardedChatError.js';
66
>
import { getEffectiveMcpServerCustomizations, McpCustomizationController, type ISdkMcpServer } from '../shared/mcpCustomizationController.js';
67
>
import { appendSdkToolResultContent, mapSessionEvents } from './mapSessionEvents.js';
68
>
import { addSimpleAttachmentDisplayKindToMimeType } from './copilotAttachmentUtils.js';
69
>
import { buildPendingEditContentUri } from './pendingEditContentStore.js';
70
>
import { AgentHostStateManager, IAgentHostStateManager } from '../agentHostStateManager.js';
71
>
import { McpAuthRequiredReason, McpServerStatus, type McpAuthRequirement, type McpServerState } from '../../common/state/protocol/channels-session/state.js';
72
>
import type { ProtectedResourceMetadata } from '../../common/state/protocol/common/state.js';
73
>
import { CopilotSlashCommandProvider } from './copilotSlashCommandProvider.js';
74
>
75
>
/**
76
>
* The full set of agent modes the Copilot SDK accepts. AHP now exposes the
77
>
* same three modes (`interactive` / `plan` / `autopilot`) on its `mode` axis,
78
>
* so the Copilot agent maps between the two views directly in
79
>
* {@link CopilotAgentSession.send} and the `session.mode_changed` listener.
80
>
*/
81
>
export type CopilotSdkMode = 'interactive' | 'plan' | 'autopilot';
82
>
type CopilotSdkAttachment = Required<MessageOptions>['attachments'][number];
83
>
type CopilotCommandInvocationResult = Awaited<ReturnType<CopilotSession['rpc']['commands']['invoke']>>;
84
>
type RuntimeSlashCommandInfo = Awaited<ReturnType<CopilotSession['rpc']['commands']['list']>>['commands'][number];
85
>
type McpAuthHandler = NonNullable<SessionConfig['onMcpAuthRequest']>;
86
>
type McpAuthRequest = Parameters<McpAuthHandler>[0];
87
>
type McpAuthResult = Awaited<ReturnType<McpAuthHandler>>;
88
>
89
>
interface IPendingMcpAuthRequest {
90
>
readonly serverName: string;
91
>
readonly resource: ProtectedResourceMetadata;
92
>
readonly requiredScopes: readonly string[];
93
>
readonly toolCalls: IMcpAuthToolCall[];
94
>
readonly deferred: DeferredPromise<McpAuthResult | null | undefined>;
95
>
}
96
>
97
>
interface IMcpAuthToolCall {
98
>
readonly turnId: string;
99
>
readonly toolCallId: string;
100
>
readonly parentToolCallId: string | undefined;
101
>
}
102
>
103
>
const COPILOT_HOME_DIRECTORY = '.copilot';
104
>
const SESSION_STATE_DIRECTORY = join(COPILOT_HOME_DIRECTORY, 'session-state');
105
>
const EMPTY_TOOL_RESULT_TEXT = '<empty />';
106
>
107
function normalizeMcpServerUrl(value: string): string | undefined {
108
if (!URL.canParse(value)) {