@mastra/fastembed 0.10.1 → 0.10.2

package/dist/index.cjs

@@ -63,3 +63,5 @@ var fastembed = Object.assign(fastEmbedProvider.textEmbeddingModel(`bge-small-en
 });
 
 exports.fastembed = fastembed;
+//# sourceMappingURL=index.cjs.map
+//# sourceMappingURL=index.cjs.map

package/dist/index.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/index.ts"],"names":["path","os","fsp","FlagEmbedding","EmbeddingModel","experimental_customProvider"],"mappings":";;;;;;;;;;;;;;;AAMA,eAAe,iBAAA,GAAoB;AACjC,EAAA,MAAM,SAAA,GAAYA,sBAAK,IAAA,CAAKC,mBAAA,CAAG,SAAQ,EAAG,QAAA,EAAU,UAAU,kBAAkB,CAAA;AAChF,EAAA,MAAMC,qBAAI,KAAA,CAAM,SAAA,EAAW,EAAE,SAAA,EAAW,MAAM,CAAA;AAC9C,EAAA,OAAO,SAAA;AACT;AAGA,eAAe,kBAAA,CAAmB,QAAkB,SAAA,EAA6C;AAC/F,EAAA,MAAM,KAAA,GAAQ,MAAMC,yBAAA,CAAc,IAAA,CAAK;AAAA,IACrC,KAAA,EAAOC,2BAAe,SAAS,CAAA;AAAA,IAC/B,QAAA,EAAU,MAAM,iBAAA;AAAkB,GACnC,CAAA;AAGD,EAAA,MAAM,UAAA,GAAa,KAAA,CAAM,KAAA,CAAM,MAAM,CAAA;AAErC,EAAA,MAAM,aAAa,EAAC;AACpB,EAAA,WAAA,MAAiB,UAAU,UAAA,EAAY;AAGrC,IAAA,UAAA,CAAW,IAAA,CAAK,GAAG,MAAA,CAAO,GAAA,CAAI,eAAa,KAAA,CAAM,IAAA,CAAK,SAAS,CAAC,CAAC,CAAA;AAAA,EACnE;AAEA,EAAA,IAAI,WAAW,MAAA,KAAW,CAAA,EAAG,MAAM,IAAI,MAAM,yBAAyB,CAAA;AAEtE,EAAA,OAAO;AAAA,IACL,UAAA,EAAY;AAAA,GACd;AACF;AAEA,IAAM,oBAAoBC,8BAAA,CAA4B;AAAA,EACpD,mBAAA,EAAqB;AAAA,IACnB,mBAAA,EAAqB;AAAA,MACnB,oBAAA,EAAsB,IAAA;AAAA,MACtB,QAAA,EAAU,WAAA;AAAA,MACV,OAAA,EAAS,mBAAA;AAAA,MACT,oBAAA,EAAsB,GAAA;AAAA,MACtB,qBAAA,EAAuB,IAAA;AAAA,MACvB,MAAM,OAAA,CAAQ,EAAE,MAAA,EAAO,EAAG;AACxB,QAAA,OAAO,kBAAA,CAAmB,QAAQ,eAAe,CAAA;AAAA,MACnD;AAAA,KACF;AAAA,IACA,kBAAA,EAAoB;AAAA,MAClB,oBAAA,EAAsB,IAAA;AAAA,MACtB,QAAA,EAAU,WAAA;AAAA,MACV,OAAA,EAAS,kBAAA;AAAA,MACT,oBAAA,EAAsB,GAAA;AAAA,MACtB,qBAAA,EAAuB,IAAA;AAAA,MACvB,MAAM,OAAA,CAAQ,EAAE,MAAA,EAAO,EAAG;AACxB,QAAA,OAAO,kBAAA,CAAmB,QAAQ,cAAc,CAAA;AAAA,MAClD;AAAA;AACF;AAEJ,CAAC,CAAA;AAEM,IAAM,YAAY,MAAA,CAAO,MAAA,CAAO,iBAAA,CAAkB,kBAAA,CAAmB,mBAAmB,CAAA,EAAG;AAAA,EAChG,KAAA,EAAO,iBAAA,CAAkB,kBAAA,CAAmB,CAAA,iBAAA,CAAmB,CAAA;AAAA,EAC/D,IAAA,EAAM,iBAAA,CAAkB,kBAAA,CAAmB,CAAA,gBAAA,CAAkB;AAC/D,CAAC","file":"index.cjs","sourcesContent":["import fsp from 'node:fs/promises';\nimport os from 'node:os';\nimport path from 'node:path';\nimport { experimental_customProvider } from 'ai';\nimport { FlagEmbedding, EmbeddingModel } from 'fastembed';\n\nasync function getModelCachePath() {\n  const cachePath = path.join(os.homedir(), '.cache', 'mastra', 'fastembed-models');\n  await fsp.mkdir(cachePath, { recursive: true });\n  return cachePath;\n}\n\n// Shared function to generate embeddings using fastembed\nasync function generateEmbeddings(values: string[], modelType: 'BGESmallENV15' | 'BGEBaseENV15') {\n  const model = await FlagEmbedding.init({\n    model: EmbeddingModel[modelType],\n    cacheDir: await getModelCachePath(),\n  });\n\n  // model.embed() returns an AsyncGenerator that processes texts in batches (default size 256)\n  const embeddings = model.embed(values);\n\n  const allResults = [];\n  for await (const result of embeddings) {\n    // result is an array of embeddings, one for each text in the batch\n    // We convert each Float32Array embedding to a regular number array\n    allResults.push(...result.map(embedding => Array.from(embedding)));\n  }\n\n  if (allResults.length === 0) throw new Error('No embeddings generated');\n\n  return {\n    embeddings: allResults,\n  };\n}\n\nconst fastEmbedProvider = experimental_customProvider({\n  textEmbeddingModels: {\n    'bge-small-en-v1.5': {\n      specificationVersion: 'v1',\n      provider: 'fastembed',\n      modelId: 'bge-small-en-v1.5',\n      maxEmbeddingsPerCall: 256,\n      supportsParallelCalls: true,\n      async doEmbed({ values }) {\n        return generateEmbeddings(values, 'BGESmallENV15');\n      },\n    },\n    'bge-base-en-v1.5': {\n      specificationVersion: 'v1',\n      provider: 'fastembed',\n      modelId: 'bge-base-en-v1.5',\n      maxEmbeddingsPerCall: 256,\n      supportsParallelCalls: true,\n      async doEmbed({ values }) {\n        return generateEmbeddings(values, 'BGEBaseENV15');\n      },\n    },\n  },\n});\n\nexport const fastembed = Object.assign(fastEmbedProvider.textEmbeddingModel(`bge-small-en-v1.5`), {\n  small: fastEmbedProvider.textEmbeddingModel(`bge-small-en-v1.5`),\n  base: fastEmbedProvider.textEmbeddingModel(`bge-base-en-v1.5`),\n});\n"]}

package/dist/index.d.ts

@@ -1 +1,5 @@
-export { fastembed } from './_tsup-dts-rollup.js';
+export declare const fastembed: import("ai").EmbeddingModel<string> & {
+    small: import("ai").EmbeddingModel<string>;
+    base: import("ai").EmbeddingModel<string>;
+};
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AA6DA,eAAO,MAAM,SAAS;;;CAGpB,CAAC"}

package/dist/index.js

@@ -55,3 +55,5 @@ var fastembed = Object.assign(fastEmbedProvider.textEmbeddingModel(`bge-small-en
 });
 
 export { fastembed };
+//# sourceMappingURL=index.js.map
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/index.ts"],"names":[],"mappings":";;;;;;;AAMA,eAAe,iBAAA,GAAoB;AACjC,EAAA,MAAM,SAAA,GAAY,KAAK,IAAA,CAAK,EAAA,CAAG,SAAQ,EAAG,QAAA,EAAU,UAAU,kBAAkB,CAAA;AAChF,EAAA,MAAM,IAAI,KAAA,CAAM,SAAA,EAAW,EAAE,SAAA,EAAW,MAAM,CAAA;AAC9C,EAAA,OAAO,SAAA;AACT;AAGA,eAAe,kBAAA,CAAmB,QAAkB,SAAA,EAA6C;AAC/F,EAAA,MAAM,KAAA,GAAQ,MAAM,aAAA,CAAc,IAAA,CAAK;AAAA,IACrC,KAAA,EAAO,eAAe,SAAS,CAAA;AAAA,IAC/B,QAAA,EAAU,MAAM,iBAAA;AAAkB,GACnC,CAAA;AAGD,EAAA,MAAM,UAAA,GAAa,KAAA,CAAM,KAAA,CAAM,MAAM,CAAA;AAErC,EAAA,MAAM,aAAa,EAAC;AACpB,EAAA,WAAA,MAAiB,UAAU,UAAA,EAAY;AAGrC,IAAA,UAAA,CAAW,IAAA,CAAK,GAAG,MAAA,CAAO,GAAA,CAAI,eAAa,KAAA,CAAM,IAAA,CAAK,SAAS,CAAC,CAAC,CAAA;AAAA,EACnE;AAEA,EAAA,IAAI,WAAW,MAAA,KAAW,CAAA,EAAG,MAAM,IAAI,MAAM,yBAAyB,CAAA;AAEtE,EAAA,OAAO;AAAA,IACL,UAAA,EAAY;AAAA,GACd;AACF;AAEA,IAAM,oBAAoB,2BAAA,CAA4B;AAAA,EACpD,mBAAA,EAAqB;AAAA,IACnB,mBAAA,EAAqB;AAAA,MACnB,oBAAA,EAAsB,IAAA;AAAA,MACtB,QAAA,EAAU,WAAA;AAAA,MACV,OAAA,EAAS,mBAAA;AAAA,MACT,oBAAA,EAAsB,GAAA;AAAA,MACtB,qBAAA,EAAuB,IAAA;AAAA,MACvB,MAAM,OAAA,CAAQ,EAAE,MAAA,EAAO,EAAG;AACxB,QAAA,OAAO,kBAAA,CAAmB,QAAQ,eAAe,CAAA;AAAA,MACnD;AAAA,KACF;AAAA,IACA,kBAAA,EAAoB;AAAA,MAClB,oBAAA,EAAsB,IAAA;AAAA,MACtB,QAAA,EAAU,WAAA;AAAA,MACV,OAAA,EAAS,kBAAA;AAAA,MACT,oBAAA,EAAsB,GAAA;AAAA,MACtB,qBAAA,EAAuB,IAAA;AAAA,MACvB,MAAM,OAAA,CAAQ,EAAE,MAAA,EAAO,EAAG;AACxB,QAAA,OAAO,kBAAA,CAAmB,QAAQ,cAAc,CAAA;AAAA,MAClD;AAAA;AACF;AAEJ,CAAC,CAAA;AAEM,IAAM,YAAY,MAAA,CAAO,MAAA,CAAO,iBAAA,CAAkB,kBAAA,CAAmB,mBAAmB,CAAA,EAAG;AAAA,EAChG,KAAA,EAAO,iBAAA,CAAkB,kBAAA,CAAmB,CAAA,iBAAA,CAAmB,CAAA;AAAA,EAC/D,IAAA,EAAM,iBAAA,CAAkB,kBAAA,CAAmB,CAAA,gBAAA,CAAkB;AAC/D,CAAC","file":"index.js","sourcesContent":["import fsp from 'node:fs/promises';\nimport os from 'node:os';\nimport path from 'node:path';\nimport { experimental_customProvider } from 'ai';\nimport { FlagEmbedding, EmbeddingModel } from 'fastembed';\n\nasync function getModelCachePath() {\n  const cachePath = path.join(os.homedir(), '.cache', 'mastra', 'fastembed-models');\n  await fsp.mkdir(cachePath, { recursive: true });\n  return cachePath;\n}\n\n// Shared function to generate embeddings using fastembed\nasync function generateEmbeddings(values: string[], modelType: 'BGESmallENV15' | 'BGEBaseENV15') {\n  const model = await FlagEmbedding.init({\n    model: EmbeddingModel[modelType],\n    cacheDir: await getModelCachePath(),\n  });\n\n  // model.embed() returns an AsyncGenerator that processes texts in batches (default size 256)\n  const embeddings = model.embed(values);\n\n  const allResults = [];\n  for await (const result of embeddings) {\n    // result is an array of embeddings, one for each text in the batch\n    // We convert each Float32Array embedding to a regular number array\n    allResults.push(...result.map(embedding => Array.from(embedding)));\n  }\n\n  if (allResults.length === 0) throw new Error('No embeddings generated');\n\n  return {\n    embeddings: allResults,\n  };\n}\n\nconst fastEmbedProvider = experimental_customProvider({\n  textEmbeddingModels: {\n    'bge-small-en-v1.5': {\n      specificationVersion: 'v1',\n      provider: 'fastembed',\n      modelId: 'bge-small-en-v1.5',\n      maxEmbeddingsPerCall: 256,\n      supportsParallelCalls: true,\n      async doEmbed({ values }) {\n        return generateEmbeddings(values, 'BGESmallENV15');\n      },\n    },\n    'bge-base-en-v1.5': {\n      specificationVersion: 'v1',\n      provider: 'fastembed',\n      modelId: 'bge-base-en-v1.5',\n      maxEmbeddingsPerCall: 256,\n      supportsParallelCalls: true,\n      async doEmbed({ values }) {\n        return generateEmbeddings(values, 'BGEBaseENV15');\n      },\n    },\n  },\n});\n\nexport const fastembed = Object.assign(fastEmbedProvider.textEmbeddingModel(`bge-small-en-v1.5`), {\n  small: fastEmbedProvider.textEmbeddingModel(`bge-small-en-v1.5`),\n  base: fastEmbedProvider.textEmbeddingModel(`bge-base-en-v1.5`),\n});\n"]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/fastembed",
-  "version": "0.10.1",
+  "version": "0.10.2",
   "license": "Apache-2.0",
   "type": "module",
   "main": "dist/index.js",
@@ -12,7 +12,7 @@
         "default": "./dist/index.js"
       },
       "require": {
-        "types": "./dist/index.d.cts",
+        "types": "./dist/index.d.ts",
         "default": "./dist/index.cjs"
       }
     },
@@ -30,11 +30,12 @@
     "tsup": "^8.5.0",
     "typescript": "^5.8.3",
     "vitest": "^3.2.4",
-    "@internal/lint": "0.0.18"
+    "@internal/lint": "0.0.27",
+    "@internal/types-builder": "0.0.2"
   },
   "scripts": {
-    "build": "tsup src/index.ts --format esm,cjs --experimental-dts --clean --treeshake=smallest",
-    "build:watch": "pnpm build --watch",
+    "build": "tsup --silent --config tsup.config.ts",
+    "build:watch": "tsup --watch --silent --config tsup.config.ts",
     "test": "vitest run"
   }
 }

package/dist/_tsup-dts-rollup.d.cts

@@ -1,8 +0,0 @@
-import { EmbeddingModel } from 'ai';
-
-export declare const fastembed: EmbeddingModel<string> & {
-    small: EmbeddingModel<string>;
-    base: EmbeddingModel<string>;
-};
-
-export { }

package/dist/_tsup-dts-rollup.d.ts

@@ -1,8 +0,0 @@
-import { EmbeddingModel } from 'ai';
-
-export declare const fastembed: EmbeddingModel<string> & {
-    small: EmbeddingModel<string>;
-    base: EmbeddingModel<string>;
-};
-
-export { }

package/dist/index.d.cts

@@ -1 +0,0 @@
-export { fastembed } from './_tsup-dts-rollup.cjs';