Selecting embedding size in BedrockEmbedding titan v1 and v2, switching dimensions dynamically via keyword arguments using langchain

0

I am trying to provide dimension argument while trying to fetch the Titan version 1 and version 2 embedding models, but the BedrockEmbedding class does not have the feature of dimension, It only provideds the default 1024 dimension. How can we solve this?

Below is how i am calling BedrockEmbedding from langchain_community.embeddings import BedrockEmbeddings

asked 10 months ago499 views
1 Answer
1

Hi,

You don't have to provide the embeddings dimensions in the query: the model that you select (via its id) will return the number of dimensions that it is programmed for.

So, if you work with multiple embedding engines, your code must be ready to work with variable numbers of dimensions depending on the model that you query.

Best,

Didier

profile pictureAWS
EXPERT
answered 10 months ago
profile picture
EXPERT
reviewed 10 months ago
profile picture
EXPERT
reviewed 10 months ago
  • Currently, our solution selects the model_id and defaults the text embedding dimensions to 1536 for Titan version 1 and 1024 for Titan version 2. However, we aim to test our dataset with various embedding sizes, such as 256, 512, and 1024.

    How can we dynamically set the embedding dimension for both Titan version 1 and version 2 using the Langchain Community BedrockEmbedding? The model keyword arguments do not seem to accept dimension specifications.

    How do we dynamically select the embedding dimension for both Titan version 1 and version 2 during testing.

You are not logged in. Log in to post an answer.

A good answer clearly answers the question and provides constructive feedback and encourages professional growth in the question asker.

Guidelines for Answering Questions