diff --git a/Dockerfile b/Dockerfile
index c91a270..8157b05 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -12,8 +12,8 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
 COPY pyproject.toml README.md ./
 COPY src/ ./src/
 
-# Install Python dependencies
-RUN pip install --no-cache-dir .
+# Install Python dependencies (including AI packages)
+RUN pip install --no-cache-dir ".[ai]"
 
 # Copy remaining files
 COPY migrations/ ./migrations/