From 47818f4f41b6d7dbe31c4fab3e312870bec79709 Mon Sep 17 00:00:00 2001 From: "Brandon Hancock (bhancock_ai)" <109994880+bhancockio@users.noreply.github.com> Date: Mon, 10 Feb 2025 12:48:12 -0500 Subject: [PATCH] updating bedrock docs (#2088) Co-authored-by: Lorenze Jay <63378463+lorenzejay@users.noreply.github.com> --- docs/concepts/memory.mdx | 32 ++++++++++++++++++++++++++++++++ 1 file changed, 32 insertions(+) diff --git a/docs/concepts/memory.mdx b/docs/concepts/memory.mdx index cdb1745ee..ae65db290 100644 --- a/docs/concepts/memory.mdx +++ b/docs/concepts/memory.mdx @@ -447,6 +447,38 @@ my_crew = Crew( ) ``` +### Using Amazon Bedrock embeddings + +```python Code +# Note: Ensure you have installed `boto3` for Bedrock embeddings to work. + +import os +import boto3 +from crewai import Crew, Agent, Task, Process + +boto3_session = boto3.Session( + region_name=os.environ.get("AWS_REGION_NAME"), + aws_access_key_id=os.environ.get("AWS_ACCESS_KEY_ID"), + aws_secret_access_key=os.environ.get("AWS_SECRET_ACCESS_KEY") +) + +my_crew = Crew( + agents=[...], + tasks=[...], + process=Process.sequential, + memory=True, + embedder={ + "provider": "bedrock", + "config":{ + "session": boto3_session, + "model": "amazon.titan-embed-text-v2:0", + "vector_dimension": 1024 + } + } + verbose=True +) +``` + ### Adding Custom Embedding Function ```python Code