108 lines
3.7 KiB
Bash
108 lines
3.7 KiB
Bash
#!/usr/bin/env bash
|
|
# 08-memorybank-setup.sh — Create Vertex AI Agent Engine (Memory Bank) instance
|
|
# Idempotent — safe to run multiple times
|
|
#
|
|
# KNOWN LIMITATION (OQ-01):
|
|
# Memory Bank is Python SDK only — no gcloud CLI equivalent.
|
|
# Requires: google-cloud-aiplatform >= 1.111.0
|
|
# Install: pip install google-cloud-aiplatform>=1.111.0
|
|
#
|
|
# KNOWN LIMITATION (OQ-06):
|
|
# Memory Bank does NOT support auto-discovery via gcloud.
|
|
# The engine resource name must be manually set in .env as MEMORY_ENGINE_NAME.
|
|
#
|
|
# Source .env before running: source .env
|
|
|
|
set -euo pipefail
|
|
|
|
: "${PROJECT_ID:?Set PROJECT_ID}"
|
|
: "${REGION:?Set REGION}"
|
|
: "${AGENT_SA:?Set AGENT_SA}"
|
|
|
|
MEMORY_INSTANCE_DISPLAY_NAME="${MEMORY_INSTANCE_DISPLAY_NAME:-${PROJECT_ID}-memory-bank}"
|
|
|
|
echo "=== 08: Setting up Vertex AI Memory Bank (Agent Engine) ==="
|
|
echo " Project : ${PROJECT_ID}"
|
|
echo " Region : ${REGION}"
|
|
echo " Instance : ${MEMORY_INSTANCE_DISPLAY_NAME}"
|
|
echo ""
|
|
|
|
bash "$(dirname "$0")/00-authcheck.sh"
|
|
|
|
gcloud services enable aiplatform.googleapis.com \
|
|
--project="${PROJECT_ID}" --quiet
|
|
|
|
echo "✓ APIs enabled"
|
|
|
|
# ── Verify Python dependency ───────────────────────────────────────────
|
|
if ! python3 -c "import vertexai; from vertexai.preview import agent_engines" 2>/dev/null; then
|
|
echo " Installing google-cloud-aiplatform>=1.111.0 ..."
|
|
pip install --quiet "google-cloud-aiplatform>=1.111.0" 2>&1 | tail -3
|
|
fi
|
|
echo "✓ Python SDK available"
|
|
|
|
# ── Create or find Memory Bank instance via Python SDK ─────────────────────
|
|
python3 - << PYEOF
|
|
import os, sys
|
|
|
|
try:
|
|
import vertexai
|
|
from vertexai.preview import agent_engines
|
|
except ImportError as e:
|
|
print(f"ERROR: Failed to import agent_engines: {e}")
|
|
print("Run: pip install google-cloud-aiplatform>=1.111.0")
|
|
sys.exit(1)
|
|
|
|
PROJECT_ID = os.environ["PROJECT_ID"]
|
|
REGION = os.environ["REGION"]
|
|
DISP_NAME = os.environ.get("MEMORY_INSTANCE_DISPLAY_NAME", f"{PROJECT_ID}-memory-bank")
|
|
|
|
vertexai.init(project=PROJECT_ID, location=REGION)
|
|
|
|
# Check if instance already exists
|
|
existing = None
|
|
try:
|
|
for eng in agent_engines.list():
|
|
if hasattr(eng, "display_name") and eng.display_name == DISP_NAME:
|
|
existing = eng
|
|
print(f"✓ Memory Bank instance already exists: {eng.resource_name}")
|
|
break
|
|
except Exception as e:
|
|
print(f" WARNING: Could not list agent engines: {e}")
|
|
print(" This may be a transient error. Proceeding with create attempt.")
|
|
|
|
if existing is None:
|
|
try:
|
|
instance = agent_engines.create(
|
|
display_name=DISP_NAME,
|
|
)
|
|
print(f"✓ Memory Bank instance created: {instance.resource_name}")
|
|
existing = instance
|
|
except Exception as e:
|
|
print(f"ERROR: Failed to create Memory Bank instance: {e}")
|
|
print("")
|
|
print("Possible causes:")
|
|
print(" 1. SDK version too old — run: pip install google-cloud-aiplatform>=1.111.0")
|
|
print(" 2. REGION not supported — try REGION=us-central1")
|
|
print(" 3. Quota not enabled — check: https://console.cloud.google.com/iam-admin/quotas")
|
|
sys.exit(1)
|
|
|
|
if existing:
|
|
resource_name = existing.resource_name
|
|
with open("/tmp/memory_engine_name.txt", "w") as f:
|
|
f.write(resource_name)
|
|
|
|
print(f"")
|
|
print(f" Memory Bank resource name : {resource_name}")
|
|
print(f"")
|
|
print(f" ACTION REQUIRED — add to .env:")
|
|
print(f" MEMORY_ENGINE_NAME={resource_name}")
|
|
PYEOF
|
|
|
|
echo ""
|
|
echo "=== 08: Memory Bank setup COMPLETE ==="
|
|
echo ""
|
|
echo " NEXT STEP: Copy the MEMORY_ENGINE_NAME value above into your .env file"
|
|
echo " Then run: bash infrastructure/05-cloudrun-deploy.sh"
|
|
echo ""
|