@jungjaehoon/mama-core 1.2.0 → 1.2.1

package/dist/config-loader.d.ts

@@ -15,6 +15,7 @@
 export interface MAMAConfig {
     modelName: string;
     embeddingDim: number;
+    quantized: boolean;
     cacheDir: string;
     [key: string]: unknown;
 }
@@ -36,6 +37,10 @@ export declare function getModelName(): string;
  * Get current embedding dimension
  */
 export declare function getEmbeddingDim(): number;
+/**
+ * Get whether to use quantized model
+ */
+export declare function getQuantized(): boolean;
 /**
  * Get current cache directory
  */
@@ -43,6 +48,7 @@ export declare function getCacheDir(): string;
 export interface ConfigUpdates {
     modelName?: string;
     embeddingDim?: number;
+    quantized?: boolean;
     cacheDir?: string;
 }
 /**

package/dist/config-loader.js

@@ -21,6 +21,7 @@ exports.DEFAULT_CONFIG = void 0;
 exports.loadConfig = loadConfig;
 exports.getModelName = getModelName;
 exports.getEmbeddingDim = getEmbeddingDim;
+exports.getQuantized = getQuantized;
 exports.getCacheDir = getCacheDir;
 exports.updateConfig = updateConfig;
 exports.getConfigPath = getConfigPath;
@@ -32,6 +33,7 @@ const debug_logger_js_1 = require("./debug-logger.js");
 exports.DEFAULT_CONFIG = {
     modelName: 'Xenova/multilingual-e5-small',
     embeddingDim: 384,
+    quantized: true,
     cacheDir: path_1.default.join(os_1.default.homedir(), '.cache', 'huggingface', 'transformers'),
 };
 // Config file path
@@ -129,6 +131,13 @@ function getEmbeddingDim() {
     const config = loadConfig();
     return config.embeddingDim;
 }
+/**
+ * Get whether to use quantized model
+ */
+function getQuantized() {
+    const config = loadConfig();
+    return config.quantized;
+}
 /**
  * Get current cache directory
  */

package/dist/embeddings.d.ts

@@ -10,7 +10,7 @@
  * @date 2025-11-20
  */
 import { embeddingCache } from './embedding-cache.js';
-import { loadConfig, getModelName, getEmbeddingDim } from './config-loader.js';
+import { loadConfig, getModelName, getEmbeddingDim, getQuantized } from './config-loader.js';
 /**
  * Decision object for enhanced embedding generation
  */
@@ -71,5 +71,5 @@ export declare function cosineSimilarity(embA: Float32Array, embB: Float32Array)
 export { embeddingCache };
 export declare const EMBEDDING_DIM: number;
 export declare const MODEL_NAME: string;
-export { loadConfig, getModelName, getEmbeddingDim };
+export { loadConfig, getModelName, getEmbeddingDim, getQuantized };
 //# sourceMappingURL=embeddings.d.ts.map

package/dist/embeddings.js

@@ -14,7 +14,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.getEmbeddingDim = exports.getModelName = exports.loadConfig = exports.MODEL_NAME = exports.EMBEDDING_DIM = exports.embeddingCache = void 0;
+exports.getQuantized = exports.getEmbeddingDim = exports.getModelName = exports.loadConfig = exports.MODEL_NAME = exports.EMBEDDING_DIM = exports.embeddingCache = void 0;
 exports.generateEmbedding = generateEmbedding;
 exports.generateEnhancedEmbedding = generateEnhancedEmbedding;
 exports.generateBatchEmbeddings = generateBatchEmbeddings;
@@ -29,6 +29,7 @@ const config_loader_js_1 = require("./config-loader.js");
 Object.defineProperty(exports, "loadConfig", { enumerable: true, get: function () { return config_loader_js_1.loadConfig; } });
 Object.defineProperty(exports, "getModelName", { enumerable: true, get: function () { return config_loader_js_1.getModelName; } });
 Object.defineProperty(exports, "getEmbeddingDim", { enumerable: true, get: function () { return config_loader_js_1.getEmbeddingDim; } });
+Object.defineProperty(exports, "getQuantized", { enumerable: true, get: function () { return config_loader_js_1.getQuantized; } });
 // Shared cache directory (not in node_modules)
 const DEFAULT_CACHE_DIR = path_1.default.join(os_1.default.homedir(), '.cache', 'huggingface', 'transformers');
 // Singleton pattern for model loading
@@ -73,11 +74,14 @@ async function loadModel() {
             const cacheDir = process.env.HF_HOME || process.env.TRANSFORMERS_CACHE || DEFAULT_CACHE_DIR;
             env.cacheDir = cacheDir;
             (0, debug_logger_js_1.info)(`[MAMA] Model cache directory: ${cacheDir}`);
-            embeddingPipeline = (await pipeline('feature-extraction', modelName));
+            const quantized = (0, config_loader_js_1.getQuantized)();
+            embeddingPipeline = (await pipeline('feature-extraction', modelName, {
+                dtype: quantized ? 'q8' : 'fp32',
+            }));
             currentModelName = modelName;
             const loadTime = Date.now() - startTime;
             const config = (0, config_loader_js_1.loadConfig)();
-            (0, progress_indicator_js_1.logComplete)(`Embedding model ready (${loadTime}ms, ${config.embeddingDim}-dim)`);
+            (0, progress_indicator_js_1.logComplete)(`Embedding model ready (${loadTime}ms, ${config.embeddingDim}-dim, ${quantized ? 'q8' : 'fp32'})`);
         }
         catch (loadErr) {
             modelLoadFailed = true;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@jungjaehoon/mama-core",
-  "version": "1.2.0",
+  "version": "1.2.1",
   "description": "MAMA Core - Shared modules for Memory-Augmented MCP Assistant",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",