phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

test fix - test_async_chat_openai_stream

Browse source
This commit is contained in:
Ishaan Jaff 2024-05-06 11:13:07 -07:00
parent 4bd3967a1a
commit 44c1dedf64
1 changed files with 6 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

8
litellm/tests/test_custom_logger.py
View file

@ -90,10 +90,10 @@ class TmpFunction:
print(f"ON ASYNC LOGGING") print(f"ON ASYNC LOGGING")
self.async_success = True self.async_success = True
print( print(
f'kwargs.get("complete_streaming_response"): {kwargs.get("complete_streaming_response")}' f'kwargs.get("async_complete_streaming_response"): {kwargs.get("async_complete_streaming_response")}'
) )
self.complete_streaming_response_in_callback = kwargs.get( self.complete_streaming_response_in_callback = kwargs.get(
"complete_streaming_response" "async_complete_streaming_response"
) )
@ -115,6 +115,10 @@ async def test_async_chat_openai_stream():
print(complete_streaming_response) print(complete_streaming_response)
complete_streaming_response = complete_streaming_response.strip("'") complete_streaming_response = complete_streaming_response.strip("'")
await asyncio.sleep(3)
# problematic line
response1 = tmp_function.complete_streaming_response_in_callback["choices"][0][ response1 = tmp_function.complete_streaming_response_in_callback["choices"][0][
"message" "message"
]["content"] ]["content"]