phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

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

Browse source
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
Download patch file Download diff file Expand all files Collapse all files

7
proxy_server_config.yaml
View file

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

5
tests/test_end_users.py
View file

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