From 60e442b16879944ee0ba5472692361d945d9e178 Mon Sep 17 00:00:00 2001 From: Adil Hafeez Date: Wed, 6 Nov 2024 15:37:13 -0800 Subject: [PATCH] fix e2e --- e2e_tests/test_prompt_gateway.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/e2e_tests/test_prompt_gateway.py b/e2e_tests/test_prompt_gateway.py index 31f305d4..809fc710 100644 --- a/e2e_tests/test_prompt_gateway.py +++ b/e2e_tests/test_prompt_gateway.py @@ -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" )