scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-07 08:26:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 36

fix(bedrock_httpx.py): fix post call success logging

Browse source
This commit is contained in:
Krrish Dholakia 2024-06-07 14:47:01 -07:00
parent 5a2a8bdea2
commit d16bf261a1
1 changed files with 2 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
litellm/llms/bedrock_httpx.py
View file

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