From c4f618207fc89a29259c294e8e23a48aa52e9745 Mon Sep 17 00:00:00 2001 From: Jason Weill Date: Tue, 9 Jan 2024 14:37:22 -0800 Subject: [PATCH] Updates dependencies, imports --- .../jupyter-ai-magics/jupyter_ai_magics/embedding_providers.py | 2 +- packages/jupyter-ai-magics/pyproject.toml | 1 + packages/jupyter-ai/pyproject.toml | 2 -- 3 files changed, 2 insertions(+), 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 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]",