Logo
Classes

BaseEmbedding

Defined in: packages/core/src/embeddings/base.ts:25

Extends

Extended by

abstract BaseEmbedding<Options>(nodes, options?): Promise

Defined in: packages/core/src/embeddings/base.ts:25

Type Parameters

Options

Options extends Record<string, unknown>

Parameters

nodes

BaseNode<Metadata>[]

options?

Options

Returns

Promise

Constructors

Constructor

protected new BaseEmbedding(transformFn?): BaseEmbedding

Defined in: packages/core/src/embeddings/base.ts:31

Parameters

transformFn?

(nodes, options?) => Promise<BaseNode<Metadata>[]>

Returns

BaseEmbedding

Overrides

TransformComponent.constructor

Properties

embedBatchSize

embedBatchSize: number = DEFAULT_EMBED_BATCH_SIZE

Defined in: packages/core/src/embeddings/base.ts:28


embedInfo?

optional embedInfo: EmbeddingInfo

Defined in: packages/core/src/embeddings/base.ts:29


id

id: string

Defined in: packages/core/src/schema/type.ts:22

Inherited from

TransformComponent.id

Methods

similarity()

similarity(embedding1, embedding2, mode): number

Defined in: packages/core/src/embeddings/base.ts:61

Parameters

embedding1

number[]

embedding2

number[]

mode

SimilarityType = SimilarityType.DEFAULT

Returns

number


getTextEmbedding()

abstract getTextEmbedding(text): Promise<number[]>

Defined in: packages/core/src/embeddings/base.ts:69

Parameters

text

string

Returns

Promise<number[]>


getQueryEmbedding()

getQueryEmbedding(query): Promise<null | number[]>

Defined in: packages/core/src/embeddings/base.ts:71

Parameters

query

MessageContentDetail

Returns

Promise<null | number[]>


getTextEmbeddings()

getTextEmbeddings(texts): Promise<number[][]>

Defined in: packages/core/src/embeddings/base.ts:85

Optionally override this method to retrieve multiple embeddings in a single request

Parameters

texts

string[]

Returns

Promise<number[][]>


getTextEmbeddingsBatch()

getTextEmbeddingsBatch(texts, options?): Promise<number[][]>

Defined in: packages/core/src/embeddings/base.ts:101

Get embeddings for a batch of texts

Parameters

texts

string[]

options?

BaseEmbeddingOptions

Returns

Promise<number[][]>


truncateMaxTokens()

truncateMaxTokens(input): string[]

Defined in: packages/core/src/embeddings/base.ts:113

Parameters

input

string[]

Returns

string[]