ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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},