Medical GraphRAG Assistant

#!/bin/bash # # Start NIM EC2 instance for demo/testing. # # This script: # 1. Starts the stopped EC2 instance # 2. Waits for it to be running # 3. Retrieves public IP # 4. Outputs NIM endpoint URL # # Usage: # ./scripts/aws/start-nim-ec2.sh set -e # Exit on error # Configuration INSTANCE_ID="${INSTANCE_ID:-i-xxxxxxxxxxxx}" # UPDATE or set INSTANCE_ID env var if [ "$INSTANCE_ID" = "i-xxxxxxxxxxxx" ]; then echo "❌ Error: INSTANCE_ID not set" echo "" echo "Set INSTANCE_ID environment variable:" echo " export INSTANCE_ID='i-xxxxxxxxxxxx'" echo "" echo "Or update scripts/aws/start-nim-ec2.sh with your instance ID" exit 1 fi echo "========================================" echo "Starting NIM EC2 Instance" echo "========================================" echo "" echo "Instance ID: $INSTANCE_ID" echo "" # Check current state CURRENT_STATE=$(aws ec2 describe-instances \ --instance-ids "$INSTANCE_ID" \ --query 'Reservations[0].Instances[0].State.Name' \ --output text) echo "Current state: $CURRENT_STATE" if [ "$CURRENT_STATE" = "running" ]; then echo "✅ Instance already running!" elif [ "$CURRENT_STATE" = "stopped" ]; then echo "" echo "Starting instance..." aws ec2 start-instances --instance-ids "$INSTANCE_ID" echo "Waiting for instance to start..." aws ec2 wait instance-running --instance-ids "$INSTANCE_ID" echo "✅ Instance started!" else echo "⚠️ Instance in unexpected state: $CURRENT_STATE" echo "Please check AWS console" exit 1 fi # Get public IP echo "" echo "Retrieving public IP..." PUBLIC_IP=$(aws ec2 describe-instances \ --instance-ids "$INSTANCE_ID" \ --query 'Reservations[0].Instances[0].PublicIpAddress' \ --output text) if [ "$PUBLIC_IP" = "None" ] || [ -z "$PUBLIC_IP" ]; then echo "❌ No public IP assigned" echo "Please check instance configuration" exit 1 fi # Wait for NIM container to be ready (startup time) echo "" echo "Waiting for NIM container to be ready (30 seconds)..." sleep 30 # Test endpoint echo "" echo "Testing NIM endpoint..." if curl -s --max-time 10 "http://$PUBLIC_IP:8000/health" > /dev/null 2>&1; then echo "✅ NIM endpoint is ready!" else echo "⚠️ NIM endpoint not ready yet (may need more time)" echo "Wait 30 seconds and try again:" echo " curl http://$PUBLIC_IP:8000/health" fi echo "" echo "========================================" echo "NIM Instance Ready!" echo "========================================" echo "" echo "Instance ID: $INSTANCE_ID" echo "Public IP: $PUBLIC_IP" echo "NIM Endpoint: http://$PUBLIC_IP:8000/v1/embeddings" echo "" echo "Set environment variables:" echo " export NIM_ENDPOINT=\"http://$PUBLIC_IP:8000/v1/embeddings\"" echo " export EMBEDDINGS_PROVIDER=\"nim\"" echo "" echo "Test endpoint:" echo " curl http://$PUBLIC_IP:8000/health" echo "" echo "Run vectorization:" echo " python src/setup/vectorize_documents.py" echo "" echo "Stop instance when done (save ~\$24/day):" echo " ./scripts/aws/stop-nim-ec2.sh"