phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 19:54:13 +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 4c7d530c2a
commit 35fd28073e
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: except Exception as e:
raise SagemakerError(status_code=500, message=f"{str(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")) response = json.loads(response["Body"].read().decode("utf8"))
## LOGGING ## LOGGING
logging_obj.post_call( logging_obj.post_call(
@ -376,6 +368,7 @@ def embedding(
original_response=response, original_response=response,
additional_args={"complete_input_dict": data}, additional_args={"complete_input_dict": data},
) )
print_verbose(f"raw model_response: {response}") print_verbose(f"raw model_response: {response}")
if "embedding" not in response: if "embedding" not in response:
raise SagemakerError(status_code=500, message="embedding not found in response") raise SagemakerError(status_code=500, message="embedding not found in response")