From 39d8645596c54e5c1495714f56ae2e7f546e1c4f Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Tue, 27 Feb 2024 14:53:46 -0800 Subject: [PATCH] test(test_proxy_custom_logger.py): fix format test --- litellm/tests/test_proxy_custom_logger.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/tests/test_proxy_custom_logger.py b/litellm/tests/test_proxy_custom_logger.py index 34e427ef4d9..64ed08897f1 100644 --- a/litellm/tests/test_proxy_custom_logger.py +++ b/litellm/tests/test_proxy_custom_logger.py @@ -274,7 +274,7 @@ def test_chat_completion_stream(client): print("\n\n decode_data", data) # Access the content of choices[0]['message']['content'] - content = data["choices"][0]["delta"]["content"] or "" + content = data["choices"][0]["delta"].get("content", None) or "" # Process the content as needed print("Content:", content)