chore(java-spring-ai-agents): improve setup and memory scripts with dynamic AWS queries

Author: Yuriy

apps/java-spring-ai-agents/scripts/01-setup.sh

@@ -67,14 +67,18 @@ else
         ~/environment/backoffice/src/main/resources/application.properties
 fi
 
-# Add basic variables if not present
-if ! grep -q "AWS_REGION=" ~/environment/.envrc 2>/dev/null; then
-    echo "export AWS_REGION=${AWS_REGION}" >> ~/environment/.envrc
+# Add basic variables to .envrc if not already in the environment
+if [ -z "${AWS_REGION}" ]; then
+    AWS_REGION=$(aws configure get region)
+fi
+if [ -z "${ACCOUNT_ID}" ]; then
+    ACCOUNT_ID=$(aws sts get-caller-identity --query Account --output text --no-cli-pager)
 fi
 
-if ! grep -q "ACCOUNT_ID=" ~/environment/.envrc 2>/dev/null; then
+grep -q "AWS_REGION=" ~/environment/.envrc 2>/dev/null || \
+    echo "export AWS_REGION=${AWS_REGION}" >> ~/environment/.envrc
+grep -q "ACCOUNT_ID=" ~/environment/.envrc 2>/dev/null || \
     echo "export ACCOUNT_ID=${ACCOUNT_ID}" >> ~/environment/.envrc
-fi
 
 ## Setting up direnv
 

apps/java-spring-ai-agents/scripts/02-memory.sh

@@ -5,43 +5,28 @@ echo "=============================================="
 echo "02-memory.sh - AgentCore Memory Setup"
 echo "=============================================="
 
-# Check if .envrc exists
-if [ ! -f ~/environment/.envrc ]; then
-    echo "Creating ~/environment/.envrc"
-    mkdir -p ~/environment
-    touch ~/environment/.envrc
-fi
-
 # Source existing environment
 source ~/environment/.envrc 2>/dev/null || true
 
 ## Creating the memory resource
 
-# Check if memory already exists
-if [ -n "${AGENTCORE_MEMORY_MEMORY_ID}" ]; then
+# Check if memory already exists in AWS
+AGENTCORE_MEMORY_MEMORY_ID=$(aws bedrock-agentcore-control list-memories --no-cli-pager \
+    --query "memories[?name=='aiagent_memory'].id | [0]" --output text 2>/dev/null || echo "")
+
+if [ -n "${AGENTCORE_MEMORY_MEMORY_ID}" ] && [ "${AGENTCORE_MEMORY_MEMORY_ID}" != "None" ]; then
     echo "Memory resource already exists: ${AGENTCORE_MEMORY_MEMORY_ID}"
     MEMORY_STATUS=$(aws bedrock-agentcore-control get-memory --memory-id "${AGENTCORE_MEMORY_MEMORY_ID}" \
         --no-cli-pager --query 'memory.status' --output text 2>/dev/null || echo "NOT_FOUND")
-
-    if [ "$MEMORY_STATUS" = "ACTIVE" ]; then
-        echo "Memory resource is ACTIVE, skipping creation"
-    elif [ "$MEMORY_STATUS" = "NOT_FOUND" ]; then
-        echo "Memory resource not found, will create new one"
-        unset AGENTCORE_MEMORY_MEMORY_ID
-    else
-        echo "Memory resource status: $MEMORY_STATUS"
-    fi
-fi
-
-if [ -z "${AGENTCORE_MEMORY_MEMORY_ID}" ]; then
+    echo "Memory resource status: ${MEMORY_STATUS}"
+else
     echo ""
     echo "## Creating the memory resource"
     echo "1. Create an AgentCore Memory resource and wait for it to become active (2-5 minutes)"
 
     AGENTCORE_MEMORY_MEMORY_ID=$(aws bedrock-agentcore-control create-memory \
         --name "aiagent_memory" --event-expiry-duration 7 \
         --no-cli-pager --query "memory.id" --output text)
-    echo "export AGENTCORE_MEMORY_MEMORY_ID=${AGENTCORE_MEMORY_MEMORY_ID}" >> ~/environment/.envrc
     echo "Created memory resource: ${AGENTCORE_MEMORY_MEMORY_ID}"
 
     echo -n "Waiting for memory"
@@ -93,8 +78,7 @@ echo "=============================================="
 echo "Memory setup complete!"
 echo "=============================================="
 echo ""
-echo "Environment variables saved to ~/environment/.envrc:"
-echo "  AGENTCORE_MEMORY_MEMORY_ID=${AGENTCORE_MEMORY_MEMORY_ID}"
+echo "Memory ID: ${AGENTCORE_MEMORY_MEMORY_ID}"
 echo ""
 echo "Application properties written to application.properties:"
 echo "  agentcore.memory.memory-id=${AGENTCORE_MEMORY_MEMORY_ID}"