-
-
Notifications
You must be signed in to change notification settings - Fork 687
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
bedrockclient: add support for new Amazon Nova models #1083
base: main
Are you sure you want to change the base?
Conversation
This commit implements support of the new Amazon Nova models: * Nova Micro * Nova Lite * Nova Pro
Hi @tmc, any chance to get a review for these changes? (: |
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.
Looks very good overall.
@@ -39,6 +39,39 @@ const ( | |||
// Languages: English (GA), Multilingual in 100+ languages (Preview). | |||
ModelAmazonTitanTextExpressV1 = "amazon.titan-text-express-v1" | |||
|
|||
// Amazon Nova Micro is a text only model that delivers the lowest latency responses at very low cost. | |||
// It is highly performant at language understanding, translation, reasoning, code completion, | |||
// brainstorming, and mathematical problem-solving. With its generation speed of over 200 tokens per second, |
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.
Maybe the language in these comments sound a little biased? I also think they could be shorter and focus more on the differences between the models.
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.
This can be put in a separate folder called testdata.
content := output.Output.Message.Content | ||
if len(content) == 0 { | ||
return nil, errors.New("no results") | ||
} else if stopReason := output.StopReason; stopReason != NovaCompletionReasonEndTurn && stopReason != NovaCompletionReasonStopSequence { |
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.
nit: the else is unnecessary.
PR Checklist
memory: add interfaces for X, Y
orutil: add whizzbang helpers
).Fixes #123
).golangci-lint
checks.This PR adds support for the new Amazon Nova models
The implementation is very similar to the anthropic provider (since they share some common structures and constraints)