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? | GeminiConfig |
Returns
Overrides
Defined in
packages/core/src/embeddings/GeminiEmbedding.ts:19
Properties
embedBatchSize
• embedBatchSize: number
= DEFAULT_EMBED_BATCH_SIZE
Inherited from
Defined in
packages/core/src/embeddings/types.ts:11
maxTokens
• Optional
maxTokens: number
Defined in
packages/core/src/embeddings/GeminiEmbedding.ts:16
model
• model: GEMINI_MODEL
Defined in
packages/core/src/embeddings/GeminiEmbedding.ts:13
session
• session: GeminiSession
Defined in
packages/core/src/embeddings/GeminiEmbedding.ts:17
temperature
• temperature: number
Defined in
packages/core/src/embeddings/GeminiEmbedding.ts:14
topP
• topP: number
Defined in
packages/core/src/embeddings/GeminiEmbedding.ts:15
Methods
getEmbedding
▸ getEmbedding(prompt
): Promise
<number
[]>
Parameters
Name | Type |
---|---|
prompt | string |
Returns
Promise
<number
[]>
Defined in
packages/core/src/embeddings/GeminiEmbedding.ts:28
getQueryEmbedding
▸ getQueryEmbedding(query
): Promise
<number
[]>
Parameters
Name | Type |
---|---|
query | string |
Returns
Promise
<number
[]>
Overrides
BaseEmbedding.getQueryEmbedding
Defined in
packages/core/src/embeddings/GeminiEmbedding.ts:40
getTextEmbedding
▸ getTextEmbedding(text
): Promise
<number
[]>
Parameters
Name | Type |
---|---|
text | string |
Returns
Promise
<number
[]>
Overrides
BaseEmbedding.getTextEmbedding
Defined in
packages/core/src/embeddings/GeminiEmbedding.ts:36
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 |