test(test_end_users.py): fix end user region routing test

This commit is contained in:
Krrish Dholakia 2024-05-11 22:42:43 -07:00
parent b4684d5132
commit 99e8f0715e
2 changed files with 5 additions and 7 deletions

View file

@ -1,9 +1,10 @@
model_list:
- model_name: gpt-3.5-turbo
litellm_params:
model: azure/gpt-35-turbo
api_base: https://my-endpoint-europe-berri-992.openai.azure.com/
api_key: os.environ/AZURE_EUROPE_API_KEY
model: gpt-3.5-turbo
region_name: "eu"
model_info:
id: "1"
- model_name: gpt-3.5-turbo
litellm_params:
model: azure/chatgpt-v-2

View file

@ -167,7 +167,4 @@ async def test_end_user_specific_region():
user=end_user_obj["user_id"],
)
assert (
result.headers.get("x-litellm-model-api-base")
== "https://my-endpoint-europe-berri-992.openai.azure.com/"
)
assert result.headers.get("x-litellm-model-id") == "1"