diff --git a/docs/environment.yml b/docs/environment.yml
index 7d3f049bb..5f0e05304 100644
--- a/docs/environment.yml
+++ b/docs/environment.yml
@@ -12,6 +12,7 @@ dependencies:
   - comm>=0.1.2,<0.2.0
   - pydantic>=2,<3
   - sidecar
+  - jupyterlite-ai
   - pip:
       - yjs-widgets>=0.4,<0.5
       - my-jupyter-shared-drive<0.2.0