0f0addb36b
Pin mem0ai[nlp]==2.0.2 and fastembed for the new hybrid-search pipeline. Drop OSS graph memory (removed upstream in 2.0.0, PR #4805): remove Neo4j service, env vars, volumes, and driver deps; mark /graph/relationships deprecated. Rewrite Memory.search/get_all/chat/health call sites to use the v2 filters={} + top_k API (entity IDs at top level now raise ValueError). Tighten MCP remove_memory ownership check to O(1) verify_memory_ownership so it doesn't silently truncate at the new top_k=20 default. Downgrade base image to python:3.12-slim for spaCy. Adds scripts/migrate_qdrant_to_v3.py (scroll+upsert with per-user count parity check) and docs/MIGRATION_RUNBOOK.md covering snapshot, dump, collection rebuild, cutover, and rollback procedures.
122 lines
3.2 KiB
Bash
Executable file
122 lines
3.2 KiB
Bash
Executable file
#!/bin/bash
|
|
|
|
set -e
|
|
|
|
COMPOSE_PROJECT_NAME="${COMPOSE_PROJECT_NAME:-mem0}"
|
|
VOLUMES=("qdrant_data")
|
|
|
|
echo "=========================================="
|
|
echo " Mem0 Setup Script"
|
|
echo "=========================================="
|
|
echo ""
|
|
|
|
check_volumes_exist() {
|
|
local exists=false
|
|
for vol in "${VOLUMES[@]}"; do
|
|
full_name="${COMPOSE_PROJECT_NAME}_${vol}"
|
|
if docker volume ls -q | grep -q "^${full_name}$"; then
|
|
exists=true
|
|
break
|
|
fi
|
|
done
|
|
echo "$exists"
|
|
}
|
|
|
|
list_existing_volumes() {
|
|
echo "Existing volumes:"
|
|
for vol in "${VOLUMES[@]}"; do
|
|
full_name="${COMPOSE_PROJECT_NAME}_${vol}"
|
|
if docker volume ls -q | grep -q "^${full_name}$"; then
|
|
size=$(docker system df -v 2>/dev/null | grep "$full_name" | awk '{print $4}' || echo "unknown")
|
|
echo " - $full_name ($size)"
|
|
fi
|
|
done
|
|
}
|
|
|
|
remove_volumes() {
|
|
echo "Stopping containers..."
|
|
docker compose down 2>/dev/null || true
|
|
|
|
echo "Removing volumes..."
|
|
for vol in "${VOLUMES[@]}"; do
|
|
full_name="${COMPOSE_PROJECT_NAME}_${vol}"
|
|
if docker volume ls -q | grep -q "^${full_name}$"; then
|
|
echo " Removing $full_name..."
|
|
docker volume rm "$full_name" 2>/dev/null || true
|
|
fi
|
|
done
|
|
echo "Volumes removed."
|
|
}
|
|
|
|
build_and_start() {
|
|
echo ""
|
|
echo "Building containers..."
|
|
docker compose build
|
|
|
|
echo ""
|
|
echo "Starting services..."
|
|
docker compose up -d
|
|
|
|
echo ""
|
|
echo "Waiting for services to be healthy..."
|
|
sleep 5
|
|
|
|
echo ""
|
|
echo "Checking health..."
|
|
curl -s http://localhost:8000/health 2>/dev/null | jq . || echo "Health check not available yet. Services may still be starting."
|
|
|
|
echo ""
|
|
echo "=========================================="
|
|
echo " Setup Complete!"
|
|
echo "=========================================="
|
|
echo ""
|
|
echo "Services:"
|
|
echo " - Backend API: http://localhost:8000"
|
|
echo " - API Docs: http://localhost:8000/docs"
|
|
echo " - Health Check: http://localhost:8000/health"
|
|
echo ""
|
|
echo "Logs: docker compose logs -f backend"
|
|
}
|
|
|
|
if [ "$(check_volumes_exist)" = "true" ]; then
|
|
echo "Existing data volumes detected!"
|
|
echo ""
|
|
list_existing_volumes
|
|
echo ""
|
|
echo "Options:"
|
|
echo " 1) Keep existing data and start services"
|
|
echo " 2) Reset everything (DELETE ALL DATA) and start fresh"
|
|
echo " 3) Exit"
|
|
echo ""
|
|
read -p "Choose an option [1/2/3]: " choice
|
|
|
|
case $choice in
|
|
1)
|
|
echo ""
|
|
echo "Keeping existing data..."
|
|
build_and_start
|
|
;;
|
|
2)
|
|
echo ""
|
|
read -p "Are you sure you want to DELETE ALL DATA? Type 'yes' to confirm: " confirm
|
|
if [ "$confirm" = "yes" ]; then
|
|
remove_volumes
|
|
build_and_start
|
|
else
|
|
echo "Aborted."
|
|
exit 1
|
|
fi
|
|
;;
|
|
3)
|
|
echo "Exiting."
|
|
exit 0
|
|
;;
|
|
*)
|
|
echo "Invalid option. Exiting."
|
|
exit 1
|
|
;;
|
|
esac
|
|
else
|
|
echo "No existing volumes found. Starting fresh setup..."
|
|
build_and_start
|
|
fi
|