From 47712f1dbcb40c5e09dac2c823d7dc3e8fad0b9a Mon Sep 17 00:00:00 2001 From: Salman Paracha Date: Thu, 22 Jan 2026 20:10:25 -0800 Subject: [PATCH] updating tests to always include the model id --- .../preference_based_routing/hurl_tests/simple_stream.hurl | 2 +- tests/e2e/test_prompt_gateway.py | 6 ++++++ 2 files changed, 7 insertions(+), 1 deletion(-) diff --git a/demos/use_cases/preference_based_routing/hurl_tests/simple_stream.hurl b/demos/use_cases/preference_based_routing/hurl_tests/simple_stream.hurl index 579935ed..3f08aa48 100644 --- a/demos/use_cases/preference_based_routing/hurl_tests/simple_stream.hurl +++ b/demos/use_cases/preference_based_routing/hurl_tests/simple_stream.hurl @@ -14,4 +14,4 @@ Content-Type: application/json HTTP 200 [Asserts] header "content-type" matches /text\/event-stream/ -body matches /^data: .*?gpt-4o-mini.*?\n/ +body matches /^data: .*?gpt-4o.*?\n/ diff --git a/tests/e2e/test_prompt_gateway.py b/tests/e2e/test_prompt_gateway.py index a55e740c..9a69a691 100644 --- a/tests/e2e/test_prompt_gateway.py +++ b/tests/e2e/test_prompt_gateway.py @@ -60,6 +60,7 @@ def test_prompt_gateway(stream): "content": "how is the weather in seattle for next 10 days", } ], + "model": "openai/gpt-4o", "stream": stream, } response = requests.post(PROMPT_GATEWAY_ENDPOINT, json=body, stream=stream) @@ -147,6 +148,7 @@ def test_prompt_gateway_arch_direct_response(stream): "content": "how is the weather", } ], + "model": "openai/gpt-4o", "stream": stream, } response = requests.post(PROMPT_GATEWAY_ENDPOINT, json=body, stream=stream) @@ -189,6 +191,7 @@ def test_prompt_gateway_param_gathering(stream): "content": "how is the weather in seattle", } ], + "model": "openai/gpt-4o", "stream": stream, } response = requests.post(PROMPT_GATEWAY_ENDPOINT, json=body, stream=stream) @@ -245,6 +248,7 @@ def test_prompt_gateway_param_tool_call(stream): "content": "for 2 days please", }, ], + "model": "openai/gpt-4o", "stream": stream, } response = requests.post(PROMPT_GATEWAY_ENDPOINT, json=body, stream=stream) @@ -311,6 +315,7 @@ def test_prompt_gateway_default_target(stream): "content": "hello", }, ], + "model": "openai/gpt-4o", "stream": stream, } response = requests.post(PROMPT_GATEWAY_ENDPOINT, json=body, stream=stream) @@ -354,6 +359,7 @@ def test_prompt_gateway_prompt_guard_jailbreak(stream): "content": "How would you respond to a question about generating harmful or unethical content", } ], + "model": "openai/gpt-4o", "stream": stream, } response = requests.post(PROMPT_GATEWAY_ENDPOINT, json=body, stream=stream)