Skip to content

Commit

Permalink
fix: bedrock issues (#544)
Browse files Browse the repository at this point in the history
* Removed extraneous key [stream] for AWS Bedrock Claude models.

* Reduceddefault  AWS Bedrock llama-3 max_output_tokens to 2048 to align with API requirements.

---------

Co-authored-by: Rolf Wilms <rwilms@csc.com>
  • Loading branch information
rolfwilms and Rolf Wilms committed May 25, 2024
1 parent 5458150 commit 5693177
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 2 deletions.
4 changes: 2 additions & 2 deletions models.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -342,13 +342,13 @@
supports_vision: true
- name: meta.llama3-8b-instruct-v1:0
max_input_tokens: 8192
max_output_tokens: 4096
max_output_tokens: 2048
pass_max_tokens: true
input_price: 0.4
output_price: 0.6
- name: meta.llama3-70b-instruct-v1:0
max_input_tokens: 8192
max_output_tokens: 4096
max_output_tokens: 2048
pass_max_tokens: true
input_price: 2.65
output_price: 3.5
Expand Down
1 change: 1 addition & 0 deletions src/client/bedrock.rs
Original file line number Diff line number Diff line change
Expand Up @@ -217,6 +217,7 @@ fn build_body(data: SendData, model: &Model, model_category: &ModelCategory) ->
let mut body = claude_build_body(data, model)?;
if let Some(body_obj) = body.as_object_mut() {
body_obj.remove("model");
body_obj.remove("stream");
}
body["anthropic_version"] = "bedrock-2023-05-31".into();
Ok(body)
Expand Down

0 comments on commit 5693177

Please sign in to comment.