- For a Mistral model with generally good performance, look at
mistral.mistral-large-2402-v1:0
. - You can play with Amazon Nova models. Use
amazon.nova-pro-v1:0
for general purpose tasks. - For Claude models, see our Claude integration.
Async usage of AWS Bedrock is not yet supported. When using
AwsBedrock
with
an Agent
, you can only use agent.run
and agent.print_response
.Authentication
Set yourAWS_ACCESS_KEY_ID
, AWS_SECRET_ACCESS_KEY
and AWS_REGION
environment variables.
Get your keys from here.
Example
UseAwsBedrock
with your Agent
:
View more examples here.
Parameters
Parameter | Type | Default | Description |
---|---|---|---|
id | str | "mistral.mistral-small-2402-v1:0" | The specific model ID used for generating responses. |
name | str | "AwsBedrock" | The name identifier for the AWS Bedrock agent. |
provider | str | "AwsBedrock" | The provider of the model. |
aws_sso_auth | Optional[bool] | False | Remove the need for access and secret keys by leveraging the current profile’s authentication. |
aws_region | Optional[str] | None | The AWS region to use for API requests. |
aws_access_key_id | Optional[str] | None | The AWS access key ID to use for authentication. |
aws_secret_access_key | Optional[str] | None | The AWS secret access key to use for authentication. |
session | Optional[Session] | None | A boto3 Session object to use for authentication. |
max_tokens | Optional[int] | None | The maximum number of tokens to generate in the response. |
temperature | Optional[float] | None | The sampling temperature to use, between 0 and 2. Higher values like 0.8 make the output more random, while lower values like 0.2 make it more focused and deterministic. |
top_p | Optional[float] | None | The nucleus sampling parameter. The model considers the results of the tokens with top_p probability mass. |
stop_sequences | Optional[List[str]] | None | A list of sequences where the API will stop generating further tokens. |
request_params | Optional[Dict[str, Any]] | None | Additional parameters for the request, provided as a dictionary. |
client_params | Optional[Dict[str, Any]] | None | Additional client parameters for initializing the AwsBedrock client, provided as a dictionary. |
client | Optional[AwsClient] | None | A pre-configured AWS client instance. |
AwsBedrock
is a subclass of the Model class and has access to the same params.