@hanhnd/agent-kit 1.0.31 → 1.0.32

package/dist/__tests__/temp-dir.d.ts

@@ -0,0 +1,4 @@
+export declare function createTempDirTracker(): {
+    makeTempDir(prefix: string): string;
+    cleanup(): void;
+};

package/dist/__tests__/temp-dir.js

@@ -0,0 +1,20 @@
+import * as fs from 'node:fs';
+import * as os from 'node:os';
+import * as path from 'node:path';
+export function createTempDirTracker() {
+    const tempDirs = [];
+    return {
+        makeTempDir(prefix) {
+            const tempDir = fs.mkdtempSync(path.join(os.tmpdir(), prefix));
+            tempDirs.push(tempDir);
+            return tempDir;
+        },
+        cleanup() {
+            while (tempDirs.length > 0) {
+                const tempDir = tempDirs.pop();
+                if (tempDir)
+                    fs.rmSync(tempDir, { recursive: true, force: true });
+            }
+        },
+    };
+}

package/dist/cli/index.d.ts

@@ -0,0 +1 @@
+export declare function runAgentKitCli(argv: string[], env: NodeJS.ProcessEnv): Promise<number>;

package/dist/cli/index.js

@@ -0,0 +1,9 @@
+import { runMemoryCli } from './memory.js';
+export async function runAgentKitCli(argv, env) {
+    const [command, ...rest] = argv;
+    if (command === 'memory') {
+        return runMemoryCli(rest, env);
+    }
+    process.stderr.write('Usage: agent-kit-cli memory <digest-file|digest-init|digest-pending>\n');
+    return 1;
+}

package/dist/cli/memory.d.ts

@@ -0,0 +1 @@
+export declare function runMemoryCli(args: string[], _env: NodeJS.ProcessEnv): Promise<number>;

package/dist/cli/memory.js

@@ -0,0 +1,77 @@
+import { initializeConversationDigestModel, digestConversationFile, digestPendingConversations, } from '../memory/digest/processor.js';
+import { isDigestWorkerInvocation, runDigestFileInWorker } from '../memory/digest/worker.js';
+import { DEFAULT_DIGEST_MAX_INPUT_CHARS, DEFAULT_DIGEST_MODEL_ID, DEFAULT_DIGEST_TIMEOUT_MS, DIGEST_WORKER_RESULT_PREFIX, } from '../memory/digest/constants.js';
+import { loadProjectSettings, resolveConversationDigestConfig } from '../tools/config.js';
+import { parseArgs, stringFlag, numberFlag } from '../utils/args.js';
+import { getWorkspaceRoot } from '../utils/utils.js';
+function writeDigestFileResult(result, workerMode) {
+    if (workerMode) {
+        process.stdout.write(DIGEST_WORKER_RESULT_PREFIX + JSON.stringify(result) + '\n');
+        return;
+    }
+    process.stdout.write(JSON.stringify(result, null, 2) + '\n');
+}
+async function cmdDigestInit(args) {
+    const parsed = parseArgs(args);
+    const modelId = stringFlag(parsed.flags, 'model') ?? DEFAULT_DIGEST_MODEL_ID;
+    const result = await initializeConversationDigestModel({
+        workspaceRoot: getWorkspaceRoot(),
+        modelId,
+        allowDownload: true,
+    });
+    process.stdout.write(JSON.stringify(result, null, 2) + '\n');
+    return result.initialized ? 0 : 1;
+}
+async function cmdDigestFile(args) {
+    const workerMode = isDigestWorkerInvocation(args);
+    const workerResult = runDigestFileInWorker(args);
+    if (workerResult !== undefined)
+        return workerResult;
+    const parsed = parseArgs(args);
+    const workspaceRoot = getWorkspaceRoot();
+    const digestConfig = resolveConversationDigestConfig(loadProjectSettings(workspaceRoot));
+    const modelId = stringFlag(parsed.flags, 'model') ?? digestConfig?.modelId ?? DEFAULT_DIGEST_MODEL_ID;
+    const inputPath = stringFlag(parsed.flags, 'input');
+    if (!inputPath) {
+        process.stderr.write('Usage: agent-kit-cli memory digest-file --input <path> --model <id>\n');
+        return 1;
+    }
+    const result = await digestConversationFile({
+        workspaceRoot,
+        inputPath,
+        modelId,
+        outDir: stringFlag(parsed.flags, 'out'),
+        maxInputChars: numberFlag(parsed.flags, 'max-input-chars', DEFAULT_DIGEST_MAX_INPUT_CHARS),
+        timeoutMs: numberFlag(parsed.flags, 'timeout-ms', DEFAULT_DIGEST_TIMEOUT_MS),
+    });
+    writeDigestFileResult(result, workerMode);
+    return 0;
+}
+async function cmdDigestPending(args) {
+    const parsed = parseArgs(args);
+    if (parsed.flags.get('hook') !== true) {
+        process.stderr.write('Usage: agent-kit-cli memory digest-pending --hook\n');
+        return 1;
+    }
+    const result = await digestPendingConversations({ workspaceRoot: getWorkspaceRoot() });
+    process.stdout.write(JSON.stringify(result) + '\n');
+    return 0;
+}
+export async function runMemoryCli(args, _env) {
+    const [command, ...rest] = args;
+    try {
+        if (command === 'digest-file')
+            return await cmdDigestFile(rest);
+        if (command === 'digest-init')
+            return await cmdDigestInit(rest);
+        if (command === 'digest-pending')
+            return await cmdDigestPending(rest);
+    }
+    catch (err) {
+        const message = err instanceof Error ? err.message : String(err);
+        process.stderr.write('[agent-kit] memory ' + (command ?? '') + ' failed: ' + message + '\n');
+        return 1;
+    }
+    process.stderr.write('Usage: agent-kit-cli memory <digest-file|digest-init|digest-pending>\n');
+    return 1;
+}

package/dist/kit-cli.d.ts

@@ -0,0 +1,2 @@
+#!/usr/bin/env node
+export {};