test(test_streaming.py): add more logging

This commit is contained in:
Krrish Dholakia
2024-02-26 21:45:20 -08:00
parent b3574f2b37
commit a428501e68
+1 -1
View File
@@ -406,7 +406,7 @@ def test_completion_palm_stream():
def test_completion_gemini_stream():
try:
litellm.set_verbose = False
litellm.set_verbose = True
print("Streaming gemini response")
messages = [
{"role": "system", "content": "You are a helpful assistant."},