Converse API with Amazon Bedrock API Key exmaple

converse_with_api_key.py

"""
In this example you get a Bedrock API key from Amazon Bedrock and load it.
It will look like: AWS_BEARER_TOKEN_BEDROCK=bedrock-api-key-.........
"""
import os
import boto3

from dotenv import load_dotenv

from pprint import pprint

# Load environment variables from a .env file
load_dotenv()
                        

# Create an Amazon Bedrock client
client = boto3.client(
    service_name="bedrock-runtime",
    region_name="us-east-1" # If you've configured a default region, you can omit this line
)

# Define the model and message
model_id = "us.anthropic.claude-3-5-haiku-20241022-v1:0"
messages = [{"role": "user", "content": [{"text": "Hello"}]}]

response = client.converse(
    modelId=model_id,
    messages=messages,
)


print(f"Agent response: {response['output']['message']['content'][0]['text']}")

print("\n")

pprint(f"Metrics: {response['metrics']}")

print(f"Stop reason: {response['stopReason']}")

pprint(f"Token usage: {response['usage']}")