From afa284dd5edee29083cfa8e750b1e6766a1a040f Mon Sep 17 00:00:00 2001 From: Salman Paracha Date: Sat, 12 Apr 2025 23:30:23 -0700 Subject: [PATCH] add llama-3.2-3b-preview to the model based on addin groq to the demo --- tests/e2e/test_prompt_gateway.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/e2e/test_prompt_gateway.py b/tests/e2e/test_prompt_gateway.py index f122ad30..9b804bae 100644 --- a/tests/e2e/test_prompt_gateway.py +++ b/tests/e2e/test_prompt_gateway.py @@ -62,7 +62,7 @@ def test_prompt_gateway(stream): # third..end chunk is summarization (role = assistant) response_json = json.loads(chunks[2]) - assert response_json.get("model").startswith("gpt-4o-mini") + assert response_json.get("model").startswith("llama-3.2-3b-preview") choices = response_json.get("choices", []) assert len(choices) > 0 assert "role" in choices[0]["delta"] @@ -71,7 +71,7 @@ def test_prompt_gateway(stream): else: response_json = response.json() - assert response_json.get("model").startswith("gpt-4o-mini") + assert response_json.get("model").startswith("llama-3.2-3b-preview") choices = response_json.get("choices", []) assert len(choices) > 0 assert "role" in choices[0]["message"] @@ -231,7 +231,7 @@ def test_prompt_gateway_param_tool_call(stream): # third..end chunk is summarization (role = assistant) response_json = json.loads(chunks[2]) - assert response_json.get("model").startswith("gpt-4o-mini") + assert response_json.get("model").startswith("llama-3.2-3b-preview") choices = response_json.get("choices", []) assert len(choices) > 0 assert "role" in choices[0]["delta"] @@ -240,7 +240,7 @@ def test_prompt_gateway_param_tool_call(stream): else: response_json = response.json() - assert response_json.get("model").startswith("gpt-4o-mini") + assert response_json.get("model").startswith("llama-3.2-3b-preview") choices = response_json.get("choices", []) assert len(choices) > 0 assert "role" in choices[0]["message"]