diff --git a/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py b/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py
index 22507acbc..1186ab556 100644
--- a/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py
+++ b/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py
@@ -4,7 +4,7 @@
 
 import io
 
-from langchain.schema import BaseLanguageModel as BaseLangchainProvider
+from langchain.schema import BaseModel as BaseLangchainProvider
 from langchain.llms import (
     AI21,
     Anthropic,
diff --git a/packages/jupyter-ai-magics/pyproject.toml b/packages/jupyter-ai-magics/pyproject.toml
index a95720ecc..7bc027507 100644
--- a/packages/jupyter-ai-magics/pyproject.toml
+++ b/packages/jupyter-ai-magics/pyproject.toml
@@ -25,7 +25,7 @@ dependencies = [
     "ipython",
     "pydantic",
     "importlib_metadata~=5.2.0",
-    "langchain~=0.0.144"
+    "langchain==0.0.158"
 ]
 
 [project.optional-dependencies]
diff --git a/packages/jupyter-ai/pyproject.toml b/packages/jupyter-ai/pyproject.toml
index 350267d15..1179d44d9 100644
--- a/packages/jupyter-ai/pyproject.toml
+++ b/packages/jupyter-ai/pyproject.toml
@@ -28,7 +28,7 @@ dependencies = [
     "openai~=0.26",
     "aiosqlite~=0.18",
     "importlib_metadata~=5.2.0",
-    "langchain==0.0.153",
+    "langchain==0.0.158",
     "tiktoken", # required for OpenAIEmbeddings
     "jupyter_ai_magics",
     "ray~=2.4.0", # Requires grpcio installation from conda