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

fix e2e

Browse source
This commit is contained in:
Adil Hafeez 2024-11-06 15:37:13 -08:00
parent 9379c4ed7e
commit 60e442b168
1 changed files with 2 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

6
e2e_tests/test_prompt_gateway.py
View file

@ -248,9 +248,7 @@ def test_prompt_gateway_default_target(stream):
choices = response_json.get("choices", [])
assert len(choices) > 0
content = choices[0]["delta"]["content"]
assert (
content == "I can help you with weather forecast or insurance claim details"
)
assert content == "I can help you with weather forecast"
else:
response_json = response.json()
assert response_json.get("model").startswith("api_server")
@ -258,5 +256,5 @@ def test_prompt_gateway_default_target(stream):
assert response_json.get("choices")[0]["message"]["role"] == "assistant"
assert (
response_json.get("choices")[0]["message"]["content"]
== "I can help you with weather forecast or insurance claim details"
== "I can help you with weather forecast"
)