bedrock_prompt_api_call.py

"""
Overview:

Make an API call to bedrock to get Prompt Results

Note: It uses IAM Role for making API call to AWS.

"""

import boto3

prompt_data = """Command: Write me a blog about making strong business decisions as a leader"""
AWS_REGION='us-east-1'

modelId = "amazon.titan-tg1-large"  
accept = "application/json"
contentType = "application/json"

bedrock_client = boto3.client(
 service_name='bedrock',
 region_name=AWS_REGION
)

body = json.dumps({"inputText": prompt_data})

response = bedrock_client.invoke_model(
    body=body, modelId=modelId, accept=accept, contentType=contentType
)
response_body = json.loads(response.get("body").read())

print(response_body.get("results")[0].get("outputText"))