Feat: Final changes to prompt
This commit is contained in:
25
agent/tools/secrets.py
Normal file
25
agent/tools/secrets.py
Normal file
@@ -0,0 +1,25 @@
|
||||
import boto3
|
||||
from botocore.exceptions import ClientError
|
||||
def get_secret():
|
||||
|
||||
secret_name = "dev/chatboteditais/apigateway/apikey"
|
||||
region_name = "us-east-1"
|
||||
|
||||
# Create a Secrets Manager client
|
||||
session = boto3.session.Session()
|
||||
client = session.client(
|
||||
service_name='secretsmanager',
|
||||
region_name=region_name
|
||||
)
|
||||
|
||||
try:
|
||||
get_secret_value_response = client.get_secret_value(
|
||||
SecretId=secret_name
|
||||
)
|
||||
except ClientError as e:
|
||||
# For a list of exceptions thrown, see
|
||||
# https://docs.aws.amazon.com/secretsmanager/latest/apireference/API_GetSecretValue.html
|
||||
raise e
|
||||
|
||||
secret = get_secret_value_response['SecretString']
|
||||
return secret
|
||||
Reference in New Issue
Block a user