Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add Amazon Bedrock Integration #113

Closed
10 tasks done
Tracked by #6667
julian-risch opened this issue Dec 18, 2023 · 3 comments
Closed
10 tasks done
Tracked by #6667

Add Amazon Bedrock Integration #113

julian-risch opened this issue Dec 18, 2023 · 3 comments
Assignees
Labels
integration:amazon-bedrock new integration Discuss the creation of a new integration in Core P1

Comments

@julian-risch
Copy link
Member

julian-risch commented Dec 18, 2023

Summary and motivation

As a user, I'd like to use models on Amazon Bedrock in my Haystack 2.0 pipelines. This was brought up in deepset-ai/haystack#6545

Detailed design

A new generator class for Amazon Bedrock with the same features as the 1.x version, which was implemented as an invocation layer. All currently available models on Bedrock should be supported and streaming should be supported for all models that support streaming. For example, we should be able to load Llama 2 Chat 13B model:

from amazon_bedrock_haystack.generators import AmazonBedrockGenerator
generator = AmazonBedrockGenerator(model_name_or_path="meta.llama2-13b-chat-v1")

Related PRs for the support in Haystack 1.x were deepset-ai/haystack#6226 and deepset-ai/haystack#6406

Checklist

If the request is accepted, ensure the following checklist is complete before closing this issue.

@dfokina
Copy link
Contributor

dfokina commented Jan 5, 2024

@lambda-science
Copy link
Contributor

Excited to see AWS Bedrock in Haystack 2.0. The generator looks implemented, I was wondering if there was an issue to follow the implementation of the AmazonBedrockEmbedder (implmented in 1.x) :)

Thanks !

@julian-risch
Copy link
Member Author

The issue for the embedder is here: #181 but at the moment nobody is working on it. Just let us know in case you would like to contribute! 🙂

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
integration:amazon-bedrock new integration Discuss the creation of a new integration in Core P1
Projects
None yet
Development

No branches or pull requests

5 participants