diff --git a/libs/langchain/langchain/llms/sagemaker_endpoint.py b/libs/langchain/langchain/llms/sagemaker_endpoint.py index 993e5c5486..1f22908c7b 100644 --- a/libs/langchain/langchain/llms/sagemaker_endpoint.py +++ b/libs/langchain/langchain/llms/sagemaker_endpoint.py @@ -1,7 +1,6 @@ from langchain_community.llms.sagemaker_endpoint import ( + LLMContentHandler, SagemakerEndpoint, ) -__all__ = [ - "SagemakerEndpoint", -] +__all__ = ["SagemakerEndpoint", "LLMContentHandler"]