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

test_completion_gemini_stream

Browse source
This commit is contained in:
Ishaan Jaff 2025-08-07 13:24:00 -07:00
parent ab2c274381
commit c90d431b66
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
tests/local_testing/test_streaming.py
View file

@ -701,7 +701,7 @@ async def test_completion_gemini_stream(sync_mode):
},
}
]
messages = [{"role": "user", "content": "What is the weather like in Boston?"}]
messages = [{"role": "user", "content": "What is the weather like in Boston, MA?. You must provide me with a tool call in your response."}]
print("testing gemini streaming")
complete_response = ""
# Add any assertions here to check the response
@ -817,7 +817,7 @@ async def test_completion_gemini_stream_accumulated_json(sync_mode):
},
}
]
messages = [{"role": "user", "content": "What is the weather like in Boston?"}]
messages = [{"role": "user", "content": "What is the weather like in Boston, MA?. You must provide me with a tool call in your response."}]
print("testing gemini streaming")
complete_response = ""
# Add any assertions here to check the response