Skip to main content

Class: OpenAIEmbedding

Hierarchy

Constructors

constructor

new OpenAIEmbedding(init?): OpenAIEmbedding

OpenAI Embedding

Parameters

NameTypeDescription
init?Partial<OpenAIEmbedding> & { azure?: AzureOpenAIConfig }initial parameters

Returns

OpenAIEmbedding

Overrides

BaseEmbedding.constructor

Defined in

packages/core/src/embeddings/OpenAIEmbedding.ts:56

Properties

additionalSessionOptions

Optional additionalSessionOptions: Omit<Partial<ClientOptions>, "apiKey" | "timeout" | "maxRetries">

other session options for OpenAI

Defined in

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


apiKey

Optional apiKey: string = undefined

api key

Defined in

packages/core/src/embeddings/OpenAIEmbedding.ts:38


dimensions

dimensions: undefined | number

number of dimensions of the resulting vector, for models that support choosing fewer dimensions. undefined will default to model default

Defined in

packages/core/src/embeddings/OpenAIEmbedding.ts:33


maxRetries

maxRetries: number

maximum number of retries, default 10

Defined in

packages/core/src/embeddings/OpenAIEmbedding.ts:40


model

model: string

embeddding model. defaults to "text-embedding-ada-002"

Defined in

packages/core/src/embeddings/OpenAIEmbedding.ts:31


session

session: OpenAISession

session object

Defined in

packages/core/src/embeddings/OpenAIEmbedding.ts:50


timeout

Optional timeout: number

timeout in ms, default 60 seconds

Defined in

packages/core/src/embeddings/OpenAIEmbedding.ts:42

Methods

getOpenAIEmbedding

getOpenAIEmbedding(input): Promise<number[]>

Parameters

NameType
inputstring

Returns

Promise<number[]>

Defined in

packages/core/src/embeddings/OpenAIEmbedding.ts:103


getQueryEmbedding

getQueryEmbedding(query): Promise<number[]>

Parameters

NameType
querystring

Returns

Promise<number[]>

Overrides

BaseEmbedding.getQueryEmbedding

Defined in

packages/core/src/embeddings/OpenAIEmbedding.ts:117


getTextEmbedding

getTextEmbedding(text): Promise<number[]>

Parameters

NameType
textstring

Returns

Promise<number[]>

Overrides

BaseEmbedding.getTextEmbedding

Defined in

packages/core/src/embeddings/OpenAIEmbedding.ts:113


similarity

similarity(embedding1, embedding2, mode?): number

Parameters

NameTypeDefault value
embedding1number[]undefined
embedding2number[]undefined
modeSimilarityTypeSimilarityType.DEFAULT

Returns

number

Inherited from

BaseEmbedding.similarity

Defined in

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


transform

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

Parameters

NameType
nodesBaseNode<Metadata>[]
_options?any

Returns

Promise<BaseNode<Metadata>[]>

Inherited from

BaseEmbedding.transform

Defined in

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