File size: 2,396 Bytes
4e23b01 | 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 | import {
type CloudAppender,
createCloudAppender,
IBootstrapService,
IConfigService,
type IDisposable,
IOAuthToolkit,
ITelemetryService,
type Scope,
} from '@moonshot-ai/agent-core-v2';
import { createKimiDeviceId } from '@moonshot-ai/kimi-code-oauth';
const SERVER_TELEMETRY_APP_NAME = 'kimi-code-cli';
const SERVER_TELEMETRY_UI_MODE = 'web';
const TELEMETRY_DISABLE_ENV = 'KIMI_DISABLE_TELEMETRY';
const TELEMETRY_DISABLE_ENV_VALUES = new Set(['1', 'true', 't', 'yes', 'y']);
const TELEMETRY_SHUTDOWN_TIMEOUT_MS = 3_000;
export interface ServerTelemetry {
readonly appender?: CloudAppender;
readonly registration?: IDisposable;
}
function isTelemetryDisabledByEnv(core: Scope): boolean {
const value = core.accessor.get(IBootstrapService).getEnv(TELEMETRY_DISABLE_ENV);
return value !== undefined && TELEMETRY_DISABLE_ENV_VALUES.has(value.trim().toLowerCase());
}
export async function initializeServerTelemetry(
core: Scope,
homeDir: string,
): Promise<ServerTelemetry> {
const service = core.accessor.get(ITelemetryService);
const config = core.accessor.get(IConfigService);
await config.ready;
const enabled = config.get('telemetry') !== false;
if (!enabled || isTelemetryDisabledByEnv(core)) return {};
const auth = core.accessor.get(IOAuthToolkit);
const appender = createCloudAppender(core.accessor, {
deviceId: createKimiDeviceId(homeDir),
appName: SERVER_TELEMETRY_APP_NAME,
uiMode: SERVER_TELEMETRY_UI_MODE,
model: config.get<string>('defaultModel') ?? undefined,
getAccessToken: async () => (await auth.getCachedAccessToken()) ?? null,
});
const registration = service.addAppender(appender);
try {
appender.startPeriodicFlush();
} catch (error) {
registration.dispose();
throw error;
}
return { appender, registration };
}
export async function shutdownServerTelemetry(
telemetry: ServerTelemetry,
deadlineMs = Date.now() + TELEMETRY_SHUTDOWN_TIMEOUT_MS,
): Promise<void> {
telemetry.registration?.dispose();
if (telemetry.appender === undefined) return;
let timer: ReturnType<typeof setTimeout> | undefined;
try {
await Promise.race([
telemetry.appender.shutdown(),
new Promise<void>((resolve) => {
timer = setTimeout(resolve, Math.max(0, deadlineMs - Date.now()));
}),
]);
} finally {
if (timer !== undefined) clearTimeout(timer);
}
}
|