Amazon Bedrock invocation example in Python

bedrock.py

import json
import boto3

my_session = boto3.session.Session()
my_region = my_session.region_name

# check you are in us-east-1 region (for now)
print(my_region)

bedrock_runtime = boto3.client('bedrock-runtime')
prompt = "Hello, how are you today?" 

payload = json.dumps({
  "inputText": prompt,
  "textGenerationConfig" : { 
        "maxTokenCount": 512,
        "stopSequences": [],
        "temperature": 0.1,  
        "topP": 0.9
    }
})

print(payload)

response = bedrock_runtime.invoke_model(
  body=payload,
  modelId="amazon.titan-tg1-large",
  accept="application/json", 
  contentType="application/json"
)

response_body = json.loads(response['body'].read())
print(response_body['results'][0]['outputText'])