Member Junction
    Preparing search index...

    Class BaseEmbeddingsAbstract

    Base class for all embedding model implementations

    Hierarchy (View Summary)

    Index

    Constructors

    Properties

    _additionalSettings: Record<string, any> = {}

    Protected property to store additional provider-specific settings

    Accessors

    • get maxEmbedTextsConcurrency(): number

      Max in-flight EmbedText calls for the default (non-batch) path. Override to tune.

      Only the per-text fallback is throttled this way — the native embedBatch path sends all texts in ONE request, so there is nothing to bound. That asymmetry is intentional: N small calls need a concurrency ceiling; a single batched call does not.

      Returns number

    • get maxEmbedTextsRetries(): number

      Extra retry attempts per text on the default (non-batch) path, on top of the initial attempt. 0 disables retry. Retrying each text a few times before giving up stops one transient 429/500 from failing the whole batch (whose failure rate otherwise scales with the text count N).

      Returns number

    • get SupportsBatchEmbeddings(): boolean

      Whether this provider's embedding model has a NATIVE batch endpoint that returns one vector per input in a single request. Defaults to false. Providers that pass an array of texts straight to their API (e.g. OpenAI, Azure, Cohere, Mistral) override this to return true and implement embedBatch. Providers without one inherit the safe per-text default below.

      Returns boolean

    Methods

    • Clear all additional settings This is useful for resetting the state of the provider or when switching between different configurations.

      Returns void

    • Default (non-batch) path: fans out one EmbedText call per text with bounded concurrency, preserving order. The 1:1 vector/text count is enforced by the dispatcher (EmbedTexts).

      Each text is first retried with bounded exponential backoff (retryEmbedText) so a lone transient failure doesn't sink the batch; only a text that STILL fails after its retries counts as failed.

      ERROR CONTRACT (deliberate — change here if a different policy is wanted): mirrors the per-provider Gemini fix. On ANY per-text failure that survives retry (EmbedText throws OR yields an empty vector) we return an EMPTY result rather than throwing, so batch pipelines that don't wrap EmbedTexts (e.g. EntityVectorSyncer) degrade gracefully instead of aborting. To make it fully fail-loud instead, replace the two emptyEmbedTextsResult(...) returns below with throw.

      Parameters

      Returns Promise<EmbedTextsResult>

    • Embeds a SINGLE text into one vector.

      ERROR CONVENTION — note the deliberate asymmetry with EmbedTexts: on failure a provider returns an empty vector: [] here (a single-item soft failure the caller inspects directly). The batch EmbedTexts is stricter and ALL-OR-NOTHING — one failed item collapses the whole result to vectors: [] rather than returning a partially-filled array — because its consumers pair vectors[i] ↔ records[i] by index and a stray empty slot is silent corruption. The one case that THROWS instead of degrading is a structural collapse (vector count ≠ text count); see EmbedTexts.

      Parameters

      Returns Promise<EmbedTextResult>

    • Embeds an array of texts, returning exactly ONE vector per input text, in input order.

      DISPATCHES on SupportsBatchEmbeddings: providers with a native batch endpoint set it to true and implement embedBatch; everyone else gets the safe per-text fallback in embedPerText. (A provider may still override this method directly for fully custom behavior.)

      Whichever path runs, the result is hard-asserted to be 1:1 with the inputs (an intentional empty result — the per-text graceful-degrade — is allowed): a native embedBatch that drops or reorders vectors, or any other collapse, throws here rather than letting a misaligned array reach index-based consumers (e.g. EntityVectorSyncer).

      Parameters

      Returns Promise<EmbedTextsResult>

    • Runs a single-text embed attempt with bounded exponential-backoff retry. A transient failure — attempt throws, or returns an empty/missing vector — is retried up to maxEmbedTextsRetries times, sleeping embedRetryBaseDelayMs * 2^(n-1) between tries. Returns the first successful result; after the final attempt returns whatever it produced (or rethrows its error) so the caller's existing empty-vector / throw handling still applies. This is what keeps one transient 429/500 from failing the whole batch.

      Parameters

      Returns Promise<EmbedTextResult>

    • Runs fn over items with at most maxConcurrency in flight at once, preserving order. The per-item await inside each worker is what bounds concurrency; parallelism comes from running up to maxConcurrency workers at the same time.

      Type Parameters

      • T

      Parameters

      • items: string[]
      • fn: (item: string, index: number) => Promise<T>
      • maxConcurrency: number

      Returns Promise<T[]>

    • Set additional provider-specific settings Subclasses should override this method to validate required settings

      Parameters

      • settings: Record<string, any>

        Provider-specific settings

      Returns void

    • Sleep helper for retry backoff, isolated so tests can override embedRetryBaseDelayMs to avoid real delays.

      Parameters

      • ms: number

      Returns Promise<void>