kimi-code-memory-mcp-server 0.1.0

package/dist/config.js

@@ -0,0 +1,63 @@
+/**
+ * Global configuration.
+ *
+ * The store root can be overridden via the MEMORY_STORE_ROOT environment
+ * variable. When unset, it defaults to ~/.kimi-code-memory/ so that the
+ * server behaves like a user-level tool even when installed as a package.
+ */
+import os from 'os';
+import path from 'path';
+export const homedir = os.homedir();
+export function getStoreRoot() {
+    const envRoot = process.env.MEMORY_STORE_ROOT;
+    if (envRoot) {
+        if (envRoot.includes('\0')) {
+            throw new Error('MEMORY_STORE_ROOT contains invalid null byte');
+        }
+        const resolved = path.resolve(envRoot);
+        if (!path.isAbsolute(resolved)) {
+            throw new Error(`MEMORY_STORE_ROOT must resolve to an absolute path: ${envRoot}`);
+        }
+        return resolved;
+    }
+    return path.join(homedir, '.kimi-code-memory');
+}
+/**
+ * Resolve the directory where Kimi Code CLI session wires are stored.
+ *
+ * Priority:
+ * 1. MEMORY_SESSIONS_ROOT environment variable (absolute path)
+ * 2. KIMI_CODE_HOME environment variable (<KIMI_CODE_HOME>/sessions)
+ * 3. Default ~/.kimi-code/sessions
+ */
+export function getSessionsRoot() {
+    if (process.env.MEMORY_SESSIONS_ROOT) {
+        const resolved = path.resolve(process.env.MEMORY_SESSIONS_ROOT);
+        if (!path.isAbsolute(resolved)) {
+            throw new Error(`MEMORY_SESSIONS_ROOT must resolve to an absolute path: ${process.env.MEMORY_SESSIONS_ROOT}`);
+        }
+        return resolved;
+    }
+    if (process.env.KIMI_CODE_HOME) {
+        return path.join(path.resolve(process.env.KIMI_CODE_HOME), 'sessions');
+    }
+    return path.join(homedir, '.kimi-code', 'sessions');
+}
+export const DEFAULT_CONTEXT_WINDOW = {
+    detailedRounds: 3,
+    defaultSummaryRounds: 2,
+    loadMoreChunkSize: 5,
+};
+export const DEFAULT_RECENT_CHANGE_LIMIT = 5;
+export const ESSENCE_SIZE_LIMIT = 15 * 1024; // 15 KB
+/**
+ * search_context 用于把相邻 turn 聚成「簇」的默认时间间隔（秒）。
+ *
+ * 「簇」是指一小段连续的来回对话，通常围绕同一个决策或讨论展开。
+ * 如果两个相邻 turn 的时间间隔超过这个值，就认为它们属于不同的簇。
+ *
+ * 90 秒适合典型的 agent-user 对话：既能区分开不同话题，又足以保留
+ * 一段流畅讨论的完整性。每次搜索可以通过 cluster_gap_seconds 参数覆盖。
+ */
+export const DEFAULT_CLUSTER_GAP_SECONDS = 90;
+//# sourceMappingURL=config.js.map

package/dist/config.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"config.js","sourceRoot":"","sources":["../src/config.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,IAAI,MAAM,MAAM,CAAC;AAExB,MAAM,CAAC,MAAM,OAAO,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC;AAEpC,MAAM,UAAU,YAAY;IAC1B,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC;IAC9C,IAAI,OAAO,EAAE,CAAC;QACZ,IAAI,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;YAC3B,MAAM,IAAI,KAAK,CAAC,8CAA8C,CAAC,CAAC;QAClE,CAAC;QACD,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QACvC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC/B,MAAM,IAAI,KAAK,CAAC,uDAAuD,OAAO,EAAE,CAAC,CAAC;QACpF,CAAC;QACD,OAAO,QAAQ,CAAC;IAClB,CAAC;IACD,OAAO,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,mBAAmB,CAAC,CAAC;AACjD,CAAC;AAED;;;;;;;GAOG;AACH,MAAM,UAAU,eAAe;IAC7B,IAAI,OAAO,CAAC,GAAG,CAAC,oBAAoB,EAAE,CAAC;QACrC,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC,CAAC;QAChE,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC/B,MAAM,IAAI,KAAK,CAAC,0DAA0D,OAAO,CAAC,GAAG,CAAC,oBAAoB,EAAE,CAAC,CAAC;QAChH,CAAC;QACD,OAAO,QAAQ,CAAC;IAClB,CAAC;IACD,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,CAAC;QAC/B,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,EAAE,UAAU,CAAC,CAAC;IACzE,CAAC;IACD,OAAO,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,YAAY,EAAE,UAAU,CAAC,CAAC;AACtD,CAAC;AAQD,MAAM,CAAC,MAAM,sBAAsB,GAAkB;IACnD,cAAc,EAAE,CAAC;IACjB,oBAAoB,EAAE,CAAC;IACvB,iBAAiB,EAAE,CAAC;CACrB,CAAC;AAEF,MAAM,CAAC,MAAM,2BAA2B,GAAG,CAAC,CAAC;AAE7C,MAAM,CAAC,MAAM,kBAAkB,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC,QAAQ;AAErD;;;;;;;;GAQG;AACH,MAAM,CAAC,MAAM,2BAA2B,GAAG,EAAE,CAAC"}

package/dist/context/wire-context.d.ts

@@ -0,0 +1,171 @@
+/**
+ * wire.jsonl parsing and context recovery.
+ *
+ * This module is intentionally Kimi Code CLI specific: it understands the
+ * wire.jsonl event schema. A future abstraction layer could make this pluggable.
+ */
+import type { ContextWindow } from '../config.js';
+import type { RefinedManager } from '../refined-manager.js';
+export interface McpConfig {
+    version: number;
+    contextWindow: ContextWindow;
+    recentChangeLimit: number;
+    sessionMappings: Record<string, SessionMapping>;
+}
+export interface SessionMapping {
+    slugDir: string;
+    discoveredAt: string;
+}
+export interface WireSession {
+    sessionId: string;
+    wire: string;
+    mtime?: number;
+    slugDir?: string;
+}
+export interface WireAction {
+    name: string;
+    args: string;
+    result: string;
+}
+export interface WireTurn {
+    turnId: string;
+    timestamp: string | null;
+    user: string;
+    agentText: string;
+    actions: WireAction[];
+}
+export interface CompactionSummary {
+    time: string | null;
+    summary: string;
+}
+export interface ContextWindowOverrides {
+    detailedRounds?: number;
+    summaryRounds?: number;
+}
+export interface SearchOptions {
+    limit?: number;
+    dateFrom?: string;
+    dateTo?: string;
+    refinedManager?: RefinedManager;
+}
+export interface SearchMatch {
+    sessionId: string;
+    turnId: number;
+    timestamp: string | null;
+    score: number;
+    user: string;
+    agent: string;
+    snippet: string;
+    actions: string[];
+}
+export interface SearchResult {
+    query: string;
+    totalMatches: number;
+    matches: SearchMatch[];
+    hits: Array<{
+        sessionId: string;
+        turn: WireTurn;
+    }>;
+    skippedSessionIds?: string[];
+}
+export interface LoadTurnOptions {
+    maxReferences?: number;
+}
+export interface TurnReference {
+    sessionId: string;
+    turnId: number;
+}
+export interface DetailedRound {
+    turnId: number;
+    timestamp: string | null;
+    user: string;
+    agent: string;
+    actions: WireAction[];
+}
+export interface SummaryRound {
+    turnId: number;
+    timestamp: string | null;
+    summary: string;
+}
+/**
+ * Load the global MCP configuration, creating defaults if absent.
+ */
+export declare function loadMcpConfig(): McpConfig;
+/**
+ * Persist the global MCP configuration.
+ */
+export declare function saveMcpConfig(config: McpConfig): void;
+/**
+ * Discover the active session for the current workspace and cache the mapping.
+ */
+export declare function discoverCurrentSession(): WireSession | null;
+/**
+ * Resolve the active session wire path, using the cached mapping if valid.
+ */
+export declare function getCurrentSessionWirePath(): WireSession | null;
+/**
+ * Find all historical sessions for the current workspace.
+ */
+export declare function findAllWorkspaceSessions(): WireSession[];
+/**
+ * Parse a wire.jsonl file and return conversation turns + session-level
+ * compaction summaries.
+ */
+export declare function parseWireFile(wirePath: string): Promise<{
+    turns: WireTurn[];
+    compactionSummaries: CompactionSummary[];
+}>;
+/**
+ * Build the default context window: last N rounds detailed, preceding M rounds
+ * summarized.
+ */
+export declare function buildContextWindow(turns: WireTurn[], overrides?: ContextWindowOverrides): {
+    detailedRounds: DetailedRound[];
+    summaryRounds: SummaryRound[];
+    totalTurns: number;
+};
+/**
+ * Load older rounds before a given turnId, summarized by default.
+ */
+export declare function loadMoreRounds(turns: WireTurn[], beforeTurnId: number, limit?: number): SummaryRound[];
+/**
+ * Search across all workspace session wires for rounds matching the query.
+ *
+ * If a RefinedManager is provided, we first query the refined SQLite index for
+ * candidate turns and only parse the wires needed to return full content. This
+ * avoids parsing every historical wire on every search. When refined data is
+ * unavailable we fall back to the original full-scan behavior.
+ */
+export declare function searchWireContext(query: string, options?: SearchOptions): Promise<{
+    query: string;
+    totalMatches: number;
+    matches: SearchMatch[];
+    hits: Array<{
+        sessionId: string;
+        turn: WireTurn;
+    }>;
+    skippedSessionIds?: undefined;
+} | {
+    query: string;
+    totalMatches: number;
+    matches: SearchMatch[];
+    hits: {
+        sessionId: string;
+        turn: WireTurn;
+    }[];
+    skippedSessionIds: string[];
+}>;
+/**
+ * Load the full detailed content of specific conversation turns.
+ */
+export declare function loadTurnContext(references: TurnReference[], options?: LoadTurnOptions): Promise<{
+    rounds: never[];
+    notFound: never[];
+    error: string;
+} | {
+    rounds: ({
+        sessionId: string;
+    } & DetailedRound)[];
+    notFound: TurnReference[];
+    error?: undefined;
+}>;