Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add Amazon Bedrock Embedding function (#1361)
https://docs.aws.amazon.com/bedrock/latest/userguide/embeddings.html ## Description of changes - New functionality - Support Amazon Bedrock embedding function ## Test plan - [ ] Tests pass locally with `pytest` for python, `yarn test` for js Tested locally by given profile_name with appropreate `~/.aws/config` ```py >>> import boto3 >>> from chromadb.utils.embedding_functions import AmazonBedrockEmbeddingFunction >>> session = boto3.Session(profile_name="myprofile", region_name="us-east-1") >>> ef = AmazonBedrockEmbeddingFunction(session=session) >>> ef(["Hello Bedrock"]) [[-0.73046875, 0.390625, 0.24511719, 0.111816406, 0.83203125, 0.79296875,...,]] ``` ## Documentation Changes Written docstrings as much as possible. --------- Co-authored-by: Ben Eggers <[email protected]>
- Loading branch information