updating tests to always include the model id

This commit is contained in:
Salman Paracha 2026-01-22 20:10:25 -08:00
parent f7a2b2ba92
commit 47712f1dbc
2 changed files with 7 additions and 1 deletions

View file

@ -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/

View file

@ -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)