Updates dependencies, imports

jupyterlab · Jan 9, 2024 · c4f6182 · c4f6182
1 parent 40563cd
commit c4f6182
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 3 deletions.
diff --git a/packages/jupyter-ai-magics/jupyter_ai_magics/embedding_providers.py b/packages/jupyter-ai-magics/jupyter_ai_magics/embedding_providers.py
@@ -8,7 +8,7 @@
     Field,
     MultiEnvAuthStrategy,
 )
-from langchain.embeddings import (
+from langchain_community.embeddings import (
     BedrockEmbeddings,
     CohereEmbeddings,
     GPT4AllEmbeddings,

diff --git a/packages/jupyter-ai-magics/pyproject.toml b/packages/jupyter-ai-magics/pyproject.toml
@@ -25,6 +25,7 @@ dependencies = [
     "importlib_metadata>=5.2.0",
     "langchain>=0.1.0,<=0.2",
     "langchain-core>=0.1.7,<0.2",
+    "langchain-community>=0.0.10,<0.1",
     "typing_extensions>=4.5.0",
     "click~=8.0",
     "jsonpath-ng>=1.5.3,<2",

diff --git a/packages/jupyter-ai/pyproject.toml b/packages/jupyter-ai/pyproject.toml
@@ -26,8 +26,6 @@ dependencies = [
     "jupyterlab~=4.0",
     "aiosqlite>=0.18",
     "importlib_metadata>=5.2.0",
-    "langchain>=0.1.0,<=0.2",
-    "langchain-core>=0.1.7,<0.2",
     "tiktoken",                  # required for OpenAIEmbeddings
     "jupyter_ai_magics",
     "dask[distributed]",