Fix to SageMakerEmbedding class (#10778)

* Fix to SageMakerEmbedding class * cr --------- Co-authored-by: Haotian Zhang <[email protected]>
run-llama · Feb 16, 2024 · a900d5e · a900d5e
1 parent 9d4c630
commit a900d5e
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/...ma-index-embeddings-sagemaker-endpoint/llama_index/embeddings/sagemaker_endpoint/utils.py b/...ma-index-embeddings-sagemaker-endpoint/llama_index/embeddings/sagemaker_endpoint/utils.py
@@ -46,5 +46,4 @@ def serialize_input(self, request: List[str], model_kwargs: dict) -> bytes:
         return request_str.encode("utf-8")
 
     def deserialize_output(self, response: "StreamingBody") -> List[List[float]]:
-        response_json = json.loads(response.read().decode("utf-8"))
-        return response_json["vectors"]
+        return json.loads(response.read().decode("utf-8"))