getTopKEmbeddings

getTopKEmbeddings(queryEmbedding, embeddings, similarityTopK, embeddingIds, similarityCutoff): [number[], any[]]

Defined in: packages/core/src/embeddings/utils.ts:76

Get the top K embeddings from a list of embeddings ordered by similarity to the query.

Parameters

queryEmbedding

number[]

embeddings

number[][]

list of embeddings to consider

similarityTopK

number = 2

max number of embeddings to return, default 2

embeddingIds

ids of embeddings in the embeddings list

null | any[]

similarityCutoff

minimum similarity score

null | number

Returns

[number[], any[]]