kiokuko 1.0.0

package/dist/mcp/request-deadline.js

@@ -0,0 +1,147 @@
+import process from 'node:process';
+export const DEFAULT_MCP_REQUEST_TIMEOUT_MS = 90_000;
+export const MIN_MCP_REQUEST_TIMEOUT_MS = 10;
+export const MAX_MCP_REQUEST_TIMEOUT_MS = 90_000;
+export const MCP_READ_TIMEOUT_MS = 30_000;
+export const MCP_EXTERNAL_IO_TIMEOUT_MS = 10_000;
+export const MCP_MUTATION_TIMEOUT_MS = 60_000;
+export class McpRequestTimeoutError extends Error {
+    operation;
+    code = 'MCP_REQUEST_TIMEOUT';
+    retryable = true;
+    constructor(operation) {
+        super('MCP request timed out');
+        this.operation = operation;
+        this.name = 'McpRequestTimeoutError';
+    }
+}
+export class McpRequestCancelledError extends Error {
+    operation;
+    code = 'MCP_REQUEST_CANCELLED';
+    retryable = false;
+    constructor(operation) {
+        super('MCP request was cancelled');
+        this.operation = operation;
+        this.name = 'McpRequestCancelledError';
+    }
+}
+export const DEFAULT_MCP_DEADLINE_POLICY = Object.freeze({
+    readMs: MCP_READ_TIMEOUT_MS,
+    externalMs: MCP_EXTERNAL_IO_TIMEOUT_MS,
+    mutationMs: MCP_MUTATION_TIMEOUT_MS,
+    hardMaxMs: DEFAULT_MCP_REQUEST_TIMEOUT_MS,
+});
+const EXTERNAL_OPERATIONS = new Set([
+    'task_prepare',
+    'task_answer',
+]);
+const MUTATION_OPERATIONS = new Set([
+    'memory_checkpoint',
+    'curator_globalize',
+]);
+function assertTimeoutMs(value, label) {
+    if (!Number.isSafeInteger(value)
+        || value < MIN_MCP_REQUEST_TIMEOUT_MS
+        || value > MAX_MCP_REQUEST_TIMEOUT_MS) {
+        throw new Error(`${label} must be an integer between ${MIN_MCP_REQUEST_TIMEOUT_MS} and ${MAX_MCP_REQUEST_TIMEOUT_MS} milliseconds`);
+    }
+    return value;
+}
+function positiveIntegerEnvironmentValue(name) {
+    const raw = process.env[name];
+    if (raw === undefined || raw.length === 0)
+        return undefined;
+    const value = Number(raw);
+    return Number.isSafeInteger(value) && value > 0 ? value : undefined;
+}
+export function createMcpDeadlinePolicy(overrides = {}) {
+    const policy = {
+        readMs: overrides.readMs
+            ?? positiveIntegerEnvironmentValue('KIOKUKO_MCP_READ_TIMEOUT_MS')
+            ?? DEFAULT_MCP_DEADLINE_POLICY.readMs,
+        externalMs: overrides.externalMs
+            ?? positiveIntegerEnvironmentValue('KIOKUKO_MCP_EXTERNAL_IO_TIMEOUT_MS')
+            ?? DEFAULT_MCP_DEADLINE_POLICY.externalMs,
+        mutationMs: overrides.mutationMs
+            ?? positiveIntegerEnvironmentValue('KIOKUKO_MCP_MUTATION_TIMEOUT_MS')
+            ?? DEFAULT_MCP_DEADLINE_POLICY.mutationMs,
+        hardMaxMs: overrides.hardMaxMs
+            ?? positiveIntegerEnvironmentValue('KIOKUKO_MCP_HARD_MAX_TIMEOUT_MS')
+            ?? DEFAULT_MCP_DEADLINE_POLICY.hardMaxMs,
+    };
+    assertTimeoutMs(policy.readMs, 'read timeout');
+    assertTimeoutMs(policy.externalMs, 'external I/O timeout');
+    assertTimeoutMs(policy.mutationMs, 'mutation timeout');
+    assertTimeoutMs(policy.hardMaxMs, 'hard maximum timeout');
+    if (policy.readMs > policy.hardMaxMs || policy.externalMs > policy.hardMaxMs || policy.mutationMs > policy.hardMaxMs) {
+        throw new Error('MCP operation timeout must not exceed the hard maximum timeout');
+    }
+    return policy;
+}
+export function operationDeadlineClass(operation) {
+    if (EXTERNAL_OPERATIONS.has(operation))
+        return 'external';
+    if (MUTATION_OPERATIONS.has(operation))
+        return 'mutation';
+    return 'read';
+}
+export function timeoutForOperation(operation, policy = DEFAULT_MCP_DEADLINE_POLICY) {
+    const validated = createMcpDeadlinePolicy(policy);
+    switch (operationDeadlineClass(operation)) {
+        case 'external': return validated.externalMs;
+        case 'mutation': return validated.mutationMs;
+        case 'read': return validated.readMs;
+    }
+}
+function abortReasonIsTimeout(reason) {
+    return reason instanceof McpRequestTimeoutError;
+}
+export async function runWithMcpDeadline({ operation, timeoutMs, policy = DEFAULT_MCP_DEADLINE_POLICY, signal, operationFn, }) {
+    const selectedTimeoutMs = assertTimeoutMs(timeoutMs ?? timeoutForOperation(operation, policy), 'MCP request timeout');
+    const deadlineAt = Date.now() + selectedTimeoutMs;
+    const controller = new AbortController();
+    const context = {
+        operation,
+        deadlineAt,
+        remainingMs: () => Math.max(0, deadlineAt - Date.now()),
+        childTimeoutMs: (requestedMs) => Math.min(assertTimeoutMs(requestedMs, 'child timeout'), Math.max(0, deadlineAt - Date.now())),
+    };
+    if (signal?.aborted)
+        throw new McpRequestCancelledError(operation);
+    return await new Promise((resolve, reject) => {
+        let settled = false;
+        let timer;
+        const cleanup = () => {
+            if (timer !== undefined)
+                clearTimeout(timer);
+            signal?.removeEventListener('abort', onCallerAbort);
+        };
+        const settle = (settlement) => {
+            if (settled)
+                return;
+            settled = true;
+            cleanup();
+            settlement();
+        };
+        const onCallerAbort = () => {
+            const error = new McpRequestCancelledError(operation);
+            controller.abort(error);
+            settle(() => reject(error));
+        };
+        const onTimeout = () => {
+            const error = new McpRequestTimeoutError(operation);
+            controller.abort(error);
+            settle(() => reject(error));
+        };
+        signal?.addEventListener('abort', onCallerAbort, { once: true });
+        timer = setTimeout(onTimeout, selectedTimeoutMs);
+        void Promise.resolve()
+            .then(() => {
+            if (controller.signal.aborted)
+                throw controller.signal.reason;
+            return operationFn(controller.signal, context);
+        })
+            .then((value) => settle(() => resolve(value)), (error) => settle(() => reject(abortReasonIsTimeout(controller.signal.reason) ? new McpRequestTimeoutError(operation) : error)));
+    });
+}
+//# sourceMappingURL=request-deadline.js.map

package/dist/mcp/request-deadline.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"request-deadline.js","sourceRoot":"","sources":["../../src/mcp/request-deadline.ts"],"names":[],"mappings":"AAAA,OAAO,OAAO,MAAM,cAAc,CAAC;AAEnC,MAAM,CAAC,MAAM,8BAA8B,GAAG,MAAM,CAAC;AACrD,MAAM,CAAC,MAAM,0BAA0B,GAAG,EAAE,CAAC;AAC7C,MAAM,CAAC,MAAM,0BAA0B,GAAG,MAAM,CAAC;AAEjD,MAAM,CAAC,MAAM,mBAAmB,GAAG,MAAM,CAAC;AAC1C,MAAM,CAAC,MAAM,0BAA0B,GAAG,MAAM,CAAC;AACjD,MAAM,CAAC,MAAM,uBAAuB,GAAG,MAAM,CAAC;AAwC9C,MAAM,OAAO,sBAAuB,SAAQ,KAAK;IAI1B;IAHZ,IAAI,GAAG,qBAA8B,CAAC;IACtC,SAAS,GAAG,IAAa,CAAC;IAEnC,YAAqB,SAA2B;QAC9C,KAAK,CAAC,uBAAuB,CAAC,CAAC;QADZ,cAAS,GAAT,SAAS,CAAkB;QAE9C,IAAI,CAAC,IAAI,GAAG,wBAAwB,CAAC;IACvC,CAAC;CACF;AAED,MAAM,OAAO,wBAAyB,SAAQ,KAAK;IAI5B;IAHZ,IAAI,GAAG,uBAAgC,CAAC;IACxC,SAAS,GAAG,KAAc,CAAC;IAEpC,YAAqB,SAA2B;QAC9C,KAAK,CAAC,2BAA2B,CAAC,CAAC;QADhB,cAAS,GAAT,SAAS,CAAkB;QAE9C,IAAI,CAAC,IAAI,GAAG,0BAA0B,CAAC;IACzC,CAAC;CACF;AAED,MAAM,CAAC,MAAM,2BAA2B,GAAgC,MAAM,CAAC,MAAM,CAAC;IACpF,MAAM,EAAE,mBAAmB;IAC3B,UAAU,EAAE,0BAA0B;IACtC,UAAU,EAAE,uBAAuB;IACnC,SAAS,EAAE,8BAA8B;CAC1C,CAAC,CAAC;AAEH,MAAM,mBAAmB,GAAkC,IAAI,GAAG,CAAC;IACjE,cAAc;IACd,aAAa;CACd,CAAC,CAAC;AAEH,MAAM,mBAAmB,GAAkC,IAAI,GAAG,CAAC;IACjE,mBAAmB;IACnB,mBAAmB;CACpB,CAAC,CAAC;AAEH,SAAS,eAAe,CAAC,KAAa,EAAE,KAAa;IACnD,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,KAAK,CAAC;WAC3B,KAAK,GAAG,0BAA0B;WAClC,KAAK,GAAG,0BAA0B,EAAE,CAAC;QACxC,MAAM,IAAI,KAAK,CAAC,GAAG,KAAK,+BAA+B,0BAA0B,QAAQ,0BAA0B,eAAe,CAAC,CAAC;IACtI,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,+BAA+B,CAAC,IAAY;IACnD,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IAC9B,IAAI,GAAG,KAAK,SAAS,IAAI,GAAG,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,SAAS,CAAC;IAC5D,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC;IAC1B,OAAO,MAAM,CAAC,aAAa,CAAC,KAAK,CAAC,IAAI,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC;AACtE,CAAC;AAED,MAAM,UAAU,uBAAuB,CAAC,YAAwC,EAAE;IAChF,MAAM,MAAM,GAAG;QACb,MAAM,EAAE,SAAS,CAAC,MAAM;eACnB,+BAA+B,CAAC,6BAA6B,CAAC;eAC9D,2BAA2B,CAAC,MAAM;QACvC,UAAU,EAAE,SAAS,CAAC,UAAU;eAC3B,+BAA+B,CAAC,oCAAoC,CAAC;eACrE,2BAA2B,CAAC,UAAU;QAC3C,UAAU,EAAE,SAAS,CAAC,UAAU;eAC3B,+BAA+B,CAAC,iCAAiC,CAAC;eAClE,2BAA2B,CAAC,UAAU;QAC3C,SAAS,EAAE,SAAS,CAAC,SAAS;eACzB,+BAA+B,CAAC,iCAAiC,CAAC;eAClE,2BAA2B,CAAC,SAAS;KAC3C,CAAC;IACF,eAAe,CAAC,MAAM,CAAC,MAAM,EAAE,cAAc,CAAC,CAAC;IAC/C,eAAe,CAAC,MAAM,CAAC,UAAU,EAAE,sBAAsB,CAAC,CAAC;IAC3D,eAAe,CAAC,MAAM,CAAC,UAAU,EAAE,kBAAkB,CAAC,CAAC;IACvD,eAAe,CAAC,MAAM,CAAC,SAAS,EAAE,sBAAsB,CAAC,CAAC;IAC1D,IAAI,MAAM,CAAC,MAAM,GAAG,MAAM,CAAC,SAAS,IAAI,MAAM,CAAC,UAAU,GAAG,MAAM,CAAC,SAAS,IAAI,MAAM,CAAC,UAAU,GAAG,MAAM,CAAC,SAAS,EAAE,CAAC;QACrH,MAAM,IAAI,KAAK,CAAC,gEAAgE,CAAC,CAAC;IACpF,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,SAA2B;IAChE,IAAI,mBAAmB,CAAC,GAAG,CAAC,SAAS,CAAC;QAAE,OAAO,UAAU,CAAC;IAC1D,IAAI,mBAAmB,CAAC,GAAG,CAAC,SAAS,CAAC;QAAE,OAAO,UAAU,CAAC;IAC1D,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,MAAM,UAAU,mBAAmB,CACjC,SAA2B,EAC3B,SAA4B,2BAA2B;IAEvD,MAAM,SAAS,GAAG,uBAAuB,CAAC,MAAM,CAAC,CAAC;IAClD,QAAQ,sBAAsB,CAAC,SAAS,CAAC,EAAE,CAAC;QAC1C,KAAK,UAAU,CAAC,CAAC,OAAO,SAAS,CAAC,UAAU,CAAC;QAC7C,KAAK,UAAU,CAAC,CAAC,OAAO,SAAS,CAAC,UAAU,CAAC;QAC7C,KAAK,MAAM,CAAC,CAAC,OAAO,SAAS,CAAC,MAAM,CAAC;IACvC,CAAC;AACH,CAAC;AAED,SAAS,oBAAoB,CAAC,MAAe;IAC3C,OAAO,MAAM,YAAY,sBAAsB,CAAC;AAClD,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,kBAAkB,CAAI,EAC1C,SAAS,EACT,SAAS,EACT,MAAM,GAAG,2BAA2B,EACpC,MAAM,EACN,WAAW,GACkB;IAC7B,MAAM,iBAAiB,GAAG,eAAe,CAAC,SAAS,IAAI,mBAAmB,CAAC,SAAS,EAAE,MAAM,CAAC,EAAE,qBAAqB,CAAC,CAAC;IACtH,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,iBAAiB,CAAC;IAClD,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;IACzC,MAAM,OAAO,GAAuB;QAClC,SAAS;QACT,UAAU;QACV,WAAW,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvD,cAAc,EAAE,CAAC,WAAW,EAAE,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,eAAe,CAAC,WAAW,EAAE,eAAe,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;KAC/H,CAAC;IAEF,IAAI,MAAM,EAAE,OAAO;QAAE,MAAM,IAAI,wBAAwB,CAAC,SAAS,CAAC,CAAC;IAEnE,OAAO,MAAM,IAAI,OAAO,CAAI,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QAC9C,IAAI,OAAO,GAAG,KAAK,CAAC;QACpB,IAAI,KAAgD,CAAC;QAErD,MAAM,OAAO,GAAG,GAAS,EAAE;YACzB,IAAI,KAAK,KAAK,SAAS;gBAAE,YAAY,CAAC,KAAK,CAAC,CAAC;YAC7C,MAAM,EAAE,mBAAmB,CAAC,OAAO,EAAE,aAAa,CAAC,CAAC;QACtD,CAAC,CAAC;QAEF,MAAM,MAAM,GAAG,CAAC,UAAsB,EAAQ,EAAE;YAC9C,IAAI,OAAO;gBAAE,OAAO;YACpB,OAAO,GAAG,IAAI,CAAC;YACf,OAAO,EAAE,CAAC;YACV,UAAU,EAAE,CAAC;QACf,CAAC,CAAC;QAEF,MAAM,aAAa,GAAG,GAAS,EAAE;YAC/B,MAAM,KAAK,GAAG,IAAI,wBAAwB,CAAC,SAAS,CAAC,CAAC;YACtD,UAAU,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YACxB,MAAM,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;QAC9B,CAAC,CAAC;QAEF,MAAM,SAAS,GAAG,GAAS,EAAE;YAC3B,MAAM,KAAK,GAAG,IAAI,sBAAsB,CAAC,SAAS,CAAC,CAAC;YACpD,UAAU,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YACxB,MAAM,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;QAC9B,CAAC,CAAC;QAEF,MAAM,EAAE,gBAAgB,CAAC,OAAO,EAAE,aAAa,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;QACjE,KAAK,GAAG,UAAU,CAAC,SAAS,EAAE,iBAAiB,CAAC,CAAC;QACjD,KAAK,OAAO,CAAC,OAAO,EAAE;aACnB,IAAI,CAAC,GAAG,EAAE;YACT,IAAI,UAAU,CAAC,MAAM,CAAC,OAAO;gBAAE,MAAM,UAAU,CAAC,MAAM,CAAC,MAAM,CAAC;YAC9D,OAAO,WAAW,CAAC,UAAU,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QACjD,CAAC,CAAC;aACD,IAAI,CACH,CAAC,KAAK,EAAE,EAAE,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EACvC,CAAC,KAAc,EAAE,EAAE,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,MAAM,CACrC,oBAAoB,CAAC,UAAU,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,sBAAsB,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,KAAK,CAC/F,CAAC,CACH,CAAC;IACN,CAAC,CAAC,CAAC;AACL,CAAC"}

package/dist/mcp/runtime-owner.d.ts

@@ -0,0 +1,27 @@
+import { type InitOptions } from '../commands/init.js';
+import { type PathEnvironment } from '../config/paths.js';
+import type { SqliteDatabase } from '../db/adapter.js';
+import { type EmbeddingDatabaseOpener } from '../embedding/backend.js';
+import type { EmbeddingConfig, EmbeddingProvider, EmbeddingRuntime, VectorSearchBackend } from '../embedding/types.js';
+export interface McpRuntimeOwnerOptions extends PathEnvironment {
+    readonly databasePath?: string;
+    readonly migrationsDirectory?: string;
+    readonly initializeDatabase?: (options: InitOptions) => unknown | PromiseLike<unknown>;
+    readonly openDatabase?: EmbeddingDatabaseOpener;
+    readonly embeddingConfig?: EmbeddingConfig;
+    readonly embeddingProvider?: EmbeddingProvider;
+    readonly embeddingBackend?: VectorSearchBackend;
+}
+export type McpDatabaseOperation<T> = (database: SqliteDatabase, runtime: EmbeddingRuntime) => Promise<T> | T;
+export interface McpDatabaseOwner {
+    withDatabase<T>(operation: McpDatabaseOperation<T>): Promise<T>;
+    close(): Promise<void>;
+}
+/** Own one MCP-process database/runtime instead of opening a connection per tool call. */
+export declare class McpRuntimeOwner implements McpDatabaseOwner {
+    #private;
+    constructor(options?: McpRuntimeOwnerOptions);
+    withDatabase<T>(operation: McpDatabaseOperation<T>): Promise<T>;
+    close(): Promise<void>;
+}
+//# sourceMappingURL=runtime-owner.d.ts.map

package/dist/mcp/runtime-owner.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"runtime-owner.d.ts","sourceRoot":"","sources":["../../src/mcp/runtime-owner.ts"],"names":[],"mappings":"AAAA,OAAO,EAAsB,KAAK,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAC3E,OAAO,EAAyB,KAAK,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAEjF,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AAEvD,OAAO,EAAyB,KAAK,uBAAuB,EAAE,MAAM,yBAAyB,CAAC;AAE9F,OAAO,KAAK,EAAE,eAAe,EAAE,iBAAiB,EAAE,gBAAgB,EAAE,mBAAmB,EAAE,MAAM,uBAAuB,CAAC;AAIvH,MAAM,WAAW,sBAAuB,SAAQ,eAAe;IAC7D,QAAQ,CAAC,YAAY,CAAC,EAAE,MAAM,CAAC;IAC/B,QAAQ,CAAC,mBAAmB,CAAC,EAAE,MAAM,CAAC;IACtC,QAAQ,CAAC,kBAAkB,CAAC,EAAE,CAAC,OAAO,EAAE,WAAW,KAAK,OAAO,GAAG,WAAW,CAAC,OAAO,CAAC,CAAC;IACvF,QAAQ,CAAC,YAAY,CAAC,EAAE,uBAAuB,CAAC;IAChD,QAAQ,CAAC,eAAe,CAAC,EAAE,eAAe,CAAC;IAC3C,QAAQ,CAAC,iBAAiB,CAAC,EAAE,iBAAiB,CAAC;IAC/C,QAAQ,CAAC,gBAAgB,CAAC,EAAE,mBAAmB,CAAC;CACjD;AAED,MAAM,MAAM,oBAAoB,CAAC,CAAC,IAAI,CACpC,QAAQ,EAAE,cAAc,EACxB,OAAO,EAAE,gBAAgB,KACtB,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;AAEpB,MAAM,WAAW,gBAAgB;IAC/B,YAAY,CAAC,CAAC,EAAE,SAAS,EAAE,oBAAoB,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;IAChE,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;CACxB;AASD,0FAA0F;AAC1F,qBAAa,eAAgB,YAAW,gBAAgB;;gBAQ1C,OAAO,GAAE,sBAA2B;IA0D1C,YAAY,CAAC,CAAC,EAAE,SAAS,EAAE,oBAAoB,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC;IAK/D,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;CAuC7B"}

package/dist/mcp/runtime-owner.js

@@ -0,0 +1,129 @@
+import { initializeDatabase } from '../commands/init.js';
+import { getGlobalDatabasePath } from '../config/paths.js';
+import { openConnection } from '../db/connection.js';
+import { KiokukoError } from '../errors.js';
+import { openEmbeddingDatabase } from '../embedding/backend.js';
+import { createEmbeddingRuntime } from '../embedding/runtime.js';
+import { createEmbeddingWorker } from '../embedding/worker.js';
+import { WriteQueue } from '../server/write-queue.js';
+/** Own one MCP-process database/runtime instead of opening a connection per tool call. */
+export class McpRuntimeOwner {
+    #options;
+    #initialization;
+    #state;
+    #closing = false;
+    #closed = false;
+    #closePromise;
+    constructor(options = {}) {
+        this.#options = options;
+    }
+    async #initialize() {
+        const databasePath = this.#options.databasePath ?? getGlobalDatabasePath(this.#options);
+        const initialize = this.#options.initializeDatabase ?? initializeDatabase;
+        await initialize({
+            databasePath,
+            ...(this.#options.migrationsDirectory === undefined ? {} : { migrationsDirectory: this.#options.migrationsDirectory }),
+        });
+        const config = this.#options.embeddingConfig;
+        const opened = await openEmbeddingDatabase(databasePath, {
+            ...(config === undefined ? {} : { config }),
+            openDatabase: this.#options.openDatabase ?? openConnection,
+            ...(this.#options.embeddingBackend === undefined ? {} : { backend: this.#options.embeddingBackend }),
+        });
+        const database = opened.database;
+        const queue = new WriteQueue(64);
+        try {
+            const runtime = createEmbeddingRuntime(database, config, {
+                ...(this.#options.embeddingProvider === undefined ? {} : { provider: this.#options.embeddingProvider }),
+                ...(opened.backend === undefined ? {} : { backend: opened.backend }),
+                enqueueWrite: (operation) => queue.enqueue(operation),
+            });
+            const worker = runtime.profileId === null ? undefined : createEmbeddingWorker({ runtime });
+            const state = { database, runtime, worker, queue };
+            this.#state = state;
+            worker?.start();
+            return state;
+        }
+        catch (error) {
+            try {
+                await queue.close();
+            }
+            catch (closeError) {
+                throw new AggregateError([error, closeError], 'MCP runtime initialization failed and its write queue could not be closed');
+            }
+            try {
+                database.close();
+            }
+            catch (closeError) {
+                throw new AggregateError([error, closeError], 'MCP runtime initialization failed and its database could not be closed');
+            }
+            throw error;
+        }
+    }
+    async #ensureState() {
+        if (this.#closed || this.#closing)
+            throw new KiokukoError('SERVICE_UNAVAILABLE', 'MCP runtime is closed');
+        if (this.#state !== undefined)
+            return this.#state;
+        if (this.#initialization === undefined) {
+            const initialization = this.#initialize();
+            this.#initialization = initialization;
+            void initialization.then(undefined, () => {
+                if (this.#initialization === initialization)
+                    this.#initialization = undefined;
+            });
+        }
+        return this.#initialization;
+    }
+    async withDatabase(operation) {
+        const state = await this.#ensureState();
+        return operation(state.database, state.runtime);
+    }
+    async close() {
+        if (this.#closePromise !== undefined)
+            return this.#closePromise;
+        this.#closing = true;
+        const attempt = (async () => {
+            const state = this.#state ?? await this.#initialization;
+            if (state === undefined) {
+                this.#closed = true;
+                return;
+            }
+            const errors = [];
+            try {
+                await state.worker?.close();
+            }
+            catch (error) {
+                errors.push(error);
+            }
+            if (state.worker === undefined) {
+                try {
+                    await state.runtime.close();
+                }
+                catch (error) {
+                    errors.push(error);
+                }
+            }
+            try {
+                await state.queue.close();
+            }
+            catch (error) {
+                errors.push(error);
+            }
+            try {
+                state.database.close();
+            }
+            catch (error) {
+                errors.push(error);
+            }
+            this.#closed = true;
+            if (errors.length === 1)
+                throw errors[0];
+            if (errors.length > 1)
+                throw new AggregateError(errors, 'MCP runtime cleanup failed');
+        })();
+        this.#closePromise = attempt;
+        return attempt;
+    }
+}
+//# sourceMappingURL=runtime-owner.js.map

package/dist/mcp/runtime-owner.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"runtime-owner.js","sourceRoot":"","sources":["../../src/mcp/runtime-owner.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAoB,MAAM,qBAAqB,CAAC;AAC3E,OAAO,EAAE,qBAAqB,EAAwB,MAAM,oBAAoB,CAAC;AACjF,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AAErD,OAAO,EAAE,YAAY,EAAE,MAAM,cAAc,CAAC;AAC5C,OAAO,EAAE,qBAAqB,EAAgC,MAAM,yBAAyB,CAAC;AAC9F,OAAO,EAAE,sBAAsB,EAAE,MAAM,yBAAyB,CAAC;AAEjE,OAAO,EAAE,qBAAqB,EAAwB,MAAM,wBAAwB,CAAC;AACrF,OAAO,EAAE,UAAU,EAAE,MAAM,0BAA0B,CAAC;AA6BtD,0FAA0F;AAC1F,MAAM,OAAO,eAAe;IACjB,QAAQ,CAAyB;IAC1C,eAAe,CAAkC;IACjD,MAAM,CAAyB;IAC/B,QAAQ,GAAG,KAAK,CAAC;IACjB,OAAO,GAAG,KAAK,CAAC;IAChB,aAAa,CAA4B;IAEzC,YAAY,UAAkC,EAAE;QAC9C,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC;IAC1B,CAAC;IAED,KAAK,CAAC,WAAW;QACf,MAAM,YAAY,GAAG,IAAI,CAAC,QAAQ,CAAC,YAAY,IAAI,qBAAqB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACxF,MAAM,UAAU,GAAG,IAAI,CAAC,QAAQ,CAAC,kBAAkB,IAAI,kBAAkB,CAAC;QAC1E,MAAM,UAAU,CAAC;YACf,YAAY;YACZ,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,mBAAmB,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,mBAAmB,EAAE,IAAI,CAAC,QAAQ,CAAC,mBAAmB,EAAE,CAAC;SACvH,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,eAAe,CAAC;QAC7C,MAAM,MAAM,GAAG,MAAM,qBAAqB,CAAC,YAAY,EAAE;YACvD,GAAG,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC;YAC3C,YAAY,EAAE,IAAI,CAAC,QAAQ,CAAC,YAAY,IAAI,cAAc;YAC1D,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,gBAAgB,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,CAAC,QAAQ,CAAC,gBAAgB,EAAE,CAAC;SACrG,CAAC,CAAC;QACH,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC;QACjC,MAAM,KAAK,GAAG,IAAI,UAAU,CAAU,EAAE,CAAC,CAAC;QAC1C,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,sBAAsB,CAAC,QAAQ,EAAE,MAAM,EAAE;gBACvD,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,iBAAiB,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,CAAC,iBAAiB,EAAE,CAAC;gBACvG,GAAG,CAAC,MAAM,CAAC,OAAO,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC;gBACpE,YAAY,EAAE,CAAI,SAAmC,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,SAAS,CAAe;aACjG,CAAC,CAAC;YACH,MAAM,MAAM,GAAG,OAAO,CAAC,SAAS,KAAK,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,qBAAqB,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC;YAC3F,MAAM,KAAK,GAAG,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,EAAE,KAAK,EAAuB,CAAC;YACxE,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;YACpB,MAAM,EAAE,KAAK,EAAE,CAAC;YAChB,OAAO,KAAK,CAAC;QACf,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC;gBACH,MAAM,KAAK,CAAC,KAAK,EAAE,CAAC;YACtB,CAAC;YAAC,OAAO,UAAU,EAAE,CAAC;gBACpB,MAAM,IAAI,cAAc,CAAC,CAAC,KAAK,EAAE,UAAU,CAAC,EAAE,2EAA2E,CAAC,CAAC;YAC7H,CAAC;YACD,IAAI,CAAC;gBACH,QAAQ,CAAC,KAAK,EAAE,CAAC;YACnB,CAAC;YAAC,OAAO,UAAU,EAAE,CAAC;gBACpB,MAAM,IAAI,cAAc,CAAC,CAAC,KAAK,EAAE,UAAU,CAAC,EAAE,wEAAwE,CAAC,CAAC;YAC1H,CAAC;YACD,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED,KAAK,CAAC,YAAY;QAChB,IAAI,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,QAAQ;YAAE,MAAM,IAAI,YAAY,CAAC,qBAAqB,EAAE,uBAAuB,CAAC,CAAC;QAC1G,IAAI,IAAI,CAAC,MAAM,KAAK,SAAS;YAAE,OAAO,IAAI,CAAC,MAAM,CAAC;QAClD,IAAI,IAAI,CAAC,eAAe,KAAK,SAAS,EAAE,CAAC;YACvC,MAAM,cAAc,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;YAC1C,IAAI,CAAC,eAAe,GAAG,cAAc,CAAC;YACtC,KAAK,cAAc,CAAC,IAAI,CAAC,SAAS,EAAE,GAAG,EAAE;gBACvC,IAAI,IAAI,CAAC,eAAe,KAAK,cAAc;oBAAE,IAAI,CAAC,eAAe,GAAG,SAAS,CAAC;YAChF,CAAC,CAAC,CAAC;QACL,CAAC;QACD,OAAO,IAAI,CAAC,eAAe,CAAC;IAC9B,CAAC;IAED,KAAK,CAAC,YAAY,CAAI,SAAkC;QACtD,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC;QACxC,OAAO,SAAS,CAAC,KAAK,CAAC,QAAQ,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;IAClD,CAAC;IAED,KAAK,CAAC,KAAK;QACT,IAAI,IAAI,CAAC,aAAa,KAAK,SAAS;YAAE,OAAO,IAAI,CAAC,aAAa,CAAC;QAChE,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;QACrB,MAAM,OAAO,GAAG,CAAC,KAAK,IAAI,EAAE;YAC1B,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,IAAI,MAAM,IAAI,CAAC,eAAe,CAAC;YACxD,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;gBACxB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;gBACpB,OAAO;YACT,CAAC;YACD,MAAM,MAAM,GAAc,EAAE,CAAC;YAC7B,IAAI,CAAC;gBACH,MAAM,KAAK,CAAC,MAAM,EAAE,KAAK,EAAE,CAAC;YAC9B,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACrB,CAAC;YACD,IAAI,KAAK,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;gBAC/B,IAAI,CAAC;oBACH,MAAM,KAAK,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;gBAC9B,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;gBACrB,CAAC;YACH,CAAC;YACD,IAAI,CAAC;gBACH,MAAM,KAAK,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;YAC5B,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACrB,CAAC;YACD,IAAI,CAAC;gBACH,KAAK,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;YACzB,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACrB,CAAC;YACD,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;YACpB,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC;gBAAE,MAAM,MAAM,CAAC,CAAC,CAAC,CAAC;YACzC,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC;gBAAE,MAAM,IAAI,cAAc,CAAC,MAAM,EAAE,4BAA4B,CAAC,CAAC;QACxF,CAAC,CAAC,EAAE,CAAC;QACL,IAAI,CAAC,aAAa,GAAG,OAAO,CAAC;QAC7B,OAAO,OAAO,CAAC;IACjB,CAAC;CACF"}

package/dist/mcp/server.d.ts

@@ -0,0 +1,24 @@
+import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
+import { type InitOptions } from '../commands/init.js';
+import type { SqliteDatabase } from '../db/adapter.js';
+import { openConnection } from '../db/connection.js';
+import type { EmbeddingProvider, EmbeddingRuntime, VectorSearchBackend } from '../embedding/types.js';
+import { type McpDeadlinePolicyOverrides } from './request-deadline.js';
+import { type McpDatabaseOwner } from './runtime-owner.js';
+export interface McpServerDependencies {
+    databasePath?: string;
+    migrationsDirectory?: string;
+    cwd?: () => string;
+    openConnection?: typeof openConnection;
+    initializeDatabase?: (options: InitOptions) => unknown | PromiseLike<unknown>;
+    fetchImpl?: typeof fetch;
+    embeddingEnvironment?: NodeJS.ProcessEnv;
+    embeddingProvider?: EmbeddingProvider;
+    embeddingBackend?: VectorSearchBackend;
+    databaseOwner?: McpDatabaseOwner;
+    deadlinePolicy?: McpDeadlinePolicyOverrides;
+}
+export declare function withDatabase<T>(dependencies: McpServerDependencies, operation: (database: SqliteDatabase, runtime?: EmbeddingRuntime) => Promise<T> | T): Promise<T>;
+export declare function createKiokukoMcpServer(dependencies?: McpServerDependencies): McpServer;
+export declare function runMcpServer(dependencies?: McpServerDependencies): Promise<void>;
+//# sourceMappingURL=server.d.ts.map

package/dist/mcp/server.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"server.d.ts","sourceRoot":"","sources":["../../src/mcp/server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAMpE,OAAO,EAAsB,KAAK,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAE3E,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AACvD,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AACrD,OAAO,KAAK,EAAE,iBAAiB,EAAE,gBAAgB,EAAE,mBAAmB,EAAE,MAAM,uBAAuB,CAAC;AAkBtG,OAAO,EAML,KAAK,0BAA0B,EAEhC,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EAAmB,KAAK,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AAE5E,MAAM,WAAW,qBAAqB;IACpC,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,mBAAmB,CAAC,EAAE,MAAM,CAAC;IAC7B,GAAG,CAAC,EAAE,MAAM,MAAM,CAAC;IACnB,cAAc,CAAC,EAAE,OAAO,cAAc,CAAC;IACvC,kBAAkB,CAAC,EAAE,CAAC,OAAO,EAAE,WAAW,KAAK,OAAO,GAAG,WAAW,CAAC,OAAO,CAAC,CAAC;IAC9E,SAAS,CAAC,EAAE,OAAO,KAAK,CAAC;IACzB,oBAAoB,CAAC,EAAE,MAAM,CAAC,UAAU,CAAC;IACzC,iBAAiB,CAAC,EAAE,iBAAiB,CAAC;IACtC,gBAAgB,CAAC,EAAE,mBAAmB,CAAC;IACvC,aAAa,CAAC,EAAE,gBAAgB,CAAC;IACjC,cAAc,CAAC,EAAE,0BAA0B,CAAC;CAC7C;AAED,wBAAsB,YAAY,CAAC,CAAC,EAClC,YAAY,EAAE,qBAAqB,EACnC,SAAS,EAAE,CAAC,QAAQ,EAAE,cAAc,EAAE,OAAO,CAAC,EAAE,gBAAgB,KAAK,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,GAClF,OAAO,CAAC,CAAC,CAAC,CAoCZ;AAkOD,wBAAgB,sBAAsB,CAAC,YAAY,GAAE,qBAA0B,GAAG,SAAS,CAuH1F;AAED,wBAAsB,YAAY,CAAC,YAAY,GAAE,qBAA0B,GAAG,OAAO,CAAC,IAAI,CAAC,CA2B1F"}