From 1cc7d4c9ebd1e3e650e12fe913f43aeffe4032c5 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Juan=20Jos=C3=A9=20Torres?= <43612178+JuanJTorres11@users.noreply.github.com> Date: Mon, 24 Jul 2023 15:35:06 -0500 Subject: [PATCH] Update SageMaker Endpoint Embeddings docs to be up to date with current requirements (#8103) - **Description:** Simple change of the Class that ContentHandler inherits from. To create an object of type SagemakerEndpointEmbeddings, the property content_handler must be of type EmbeddingsContentHandler not ContentHandlerBase anymore, - **Twitter handle:** @Juanjo_Torres11 Co-authored-by: Bagatur --- .../integrations/text_embedding/sagemaker-endpoint.ipynb | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/docs/extras/integrations/text_embedding/sagemaker-endpoint.ipynb b/docs/extras/integrations/text_embedding/sagemaker-endpoint.ipynb index 994556ce76..96d09be4b1 100644 --- a/docs/extras/integrations/text_embedding/sagemaker-endpoint.ipynb +++ b/docs/extras/integrations/text_embedding/sagemaker-endpoint.ipynb @@ -39,11 +39,11 @@ "source": [ "from typing import Dict, List\n", "from langchain.embeddings import SagemakerEndpointEmbeddings\n", - "from langchain.llms.sagemaker_endpoint import ContentHandlerBase\n", + "from langchain.embeddings.sagemaker_endpoint import EmbeddingsContentHandler\n", "import json\n", "\n", "\n", - "class ContentHandler(ContentHandlerBase):\n", + "class ContentHandler(EmbeddingsContentHandler):\n", " content_type = \"application/json\"\n", " accepts = \"application/json\"\n", "\n",