Fix import deprecations

jupyterlab · Jun 11, 2024 · 6a1a56c · 6a1a56c
1 parent 6f32acc
commit 6a1a56c
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py b/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py
@@ -20,8 +20,6 @@
 
 from jsonpath_ng import parse
 from langchain.chat_models.base import BaseChatModel
-from langchain.llms.sagemaker_endpoint import LLMContentHandler
-from langchain.llms.utils import enforce_stop_tokens
 from langchain.prompts import (
     ChatPromptTemplate,
     HumanMessagePromptTemplate,
@@ -44,6 +42,7 @@
     SagemakerEndpoint,
     Together,
 )
+from langchain_community.llms.sagemaker_endpoint import LLMContentHandler
 
 # this is necessary because `langchain.pydantic_v1.main` does not include
 # `ModelMetaclass`, as it is not listed in `__all__` by the `pydantic.main`