From d16bf261a1832ebda30c2aa8b319f4481dd84098 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Fri, 7 Jun 2024 14:47:01 -0700 Subject: [PATCH] fix(bedrock_httpx.py): fix post call success logging --- litellm/llms/bedrock_httpx.py | 11 ++--------- 1 file changed, 2 insertions(+), 9 deletions(-) diff --git a/litellm/llms/bedrock_httpx.py b/litellm/llms/bedrock_httpx.py index 59945a5857..afc2657610 100644 --- a/litellm/llms/bedrock_httpx.py +++ b/litellm/llms/bedrock_httpx.py @@ -178,7 +178,7 @@ async def make_call( logging_obj.post_call( input=messages, api_key="", - original_response=completion_stream, # Pass the completion stream for logging + original_response="first stream response received", additional_args={"complete_input_dict": data}, ) @@ -209,7 +209,7 @@ def make_sync_call( logging_obj.post_call( input=messages, api_key="", - original_response=completion_stream, # Pass the completion stream for logging + original_response="first stream response received", additional_args={"complete_input_dict": data}, ) @@ -1805,13 +1805,6 @@ class BedrockConverseLLM(BaseLLM): logging_obj=logging_obj, ) - ## LOGGING - logging_obj.post_call( - input=messages, - api_key="", - original_response=streaming_response, - additional_args={"complete_input_dict": data}, - ) return streaming_response ### COMPLETION