Skip to content

Commit ab3e9eb

Browse files
authored
Fix import deprecations (#824)
1 parent 6f32acc commit ab3e9eb

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

packages/jupyter-ai-magics/jupyter_ai_magics/providers.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,8 +20,6 @@
2020

2121
from jsonpath_ng import parse
2222
from langchain.chat_models.base import BaseChatModel
23-
from langchain.llms.sagemaker_endpoint import LLMContentHandler
24-
from langchain.llms.utils import enforce_stop_tokens
2523
from langchain.prompts import (
2624
ChatPromptTemplate,
2725
HumanMessagePromptTemplate,
@@ -44,6 +42,7 @@
4442
SagemakerEndpoint,
4543
Together,
4644
)
45+
from langchain_community.llms.sagemaker_endpoint import LLMContentHandler
4746

4847
# this is necessary because `langchain.pydantic_v1.main` does not include
4948
# `ModelMetaclass`, as it is not listed in `__all__` by the `pydantic.main`

0 commit comments

Comments
 (0)