File size: 3,302 Bytes
68d7816 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 | import { createDecorator } from '#/_base/di/instantiation';
import type { ITaskHandle } from '#/app/task/task';
import type {
AgentTask,
AgentTaskInfo,
AgentTaskInfoBase,
AgentTaskStatus,
} from './types';
export { AgentTaskPersistence } from './persist';
export type {
AgentTask,
AgentTaskInfo,
AgentTaskInfoBase,
AgentTaskKind,
AgentTaskStatus,
} from './types';
export interface AgentTaskLoadOptions {
readonly replace?: boolean;
}
export interface AgentTaskOutputSnapshot {
readonly outputPath?: string;
readonly outputSizeBytes: number;
readonly previewBytes: number;
readonly truncated: boolean;
readonly fullOutputAvailable: boolean;
readonly preview: string;
}
export interface RegisterAgentTaskOptions {
readonly detached?: boolean;
readonly timeoutMs?: number;
readonly detachTimeoutMs?: number;
readonly autoBackgroundOnTimeout?: boolean;
readonly signal?: AbortSignal;
}
export type ForegroundTaskReleaseReason = 'detached' | 'timeout_detached' | 'terminal';
export interface AgentTaskTrackOptions {
readonly idPrefix?: string;
readonly description: string;
readonly detached?: boolean;
readonly timeoutMs?: number;
readonly detachTimeoutMs?: number;
readonly signal?: AbortSignal;
readonly forceStop?: () => Promise<void>;
readonly onDetach?: () => void;
readonly toInfo: (base: AgentTaskInfoBase) => AgentTaskInfo;
}
export interface IAgentTaskEntry {
readonly taskId: string;
readonly onDidDetach: Promise<ForegroundTaskReleaseReason>;
}
export interface AgentTaskNotificationContext {
readonly agentId: string;
readonly notificationType: string;
readonly title: string;
readonly body: string;
readonly severity: 'info' | 'warning';
readonly sourceKind: string;
readonly sourceId: string;
}
export interface AgentTaskWaitDelivery {
readonly taskId: string;
readonly status: AgentTaskStatus;
}
export interface IAgentTaskService {
readonly _serviceBrand: undefined;
track(handle: ITaskHandle, options: AgentTaskTrackOptions): IAgentTaskEntry;
registerTask(task: AgentTask, options?: RegisterAgentTaskOptions): string;
getTask(taskId: string): AgentTaskInfo | undefined;
list(activeOnly?: boolean, limit?: number): readonly AgentTaskInfo[];
persistOutput(taskId: string): void;
getOutputSnapshot(
taskId: string,
maxPreviewBytes: number,
): Promise<AgentTaskOutputSnapshot>;
readOutput(taskId: string, tail?: number): Promise<string>;
suppressTerminalNotification(taskId: string): Promise<void>;
suppressAllTerminalNotifications(): Promise<void>;
markTasksDeliveredViaWait(tasks: readonly AgentTaskWaitDelivery[]): void;
detach(taskId: string): AgentTaskInfo | undefined;
stop(taskId: string, reason?: string): Promise<AgentTaskInfo | undefined>;
stopByUser(taskId: string): Promise<AgentTaskInfo | undefined>;
stopAll(reason?: string): Promise<readonly AgentTaskInfo[]>;
stopAllOnExit(reason: string): Promise<readonly AgentTaskInfo[]>;
wait(
taskId: string,
timeoutMs?: number,
signal?: AbortSignal,
): Promise<AgentTaskInfo | undefined>;
waitForForegroundRelease(
taskId: string,
): Promise<ForegroundTaskReleaseReason | undefined>;
}
export const IAgentTaskService =
createDecorator<IAgentTaskService>('agentTaskService');
|