Bedrock

Amazon Bedrock is a fully managed service that makes FMs from leading AI startups and Amazon available via an API, so you can choose from a wide range of FMs to find the model that is best suited for your use case.

%pip install boto3

from langchain.embeddings import BedrockEmbeddings

embeddings = BedrockEmbeddings(
    credentials_profile_name="bedrock-admin", region_name="us-east-1"
)

API Reference:

BedrockEmbeddings

embeddings.embed_query("This is a content of the document")

embeddings.embed_documents(["This is a content of the document", "This is another document"])

# async embed query
await embeddings.aembed_query("This is a content of the document")

# async embed documents
await embeddings.aembed_documents(["This is a content of the document", "This is another document"])