From b2c21f3e5748b168141a2362af15092b936d3b5c Mon Sep 17 00:00:00 2001 From: Michael Chin Date: Sat, 15 Feb 2025 18:34:56 -0800 Subject: [PATCH] docs: Update SagemakerEndpoint examples (#29814) Related issue: https://github.com/langchain-ai/langchain-aws/issues/361 Updated the AWS `SagemakerEndpoint` LLM documentation to import from `langchain-aws`. --- docs/docs/integrations/llms/sagemaker.ipynb | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/docs/docs/integrations/llms/sagemaker.ipynb b/docs/docs/integrations/llms/sagemaker.ipynb index 387f91f2edd..f07a8ac311b 100644 --- a/docs/docs/integrations/llms/sagemaker.ipynb +++ b/docs/docs/integrations/llms/sagemaker.ipynb @@ -104,8 +104,8 @@ "\n", "import boto3\n", "from langchain.chains.question_answering import load_qa_chain\n", - "from langchain_community.llms import SagemakerEndpoint\n", - "from langchain_community.llms.sagemaker_endpoint import LLMContentHandler\n", + "from langchain_aws.llms import SagemakerEndpoint\n", + "from langchain_aws.llms.sagemaker_endpoint import LLMContentHandler\n", "from langchain_core.prompts import PromptTemplate\n", "\n", "query = \"\"\"How long was Elizabeth hospitalized?\n", @@ -174,8 +174,8 @@ "from typing import Dict\n", "\n", "from langchain.chains.question_answering import load_qa_chain\n", - "from langchain_community.llms import SagemakerEndpoint\n", - "from langchain_community.llms.sagemaker_endpoint import LLMContentHandler\n", + "from langchain_aws.llms import SagemakerEndpoint\n", + "from langchain_aws.llms.sagemaker_endpoint import LLMContentHandler\n", "from langchain_core.prompts import PromptTemplate\n", "\n", "query = \"\"\"How long was Elizabeth hospitalized?\n",