@hanhnd/agent-kit 1.0.36 → 1.0.37

package/dist/cli.js

@@ -15019,6 +15019,11 @@ import * as os from "os";
 import * as path3 from "path";
 
 // src/services/memory/constants.ts
+var EMBEDDING_MODEL_DIMENSIONS = {
+  ["tiny" /* TINY */]: 384,
+  ["base" /* BASE */]: 384,
+  ["large" /* LARGE */]: 768
+};
 var DEFAULT_MEMORY_CONFIG = {
   enabled: false,
   wikiDir: "",
@@ -15026,7 +15031,7 @@ var DEFAULT_MEMORY_CONFIG = {
   chunkSize: 1500,
   overlapLines: 2,
   embeddingModel: "base" /* BASE */,
-  vectorDimension: 384
+  vectorDimension: EMBEDDING_MODEL_DIMENSIONS["base" /* BASE */]
 };
 var LOCK_RETRY_MS = 50;
 var LOCK_TIMEOUT_MS = 500;
@@ -15035,13 +15040,20 @@ var RRF_K = 60;
 // src/core/config/index.ts
 function resolveMemoryConfig(settings, workspaceRoot) {
   const override = settings.memory ?? {};
+  const embeddingModel = isEmbeddingModelName(override.embeddingModel) ? override.embeddingModel : DEFAULT_MEMORY_CONFIG.embeddingModel;
+  const vectorDimension = typeof override.vectorDimension === "number" && Number.isFinite(override.vectorDimension) && override.vectorDimension > 0 ? override.vectorDimension : EMBEDDING_MODEL_DIMENSIONS[embeddingModel];
   return {
     ...DEFAULT_MEMORY_CONFIG,
     ...override,
     enabled: override.enabled ?? DEFAULT_MEMORY_CONFIG.enabled,
+    embeddingModel,
+    vectorDimension,
     wikiDir: typeof override.wikiDir === "string" && override.wikiDir.length > 0 ? override.wikiDir : path3.join(workspaceRoot, ".agent-kit", "wiki")
   };
 }
+function isEmbeddingModelName(value) {
+  return value === "tiny" /* TINY */ || value === "base" /* BASE */ || value === "large" /* LARGE */;
+}
 function getGlobalSettingsFile() {
   return path3.join(os.homedir(), ".agent-kit", "settings.json");
 }
@@ -15313,7 +15325,7 @@ var import_stopword = __toESM(require_stopword_cjs(), 1);
 import Database from "libsql";
 import * as fs5 from "fs";
 import * as path7 from "path";
-var SCHEMA_VERSION = 3;
+var SCHEMA_VERSION = 4;
 var LibsqlDatabase = Database;
 var StoreError = class extends Error {
   constructor(message, cause) {
@@ -15356,7 +15368,8 @@ var MemoryStore = class {
   createSchema(db) {
     try {
       const current = this.readUserVersion(db);
-      if (current < SCHEMA_VERSION) {
+      const currentVectorDimension = current >= SCHEMA_VERSION ? this.readEmbeddingColumnDimension(db) : void 0;
+      if (current < SCHEMA_VERSION || currentVectorDimension !== void 0 && currentVectorDimension !== this.config.vectorDimension) {
         db.exec(`
           DROP TRIGGER IF EXISTS memory_chunks_ai;
           DROP TRIGGER IF EXISTS memory_chunks_ad;
@@ -15413,6 +15426,22 @@ var MemoryStore = class {
       throw new StoreError("schema migration failed", err);
     }
   }
+  readEmbeddingColumnDimension(db) {
+    try {
+      const rows = db.pragma("table_info(memory_chunks)");
+      if (!Array.isArray(rows)) return void 0;
+      const embeddingColumn = rows.find((row) => {
+        if (!row || typeof row !== "object") return false;
+        const candidate = row;
+        return candidate.name === "embedding" && typeof candidate.type === "string";
+      });
+      const match = /^F32_BLOB\((\d+)\)$/i.exec(embeddingColumn?.type.trim() ?? "");
+      if (!match) return void 0;
+      return Number.parseInt(match[1], 10);
+    } catch {
+      return void 0;
+    }
+  }
   readUserVersion(db) {
     const value = db.pragma("user_version", { simple: true });
     if (typeof value === "number") return value;

package/dist/core/config/index.d.ts

@@ -2,7 +2,7 @@
  * Configuration utilities
  * Shared configuration constants and loaders
  */
-import type { MemoryConfig } from '../../services/memory/types.js';
+import { type MemoryConfig } from '../../services/memory/types.js';
 import type { ConversationDigestSettings } from '../../services/digest/types.js';
 /**
  * Default file extensions - expanded to support more languages

package/dist/core/config/index.js

@@ -5,7 +5,8 @@
 import * as fs from 'fs';
 import * as os from 'os';
 import * as path from 'path';
-import { DEFAULT_MEMORY_CONFIG } from '../../services/memory/constants.js';
+import { DEFAULT_MEMORY_CONFIG, EMBEDDING_MODEL_DIMENSIONS } from '../../services/memory/constants.js';
+import { EmbeddingModelName } from '../../services/memory/types.js';
 import { atomicWriteJsonFile } from '../../utils/files.js';
 /**
  * Default file extensions - expanded to support more languages
@@ -40,15 +41,28 @@ export const DEFAULT_EXTENSIONS = [
  */
 export function resolveMemoryConfig(settings, workspaceRoot) {
     const override = settings.memory ?? {};
+    const embeddingModel = isEmbeddingModelName(override.embeddingModel)
+        ? override.embeddingModel
+        : DEFAULT_MEMORY_CONFIG.embeddingModel;
+    const vectorDimension = typeof override.vectorDimension === 'number' &&
+        Number.isFinite(override.vectorDimension) &&
+        override.vectorDimension > 0
+        ? override.vectorDimension
+        : EMBEDDING_MODEL_DIMENSIONS[embeddingModel];
     return {
         ...DEFAULT_MEMORY_CONFIG,
         ...override,
         enabled: override.enabled ?? DEFAULT_MEMORY_CONFIG.enabled,
+        embeddingModel,
+        vectorDimension,
         wikiDir: typeof override.wikiDir === 'string' && override.wikiDir.length > 0
             ? override.wikiDir
             : path.join(workspaceRoot, '.agent-kit', 'wiki'),
     };
 }
+function isEmbeddingModelName(value) {
+    return value === EmbeddingModelName.TINY || value === EmbeddingModelName.BASE || value === EmbeddingModelName.LARGE;
+}
 export function loadProjectSettings(projectDir) {
     const settingsPath = path.join(projectDir, '.agent-kit', 'settings.json');
     if (fs.existsSync(settingsPath)) {

package/dist/server.js

@@ -35903,6 +35903,11 @@ import * as path2 from "path";
 var SOURCE_TYPES = ["digest", "concept", "entity", "preference", "wiki"];
 
 // src/services/memory/constants.ts
+var EMBEDDING_MODEL_DIMENSIONS = {
+  ["tiny" /* TINY */]: 384,
+  ["base" /* BASE */]: 384,
+  ["large" /* LARGE */]: 768
+};
 var DEFAULT_MEMORY_CONFIG = {
   enabled: false,
   wikiDir: "",
@@ -35910,7 +35915,7 @@ var DEFAULT_MEMORY_CONFIG = {
   chunkSize: 1500,
   overlapLines: 2,
   embeddingModel: "base" /* BASE */,
-  vectorDimension: 384
+  vectorDimension: EMBEDDING_MODEL_DIMENSIONS["base" /* BASE */]
 };
 var LOCK_RETRY_MS = 50;
 var LOCK_TIMEOUT_MS = 500;
@@ -35985,13 +35990,20 @@ function tryAcquireProcessLock(lockPath) {
 // src/core/config/index.ts
 function resolveMemoryConfig(settings, workspaceRoot2) {
   const override = settings.memory ?? {};
+  const embeddingModel = isEmbeddingModelName(override.embeddingModel) ? override.embeddingModel : DEFAULT_MEMORY_CONFIG.embeddingModel;
+  const vectorDimension = typeof override.vectorDimension === "number" && Number.isFinite(override.vectorDimension) && override.vectorDimension > 0 ? override.vectorDimension : EMBEDDING_MODEL_DIMENSIONS[embeddingModel];
   return {
     ...DEFAULT_MEMORY_CONFIG,
     ...override,
     enabled: override.enabled ?? DEFAULT_MEMORY_CONFIG.enabled,
+    embeddingModel,
+    vectorDimension,
     wikiDir: typeof override.wikiDir === "string" && override.wikiDir.length > 0 ? override.wikiDir : path2.join(workspaceRoot2, ".agent-kit", "wiki")
   };
 }
+function isEmbeddingModelName(value) {
+  return value === "tiny" /* TINY */ || value === "base" /* BASE */ || value === "large" /* LARGE */;
+}
 function getGlobalSettingsFile() {
   return path2.join(os.homedir(), ".agent-kit", "settings.json");
 }
@@ -37178,7 +37190,7 @@ var import_stopword = __toESM(require_stopword_cjs(), 1);
 import Database from "libsql";
 import * as fs7 from "fs";
 import * as path9 from "path";
-var SCHEMA_VERSION = 3;
+var SCHEMA_VERSION = 4;
 var LibsqlDatabase = Database;
 var StoreError = class extends Error {
   constructor(message, cause) {
@@ -37221,7 +37233,8 @@ var MemoryStore = class {
   createSchema(db) {
     try {
       const current = this.readUserVersion(db);
-      if (current < SCHEMA_VERSION) {
+      const currentVectorDimension = current >= SCHEMA_VERSION ? this.readEmbeddingColumnDimension(db) : void 0;
+      if (current < SCHEMA_VERSION || currentVectorDimension !== void 0 && currentVectorDimension !== this.config.vectorDimension) {
         db.exec(`
           DROP TRIGGER IF EXISTS memory_chunks_ai;
           DROP TRIGGER IF EXISTS memory_chunks_ad;
@@ -37278,6 +37291,22 @@ var MemoryStore = class {
       throw new StoreError("schema migration failed", err);
     }
   }
+  readEmbeddingColumnDimension(db) {
+    try {
+      const rows = db.pragma("table_info(memory_chunks)");
+      if (!Array.isArray(rows)) return void 0;
+      const embeddingColumn = rows.find((row) => {
+        if (!row || typeof row !== "object") return false;
+        const candidate = row;
+        return candidate.name === "embedding" && typeof candidate.type === "string";
+      });
+      const match = /^F32_BLOB\((\d+)\)$/i.exec(embeddingColumn?.type.trim() ?? "");
+      if (!match) return void 0;
+      return Number.parseInt(match[1], 10);
+    } catch {
+      return void 0;
+    }
+  }
   readUserVersion(db) {
     const value = db.pragma("user_version", { simple: true });
     if (typeof value === "number") return value;

package/dist/services/memory/constants.d.ts

@@ -1,4 +1,5 @@
-import { type MemoryConfig } from './types.js';
+import { type MemoryConfig, EmbeddingModelName } from './types.js';
+export declare const EMBEDDING_MODEL_DIMENSIONS: Record<EmbeddingModelName, number>;
 export declare const DEFAULT_MEMORY_CONFIG: MemoryConfig;
 export declare const LOCK_RETRY_MS = 50;
 export declare const LOCK_TIMEOUT_MS = 500;

package/dist/services/memory/constants.js

@@ -1,4 +1,9 @@
 import { EmbeddingModelName } from './types.js';
+export const EMBEDDING_MODEL_DIMENSIONS = {
+    [EmbeddingModelName.TINY]: 384,
+    [EmbeddingModelName.BASE]: 384,
+    [EmbeddingModelName.LARGE]: 768,
+};
 export const DEFAULT_MEMORY_CONFIG = {
     enabled: false,
     wikiDir: '',
@@ -6,7 +11,7 @@ export const DEFAULT_MEMORY_CONFIG = {
     chunkSize: 1500,
     overlapLines: 2,
     embeddingModel: EmbeddingModelName.BASE,
-    vectorDimension: 384,
+    vectorDimension: EMBEDDING_MODEL_DIMENSIONS[EmbeddingModelName.BASE],
 };
 export const LOCK_RETRY_MS = 50;
 export const LOCK_TIMEOUT_MS = 500;

package/dist/services/memory/memory.test.js

@@ -402,4 +402,14 @@ describe('resolveMemoryConfig', () => {
         const config = resolveMemoryConfig({ memory: { wikiDir: '/custom/wiki' } }, '/repo');
         assert.equal(config.wikiDir, '/custom/wiki');
     });
+    test('derives vector dimension from embedding model', () => {
+        const config = resolveMemoryConfig({ memory: { embeddingModel: EmbeddingModelName.LARGE } }, '/repo');
+        assert.equal(config.embeddingModel, EmbeddingModelName.LARGE);
+        assert.equal(config.vectorDimension, 768);
+    });
+    test('preserves explicit vector dimension override', () => {
+        const config = resolveMemoryConfig({ memory: { embeddingModel: EmbeddingModelName.LARGE, vectorDimension: 1024 } }, '/repo');
+        assert.equal(config.embeddingModel, EmbeddingModelName.LARGE);
+        assert.equal(config.vectorDimension, 1024);
+    });
 });

package/dist/services/memory/store.d.ts

@@ -1,5 +1,5 @@
 import type { MemoryChunk, MemoryConfig, RecentSource, SourceType } from './types.js';
-export declare const SCHEMA_VERSION = 3;
+export declare const SCHEMA_VERSION = 4;
 export declare class StoreError extends Error {
     constructor(message: string, cause?: unknown);
 }
@@ -9,6 +9,7 @@ export declare class MemoryStore {
     constructor(dbPath: string, config: MemoryConfig);
     private openDatabase;
     private createSchema;
+    private readEmbeddingColumnDimension;
     private readUserVersion;
     upsert(chunks: MemoryChunk[], embeddings: Float32Array[]): void;
     private toVectorBinding;

package/dist/services/memory/store.js

@@ -2,7 +2,7 @@ import Database from 'libsql';
 import * as fs from 'fs';
 import * as path from 'path';
 import { eng, removeStopwords } from 'stopword';
-export const SCHEMA_VERSION = 3;
+export const SCHEMA_VERSION = 4;
 const LibsqlDatabase = Database;
 export class StoreError extends Error {
     constructor(message, cause) {
@@ -51,7 +51,9 @@ export class MemoryStore {
     createSchema(db) {
         try {
             const current = this.readUserVersion(db);
-            if (current < SCHEMA_VERSION) {
+            const currentVectorDimension = current >= SCHEMA_VERSION ? this.readEmbeddingColumnDimension(db) : undefined;
+            if (current < SCHEMA_VERSION ||
+                (currentVectorDimension !== undefined && currentVectorDimension !== this.config.vectorDimension)) {
                 db.exec(`
           DROP TRIGGER IF EXISTS memory_chunks_ai;
           DROP TRIGGER IF EXISTS memory_chunks_ad;
@@ -109,6 +111,26 @@ export class MemoryStore {
             throw new StoreError('schema migration failed', err);
         }
     }
+    readEmbeddingColumnDimension(db) {
+        try {
+            const rows = db.pragma('table_info(memory_chunks)');
+            if (!Array.isArray(rows))
+                return undefined;
+            const embeddingColumn = rows.find((row) => {
+                if (!row || typeof row !== 'object')
+                    return false;
+                const candidate = row;
+                return candidate.name === 'embedding' && typeof candidate.type === 'string';
+            });
+            const match = /^F32_BLOB\((\d+)\)$/i.exec(embeddingColumn?.type.trim() ?? '');
+            if (!match)
+                return undefined;
+            return Number.parseInt(match[1], 10);
+        }
+        catch {
+            return undefined;
+        }
+    }
     readUserVersion(db) {
         const value = db.pragma('user_version', { simple: true });
         if (typeof value === 'number')

package/dist/services/memory/store.test.js

@@ -387,4 +387,47 @@ describe('MemoryStore', () => {
             fs.rmSync(migrationDir, { recursive: true, force: true });
         }
     });
+    test('recreates schema when existing vector dimension differs from config', () => {
+        const dimensionDir = fs.mkdtempSync(path.join(os.tmpdir(), 'memory-dimension-migration-'));
+        const dimensionDbPath = path.join(dimensionDir, 'index.db');
+        const baseStore = new MemoryStore(dimensionDbPath, { ...TEST_CONFIG, wikiDir: dimensionDir });
+        try {
+            baseStore.upsert([makeChunk({ id: 'old-dimension-row', source: 'dimension.md' })], [makeEmbedding(17)]);
+        }
+        finally {
+            baseStore.close();
+        }
+        const largeConfig = {
+            ...TEST_CONFIG,
+            wikiDir: dimensionDir,
+            embeddingModel: EmbeddingModelName.LARGE,
+            vectorDimension: 768,
+        };
+        const largeStore = new MemoryStore(dimensionDbPath, largeConfig);
+        try {
+            const verifyDb = new Database(dimensionDbPath, { readonly: true });
+            try {
+                const columns = verifyDb.prepare(`PRAGMA table_info(memory_chunks)`).all();
+                const embeddingColumn = columns.find((column) => column.name === 'embedding');
+                assert.equal(embeddingColumn?.type, 'F32_BLOB(768)');
+                const rowCount = verifyDb.prepare(`SELECT COUNT(*) AS count FROM memory_chunks`).get();
+                assert.equal(rowCount.count, 0);
+            }
+            finally {
+                verifyDb.close();
+            }
+            const chunk = makeChunk({
+                id: 'new-dimension-row',
+                source: 'dimension.md',
+                content: 'large dimension vector can be indexed',
+            });
+            largeStore.upsert([chunk], [makeEmbedding(18, 768)]);
+            const results = largeStore.searchBm25('large dimension vector', 5);
+            assert.ok(results.some((result) => result.id === 'new-dimension-row'));
+        }
+        finally {
+            largeStore.close();
+            fs.rmSync(dimensionDir, { recursive: true, force: true });
+        }
+    });
 });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@hanhnd/agent-kit",
-  "version": "1.0.36",
+  "version": "1.0.37",
   "description": "Agent Kit MCP server for software development agent workflows",
   "type": "module",
   "main": "dist/server.js",