Class: GeminiEmbedding
GeminiEmbedding is an alias for Gemini that implements the BaseEmbedding interface.
Hierarchy
-
↳
GeminiEmbedding
Constructors
constructor
• new GeminiEmbedding(init?): GeminiEmbedding
Parameters
| Name | Type |
|---|---|
init? | Partial<GeminiEmbedding> |
Returns
Overrides
Defined in
packages/core/src/embeddings/GeminiEmbedding.ts:16
Properties
embedBatchSize
• embedBatchSize: number = DEFAULT_EMBED_BATCH_SIZE
Inherited from
Defined in
packages/core/src/embeddings/types.ts:11
model
• model: GEMINI_EMBEDDING_MODEL
Defined in
packages/core/src/embeddings/GeminiEmbedding.ts:13
session
• session: GeminiSession
Defined in
packages/core/src/embeddings/GeminiEmbedding.ts:14
Methods
getEmbedding
▸ getEmbedding(prompt): Promise<number[]>
Parameters
| Name | Type |
|---|---|
prompt | string |
Returns
Promise<number[]>
Defined in
packages/core/src/embeddings/GeminiEmbedding.ts:22
getQueryEmbedding
▸ getQueryEmbedding(query): Promise<number[]>
Parameters
| Name | Type |
|---|---|
query | string |
Returns
Promise<number[]>
Overrides
BaseEmbedding.getQueryEmbedding
Defined in
packages/core/src/embeddings/GeminiEmbedding.ts:34
getTextEmbedding
▸ getTextEmbedding(text): Promise<number[]>
Parameters
| Name | Type |
|---|---|
text | string |
Returns
Promise<number[]>
Overrides
BaseEmbedding.getTextEmbedding
Defined in
packages/core/src/embeddings/GeminiEmbedding.ts:30
getTextEmbeddings
▸ getTextEmbeddings(texts): Promise<number[][]>
Optionally override this method to retrieve multiple embeddings in a single request
Parameters
| Name | Type |
|---|---|
texts | string[] |
Returns
Promise<number[][]>
Inherited from
BaseEmbedding.getTextEmbeddings
Defined in
packages/core/src/embeddings/types.ts:28
getTextEmbeddingsBatch
▸ getTextEmbeddingsBatch(texts, options?): Promise<number[][]>
Get embeddings for a batch of texts
Parameters
| Name | Type |
|---|---|
texts | string[] |
options? | Object |
options.logProgress? | boolean |
Returns
Promise<number[][]>
Inherited from
BaseEmbedding.getTextEmbeddingsBatch
Defined in
packages/core/src/embeddings/types.ts:44
similarity
▸ similarity(embedding1, embedding2, mode?): number
Parameters
| Name | Type | Default value |
|---|---|---|
embedding1 | number[] | undefined |
embedding2 | number[] | undefined |
mode | SimilarityType | SimilarityType.DEFAULT |
Returns
number
Inherited from
Defined in
packages/core/src/embeddings/types.ts:13
transform
▸ transform(nodes, _options?): Promise<BaseNode<Metadata>[]>
Parameters
| Name | Type |
|---|---|
nodes | BaseNode<Metadata>[] |
_options? | any |