diff --git a/packages/jupyter-ai-magics/jupyter_ai_magics/embedding_providers.py b/packages/jupyter-ai-magics/jupyter_ai_magics/embedding_providers.py index adabd97a8..24922b2a8 100644 --- 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 index 57fc6771b..d7e220f1b 100644 --- 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 index 484203114..073abb8de 100644 --- 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]",