diff --git a/jupyter-ai.yml b/jupyter-ai.yml
index f489500..68f8624 100644
--- a/jupyter-ai.yml
+++ b/jupyter-ai.yml
@@ -7,7 +7,7 @@ dependencies:
   - jupyter-resource-usage
   - jupyter-vscode-proxy
   - pypdf
-  - langchain-ollama
+  - langchain
   - streamlit
   - ibis-duckdb 
   - pip