We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 6f32acc commit ab3e9ebCopy full SHA for ab3e9eb
packages/jupyter-ai-magics/jupyter_ai_magics/providers.py
@@ -20,8 +20,6 @@
20
21
from jsonpath_ng import parse
22
from langchain.chat_models.base import BaseChatModel
23
-from langchain.llms.sagemaker_endpoint import LLMContentHandler
24
-from langchain.llms.utils import enforce_stop_tokens
25
from langchain.prompts import (
26
ChatPromptTemplate,
27
HumanMessagePromptTemplate,
@@ -44,6 +42,7 @@
44
42
SagemakerEndpoint,
45
43
Together,
46
)
+from langchain_community.llms.sagemaker_endpoint import LLMContentHandler
47
48
# this is necessary because `langchain.pydantic_v1.main` does not include
49
# `ModelMetaclass`, as it is not listed in `__all__` by the `pydantic.main`
0 commit comments