-
-
Notifications
You must be signed in to change notification settings - Fork 341
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
Migrate to langchain-aws
for AWS providers
#909
Conversation
6ffa205
to
a477538
Compare
packages/jupyter-ai-magics/jupyter_ai_magics/partner_providers/aws.py
Outdated
Show resolved
Hide resolved
Co-authored-by: Jason Weill <[email protected]>
packages/jupyter-ai-magics/jupyter_ai_magics/partner_providers/aws.py
Outdated
Show resolved
Hide resolved
packages/jupyter-ai-magics/jupyter_ai_magics/partner_providers/aws.py
Outdated
Show resolved
Hide resolved
Co-authored-by: Piyush Jain <[email protected]>
for more information, see https://pre-commit.ci
for more information, see https://pre-commit.ci
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Reviewed the code, tested examples for chat, learn, ask, export, using different LLMs such as Claude and Mistral, which are moved to aws.py
.
One small non-blocking (non-regressive) issue relating to the SmEndpointProvider
class. Tried it with Llama3.1 and it works in the magics,
But throws an error in the chat invocation.
This may be related to the new Llama3.1 models and to be further explored.
* migrate to langchain-aws * pre-commit * update aws provider dependencies in docs * correct SM endpoints docs URL Co-authored-by: Jason Weill <[email protected]> * add new Cohere model IDs to BedrockEmbeddings Co-authored-by: Piyush Jain <[email protected]> * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * use BedrockLLM instead of Bedrock class * add Amazon, Meta, Mistral models to BedrockChatProvider * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --------- Co-authored-by: Jason Weill <[email protected]> Co-authored-by: Piyush Jain <[email protected]> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Description
langchain-aws
package.Demo
Screen.Recording.2024-07-24.at.3.39.54.PM.mov