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

test(test_proxy_custom_logger.py): fix proxy test

Browse source
This commit is contained in:
Krrish Dholakia 2024-02-27 14:52:45 -08:00
parent d8669dea64
commit 411787963b
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/tests/test_proxy_custom_logger.py
View file

@ -274,7 +274,7 @@ def test_chat_completion_stream(client):
print("\n\n decode_data", data) print("\n\n decode_data", data)
# Access the content of choices[0]['message']['content'] # 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 # Process the content as needed
print("Content:", content) print("Content:", content)