diff --git a/jupyter-python/Dockerfile b/jupyter-python/Dockerfile
index eb316f5..a91478d 100644
--- a/jupyter-python/Dockerfile
+++ b/jupyter-python/Dockerfile
@@ -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"