@mastra/fastembed 1.3.6-alpha.1 → 1.3.7-alpha.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
package/dist/index.d.ts CHANGED
@@ -1,11 +1,11 @@
1
- import type { EmbeddingModelV3 } from './_types/@internal_ai-v6/dist/index.js';
2
- import type { EmbeddingModel as EmbeddingModelV2 } from './_types/@internal_ai-sdk-v5/dist/index.js';
3
- import type { EmbeddingModel as EmbeddingModelV1 } from './_types/@internal_ai-sdk-v4/dist/index.js';
1
+ import type { EmbeddingModelV3 } from './_types/@internal_ai-v6@0.0.95/dist/index.js';
2
+ import type { EmbeddingModel as EmbeddingModelV2 } from './_types/@internal_ai-sdk-v5@0.0.95/dist/index.js';
3
+ import type { EmbeddingModel as EmbeddingModelV1 } from './_types/@internal_ai-sdk-v4@0.0.95/dist/index.js';
4
4
  export { EmbeddingModel, FlagEmbedding, SparseTextEmbedding, SparseEmbeddingModel, ExecutionProvider, } from './fastembed.js';
5
5
  export type { SparseVector, InitOptions, InitSparseOptions } from './fastembed.js';
6
- export type { EmbeddingModel as EmbeddingModelV1 } from './_types/@internal_ai-sdk-v4/dist/index.js';
7
- export type { EmbeddingModel as EmbeddingModelV2 } from './_types/@internal_ai-sdk-v5/dist/index.js';
8
- export type { EmbeddingModel as EmbeddingModelV3 } from './_types/@internal_ai-v6/dist/index.js';
6
+ export type { EmbeddingModel as EmbeddingModelV1 } from './_types/@internal_ai-sdk-v4@0.0.95/dist/index.js';
7
+ export type { EmbeddingModel as EmbeddingModelV2 } from './_types/@internal_ai-sdk-v5@0.0.95/dist/index.js';
8
+ export type { EmbeddingModel as EmbeddingModelV3 } from './_types/@internal_ai-v6@0.0.95/dist/index.js';
9
9
  /**
10
10
  * Pre-download fastembed models without creating ONNX sessions.
11
11
  * Call this before running tests in parallel to avoid concurrent download races.
@@ -1 +1 @@
1
- {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,iBAAiB,CAAC;AACxD,OAAO,KAAK,EAAE,cAAc,IAAI,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAC9E,OAAO,KAAK,EAAE,cAAc,IAAI,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAI9E,OAAO,EACL,cAAc,EACd,aAAa,EACb,mBAAmB,EACnB,oBAAoB,EACpB,iBAAiB,GAClB,MAAM,gBAAgB,CAAC;AACxB,YAAY,EAAE,YAAY,EAAE,WAAW,EAAE,iBAAiB,EAAE,MAAM,gBAAgB,CAAC;AAEnF,YAAY,EAAE,cAAc,IAAI,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAC9E,YAAY,EAAE,cAAc,IAAI,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAC9E,YAAY,EAAE,cAAc,IAAI,gBAAgB,EAAE,MAAM,iBAAiB,CAAC;AAE1E;;;GAGG;AACH,wBAAsB,MAAM,kBAE3B;AAuJD,eAAO,MAAM,SAAS,EAAE,gBAAgB,GAAG;IACzC,KAAK,EAAE,gBAAgB,CAAC;IACxB,IAAI,EAAE,gBAAgB,CAAC;IACvB,wBAAwB,EAAE,gBAAgB,CAAC;IAC3C,0BAA0B,EAAE,gBAAgB,CAAC;IAC7C,OAAO,EAAE,gBAAgB,CAAC,MAAM,CAAC,CAAC;IAClC,MAAM,EAAE,gBAAgB,CAAC,MAAM,CAAC,CAAC;IACjC,WAAW,EAAE,gBAAgB,CAAC,MAAM,CAAC,CAAC;IACtC,UAAU,EAAE,gBAAgB,CAAC,MAAM,CAAC,CAAC;CAUrC,CAAC"}
1
+ {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,iBAAiB,CAAC;AACxD,OAAO,KAAK,EAAE,cAAc,IAAI,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAC9E,OAAO,KAAK,EAAE,cAAc,IAAI,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAI9E,OAAO,EACL,cAAc,EACd,aAAa,EACb,mBAAmB,EACnB,oBAAoB,EACpB,iBAAiB,GAClB,MAAM,gBAAgB,CAAC;AACxB,YAAY,EAAE,YAAY,EAAE,WAAW,EAAE,iBAAiB,EAAE,MAAM,gBAAgB,CAAC;AAEnF,YAAY,EAAE,cAAc,IAAI,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAC9E,YAAY,EAAE,cAAc,IAAI,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAC9E,YAAY,EAAE,cAAc,IAAI,gBAAgB,EAAE,MAAM,iBAAiB,CAAC;AAE1E;;;GAGG;AACH,wBAAsB,MAAM,kBAE3B;AAsKD,eAAO,MAAM,SAAS,EAAE,gBAAgB,GAAG;IACzC,KAAK,EAAE,gBAAgB,CAAC;IACxB,IAAI,EAAE,gBAAgB,CAAC;IACvB,wBAAwB,EAAE,gBAAgB,CAAC;IAC3C,0BAA0B,EAAE,gBAAgB,CAAC;IAC7C,OAAO,EAAE,gBAAgB,CAAC,MAAM,CAAC,CAAC;IAClC,MAAM,EAAE,gBAAgB,CAAC,MAAM,CAAC,CAAC;IACjC,WAAW,EAAE,gBAAgB,CAAC,MAAM,CAAC,CAAC;IACtC,UAAU,EAAE,gBAAgB,CAAC,MAAM,CAAC,CAAC;CAUrC,CAAC"}
package/dist/index.js CHANGED
@@ -9480,7 +9480,7 @@ ZodOptional.create;
9480
9480
  ZodNullable.create;
9481
9481
  ZodEffects.createWithPreprocess;
9482
9482
  ZodPipeline.create;
9483
- // ../_vendored/ai_v5/dist/dist-DRvrQv6K.js
9483
+ // ../_vendored/ai_v5/dist/dist-BYnbJEqC.js
9484
9484
  var marker$1$6 = "vercel.ai.error";
9485
9485
  var symbol$1$6 = Symbol.for(marker$1$6);
9486
9486
  var _a$1$6;
@@ -10698,6 +10698,20 @@ function asSchema$2(schema) {
10698
10698
  }) : isSchema$2(schema) ? schema : typeof schema === "function" ? schema() : zodSchema$2(schema);
10699
10699
  }
10700
10700
  var { btoa: btoa$3, atob: atob$5 } = globalThis;
10701
+ var typedArrayTag$1 = Object.getOwnPropertyDescriptor(Object.getPrototypeOf(Uint8Array.prototype), Symbol.toStringTag).get;
10702
+ var arrayBufferByteLength$1 = Object.getOwnPropertyDescriptor(ArrayBuffer.prototype, "byteLength").get;
10703
+ function isUint8Array$1(value) {
10704
+ return ArrayBuffer.isView(value) && typedArrayTag$1.call(value) === "Uint8Array";
10705
+ }
10706
+ function isArrayBuffer$1(value) {
10707
+ if (typeof value !== "object" || value === null || !("byteLength" in value)) return false;
10708
+ try {
10709
+ arrayBufferByteLength$1.call(value);
10710
+ return true;
10711
+ } catch (e) {
10712
+ return false;
10713
+ }
10714
+ }
10701
10715
  // ../_vendored/ai_v5/dist/index.js
10702
10716
  /** only globals that common to node and browsers are allowed */
10703
10717
  var _globalThis$1 = typeof globalThis === "object" ? globalThis : global;
@@ -11515,8 +11529,8 @@ var NoObjectGeneratedError$2 = class extends AISDKError$2 {
11515
11529
  _a7$4 = symbol7$4;
11516
11530
  var dataContentSchema$2 = union([
11517
11531
  string(),
11518
- _instanceof(Uint8Array),
11519
- _instanceof(ArrayBuffer),
11532
+ custom(isUint8Array$1, { message: "Must be a Uint8Array" }),
11533
+ custom(isArrayBuffer$1, { message: "Must be an ArrayBuffer" }),
11520
11534
  custom((value) => {
11521
11535
  var _a17, _b;
11522
11536
  return (_b = (_a17 = globalThis.Buffer) == null ? void 0 : _a17.isBuffer(value)) != null ? _b : false;
@@ -15275,7 +15289,7 @@ function trimStartOfStream() {
15275
15289
  return text2;
15276
15290
  };
15277
15291
  }
15278
- // ../_vendored/ai_v6/dist/dist-Dl1sIpsh.js
15292
+ // ../_vendored/ai_v6/dist/dist-DwNW4Hp7.js
15279
15293
  var marker$1 = "vercel.ai.error";
15280
15294
  var symbol$1 = Symbol.for(marker$1);
15281
15295
  var _a$1;
@@ -15425,6 +15439,20 @@ Error message: ${getErrorMessage$1(cause)}`,
15425
15439
  };
15426
15440
  TransformStream;
15427
15441
  var { btoa: btoa$1, atob: atob$2 } = globalThis;
15442
+ var typedArrayTag = Object.getOwnPropertyDescriptor(Object.getPrototypeOf(Uint8Array.prototype), Symbol.toStringTag).get;
15443
+ var arrayBufferByteLength = Object.getOwnPropertyDescriptor(ArrayBuffer.prototype, "byteLength").get;
15444
+ function isUint8Array(value) {
15445
+ return ArrayBuffer.isView(value) && typedArrayTag.call(value) === "Uint8Array";
15446
+ }
15447
+ function isArrayBuffer(value) {
15448
+ if (typeof value !== "object" || value === null || !("byteLength" in value)) return false;
15449
+ try {
15450
+ arrayBufferByteLength.call(value);
15451
+ return true;
15452
+ } catch (e) {
15453
+ return false;
15454
+ }
15455
+ }
15428
15456
  var createIdGenerator = ({ prefix, size = 16, alphabet = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz", separator = "-" } = {}) => {
15429
15457
  const generator = () => {
15430
15458
  const alphabetLength = alphabet.length;
@@ -17481,8 +17509,8 @@ function asTranscriptionModelV3(model) {
17481
17509
  }
17482
17510
  var dataContentSchema = union([
17483
17511
  string(),
17484
- _instanceof(Uint8Array),
17485
- _instanceof(ArrayBuffer),
17512
+ custom(isUint8Array, { message: "Must be a Uint8Array" }),
17513
+ custom(isArrayBuffer, { message: "Must be an ArrayBuffer" }),
17486
17514
  custom((value) => {
17487
17515
  var _a23, _b;
17488
17516
  return (_b = (_a23 = globalThis.Buffer) == null ? void 0 : _a23.isBuffer(value)) != null ? _b : false;
@@ -18772,6 +18800,12 @@ async function generateEmbeddingsWithWarnings(values, modelType, prefix) {
18772
18800
  warnings
18773
18801
  };
18774
18802
  }
18803
+ /**
18804
+ * The token window of every fastembed model here. Set as `maxInputTokens` on each model object:
18805
+ * Mastra's semantic recall reads it to size message chunks, so long messages are split instead of
18806
+ * having everything past the window silently dropped from the embedding.
18807
+ */
18808
+ const FASTEMBED_INPUT_LIMIT = { maxInputTokens: 512 };
18775
18809
  const fastEmbedLegacyProvider = customProvider$1({ textEmbeddingModels: {
18776
18810
  "bge-small-en-v1.5": {
18777
18811
  specificationVersion: "v1",
@@ -18779,6 +18813,7 @@ const fastEmbedLegacyProvider = customProvider$1({ textEmbeddingModels: {
18779
18813
  modelId: "bge-small-en-v1.5",
18780
18814
  maxEmbeddingsPerCall: 256,
18781
18815
  supportsParallelCalls: true,
18816
+ ...FASTEMBED_INPUT_LIMIT,
18782
18817
  async doEmbed({ values }) {
18783
18818
  return generateEmbeddings(values, "BGESmallENV15");
18784
18819
  }
@@ -18789,6 +18824,7 @@ const fastEmbedLegacyProvider = customProvider$1({ textEmbeddingModels: {
18789
18824
  modelId: "bge-base-en-v1.5",
18790
18825
  maxEmbeddingsPerCall: 256,
18791
18826
  supportsParallelCalls: true,
18827
+ ...FASTEMBED_INPUT_LIMIT,
18792
18828
  async doEmbed({ values }) {
18793
18829
  return generateEmbeddings(values, "BGEBaseENV15");
18794
18830
  }
@@ -18801,6 +18837,7 @@ const fastEmbedProviderV2 = customProvider$2({ textEmbeddingModels: {
18801
18837
  modelId: "bge-small-en-v1.5",
18802
18838
  maxEmbeddingsPerCall: 256,
18803
18839
  supportsParallelCalls: true,
18840
+ ...FASTEMBED_INPUT_LIMIT,
18804
18841
  async doEmbed({ values }) {
18805
18842
  return generateEmbeddings(values, "BGESmallENV15");
18806
18843
  }
@@ -18811,6 +18848,7 @@ const fastEmbedProviderV2 = customProvider$2({ textEmbeddingModels: {
18811
18848
  modelId: "bge-base-en-v1.5",
18812
18849
  maxEmbeddingsPerCall: 256,
18813
18850
  supportsParallelCalls: true,
18851
+ ...FASTEMBED_INPUT_LIMIT,
18814
18852
  async doEmbed({ values }) {
18815
18853
  return generateEmbeddings(values, "BGEBaseENV15");
18816
18854
  }
@@ -18823,6 +18861,7 @@ const fastEmbedProviderV3 = customProvider({ embeddingModels: {
18823
18861
  modelId: "bge-small-en-v1.5",
18824
18862
  maxEmbeddingsPerCall: 256,
18825
18863
  supportsParallelCalls: true,
18864
+ ...FASTEMBED_INPUT_LIMIT,
18826
18865
  async doEmbed({ values }) {
18827
18866
  return generateEmbeddingsWithWarnings(values, "BGESmallENV15");
18828
18867
  }
@@ -18833,6 +18872,7 @@ const fastEmbedProviderV3 = customProvider({ embeddingModels: {
18833
18872
  modelId: "bge-base-en-v1.5",
18834
18873
  maxEmbeddingsPerCall: 256,
18835
18874
  supportsParallelCalls: true,
18875
+ ...FASTEMBED_INPUT_LIMIT,
18836
18876
  async doEmbed({ values }) {
18837
18877
  return generateEmbeddingsWithWarnings(values, "BGEBaseENV15");
18838
18878
  }
@@ -18843,6 +18883,7 @@ const fastEmbedProviderV3 = customProvider({ embeddingModels: {
18843
18883
  modelId: "multilingual-e5-large-query",
18844
18884
  maxEmbeddingsPerCall: 256,
18845
18885
  supportsParallelCalls: true,
18886
+ ...FASTEMBED_INPUT_LIMIT,
18846
18887
  async doEmbed({ values }) {
18847
18888
  return generateEmbeddingsWithWarnings(values, "MLE5Large", "query");
18848
18889
  }
@@ -18853,6 +18894,7 @@ const fastEmbedProviderV3 = customProvider({ embeddingModels: {
18853
18894
  modelId: "multilingual-e5-large-passage",
18854
18895
  maxEmbeddingsPerCall: 256,
18855
18896
  supportsParallelCalls: true,
18897
+ ...FASTEMBED_INPUT_LIMIT,
18856
18898
  async doEmbed({ values }) {
18857
18899
  return generateEmbeddingsWithWarnings(values, "MLE5Large", "passage");
18858
18900
  }