Skip to main content

Class: MistralAIEmbedding

Extends

Constructors

new MistralAIEmbedding()

new MistralAIEmbedding(init?): MistralAIEmbedding

Parameters

init?: Partial <MistralAIEmbedding>

Returns

MistralAIEmbedding

Overrides

BaseEmbedding . constructor

Source

packages/core/src/embeddings/MistralAIEmbedding.ts:14

Properties

apiKey?

optional apiKey: string

Source

packages/core/src/embeddings/MistralAIEmbedding.ts:10


embedBatchSize

embedBatchSize: number = DEFAULT_EMBED_BATCH_SIZE

Inherited from

BaseEmbedding . embedBatchSize

Source

packages/core/src/embeddings/types.ts:11


model

model: MISTRAL_EMBED

Source

packages/core/src/embeddings/MistralAIEmbedding.ts:9


session

private session: MistralAISession

Source

packages/core/src/embeddings/MistralAIEmbedding.ts:12

Methods

getMistralAIEmbedding()

private getMistralAIEmbedding(input): Promise<any>

Parameters

input: string

Returns

Promise<any>

Source

packages/core/src/embeddings/MistralAIEmbedding.ts:20


getQueryEmbedding()

getQueryEmbedding(query): Promise<number[]>

Parameters

query: string

Returns

Promise<number[]>

Overrides

BaseEmbedding . getQueryEmbedding

Source

packages/core/src/embeddings/MistralAIEmbedding.ts:34


getTextEmbedding()

getTextEmbedding(text): Promise<number[]>

Parameters

text: string

Returns

Promise<number[]>

Overrides

BaseEmbedding . getTextEmbedding

Source

packages/core/src/embeddings/MistralAIEmbedding.ts:30


getTextEmbeddings()

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

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

Parameters

texts: string[]

Returns

Promise<number[][]>

Inherited from

BaseEmbedding . getTextEmbeddings

Source

packages/core/src/embeddings/types.ts:28


getTextEmbeddingsBatch()

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

Get embeddings for a batch of texts

Parameters

texts: string[]

options?

options.logProgress?: boolean

Returns

Promise<number[][]>

Inherited from

BaseEmbedding . getTextEmbeddingsBatch

Source

packages/core/src/embeddings/types.ts:44


similarity()

similarity(embedding1, embedding2, mode): number

Parameters

embedding1: number[]

embedding2: number[]

mode: SimilarityType= SimilarityType.DEFAULT

Returns

number

Inherited from

BaseEmbedding . similarity

Source

packages/core/src/embeddings/types.ts:13


transform()

transform(nodes, _options?): Promise <BaseNode <Metadata>[]>

Parameters

nodes: BaseNode <Metadata>[]

_options?: any

Returns

Promise <BaseNode <Metadata>[]>

Inherited from

BaseEmbedding . transform

Source

packages/core/src/embeddings/types.ts:58