From 3577857ed1ebde29bf9ab3502f81962f7f7483de Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Sat, 6 Jan 2024 21:52:58 +0530 Subject: [PATCH] fix(sagemaker.py): fix the post-call logging logic --- litellm/llms/sagemaker.py | 9 +-------- 1 file changed, 1 insertion(+), 8 deletions(-) diff --git a/litellm/llms/sagemaker.py b/litellm/llms/sagemaker.py index 2dfe7c8cb88..1d341e7e92d 100644 --- a/litellm/llms/sagemaker.py +++ b/litellm/llms/sagemaker.py @@ -360,14 +360,6 @@ def embedding( except Exception as e: raise SagemakerError(status_code=500, message=f"{str(e)}") - ## LOGGING - logging_obj.post_call( - input=input, - api_key="", - additional_args={"complete_input_dict": data}, - original_response=response, - ) - response = json.loads(response["Body"].read().decode("utf8")) ## LOGGING logging_obj.post_call( @@ -376,6 +368,7 @@ def embedding( original_response=response, additional_args={"complete_input_dict": data}, ) + print_verbose(f"raw model_response: {response}") if "embedding" not in response: raise SagemakerError(status_code=500, message="embedding not found in response")