To use AWS Bedrock embedding models, you need the appropriate AWS credentials and permissions. Python uses boto3, and TypeScript uses @aws-sdk/client-bedrock-runtime.

Both SDKs support the Amazon Titan and Cohere embedding model families.

Setup

• Model access is automatic: Bedrock enables serverless foundation models on first invocation in AWS commercial regions, and the Model access page has been retired. Cohere models are served from AWS Marketplace, so an account's first invocation must come from a principal with the aws-marketplace:Subscribe permission; after that, any user in the account can invoke them. Browse the models available to you in the Bedrock model catalog.

• Install the AWS client for your language:

  <CodeGroup> ```bash Python pip install boto3 ```
  bashCopy

  npm install @aws-sdk/client-bedrock-runtime
  

  </CodeGroup>

  In TypeScript this package is an optional peer dependency, so it is only required when you actually use the Bedrock embedder.

• Set up environment variables for authentication:

  bashCopy

  export AWS_REGION=us-east-1
  export AWS_ACCESS_KEY_ID=your-access-key
  export AWS_SECRET_ACCESS_KEY=your-secret-key
  

Both SDKs fall back to the standard AWS credential chain (environment variables, shared config, SSO, or an instance role) when you do not pass credentials in the config, so you rarely need to hardcode keys. See the boto3 credentials guide for the Python resolution order.

Usage

For LLM if needed

os.environ["OPENAI_API_KEY"] = "your-openai-api-key"

AWS credentials

os.environ["AWS_REGION"] = "us-west-2" os.environ["AWS_ACCESS_KEY_ID"] = "your-access-key" os.environ["AWS_SECRET_ACCESS_KEY"] = "your-secret-key"

config = { "embedder": { "provider": "aws_bedrock", "config": { "model": "amazon.titan-embed-text-v2:0" } } }

m = Memory.from_config(config) messages = [ {"role": "user", "content": "I'm planning to watch a movie tonight. Any recommendations?"}, {"role": "assistant", "content": "How about thriller movies? They can be quite engaging."}, {"role": "user", "content": "I'm not a big fan of thriller movies but I love sci-fi movies."}, {"role": "assistant", "content": "Got it! I'll avoid thriller recommendations and suggest sci-fi movies in the future."} ] m.add(messages, user_id="alice")

```typescript TypeScript
import { Memory } from "mem0ai/oss";

// Credentials are read from the AWS default chain (AWS_REGION,
// AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY, SSO, or an instance role).
const memory = new Memory({
  embedder: {
    provider: "aws_bedrock",
    config: {
      model: "amazon.titan-embed-text-v2:0",
      awsRegion: "us-west-2",
    },
  },
});

const messages = [
  { role: "user", content: "I'm planning to watch a movie tonight. Any recommendations?" },
  { role: "assistant", content: "How about thriller movies? They can be quite engaging." },
  { role: "user", content: "I'm not a big fan of thriller movies but I love sci-fi movies." },
  { role: "assistant", content: "Got it! I'll avoid thriller recommendations and suggest sci-fi movies in the future." },
];
await memory.add(messages, { userId: "alice" });

Choosing a model

Custom output sizes are model specific. In Python, only Titan Text Embeddings V2 accepts one. In TypeScript, Titan Text Embeddings V2 and Cohere Embed v4 both do, and embeddingDims is ignored on Titan V1 and on Cohere v3, which have no such parameter. When you do set it, make sure your vector store dimension matches, otherwise inserts will fail.

Bedrock caps a Cohere embedding call at 96 texts. The TypeScript SDK splits larger batches into multiple requests for you, so a 200 text batch becomes 3 calls.

Config

Here are the parameters available for configuring AWS Bedrock embedder:

Omit the three credential fields to use the AWS default credential chain. If you do pass them, awsAccessKeyId and awsSecretAccessKey are both required. </Tab> </Tabs>