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

fix ai21 test

Browse source
This commit is contained in:
Ishaan Jaff 2025-05-07 21:26:35 -07:00
parent 764e9a1934
commit 580e221000
2 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
tests/local_testing/test_completion.py
View file

@ -4140,7 +4140,7 @@ async def test_completion_ai21_chat():
litellm.set_verbose = True
try:
response = await litellm.acompletion(
model="jamba-1.5-large",
model="jamba-mini",
user="ishaan",
tool_choice="auto",
seed=123,

2
tests/local_testing/test_streaming.py
View file

@ -553,7 +553,7 @@ async def test_completion_predibase_streaming(sync_mode):
async def test_completion_ai21_stream():
litellm.set_verbose = True
response = await litellm.acompletion(
model="ai21_chat/jamba-1.5-large",
model="ai21_chat/jamba-mini",
user="ishaan",
stream=True,
seed=123,