diff --git a/docker-compose.yml b/docker-compose.yml
index 3a3df98..a34600d 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -418,6 +418,8 @@ services:
       - "com.centurylinklabs.watchtower.enable=true"
     ports:
       - "${AMS_PORT:-3004}:3004"
+    environment:
+      - MEMORY_DATABASE_URL=${MEMORY_DATABASE_URL:-postgresql://easyai:easyai2025@easyai-pgvector:5432/easyai_memory}
     networks:
       comfyai:
         ipv4_address: 172.21.0.16