Remove unnecessary comments

Author: abrookins

agent_memory_server/cli.py

@@ -215,7 +215,6 @@ def schedule_task(task_path: str, args: list[str]):
             sys.exit(1)
 
     async def setup_and_run_task():
-        # Initialize Redis connection
         await get_redis_conn()
 
         # Import the task function
@@ -284,8 +283,6 @@ async def _ensure_stream_and_group():
                 raise
 
     async def _run_worker():
-        # Ensure Redis stream/consumer group exists before starting worker
-        # Index will be created automatically when needed
         await _ensure_stream_and_group()
         await get_redis_conn()
         await Worker.run(

agent_memory_server/long_term_memory.py

@@ -611,9 +611,6 @@ async def compact_long_term_memories(
         index_name = Keys.search_index_name()
         logger.info(f"Using index '{index_name}' for semantic duplicate compaction.")
 
-        # Index will be created automatically when we add memories if it doesn't exist
-        # No need to check or create it explicitly
-
         # Get all memories using the vector store adapter
         try:
             # Convert filters to adapter format

agent_memory_server/main.py

@@ -75,7 +75,6 @@ async def lifespan(app: FastAPI):
         )
 
     # Set up Redis connection if long-term memory is enabled
-    # The VectorStore adapter will create the index automatically when needed
     if settings.long_term_memory:
         await get_redis_conn()
 

agent_memory_server/mcp.py

@@ -158,10 +158,9 @@ async def call_tool(self, name, arguments):
         return await super().call_tool(name, arguments)
 
     async def run_sse_async(self):
-        """Start SSE server. Index will be created automatically when needed."""
+        """Start SSE server."""
         from agent_memory_server.utils.redis import get_redis_conn
 
-        # Initialize Redis connection
         await get_redis_conn()
 
         # Run the SSE server using our custom implementation
@@ -173,10 +172,9 @@ async def run_sse_async(self):
         ).serve()
 
     async def run_stdio_async(self):
-        """Start STDIO MCP server. Index will be created automatically when needed."""
+        """Start STDIO MCP server."""
         from agent_memory_server.utils.redis import get_redis_conn
 
-        # Initialize Redis connection
         await get_redis_conn()
         return await super().run_stdio_async()
 

docker-compose-task-workers.yml

@@ -0,0 +1,82 @@
+services:
+  # For testing a production-like setup, you can run this API and the
+  # task-worker container. This API container does NOT use --no-worker, so when
+  # it starts background work, the task-worker will process those tasks.
+  api:
+    build:
+      context: .
+      dockerfile: Dockerfile
+    ports:
+      - "8000:8000"
+    environment:
+      - REDIS_URL=redis://redis:6379
+      - PORT=8000
+      # Add your API keys here or use a .env file
+      - OPENAI_API_KEY=${OPENAI_API_KEY}
+      - ANTHROPIC_API_KEY=${ANTHROPIC_API_KEY}
+      # Optional configurations with defaults
+      - LONG_TERM_MEMORY=True
+      - GENERATION_MODEL=gpt-4o-mini
+      - EMBEDDING_MODEL=text-embedding-3-small
+      - ENABLE_TOPIC_EXTRACTION=True
+      - ENABLE_NER=True
+    depends_on:
+      - redis
+    volumes:
+      - ./agent_memory_server:/app/agent_memory_server
+    healthcheck:
+      test: [ "CMD", "curl", "-f", "http://localhost:8000/v1/health" ]
+      interval: 30s
+      timeout: 10s
+      retries: 3
+    command: ["agent-memory", "api", "--host", "0.0.0.0", "--port", "8000"]
+
+
+  mcp:
+    build:
+      context: .
+      dockerfile: Dockerfile
+    environment:
+      - REDIS_URL=redis://redis:6379
+      - PORT=9050
+      # Add your API keys here or use a .env file
+      - OPENAI_API_KEY=${OPENAI_API_KEY}
+      - ANTHROPIC_API_KEY=${ANTHROPIC_API_KEY}
+    ports:
+      - "9050:9000"
+    depends_on:
+      - redis
+    command: ["agent-memory", "mcp", "--mode", "sse"]
+
+  task-worker:
+    build:
+      context: .
+      dockerfile: Dockerfile
+    environment:
+      - REDIS_URL=redis://redis:6379
+      # Add your API keys here or use a .env file
+      - OPENAI_API_KEY=${OPENAI_API_KEY}
+      - ANTHROPIC_API_KEY=${ANTHROPIC_API_KEY}
+      # Optional configurations with defaults
+    depends_on:
+      - redis
+    command: ["agent-memory", "task-worker"]
+    volumes:
+      - ./agent_memory_server:/app/agent_memory_server
+    restart: unless-stopped
+
+  redis:
+    image: redis:8
+    ports:
+      - "16380:6379" # Redis port
+    volumes:
+      - redis_data:/data
+    command: redis-server --save "30 1" --loglevel warning --appendonly no --stop-writes-on-bgsave-error no
+    healthcheck:
+      test: [ "CMD", "redis-cli", "ping" ]
+      interval: 30s
+      timeout: 10s
+      retries: 3
+
+volumes:
+  redis_data:

tests/test_memory_compaction.py

@@ -126,8 +126,6 @@ async def test_hash_deduplication_integration(
     # Clear all data to ensure clean test environment
     await async_redis_client.flushdb()
 
-    # Index will be created automatically when we add memories
-
     # Stub merge to return first memory unchanged
     async def dummy_merge(memories, llm_client=None):
         memory = memories[0]
@@ -226,8 +224,6 @@ async def test_semantic_deduplication_integration(
     # Clear all data to ensure clean test environment
     await async_redis_client.flushdb()
 
-    # Index will be created automatically when we add memories
-
     # Stub merge to return first memory
     async def dummy_merge(memories, llm_client=None):
         memory = memories[0]
@@ -302,8 +298,6 @@ async def test_full_compaction_integration(
     # Clear all data to ensure clean test environment
     await async_redis_client.flushdb()
 
-    # Index will be created automatically when we add memories
-
     async def dummy_merge(memories, llm_client=None):
         memory = memories[0]
         memory.memory_hash = generate_memory_hash(memory)