Merge branch 'master' into patch-1

intuit · Oct 17, 2023 · 975fc67 · 975fc67
2 parents b2b8a1f + 4dbb8d9
commit 975fc67
Show file tree

Hide file tree

Showing 2 changed files with 33 additions and 25 deletions.
diff --git a/install.py b/install.py
@@ -2,53 +2,60 @@
 
 import subprocess
 
+def execute_subprocess(command_list):
+    print(f'Executing command list: {command_list}')
+    try:
+        subprocess.run(command_list, shell=False, check=True)
+    except subprocess.CalledProcessError as e:
+        print(f'Error: {e}')
+        raise Exception(e)
+
 
 def main():
-    package_command = 'sam package --template-file template.yaml --output-template-file deploy.yaml'
-    deploy_command = 'sam deploy --template-file deploy.yaml --capabilities CAPABILITY_NAMED_IAM '
+    package_command_list = ['sam', 'package', '--template-file', 'template.yaml', '--output-template-file', 'deploy.yaml']
+    deploy_command_list = ['sam', 'deploy', '--template-file', 'deploy.yaml', '--capabilities', 'CAPABILITY_NAMED_IAM']
     s3_bucket = input('Enter the s3 bucket name created as part of pre-requisite: ')
     if s3_bucket:
-        package_command = package_command + ' --s3-bucket ' + s3_bucket
+        package_command_list.append('--s3-bucket')
+        package_command_list.append(s3_bucket)
     region = input('Enter the region [for instance, us-west-2]: ')
     if region:
-        package_command = package_command + ' --region ' + region
-        deploy_command = deploy_command + ' --region ' + region
+        package_command_list.append('--region')
+        package_command_list.append(region)
+
+        deploy_command_list.append('--region')
+        deploy_command_list.append(region)
         print()
-        print ('packaging trapheus for use ...')
-        try:
-            subprocess.run(package_command, shell=True, check=True)
-        except subprocess.CalledProcessError as e:
-            print(f'Error: {e}')
+        print ('packaging trapheus for use...')
+        execute_subprocess(package_command_list)
 
 
     stack_name = input('Enter a stack name: ')
     if stack_name:
-        deploy_command = deploy_command + ' --stack-name ' + stack_name
-    deploy_command = deploy_command + ' --parameter-overrides'
+        deploy_command_list.append('--stack-name')
+        deploy_command_list.append(stack_name)
+        deploy_command_list.append('--parameter-overrides')
     sender_email = input('Enter sender email to send email FROM in case of failure: ')
     if sender_email:
-        deploy_command = deploy_command + ' SenderEmail=' + sender_email
-    recipient_email = input('Enter recipeint email to send email TO in case of failure: ')
+        deploy_command_list.append('SenderEmail=' + sender_email)
+    recipient_email = input('Enter recipient email to send email TO in case of failure: ')
     if recipient_email:
-        deploy_command = deploy_command + ' RecipientEmail=' + recipient_email
+        deploy_command_list.append('RecipientEmail=' + recipient_email)
     slack_webhook_urls = input('Enter slack webhooks to publish failure notifications to: ')
     if slack_webhook_urls:
-        deploy_command = deploy_command + ' --SlackWebhookUrls=' + slack_webhook_urls
+        deploy_command_list.append('--SlackWebhookUrls=' + slack_webhook_urls)
     vpc = input('Are ypu using vpc[y/n]: ')
     if vpc == 'y' or vpc == 'Y':
-        deploy_command = deploy_command + ' UseVPCAndSubnets=true'
+        deploy_command_list.append('UseVPCAndSubnets=true')
         vpc_id = input('Enter vpc ID: ')
         if vpc_id:
-            deploy_command = deploy_command + ' vpcId=' + vpc_id
-            subnets = input('Enter comman seperated list of PRIVATE subnets: ')
+            deploy_command_list.append('vpcId=' + vpc_id)
+            subnets = input('Enter comma seperated list of PRIVATE subnets: ')
             if subnets:
-                deploy_command = deploy_command + ' Subnets=' +subnets
+                deploy_command_list.append('Subnets=' +subnets)
     print()
-    print('Deploying trapheus to AWS ...')
-    try:
-        subprocess.run(deploy_command, shell=True, check=True)
-    except subprocess.CalledProcessError as e:
-        print(f'Error: {e}')
+    print('Deploying trapheus to AWS...')
+    execute_subprocess(deploy_command_list)
 
 
 if __name__ == "__main__":

diff --git a/labs/TrapheusAI/requirements.txt b/labs/TrapheusAI/requirements.txt
@@ -15,3 +15,4 @@ streamlit-agraph
 streamlit-extras
 pandasai
 tk
+safetensors