fixes

Dockerfile

@@ -19,8 +19,10 @@ RUN apt-get update && \
     rm -rf /var/lib/apt/lists/*
 
 COPY demo/requirements.txt demo/requirements.txt
+COPY mcp mcp
 RUN pip install --upgrade pip && \
-    pip install --pre -r demo/requirements.txt
+    pip install --pre -r demo/requirements.txt && \
+    pip install -e ./mcp
 
 COPY demo demo
 

demo/health.py

@@ -1,7 +1,6 @@
 from __future__ import annotations
 
 import html
-import importlib.util
 import re
 import os
 import shutil
@@ -10,6 +9,7 @@ import asyncio
 import sys
 from pathlib import Path
 import logging
+import importlib.util
 from dataclasses import dataclass
 from itertools import zip_longest
 from typing import Iterable
@@ -120,18 +120,19 @@ def _check_mcp_health() -> ToolStatus:
     env = os.environ.copy()
     env["PYTHONPATH"] = py_path
 
-    server_path = mcp_src / "aileen3_mcp" / "server.py"
+    server_entry = ["-m", "aileen3_mcp.server"]
+
     logging.warning(
         "MCP probe spawning server: cmd=%s args=%s PYTHONPATH=%s cwd=%s",
         sys.executable,
-        [str(server_path)],
+        server_entry,
         py_path,
         repo_root,
     )
 
     transport = StdioTransport(
         command=sys.executable,
-        args=[str(server_path)],
+        args=server_entry,
         env=env,
         cwd=str(repo_root),
     )