engineering/opencompletion.com
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

support passing aws profile on cli

Browse source 
modified:   app.py
This commit is contained in:
Russell Ballestrini 2023-10-30 10:24:59 -04:00
parent 6a2db62801
commit 2a4630a415
1 changed files with 13 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

15
app.py
View file

@ -27,6 +27,13 @@ app.config["SQLALCHEMY_TRACK_MODIFICATIONS"] = False
db = SQLAlchemy(app)
# Create an argument parser for aws profile.
import argparse
parser = argparse.ArgumentParser(description='Script to perform some AWS operations')
parser.add_argument('--profile', help='AWS profile name')
args = parser.parse_args()
profile_name = args.profile
class Message(db.Model):
id = db.Column(db.Integer, primary_key=True)
username = db.Column(db.String(128), nullable=False)
@ -147,8 +154,12 @@ def chat_claude(username, room, message):
# append the new message.
chat_history += f"Human: {username}: {message}\n\nAssistant:"
# Initialize the Bedrock client using boto3
client = boto3.client("bedrock-runtime", region_name="us-east-1")
# Initialize the Bedrock client using boto3 and profile name.
if profile_name:
session = boto3.Session(profile_name=profile_name)
client = session.client("bedrock-runtime", region_name="us-east-1")
else:
client = boto3.client("bedrock-runtime", region_name="us-east-1")
# Define the request parameters
params = {