update flow for the docker image
This commit is contained in:
parent
b368ee707f
commit
f0db050274
3 changed files with 76 additions and 2 deletions
|
|
@ -4,7 +4,6 @@ AWS_REGION=${region:-eu-west-1}
|
||||||
AWS_DEPLOYMENT_ACCOUNT=${account:-463722570299}
|
AWS_DEPLOYMENT_ACCOUNT=${account:-463722570299}
|
||||||
AWS_REPOSITORY=${repo:-"${AWS_DEPLOYMENT_ACCOUNT}.dkr.ecr.${AWS_REGION}.amazonaws.com"}
|
AWS_REPOSITORY=${repo:-"${AWS_DEPLOYMENT_ACCOUNT}.dkr.ecr.${AWS_REGION}.amazonaws.com"}
|
||||||
|
|
||||||
|
|
||||||
STAGE=${stage:-"dev"}
|
STAGE=${stage:-"dev"}
|
||||||
SHA_SHORT="$(git rev-parse --short HEAD)"
|
SHA_SHORT="$(git rev-parse --short HEAD)"
|
||||||
CUR_DATE="$(date +%Y%m%d%H%M%S)"
|
CUR_DATE="$(date +%Y%m%d%H%M%S)"
|
||||||
|
|
|
||||||
|
|
@ -1,6 +1,6 @@
|
||||||
#!/bin/bash
|
#!/bin/bash
|
||||||
export ENVIRONMENT
|
export ENVIRONMENT
|
||||||
#python fetch_secret.py
|
python fetch_secret.py
|
||||||
|
|
||||||
# Start Gunicorn
|
# Start Gunicorn
|
||||||
gunicorn -w 2 -k uvicorn.workers.UvicornWorker -t 120 --bind=0.0.0.0:8000 --bind=0.0.0.0:443 --log-level debug api:app
|
gunicorn -w 2 -k uvicorn.workers.UvicornWorker -t 120 --bind=0.0.0.0:8000 --bind=0.0.0.0:443 --log-level debug api:app
|
||||||
75
level_2/fetch_secret.py
Normal file
75
level_2/fetch_secret.py
Normal file
|
|
@ -0,0 +1,75 @@
|
||||||
|
import os
|
||||||
|
from dotenv import load_dotenv
|
||||||
|
from api import start_api_server
|
||||||
|
|
||||||
|
# API_ENABLED = os.environ.get("API_ENABLED", "False").lower() == "true"
|
||||||
|
import boto3
|
||||||
|
|
||||||
|
environment = os.getenv("AWS_ENV", "dev")
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
def fetch_secret(secret_name, region_name, env_file_path):
|
||||||
|
session = boto3.session.Session()
|
||||||
|
client = session.client(service_name="secretsmanager", region_name=region_name)
|
||||||
|
|
||||||
|
try:
|
||||||
|
response = client.get_secret_value(SecretId=secret_name)
|
||||||
|
except Exception as e:
|
||||||
|
print(f"Error retrieving secret: {e}")
|
||||||
|
return None
|
||||||
|
|
||||||
|
if "SecretString" in response:
|
||||||
|
secret = response["SecretString"]
|
||||||
|
else:
|
||||||
|
secret = response["SecretBinary"]
|
||||||
|
|
||||||
|
with open(env_file_path, "w") as env_file:
|
||||||
|
env_file.write(secret)
|
||||||
|
|
||||||
|
if os.path.exists(env_file_path):
|
||||||
|
print(f"The .env file is located at: {os.path.abspath(env_file_path)}")
|
||||||
|
load_dotenv()
|
||||||
|
PINECONE_API_KEY = os.getenv("PINECONE_API_KEY", "")
|
||||||
|
|
||||||
|
print("LEN OF PINECONE_API_KEY", len(PINECONE_API_KEY))
|
||||||
|
else:
|
||||||
|
print("The .env file was not found.")
|
||||||
|
return "Success in loading env files"
|
||||||
|
|
||||||
|
|
||||||
|
env_file = ".env"
|
||||||
|
if os.path.exists(env_file):
|
||||||
|
# Load default environment variables (.env)
|
||||||
|
load_dotenv()
|
||||||
|
print("Talk to the AI!")
|
||||||
|
|
||||||
|
|
||||||
|
else:
|
||||||
|
secrets = fetch_secret(
|
||||||
|
f"promethai-{environment}-backend-secretso-promethaijs-dotenv",
|
||||||
|
"eu-west-1",
|
||||||
|
".env",
|
||||||
|
)
|
||||||
|
if secrets:
|
||||||
|
print(secrets)
|
||||||
|
load_dotenv()
|
||||||
|
|
||||||
|
|
||||||
|
# Check if "dev" is present in the task ARN
|
||||||
|
if "dev" in environment:
|
||||||
|
# Fetch the secret
|
||||||
|
secrets = fetch_secret(
|
||||||
|
f"promethai-dev-backend-secretso-promethaijs-dotenv",
|
||||||
|
"eu-west-1",
|
||||||
|
".env",
|
||||||
|
)
|
||||||
|
load_dotenv()
|
||||||
|
elif "prd" in environment:
|
||||||
|
# Fetch the secret
|
||||||
|
secrets = fetch_secret(
|
||||||
|
f"promethai-prd-backend-secretso-promethaijs-dotenv",
|
||||||
|
"eu-west-1",
|
||||||
|
".env",
|
||||||
|
)
|
||||||
|
load_dotenv()
|
||||||
Loading…
Add table
Reference in a new issue