Merge pull request #28 from ThirdSpaceLearning/improve-session-creation-action

run test session based on the branch

Author: redvex

.github/workflows/create_session.yaml

@@ -37,12 +37,28 @@ jobs:
         id: branch_name
         run: |
           echo "BRANCH_NAME=$(echo ${GITHUB_REF#refs/heads/})" >> $GITHUB_ENV
-          echo "TEST_SESSION_STATE_MACHINE_ARN=${{ secrets.TEST_SESSION_STATE_MACHINE_ARN }}" >> $GITHUB_ENV
           echo "AI_TUTOR_ID=${{ secrets.AI_TUTOR_ID }}" >> $GITHUB_ENV
           echo "START_SESSION=${{ github.event.inputs.start_session == 'yes' }}" >> $GITHUB_ENV
           echo "NOW=$(date -u +"%Y-%m-%dT%H:%M:%SZ")" >> $GITHUB_ENV
 
-      - name: configure aws credentials
+      - name: Set State machine for main branch
+        if: env.BRANCH_NAME == 'main'
+        run: echo "TEST_SESSION_STATE_MACHINE_ARN=${{ secrets.PROD_TEST_SESSION_STATE_MACHINE_ARN }}" >> $GITHUB_ENV
+      
+      - name: Set AWS Role for main branch
+        if: env.BRANCH_NAME == 'main'
+        uses: aws-actions/configure-aws-credentials@v4
+        with:
+          role-to-assume: arn:aws:iam::332632916255:role/prod-deployer
+          role-session-name: prompt-library-action
+          aws-region: eu-west-1
+
+      - name: Set State machine for main branch
+        if: env.BRANCH_NAME != 'main'
+        run: echo "TEST_SESSION_STATE_MACHINE_ARN=${{ secrets.NONPROD_TEST_SESSION_STATE_MACHINE_ARN }}" >> $GITHUB_ENV
+
+      - name: Set AWS Role for non-main branches
+        if: env.BRANCH_NAME != 'main'
         uses: aws-actions/configure-aws-credentials@v4
         with:
           role-to-assume: arn:aws:iam::267739230747:role/nonprod-deployer