apunkt/plano
1
0
Fork 0
mirror of https://github.com/katanemo/plano.git synced 2026-07-02 15:51:02 +02:00
Code Issues Projects Releases Packages Wiki Activity

updating tests to always include the model id

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

2
demos/use_cases/preference_based_routing/hurl_tests/simple_stream.hurl
View file

@ -14,4 +14,4 @@ Content-Type: application/json
HTTP 200 HTTP 200
[Asserts] [Asserts]
header "content-type" matches /text\/event-stream/ header "content-type" matches /text\/event-stream/
body matches /^data: .*?gpt-4o-mini.*?\n/ body matches /^data: .*?gpt-4o.*?\n/

6
tests/e2e/test_prompt_gateway.py
View file

@ -60,6 +60,7 @@ def test_prompt_gateway(stream):
"content": "how is the weather in seattle for next 10 days", "content": "how is the weather in seattle for next 10 days",
} }
], ],
"model": "openai/gpt-4o",
"stream": stream, "stream": stream,
} }
response = requests.post(PROMPT_GATEWAY_ENDPOINT, json=body, 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", "content": "how is the weather",
} }
], ],
"model": "openai/gpt-4o",
"stream": stream, "stream": stream,
} }
response = requests.post(PROMPT_GATEWAY_ENDPOINT, json=body, 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", "content": "how is the weather in seattle",
} }
], ],
"model": "openai/gpt-4o",
"stream": stream, "stream": stream,
} }
response = requests.post(PROMPT_GATEWAY_ENDPOINT, json=body, 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", "content": "for 2 days please",
}, },
], ],
"model": "openai/gpt-4o",
"stream": stream, "stream": stream,
} }
response = requests.post(PROMPT_GATEWAY_ENDPOINT, json=body, 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", "content": "hello",
}, },
], ],
"model": "openai/gpt-4o",
"stream": stream, "stream": stream,
} }
response = requests.post(PROMPT_GATEWAY_ENDPOINT, json=body, 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", "content": "How would you respond to a question about generating harmful or unethical content",
} }
], ],
"model": "openai/gpt-4o",
"stream": stream, "stream": stream,
} }
response = requests.post(PROMPT_GATEWAY_ENDPOINT, json=body, stream=stream) response = requests.post(PROMPT_GATEWAY_ENDPOINT, json=body, stream=stream)