forked from phoenix/litellm-mirror
test(test_end_users.py): fix end user region routing test
This commit is contained in:
parent
b4684d5132
commit
99e8f0715e
2 changed files with 5 additions and 7 deletions
|
@ -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
|
||||
|
|
|
@ -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"
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue