DeepInfraEmbedding

Defined in: providers/deepinfra/src/embedding.ts:36

DeepInfraEmbedding is an alias for DeepInfra that implements the BaseEmbedding interface.

Extends

• BaseEmbedding

DeepInfraEmbedding<Options>(nodes, options?): Promise<BaseNode<Metadata>[]>

Defined in: providers/deepinfra/src/embedding.ts:36

DeepInfraEmbedding is an alias for DeepInfra that implements the BaseEmbedding interface.

Type Parameters

• Options extends Record<string, unknown>

Parameters

nodes

BaseNode<Metadata>[]

options?

Options

Returns

Promise<BaseNode<Metadata>[]>

Constructors

new DeepInfraEmbedding()

new DeepInfraEmbedding(init?): DeepInfraEmbedding

Defined in: providers/deepinfra/src/embedding.ts:76

Parameters

init?

Partial<DeepInfraEmbedding>

Returns

DeepInfraEmbedding

Overrides

BaseEmbedding.constructor

Properties

model

model: string

Defined in: providers/deepinfra/src/embedding.ts:42

DeepInfra model to use

Default

"sentence-transformers/clip-ViT-B-32"

See

https://deepinfra.com/models/embeddings

apiToken

apiToken: string

Defined in: providers/deepinfra/src/embedding.ts:50

DeepInfra API token

See

https://deepinfra.com/dash/api_keys If not provided, it will try to get the token from the environment variable DEEPINFRA_API_TOKEN

queryPrefix

queryPrefix: string

Defined in: providers/deepinfra/src/embedding.ts:56

Prefix to add to the query

Default

""

textPrefix

textPrefix: string

Defined in: providers/deepinfra/src/embedding.ts:62

Prefix to add to the text

Default

""

maxRetries

maxRetries: number

Defined in: providers/deepinfra/src/embedding.ts:68

Default

5

timeout

timeout: number

Defined in: providers/deepinfra/src/embedding.ts:74

Default

60 * 1000

Methods

getTextEmbedding()

getTextEmbedding(text): Promise<number[]>

Defined in: providers/deepinfra/src/embedding.ts:87

Parameters

text

string

Returns

Promise<number[]>

Overrides

BaseEmbedding.getTextEmbedding

getQueryEmbedding()

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

Defined in: providers/deepinfra/src/embedding.ts:93

Parameters

query

MessageContentDetail

Returns

Promise<null | number[]>

Overrides

BaseEmbedding.getQueryEmbedding

getTextEmbeddings()

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

Defined in: providers/deepinfra/src/embedding.ts:106

Parameters

texts

string[]

Returns

Promise<number[][]>

Overrides

BaseEmbedding.getTextEmbeddings

getQueryEmbeddings()

getQueryEmbeddings(queries): Promise<number[][]>

Defined in: providers/deepinfra/src/embedding.ts:111

Parameters

queries

string[]

Returns

Promise<number[][]>