diff --git a/jupyter-python/Dockerfile b/jupyter-python/Dockerfile
index eb316f5..0d57128 100644
--- a/jupyter-python/Dockerfile
+++ b/jupyter-python/Dockerfile
@@ -1,7 +1,7 @@
 FROM alpine:3.19 AS build
 RUN --mount=type=cache,target=/var/cache/apk/,sharing=locked \
     apk update &&\
-    apk add python3 python3-dev pipx gcc g++ musl-dev linux-headers libffi-dev cargo cmake autoconf
+    apk add python3 python3-dev pipx gcc g++ musl-dev linux-headers libffi-dev cargo cmake autoconf swig
 RUN adduser jovyan -u 1000 --system -s /bin/sh -h /home/jovyan
 USER jovyan
 WORKDIR /home/jovyan
@@ -17,6 +17,9 @@ RUN venv/bin/pip install notebook
 RUN venv/bin/pip install ipywidgets
 RUN venv/bin/pip install jupyterlab-lsp
 RUN venv/bin/pip install python-lsp-server[yapf,rope,pyflakes] 
+RUN venv/bin/pip install jupyter-ai
+RUN venv/bin/pip install langchain-openai 
+
 #RUN venv/bin/pip install pylsp-mypy
 ENV PATH="/home/jovyan/venv/bin/:$PATH"