phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-24 18:24:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(sagemaker.py): fix the post-call logging logic

Browse source
This commit is contained in:
Krrish Dholakia 2024-01-06 21:52:58 +05:30
parent f2ad13af65
commit 3577857ed1
1 changed files with 1 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

9
litellm/llms/sagemaker.py
View file

@ -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")