scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-06 08:16:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 78

fix test test_router_completion_streaming

Browse source
This commit is contained in:
Ishaan Jaff 2024-06-21 21:02:34 -07:00
parent b4d95e740e
commit 24addd8a4f
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
litellm/tests/test_tpm_rpm_routing_v2.py
View file

@ -394,7 +394,7 @@ async def test_router_completion_streaming():
"model": "azure/gpt-turbo",
"api_key": "os.environ/AZURE_FRANCE_API_KEY",
"api_base": "https://openai-france-1234.openai.azure.com",
"rpm": 1440,
"tpm": 1440,
"mock_response": "Hello world",
},
"model_info": {"id": 1},
@ -405,7 +405,7 @@ async def test_router_completion_streaming():
"model": "azure/gpt-35-turbo",
"api_key": "os.environ/AZURE_EUROPE_API_KEY",
"api_base": "https://my-endpoint-europe-berri-992.openai.azure.com",
"rpm": 6,
"tpm": 6,
"mock_response": "Hello world",
},
"model_info": {"id": 2},