@botiverse/kimi-code-sdk 0.9.4 → 0.16.0

package/NOTICE.md

@@ -1,5 +1,5 @@
 # NOTICE
 
 @botiverse/kimi-code-sdk is a repackage of the built **@moonshot-ai/kimi-code-sdk** node-sdk
-from [MoonshotAI/kimi-code](https://github.com/MoonshotAI/kimi-code) at `@moonshot-ai/kimi-code@0.15.0`, distributed under upstream's MIT License (see LICENSE). Sibling packages are bundled into `dist`.
+from [MoonshotAI/kimi-code](https://github.com/MoonshotAI/kimi-code) at `@moonshot-ai/kimi-code@0.16.0`, distributed under upstream's MIT License (see LICENSE). Sibling packages are bundled into `dist`.
 Mirror + provenance: https://github.com/botiverse/kimi-code-sdk

package/README.md

@@ -1,6 +1,6 @@
 # @botiverse/kimi-code-sdk
 
-Built repackage of `@moonshot-ai/kimi-code-sdk` (kimi-code @moonshot-ai/kimi-code@0.15.0) for Slock/Botiverse.
+Built repackage of `@moonshot-ai/kimi-code-sdk` (kimi-code @moonshot-ai/kimi-code@0.16.0) for Slock/Botiverse.
 Read-only mirror + release notes: https://github.com/botiverse/kimi-code-sdk
 
 > Not affiliated with Moonshot AI. MIT (see LICENSE).

package/dist/index.d.mts

@@ -78,6 +78,7 @@ declare class Agent {
     readonly log: Logger;
     readonly telemetry: TelemetryClient;
     readonly experimentalFlags: ExperimentalFlagResolver;
+    readonly llmRequestLogger: LlmRequestLogger;
     readonly blobStore: BlobStore | undefined;
     readonly records: AgentRecords;
     readonly fullCompaction: FullCompaction;
@@ -96,15 +97,12 @@ declare class Agent {
     readonly cron: CronManager | null;
     readonly goal: GoalMode;
     readonly replayBuilder: ReplayBuilder;
-    private lastLlmConfigLogSignature?;
     constructor(options: AgentOptions);
     setKaos(kaos: Kaos): void;
     get generate(): typeof generate;
     get llm(): KosongLLM;
-    private logLlmRequest;
-    private logLlmConfigIfChanged;
     useProfile(profile: ResolvedAgentProfile, context?: PreparedSystemPromptContext): void;
-    resume(): Promise<{
+    resume(options?: AgentRecordsReplayOptions): Promise<{
         warning?: string;
     }>;
     get rpcMethods(): PromisableMethods<AgentAPI>;
@@ -215,6 +213,7 @@ declare interface AgentOptions {
     readonly telemetry?: TelemetryClient | undefined;
     readonly pluginSessionStarts?: readonly EnabledPluginSessionStart[];
     readonly experimentalFlags?: ExperimentalFlagResolver;
+    readonly replay?: ReplayBuilderOptions;
 }
 
 declare type AgentRecord = {
@@ -326,13 +325,17 @@ declare class AgentRecords {
     constructor(agent: Agent, persistence?: AgentRecordPersistence | undefined);
     get restoring(): RestoringContext | null;
     logRecord(record: AgentRecord): void;
-    restore(record: AgentRecord): void;
-    replay(): Promise<{
+    restore(record: AgentRecord): boolean;
+    replay(options?: AgentRecordsReplayOptions): Promise<{
         warning?: string;
     }>;
     flush(): Promise<void>;
 }
 
+declare interface AgentRecordsReplayOptions {
+    readonly rewriteMigratedRecords?: boolean;
+}
+
 export declare type AgentReplayRecord = {
     readonly time: number;
 } & AgentReplayRecordPayload;
@@ -918,10 +921,6 @@ declare interface CloseSessionPayload {
     readonly sessionId: string;
 }
 
-declare interface CompactedHistory {
-    text: string;
-}
-
 declare interface CompactionBeginData {
     instruction?: string;
     source: CompactionSource;
@@ -957,8 +956,6 @@ declare interface CompactionSummaryOrigin {
     readonly kind: 'compaction_summary';
 }
 
-declare type CompactionTelemetryTrigger = CompactionBeginData['source'] | 'manual-with-prompt' | 'unknown';
-
 export declare interface CompactOptions {
     readonly instruction?: string | undefined;
 }
@@ -1833,20 +1830,14 @@ declare class FullCompaction {
     protected compactionCountInTurn: number;
     protected compacting: {
         abortController: AbortController;
-        startedAt: number;
-        telemetryTrigger: CompactionTelemetryTrigger;
         promise: Promise<void>;
         blockedByTurn: boolean;
     } | null;
-    protected _compactedHistory: CompactedHistory[];
     protected readonly strategy: CompactionStrategy;
     constructor(agent: Agent, strategy?: CompactionStrategy);
     get isCompacting(): boolean;
-    get compactedHistory(): readonly CompactedHistory[];
     begin(data: Readonly<CompactionBeginData>): void;
-    private startCompactionWorker;
     cancel(): void;
-    private markCanceled;
     markCompleted(): void;
     private get tokenCountWithPending();
     resetForTurn(): void;
@@ -1857,6 +1848,7 @@ declare class FullCompaction {
     private beginAutoCompaction;
     private block;
     private compactionWorker;
+    private compactionRound;
     private triggerPreCompactHook;
     private triggerPostCompactHook;
     private postProcessSummary;
@@ -3531,7 +3523,6 @@ declare class KosongLLM implements LLM {
 
 declare interface KosongLLMConfig {
     readonly provider: ChatProvider;
-    readonly modelName: string;
     readonly systemPrompt: string;
     readonly capability?: ModelCapability | undefined;
     /**
@@ -3569,7 +3560,7 @@ declare interface LLMChatParams {
     messages: Message[];
     tools: readonly Tool[];
     signal: AbortSignal;
-    requestLogContext?: LLMRequestLogContext;
+    requestLogFields?: LLMRequestLogFields;
     onTextDelta?: ((delta: string) => void) | undefined;
     onThinkDelta?: ((delta: string) => void) | undefined;
     onToolCallDelta?: ((delta: ToolCallDelta) => void) | undefined;
@@ -3597,12 +3588,23 @@ declare interface LLMChatResponse {
     streamTiming?: LLMStreamTiming;
 }
 
-declare interface LLMRequestLogContext {
-    readonly turnId?: string;
-    readonly step?: number;
-    readonly stepUuid?: string;
-    readonly attempt?: number;
-    readonly maxAttempts?: number;
+declare interface LLMRequestLogFields {
+    readonly turnStep: string;
+    readonly attempt?: string;
+}
+
+declare class LlmRequestLogger {
+    private readonly log;
+    private lastConfigLogSignature;
+    constructor(log: Logger);
+    logRequest(input: {
+        readonly provider: ChatProvider;
+        readonly modelAlias?: string;
+        readonly systemPrompt: string;
+        readonly tools: readonly Tool[];
+        readonly messages: readonly Message[];
+        readonly fields: LLMRequestLogFields | undefined;
+    }): void;
 }
 
 declare interface LLMStreamTiming {
@@ -4798,16 +4800,30 @@ declare type RenameSessionRequest = SessionScopedPayload<RenameSessionPayload>;
 
 declare class ReplayBuilder {
     readonly agent: Agent;
+    private readonly options;
     postRestoring: boolean;
     captureLiveRecords: boolean;
     protected readonly records: AgentReplayRecord[];
-    constructor(agent: Agent);
+    private frozen;
+    private segmentStart;
+    constructor(agent: Agent, options?: ReplayBuilderOptions);
     push(record: AgentReplayRecordPayload): void;
     patchLast<T extends AgentReplayRecord['type']>(type: T, patch: Partial<Extract<AgentReplayRecord, {
         type: T;
     }>>): void;
     removeLastMessages(removedMessages: ReadonlySet<ContextMessage>): void;
+    finishRestoringRecord(type: string): boolean;
     buildResult(): readonly AgentReplayRecord[];
+    private removeMessagesFrom;
+}
+
+declare interface ReplayBuilderOptions {
+    readonly range?: ReplayRangeOptions;
+}
+
+declare interface ReplayRangeOptions {
+    readonly start?: number;
+    readonly count?: number;
 }
 
 declare interface ResolvedAgentProfile {

package/dist/index.mjs

@@ -3,7 +3,6 @@ import { dirname as __cjsShimDirname } from 'node:path';
 const __filename = __cjsShimFileURLToPath(import.meta.url);
 const __dirname = __cjsShimDirname(__filename);
 import { createRequire } from "node:module";
-import { createHash, randomBytes, randomInt, randomUUID } from "node:crypto";
 import I, { createWriteStream } from "fs";
 import Kr, { EventEmitter } from "events";
 import * as path$1$1 from "path";
@@ -21,6 +20,7 @@ import * as path$9 from "node:path";
 import path, { basename, dirname as dirname$1, join, posix, win32 } from "node:path";
 import { Blob as Blob$1, File as File$1 } from "node:buffer";
 import jn, { access, appendFile, chmod, copyFile, cp, lstat, mkdir, mkdtemp, open, readFile, readdir, realpath, rename, rm, stat, unlink, writeFile as writeFile$1 } from "node:fs/promises";
+import { createHash, randomBytes, randomInt, randomUUID } from "node:crypto";
 import { createControlledPromise, objectMap, sleep, uniq } from "@antfu/utils";
 import { z } from "zod";
 import { execFile, execFileSync, spawn } from "node:child_process";
@@ -1014,6 +1014,17 @@ function mergeInPlace(target, source) {
 function extractText(message, sep = "") {
 	return message.content.filter((part) => part.type === "text").map((part) => part.text).join(sep);
 }
+/** Create a simple user message with a single text part. */
+function createUserMessage(content) {
+	return {
+		role: "user",
+		content: [{
+			type: "text",
+			text: content
+		}],
+		toolCalls: []
+	};
+}
 /** Create a tool result message. */
 function createToolMessage(toolCallId, output) {
 	return {
@@ -9351,8 +9362,7 @@ var AnthropicChatProvider = class {
 		this._stream = options.stream ?? true;
 		this._metadata = options.metadata;
 		this._adaptiveThinking = options.adaptiveThinking;
-		const apiKey = options.apiKey ?? process.env["ANTHROPIC_API_KEY"];
-		this._apiKey = apiKey === void 0 || apiKey.length === 0 ? void 0 : apiKey;
+		this._apiKey = options.apiKey === void 0 || options.apiKey.length === 0 ? void 0 : options.apiKey;
 		this._baseUrl = options.baseUrl;
 		this._defaultHeaders = options.defaultHeaders;
 		this._clientFactory = options.clientFactory;
@@ -9458,13 +9468,38 @@ var AnthropicChatProvider = class {
 		return resolveAuthBackedClient({
 			cachedClient: this._client,
 			clientFactory: this._clientFactory
-		}, auth, (a) => this._buildClient(requireProviderApiKey("AnthropicChatProvider", a, this._apiKey)));
+		}, auth, (a) => this._buildClient(this._requireApiKey(a)));
+	}
+	_requireApiKey(auth) {
+		const apiKey = auth?.apiKey ?? this._apiKey;
+		if (apiKey === void 0 || apiKey.length === 0) throw new ChatProviderError("AnthropicChatProvider: apiKey is required. Provide it via constructor options, options.auth.apiKey on each request, or an OAuth login. The Anthropic adapter does not read shell API-key environment variables.");
+		return apiKey;
+	}
+	_anthropicCustomHeaderEnvNames() {
+		const customHeaders = process.env["ANTHROPIC_CUSTOM_HEADERS"];
+		if (customHeaders === void 0 || customHeaders.length === 0) return [];
+		const names = [];
+		for (const line of customHeaders.split("\n")) {
+			const colonIndex = line.indexOf(":");
+			if (colonIndex < 0) continue;
+			const name = line.slice(0, colonIndex).trim().toLowerCase();
+			if (name.length > 0) names.push(name);
+		}
+		return names;
+	}
+	_buildDefaultHeaders(apiKey) {
+		const defaultHeaders = { authorization: null };
+		for (const name of this._anthropicCustomHeaderEnvNames()) defaultHeaders[name] = null;
+		for (const [name, value] of Object.entries(this._defaultHeaders ?? {})) defaultHeaders[name.toLowerCase()] = value;
+		defaultHeaders["x-api-key"] = apiKey;
+		return defaultHeaders;
 	}
 	_buildClient(apiKey) {
 		return new Anthropic({
 			apiKey,
-			baseURL: this._baseUrl,
-			defaultHeaders: this._defaultHeaders
+			authToken: null,
+			baseURL: this._baseUrl ?? null,
+			defaultHeaders: this._buildDefaultHeaders(apiKey)
 		});
 	}
 	withThinking(effort) {
@@ -49791,14 +49826,12 @@ function logRequestFailure(input, error, attempt, maxAttempts) {
 	});
 }
 function paramsForAttempt(input, attempt, maxAttempts) {
+	const turnStep = `${input.turnId}.${String(input.currentStep)}`;
 	return {
 		...input.params,
-		requestLogContext: {
-			turnId: input.turnId,
-			step: input.currentStep,
-			stepUuid: input.stepUuid,
-			attempt,
-			maxAttempts
+		requestLogFields: attempt === 1 ? { turnStep } : {
+			turnStep,
+			attempt: `${String(attempt)}/${String(maxAttempts)}`
 		}
 	};
 }
@@ -54712,6 +54745,7 @@ function renderPrompt(template, vars) {
 }
 //#endregion
 //#region ../agent-core/src/utils/tokens.ts
+const messageTokenEstimateCache = /* @__PURE__ */ new WeakMap();
 /**
 * Estimate token count from text using a character-based heuristic.
 *   - ASCII (~4 chars per token)
@@ -54733,12 +54767,15 @@ function estimateTokensForMessages(messages) {
 	return total;
 }
 function estimateTokensForMessage(message) {
+	const cached = messageTokenEstimateCache.get(message);
+	if (cached !== void 0) return cached;
 	let total = estimateTokens(message.role);
 	total += estimateTokensForContentParts(message.content);
 	if (message.toolCalls !== void 0) for (const call of message.toolCalls) {
 		total += estimateTokens(call.name);
 		total += estimateTokens(JSON.stringify(call.arguments));
 	}
+	messageTokenEstimateCache.set(message, total);
 	return total;
 }
 function estimateTokensForContentParts(parts) {
@@ -54807,77 +54844,6 @@ function applyCompletionBudget(args) {
 //#region ../agent-core/src/agent/compaction/compaction-instruction.md?raw
 var compaction_instruction_default = "\n--- This message is a direct task, not part of the above conversation ---\n\nYou are now given a task to compact this conversation context according to specific priorities and output requirements.\n\nOutput text only. DO NOT CALL ANY TOOLS. Calling tools will be rejected and fails the task. You already have all the information you need in the conversation history. You have only one chance.\n\nThe goal of compaction is to keep essential code patterns, technical details, and architectural decisions for continuing development without losing context after the above messages are cleared work.\n\n{{ customInstruction }}\n\n<!-- Compression Priorities (in order) -->\n\n1. **Current Task State**: What is being worked on RIGHT NOW\n2. **Errors & Solutions**: All encountered errors and their resolutions\n3. **Code Evolution**: Final working versions only (remove intermediate attempts)\n4. **System Context**: Project structure, dependencies, environment setup\n5. **Design Decisions**: Architectural choices and their rationale\n6. **TODO Items**: Unfinished tasks and known issues\n\n<!-- Required Output Structure -->\n\n## Current Focus\n\n[What we're working on now]\n\n## Environment\n\n- [Key setup/config points]\n- ...\n\n## Completed Tasks\n\n- [Task]: [Brief outcome]\n- ...\n\n## Active Issues\n\n- [Issue]: [Status/Next steps]\n- ...\n\n## Code State\n\n### [Critical file name]\n\n[Brief description of the file's purpose and current state]\n\n```\n[The latest version of critical code snippets in this file, <20 lines]\n```\n\n### [Critical file name]\n\n- [Useful classes/methods/functions]: [Brief description/usage]\n- ...\n\n<!-- Omit non-critical code, intermediate attempts, and resolved errors -->\n\n## Important Context\n\n- [Any crucial information not covered above]\n- ...\n\n## All User Messages\n\n- [Detailed non tool use user message]\n- ...\n\n<!-- Must output a summary matching the above template in the **final answer**, not in thinking. -->\n";
 //#endregion
-//#region ../agent-core/src/agent/compaction/render-messages.ts
-function renderMessagesToText(messages) {
-	return messages.map((message, index) => renderMessageToText(message, index)).join("\n\n");
-}
-function renderMessageToText(message, index) {
-	const header = [`message ${String(index + 1)}`, `role=${message.role}`];
-	if (message.name !== void 0) header.push(`name=${JSON.stringify(message.name)}`);
-	if (message.toolCallId !== void 0) header.push(`toolCallId=${JSON.stringify(message.toolCallId)}`);
-	if (message.partial === true) header.push("partial=true");
-	const lines = [`--- ${header.join(" ")} ---`];
-	if (message.content.length === 0) lines.push("[empty content]");
-	else lines.push(...message.content.map(renderContentPartToText));
-	if (message.toolCalls.length > 0) {
-		lines.push("tool calls:");
-		for (const toolCall of message.toolCalls) lines.push(renderToolCallToText(toolCall));
-	}
-	return lines.join("\n");
-}
-function renderContentPartToText(part) {
-	switch (part.type) {
-		case "text": return renderBlock("text", part.text);
-		case "think": return renderBlock("think", part.think);
-		case "image_url": return renderMediaPart("image_url", part.imageUrl.url, part.imageUrl.id);
-		case "audio_url": return renderMediaPart("audio_url", part.audioUrl.url, part.audioUrl.id);
-		case "video_url": return renderMediaPart("video_url", part.videoUrl.url, part.videoUrl.id);
-		default: return renderBlock("content", stringifyJsonish(part));
-	}
-}
-function renderToolCallToText(toolCall) {
-	const lines = [`- ${toolCall.id}: ${toolCall.name}`, renderBlock("arguments", renderToolCallArguments(toolCall.arguments))];
-	if (toolCall.extras !== void 0) lines.push(renderBlock("extras", stringifyJsonish(toolCall.extras)));
-	return lines.join("\n");
-}
-function renderToolCallArguments(args) {
-	if (args === null) return "null";
-	try {
-		return stringifyJsonish(JSON.parse(args));
-	} catch {
-		return args;
-	}
-}
-function renderMediaPart(type, url, id) {
-	if (id === void 0) return `${type}: ${url}`;
-	return `${type}: ${url} (id=${id})`;
-}
-function renderBlock(label, value) {
-	return `${label}:\n${indentBlock(value)}`;
-}
-function indentBlock(value) {
-	if (value.length === 0) return "  ";
-	return value.split("\n").map((line) => `  ${line}`).join("\n");
-}
-function stringifyJsonish(value) {
-	const seen = /* @__PURE__ */ new WeakSet();
-	const replacer = (_key, nested) => {
-		if (typeof nested === "bigint") return `${nested.toString()}n`;
-		if (typeof nested === "function") return `[Function ${nested.name || "anonymous"}]`;
-		if (typeof nested === "symbol") return nested.toString();
-		if (nested !== null && typeof nested === "object") {
-			if (seen.has(nested)) return "[Circular]";
-			seen.add(nested);
-		}
-		return nested;
-	};
-	try {
-		return JSON.stringify(value, replacer, 2) ?? String(value);
-	} catch {
-		return String(value);
-	}
-}
-//#endregion
 //#region ../agent-core/src/tools/support/input-schema.ts
 /**
 * Shared helper for deriving the JSON Schema that a tool advertises to the
@@ -55050,7 +55016,7 @@ var DefaultCompactionStrategy = class {
 	}
 	computeCompactCount(messages, source) {
 		if (source === "manual") {
-			for (let i = messages.length - 1; i > 0; i--) if (canSplitAfter(messages, i)) return i + 1;
+			for (let i = messages.length - 1; i > 0; i--) if (canSplitAfter(messages, i)) return this.fitCompactCountToWindow(messages, i + 1);
 			return 0;
 		}
 		let recentMessages = 1;
@@ -55065,7 +55031,7 @@ var DefaultCompactionStrategy = class {
 			if (canSplitAfter(messages, splitIndex)) bestN = splitIndex + 1;
 			if ((recentMessages >= this.config.maxRecentMessages || recentUserMessages >= this.config.maxRecentUserMessages || recentSize >= this.maxSize * this.config.maxRecentSizeRatio) && bestN !== void 0) break;
 		}
-		return bestN ?? 0;
+		return this.fitCompactCountToWindow(messages, bestN ?? 0);
 	}
 	reduceCompactOnOverflow(messages) {
 		const minReducedSize = Math.max(1, Math.ceil(this.maxSize * this.config.minOverflowReductionRatio));
@@ -55080,6 +55046,20 @@ var DefaultCompactionStrategy = class {
 		}
 		return bestN ?? messages.length;
 	}
+	fitCompactCountToWindow(messages, compactedCount) {
+		if (this.maxSize <= 0 || compactedCount <= 0) return compactedCount;
+		let compactedSize = 0;
+		for (let i = 0; i < compactedCount; i++) compactedSize += estimateTokensForMessage(messages[i]);
+		if (compactedSize <= this.maxSize) return compactedCount;
+		let bestN;
+		for (let n = compactedCount - 1; n > 0; n--) {
+			compactedSize -= estimateTokensForMessage(messages[n]);
+			if (!canSplitAfter(messages, n - 1)) continue;
+			bestN = n;
+			if (compactedSize <= this.maxSize) return n;
+		}
+		return bestN ?? compactedCount;
+	}
 	get checkAfterStep() {
 		return this.config.triggerRatio !== this.config.blockRatio;
 	}
@@ -55135,7 +55115,6 @@ var FullCompaction = class {
 	agent;
 	compactionCountInTurn = 0;
 	compacting = null;
-	_compactedHistory = [];
 	strategy;
 	constructor(agent, strategy) {
 		this.agent = agent;
@@ -55147,9 +55126,6 @@ var FullCompaction = class {
 	get isCompacting() {
 		return this.compacting !== null;
 	}
-	get compactedHistory() {
-		return this._compactedHistory;
-	}
 	begin(data) {
 		if (this.compacting) return;
 		if (data.source === "manual") this.compactionCountInTurn = 0;
@@ -55168,29 +55144,19 @@ var FullCompaction = class {
 			type: "full_compaction.begin",
 			...data
 		});
-		this.startCompactionWorker(data, compactedCount);
-	}
-	startCompactionWorker(data, compactedCount) {
-		const abortController = new AbortController();
 		this.agent.emitEvent({
 			type: "compaction.started",
 			trigger: data.source,
 			instruction: data.instruction
 		});
-		const active = {
+		const abortController = new AbortController();
+		this.compacting = {
 			abortController,
-			startedAt: Date.now(),
-			telemetryTrigger: compactionTelemetryTrigger(data.source, data.instruction),
-			promise: Promise.resolve(),
+			promise: this.compactionWorker(abortController.signal, data, compactedCount),
 			blockedByTurn: false
 		};
-		this.compacting = active;
-		active.promise = this.compactionWorker(abortController.signal, data, compactedCount);
 	}
 	cancel() {
-		this.markCanceled();
-	}
-	markCanceled() {
 		this.agent.replayBuilder.patchLast("compaction", { result: "cancelled" });
 		if (!this.compacting) return;
 		this.agent.records.logRecord({ type: "full_compaction.cancel" });
@@ -55201,7 +55167,6 @@ var FullCompaction = class {
 	markCompleted() {
 		this.agent.records.logRecord({ type: "full_compaction.complete" });
 		this.compacting = null;
-		this._compactedHistory.push({ text: renderMessagesToText(this.agent.context.history) });
 	}
 	get tokenCountWithPending() {
 		return this.agent.context.tokenCountWithPending;
@@ -55252,7 +55217,46 @@ var FullCompaction = class {
 			await active.promise;
 		}
 	}
-	async compactionWorker(signal, data, initialCompactedCount) {
+	async compactionWorker(signal, data, compactedCount) {
+		try {
+			const finalResult = {
+				summary: "",
+				compactedCount: 1,
+				tokensBefore: 0,
+				tokensAfter: 0
+			};
+			for (let round = 1;; round++) {
+				const result = await this.compactionRound(round, signal, data, compactedCount);
+				if (!result) return;
+				finalResult.summary = result.summary;
+				finalResult.compactedCount += result.compactedCount - 1;
+				finalResult.tokensBefore += result.tokensBefore - finalResult.tokensAfter;
+				finalResult.tokensAfter = result.tokensAfter;
+				if (result.tokensBefore - result.tokensAfter < 1024) break;
+				if (!this.strategy.shouldBlock(result.tokensAfter)) break;
+				compactedCount = this.strategy.computeCompactCount(this.agent.context.history, data.source);
+				if (compactedCount === 0) break;
+			}
+			this.markCompleted();
+			this.agent.emitEvent({
+				type: "compaction.completed",
+				result: finalResult
+			});
+			await this.agent.injection.injectGoal();
+			this.triggerPostCompactHook(data, finalResult);
+		} catch (error) {
+			if (isAbortError(error)) return;
+			const blockedByTurn = this.compacting?.blockedByTurn === true;
+			this.cancel();
+			this.agent.log.error("compaction failed", { error });
+			if (blockedByTurn) throw error;
+			this.agent.emitEvent({
+				type: "error",
+				...toKimiErrorPayload(error)
+			});
+		}
+	}
+	async compactionRound(round, signal, data, initialCompactedCount) {
 		const startedAt = Date.now();
 		const originalHistory = [...this.agent.context.history];
 		const tokensBefore = estimateTokensForMessages(originalHistory);
@@ -55271,14 +55275,7 @@ var FullCompaction = class {
 			let summary;
 			while (true) {
 				const messagesToCompact = originalHistory.slice(0, compactedCount);
-				const messages = [...this.agent.context.project(messagesToCompact), {
-					role: "user",
-					content: [{
-						type: "text",
-						text: COMPACTION_INSTRUCTION(data.instruction)
-					}],
-					toolCalls: []
-				}];
+				const messages = [...this.agent.context.project(messagesToCompact), createUserMessage(renderPrompt(compaction_instruction_default, { customInstruction: data.instruction ?? "" }))];
 				try {
 					const response = await this.agent.generate(provider, this.agent.config.systemPrompt, [...this.agent.tools.loopTools], messages, void 0, { signal });
 					if (response.finishReason === "truncated") throw new CompactionTruncatedError();
@@ -55308,51 +55305,30 @@ var FullCompaction = class {
 				tokensBefore,
 				tokensAfter
 			};
-			const active = this.compacting;
 			this.agent.telemetry.track("compaction_finished", {
-				trigger_type: active.telemetryTrigger,
-				before_tokens: result.tokensBefore,
-				after_tokens: result.tokensAfter,
-				duration_ms: Date.now() - active.startedAt,
-				compacted_count: result.compactedCount,
-				retry_count: retryCount,
-				...usage
-			});
-			this.markCompleted();
-			this.agent.emitEvent({
-				type: "compaction.completed",
-				result
+				tokensBefore: result.tokensBefore,
+				tokensAfter: result.tokensAfter,
+				duration: Date.now() - startedAt,
+				compactedCount: result.compactedCount,
+				retryCount,
+				round,
+				...usage,
+				...data
 			});
 			this.agent.context.applyCompaction(result);
-			await this.agent.injection.injectGoal();
-			this.triggerPostCompactHook(data, result);
+			return result;
 		} catch (error) {
-			if (!isAbortError(error)) {
-				const blockedByTurn = this.compacting?.blockedByTurn === true;
-				this.agent.log.error("compaction failed", {
-					code: isKimiError(error) ? error.code : void 0,
-					error
-				});
-				this.markCanceled();
-				if (!blockedByTurn) {
-					const payload = isKimiError(error) && error.code === ErrorCodes.AUTH_LOGIN_REQUIRED ? toKimiErrorPayload(error) : makeErrorPayload(ErrorCodes.COMPACTION_FAILED, String(error));
-					this.agent.emitEvent({
-						type: "error",
-						...payload
-					});
-				}
-				this.agent.telemetry.track("compaction_failed", {
-					trigger_type: compactionTelemetryTrigger(data.source, data.instruction),
-					before_tokens: tokensBefore,
-					duration_ms: Date.now() - startedAt,
-					retry_count: retryCount,
-					error_type: error instanceof Error ? error.name : "Unknown"
-				});
-				if (blockedByTurn) {
-					if (isKimiError(error) && error.code === ErrorCodes.AUTH_LOGIN_REQUIRED) throw error;
-					throw new KimiError(ErrorCodes.COMPACTION_FAILED, String(error), { cause: error });
-				}
-			}
+			if (isAbortError(error)) return;
+			this.agent.telemetry.track("compaction_failed", {
+				...data,
+				tokensBefore,
+				duration: Date.now() - startedAt,
+				round,
+				retryCount,
+				errorType: error instanceof Error ? error.name : "Unknown"
+			});
+			if (isKimiError(error) && error.code === ErrorCodes.AUTH_LOGIN_REQUIRED) throw error;
+			throw new KimiError(ErrorCodes.COMPACTION_FAILED, String(error), { cause: error });
 		}
 	}
 	async triggerPreCompactHook(data, tokenCount, signal) {
@@ -55388,12 +55364,6 @@ function extractCompactionSummary(response) {
 	if (summary.trim().length === 0) throw new APIEmptyResponseError("The compaction response did not contain a non-empty summary.");
 	return summary;
 }
-const COMPACTION_INSTRUCTION = (customInstruction = "") => renderPrompt(compaction_instruction_default, { customInstruction });
-function compactionTelemetryTrigger(trigger, instruction) {
-	if (trigger === void 0) return "unknown";
-	if (trigger === "manual" && instruction !== void 0 && instruction.length > 0) return "manual-with-prompt";
-	return trigger;
-}
 //#endregion
 //#region ../agent-core/src/agent/compaction/micro.ts
 const DEFAULT_CONFIG = {
@@ -56771,7 +56741,7 @@ var CronManager = class {
 	}
 };
 //#endregion
-//#region ../agent-core/src/agent/kimi-env-params.ts
+//#region ../agent-core/src/config/kimi-env-params.ts
 /**
 * Apply Kimi sampling params (`KIMI_MODEL_TEMPERATURE`, `KIMI_MODEL_TOP_P`) from
 * the environment to a chat provider. Applied at provider construction
@@ -56795,7 +56765,7 @@ function applyKimiEnvSamplingParams(provider, env = process.env) {
 }
 /**
 * Apply the Moonshot preserved-thinking passthrough (`KIMI_MODEL_THINKING_KEEP`
-* -> `thinking.keep`) to a chat provider. Applied in `Agent.llm` after
+* -> `thinking.keep`) to a chat provider. Applied in `ConfigState.provider` after
 * `withThinking`, and only while thinking is on — otherwise the API would
 * receive a `thinking.keep` with no accompanying `thinking.type` it honors.
 * (Compaction uses a raw provider with thinking off, so it correctly skips this.)
@@ -62523,17 +62493,20 @@ var AgentRecords = class {
 		this._restoring = { time: record.time ?? Date.now() };
 		try {
 			restoreAgentRecord(this.agent, record);
+			return this.agent.replayBuilder.finishRestoringRecord(record.type);
 		} finally {
 			this._restoring = null;
 		}
 	}
-	async replay() {
+	async replay(options = {}) {
 		if (!this.persistence) throw new Error("No persistence provided for AgentRecords");
+		const rewriteMigratedRecords = options.rewriteMigratedRecords ?? true;
 		let migrations = [];
 		let hasMetadata = false;
 		let shouldRewrite = false;
 		let warning;
-		const replayedRecords = [];
+		const replayedRecords = rewriteMigratedRecords ? [] : void 0;
+		let completed = true;
 		for await (const record of this.persistence.read()) {
 			if (!hasMetadata) {
 				if (record.type !== "metadata") throw new Error("AgentRecords replay expected metadata as the first record");
@@ -62553,14 +62526,17 @@ var AgentRecords = class {
 				...migratedRecord,
 				protocol_version: "1.4"
 			};
-			replayedRecords.push(migratedRecord);
-			this.restore(migratedRecord);
+			replayedRecords?.push(migratedRecord);
+			if (this.restore(migratedRecord)) {
+				completed = false;
+				break;
+			}
 		}
-		if (shouldRewrite) {
+		if (completed && shouldRewrite && replayedRecords !== void 0) {
 			this.persistence.rewrite(replayedRecords);
 			await this.persistence.flush();
 		}
-		if (this.agent.blobStore !== void 0) for (const msg of this.agent.context.history) await this.agent.blobStore.rehydrateParts(msg.content);
+		if (completed && this.agent.blobStore !== void 0) for (const msg of this.agent.context.history) await this.agent.blobStore.rehydrateParts(msg.content);
 		return { warning };
 	}
 	async flush() {
@@ -62569,36 +62545,78 @@ var AgentRecords = class {
 };
 //#endregion
 //#region ../agent-core/src/agent/replay/index.ts
+const UNDO_BOUNDARY_RECORD_TYPES = new Set(["context.clear", "context.apply_compaction"]);
 var ReplayBuilder = class {
 	agent;
+	options;
 	postRestoring = false;
 	captureLiveRecords = false;
 	records = [];
-	constructor(agent) {
+	frozen = false;
+	segmentStart = 0;
+	constructor(agent, options = {}) {
 		this.agent = agent;
+		this.options = options;
 	}
 	push(record) {
-		if (this.captureLiveRecords || this.agent.records.restoring || this.postRestoring) this.records.push({
-			...record,
-			time: this.agent.records.restoring?.time ?? Date.now()
-		});
+		if (this.captureLiveRecords || this.agent.records.restoring || this.postRestoring) {
+			if (this.frozen) return;
+			const stamped = {
+				...record,
+				time: this.agent.records.restoring?.time ?? Date.now()
+			};
+			this.records.push(stamped);
+		}
 	}
 	patchLast(type, patch) {
+		if (this.frozen) return;
 		if (this.agent.records.restoring) {
 			const last = this.records.at(-1);
 			if (last && last.type === type) Object.assign(last, patch);
 		}
 	}
 	removeLastMessages(removedMessages) {
+		if (this.frozen) return;
 		if (removedMessages.size === 0) return;
-		for (let i = this.records.length - 1; i >= 0; i--) {
-			const record = this.records[i];
-			if (record.type === "message" && removedMessages.has(record.message)) this.records.splice(i, 1);
+		this.removeMessagesFrom(this.records, removedMessages);
+	}
+	finishRestoringRecord(type) {
+		const range = this.options.range;
+		if (range === void 0) return false;
+		if (this.frozen) return true;
+		if (!UNDO_BOUNDARY_RECORD_TYPES.has(type)) return false;
+		if (range.start === void 0) return false;
+		const start = range.start;
+		const nextSegmentStart = this.segmentStart + this.records.length;
+		if (nextSegmentStart > start) {
+			this.frozen = true;
+			return true;
 		}
+		this.segmentStart = nextSegmentStart;
+		this.records.splice(0);
+		return false;
 	}
 	buildResult() {
+		const range = this.options.range;
+		if (range !== void 0) {
+			if (range.start === void 0 && range.count !== void 0) {
+				const offset = Math.max(0, this.records.length - range.count);
+				return this.records.slice(offset);
+			}
+			const start = range.start ?? 0;
+			const offset = Math.max(0, start - this.segmentStart);
+			const count = range.count;
+			const end = count === void 0 ? void 0 : offset + count;
+			return this.records.slice(offset, end);
+		}
 		return this.records;
 	}
+	removeMessagesFrom(records, removedMessages) {
+		for (let i = records.length - 1; i >= 0; i--) {
+			const record = records[i];
+			if (record.type === "message" && removedMessages.has(record.message)) records.splice(i, 1);
+		}
+	}
 };
 //#endregion
 //#region ../../node_modules/.pnpm/js-yaml@4.1.1/node_modules/js-yaml/dist/js-yaml.mjs
@@ -91476,7 +91494,6 @@ var ToolManager = class {
 *     derives loop control from the normalized response shape, not from the
 *     provider's finish-reason spelling.
 */
-const GENERATE_REQUEST_LOG_CONTEXT = "__kimiRequestLogContext";
 var KosongLLM = class {
 	systemPrompt;
 	modelName;
@@ -91486,7 +91503,7 @@ var KosongLLM = class {
 	completionBudgetConfig;
 	constructor(config) {
 		this.provider = config.provider;
-		this.modelName = config.modelName;
+		this.modelName = config.provider.modelName;
 		this.systemPrompt = config.systemPrompt;
 		this.capability = config.capability;
 		this.generate = config.generate ?? generate;
@@ -91503,7 +91520,7 @@ var KosongLLM = class {
 			streamEndedAt = Date.now();
 		};
 		const markStreamOutput = () => {
-			if (firstChunkAt === void 0) firstChunkAt = Date.now();
+			firstChunkAt ??= Date.now();
 		};
 		const callbacks = buildKosongCallbacks(params, markStreamOutput);
 		const effectiveProvider = applyCompletionBudget({
@@ -91511,10 +91528,13 @@ var KosongLLM = class {
 			budget: this.completionBudgetConfig,
 			capability: this.capability
 		});
-		const result = await this.generate(effectiveProvider, this.systemPrompt, [...params.tools], params.messages, callbacks, generateOptions(params, {
+		const options = {
+			signal: params.signal,
 			onRequestStart: markRequestStart,
-			onStreamEnd: markStreamEnd
-		}));
+			onStreamEnd: markStreamEnd,
+			requestLogFields: params.requestLogFields
+		};
+		const result = await this.generate(effectiveProvider, this.systemPrompt, [...params.tools], params.messages, callbacks, options);
 		if (params.onTextPart !== void 0 || params.onThinkPart !== void 0) {
 			for (const part of result.message.content) if (part.type === "text" && params.onTextPart !== void 0) await params.onTextPart(part);
 			else if (part.type === "think" && params.onThinkPart !== void 0) await params.onThinkPart(part);
@@ -91538,14 +91558,6 @@ function buildStreamTiming(requestStartedAt, firstChunkAt, streamEndedAt) {
 		streamDurationMs: Math.max(0, outputEndedAt - firstChunkAt)
 	};
 }
-function generateOptions(params, hooks) {
-	return {
-		signal: params.signal,
-		onRequestStart: hooks.onRequestStart,
-		onStreamEnd: hooks.onStreamEnd,
-		[GENERATE_REQUEST_LOG_CONTEXT]: params.requestLogContext
-	};
-}
 function buildKosongCallbacks(params, markStreamOutput) {
 	const toolCallIdentities = /* @__PURE__ */ new Map();
 	const pendingIndexedToolCallDeltas = /* @__PURE__ */ new Map();
@@ -91674,6 +91686,64 @@ function totalUsage(byModel) {
 	return total;
 }
 //#endregion
+//#region ../agent-core/src/agent/llm-request-logger.ts
+var LlmRequestLogger = class {
+	log;
+	lastConfigLogSignature;
+	constructor(log) {
+		this.log = log;
+	}
+	logRequest(input) {
+		const { provider, modelAlias, systemPrompt, tools, messages, fields } = input;
+		const requestLogFields = fields ?? {};
+		const config = {
+			provider: provider.name,
+			model: provider.modelName,
+			modelAlias,
+			thinkingEffort: provider.thinkingEffort ?? void 0,
+			systemPromptChars: systemPrompt.length,
+			toolCount: tools.length
+		};
+		const signature = JSON.stringify({
+			...config,
+			systemPromptHash: fingerprint(systemPrompt),
+			toolsHash: fingerprint(JSON.stringify(toolSignature(tools)))
+		});
+		if (signature !== this.lastConfigLogSignature) {
+			this.lastConfigLogSignature = signature;
+			this.log.info("llm config", {
+				...requestLogFields,
+				...config
+			});
+		}
+		const partialMessageCount = messages.filter((message) => message.partial === true).length;
+		const requestFields = { ...requestLogFields };
+		if (partialMessageCount > 0) requestFields.partialMessageCount = partialMessageCount;
+		this.log.info("llm request", requestFields);
+	}
+};
+function splitGenerateOptions(options) {
+	if (options === void 0) return {
+		requestLogFields: void 0,
+		generateOptions: void 0
+	};
+	const { requestLogFields, ...generateOptions } = options;
+	return {
+		requestLogFields,
+		generateOptions
+	};
+}
+function toolSignature(tools) {
+	return tools.map(({ name, description, parameters }) => ({
+		name,
+		description,
+		parameters
+	}));
+}
+function fingerprint(content) {
+	return createHash("sha256").update(content).digest("hex");
+}
+//#endregion
 //#region ../agent-core/src/agent/index.ts
 var Agent$1 = class {
 	type;
@@ -91694,6 +91764,7 @@ var Agent$1 = class {
 	log;
 	telemetry;
 	experimentalFlags;
+	llmRequestLogger;
 	blobStore;
 	records;
 	fullCompaction;
@@ -91712,7 +91783,6 @@ var Agent$1 = class {
 	cron;
 	goal;
 	replayBuilder;
-	lastLlmConfigLogSignature;
 	constructor(options) {
 		this.type = options.type ?? "main";
 		this._kaos = options.kaos;
@@ -91729,6 +91799,7 @@ var Agent$1 = class {
 		this.log = options.log ?? log;
 		this.telemetry = options.telemetry ?? noopTelemetryClient;
 		this.experimentalFlags = options.experimentalFlags ?? new FlagResolver();
+		this.llmRequestLogger = new LlmRequestLogger(this.log);
 		this.blobStore = options.homedir ? new BlobStore({ blobsDir: join$1(options.homedir, "blobs") }) : void 0;
 		this.records = new AgentRecords(this, options.persistence ?? (options.homedir ? new FileSystemAgentRecordPersistence(join$1(options.homedir, "wire.jsonl"), {
 			onError: (error) => {
@@ -91751,35 +91822,38 @@ var Agent$1 = class {
 		this.background = new BackgroundManager(this, this.homedir === void 0 ? void 0 : new BackgroundTaskPersistence(this.homedir));
 		this.cron = this.type === "sub" ? null : new CronManager(this);
 		this.goal = new GoalMode(this);
-		this.replayBuilder = new ReplayBuilder(this);
+		this.replayBuilder = new ReplayBuilder(this, options.replay);
 	}
 	setKaos(kaos) {
 		this._kaos = kaos;
 	}
 	get generate() {
 		return async (provider, systemPrompt, tools, history, callbacks, options) => {
-			if (options?.auth !== void 0) {
-				this.logLlmRequest(provider, systemPrompt, tools, history, options);
-				return this.rawGenerate(provider, systemPrompt, tools, history, callbacks, options);
-			}
+			const { requestLogFields, generateOptions } = splitGenerateOptions(options);
 			const modelAlias = this.config.modelAlias;
+			const run = (requestOptions) => {
+				this.llmRequestLogger.logRequest({
+					provider,
+					modelAlias,
+					systemPrompt,
+					tools,
+					messages: history,
+					fields: requestLogFields
+				});
+				return this.rawGenerate(provider, systemPrompt, tools, history, callbacks, requestOptions);
+			};
+			if (generateOptions?.auth !== void 0) return run(generateOptions);
 			const withAuth = modelAlias === void 0 ? void 0 : this.modelProvider?.resolveAuth?.(modelAlias, { log: this.log });
-			if (withAuth === void 0) {
-				this.logLlmRequest(provider, systemPrompt, tools, history, options);
-				return this.rawGenerate(provider, systemPrompt, tools, history, callbacks, options);
-			}
+			if (withAuth === void 0) return run(generateOptions);
 			return withAuth((auth) => {
-				const requestOptions = {
-					...options,
+				return run({
+					...generateOptions,
 					auth
-				};
-				this.logLlmRequest(provider, systemPrompt, tools, history, requestOptions);
-				return this.rawGenerate(provider, systemPrompt, tools, history, callbacks, requestOptions);
+				});
 			});
 		};
 	}
 	get llm() {
-		const model = this.config.model;
 		const provider = this.config.provider;
 		const loopControl = this.kimiConfig?.loopControl;
 		const completionBudgetConfig = resolveCompletionBudget({
@@ -91788,34 +91862,12 @@ var Agent$1 = class {
 		});
 		return new KosongLLM({
 			provider,
-			modelName: model,
 			systemPrompt: this.config.systemPrompt,
 			capability: this.config.modelCapabilities,
 			generate: this.generate,
 			completionBudgetConfig
 		});
 	}
-	logLlmRequest(provider, systemPrompt, tools, history, options) {
-		const context = buildLlmRequestContext(options);
-		const configMetadata = buildLlmConfigMetadata(provider, this.config.modelAlias, systemPrompt, tools);
-		this.logLlmConfigIfChanged(context, configMetadata, buildLlmConfigSignature(configMetadata, systemPrompt, tools));
-		let partialMessageCount = 0;
-		for (const message of history) if (message.partial === true) partialMessageCount += 1;
-		const requestMetadata = {};
-		if (partialMessageCount > 0) requestMetadata.partialMessageCount = partialMessageCount;
-		this.log.info("llm request", {
-			...context,
-			...requestMetadata
-		});
-	}
-	logLlmConfigIfChanged(context, metadata, signature) {
-		if (signature === this.lastLlmConfigLogSignature) return;
-		this.lastLlmConfigLogSignature = signature;
-		this.log.info("llm config", {
-			...context,
-			...metadata
-		});
-	}
 	useProfile(profile, context) {
 		const systemPrompt = profile.systemPrompt({
 			osEnv: this.kaos.osEnv,
@@ -91830,8 +91882,8 @@ var Agent$1 = class {
 		});
 		this.tools.setActiveTools(profile.tools);
 	}
-	async resume() {
-		const result = await this.records.replay();
+	async resume(options) {
+		const result = await this.records.replay(options);
 		try {
 			this.replayBuilder.postRestoring = true;
 			this.goal.normalizeAfterReplay();
@@ -91988,41 +92040,6 @@ var Agent$1 = class {
 		});
 	}
 };
-function buildLlmRequestContext(options) {
-	const context = requestLogContext(options);
-	if (context === void 0) return {};
-	const fields = { turnStep: context.turnId === void 0 || context.step === void 0 ? void 0 : `${context.turnId}.${String(context.step)}` };
-	if (context.attempt !== void 0 && context.maxAttempts !== void 0 && context.attempt > 1) fields.attempt = `${String(context.attempt)}/${String(context.maxAttempts)}`;
-	return fields;
-}
-function buildLlmConfigMetadata(provider, modelAlias, systemPrompt, tools) {
-	return {
-		provider: provider.name,
-		model: provider.modelName,
-		modelAlias,
-		thinkingEffort: provider.thinkingEffort ?? void 0,
-		systemPromptChars: systemPrompt.length,
-		toolCount: tools.length
-	};
-}
-function buildLlmConfigSignature(metadata, systemPrompt, tools) {
-	const toolsForSignature = tools.map(({ name, description, parameters }) => ({
-		name,
-		description,
-		parameters
-	}));
-	return JSON.stringify({
-		...metadata,
-		systemPromptHash: fingerprint(systemPrompt),
-		toolsHash: fingerprint(JSON.stringify(toolsForSignature))
-	});
-}
-function fingerprint(content) {
-	return createHash("sha256").update(content).digest("hex");
-}
-function requestLogContext(options) {
-	return options?.[GENERATE_REQUEST_LOG_CONTEXT];
-}
 //#endregion
 //#region ../agent-core/src/rpc/types.ts
 function proxyWithExtraPayload(methods, extraPayload) {
@@ -122945,7 +122962,11 @@ var Session$1 = class {
 			defaultValue: false,
 			parseEnv: parseBooleanEnv
 		})) return;
-		await Promise.all(Array.from(this.readyAgents(), (agent) => agent.background.stopAll("Session closed")));
+		await Promise.all(Array.from(this.readyAgents(), async (agent) => {
+			const activeTasks = agent.background.list(true);
+			await Promise.all(activeTasks.map((task) => agent.background.suppressTerminalNotification(task.taskId)));
+			await agent.background.stopAll("Session closed");
+		}));
 	}
 	async createAgent(config, options = {}) {
 		await this.skillsReady;
@@ -138706,6 +138727,7 @@ var BufferedReadable = class extends Readable {
 		this._source.off("end", this._onEnd);
 		this._source.off("close", this._onClose);
 		this._source.off("error", this._onError);
+		this._source.destroy();
 		callback(error);
 	}
 	_onData = (chunk) => {

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@botiverse/kimi-code-sdk",
-  "version": "0.9.4",
-  "description": "Botiverse repackage of @moonshot-ai/kimi-code-sdk (from kimi-code @moonshot-ai/kimi-code@0.15.0) — built node-sdk bundle for Slock. Mirror of MoonshotAI/kimi-code.",
+  "version": "0.16.0",
+  "description": "Botiverse repackage of @moonshot-ai/kimi-code-sdk (from kimi-code @moonshot-ai/kimi-code@0.16.0) — built node-sdk bundle for Slock. Mirror of MoonshotAI/kimi-code.",
   "license": "MIT",
   "author": "Moonshot AI",
   "homepage": "https://github.com/botiverse/kimi-code-sdk#readme",
@@ -54,7 +54,7 @@
   "types": "./dist/index.d.mts",
   "botiverse": {
     "upstream": "@moonshot-ai/kimi-code-sdk",
-    "kimiRelease": "@moonshot-ai/kimi-code@0.15.0",
+    "kimiRelease": "@moonshot-ai/kimi-code@0.16.0",
     "source": "https://github.com/MoonshotAI/kimi-code"
   }
 }