Python script to write SQL using Amazon Bedrock Generative AI

#=====================================================================
# Bedrock1.py:
# Usage: python3 Bedrock1.py
# Usage: python3 Bedrock1.py WriteSQL "top ten customers by sales"
# Usage: python3 Bedrock1.py WriteSQL "get the top ten customers by sales but print all fields from account object include costofliving"
#=====================================================================
import boto3
import json
import base64
import sys

# Global Variables
client = boto3.client("bedrock-runtime")
strBedrockConfig =  " "maxTokens":200,"temperature":0.7,"topP":1,"stopSequences":[]," + 
" "countPenalty":{"scale":0}," + 
" "presencePenalty":{"scale":0}," + 
" "frequencyPenalty":{"scale":0}"

def main():
WriteSQL(sys.argv[1])
# End main()

def WriteSQL(lstArgs):
strQuestion = lstArgs[1]
strQuestion = strQuestion.replace("n","")
strQuestion = "write sql for " + strQuestion
input_data = {
"modelId":"ai21.j2-mid-v1",
"contentType":"application/json",
"accept":"*/*",
"body":"{"prompt":"" + strQuestion + "", " + strBedrockConfig + "}"
}
response = client.invoke_model(contentType='application/json',body=input_data['body'], modelId=input_data['modelId'])
Data = json.loads(response['body'].read().decode('utf-8'))
print(Data['completions'][0]['data']['text'])
# End WriteSQL()


if __name__ == "__main__":
main()