Determine embedding size with Titan Embedding v2 model

[bigbernnn]

Support for Titan Embedding v2 with changing embedding sizes based on this blog. This change lets the user specify dimension.

from langchain_aws.embeddings import BedrockEmbeddings

prompt_data = """Priority should be funding retirement through ROTH/IRA/401K over HSA extra.  
You need to fund your HSA for reasonable and expected medical expenses. """

bedrock_embedding_model_id = "amazon.titan-embed-text-v2:0"

embed_model = BedrockEmbeddings(model_id=bedrock_embedding_model_id)
response = embed_model.embed_documents([prompt_data], 256, True) 
response = embed_model.embed_query(prompt_data, 512, True)

[3coins]

Looks good!