tests: fixed blinking test

ErikBjare · Aug 29, 2024 · c6953b3 · c6953b3
1 parent 5f00b9c
commit c6953b3
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/test_server.py b/tests/test_server.py
@@ -77,5 +77,8 @@ def test_api_conversation_generate(conv: str, client: FlaskClient):
     )
     assert response.status_code == 200
     msgs = response.get_json()
-    assert len(msgs) == 1
+    assert len(msgs) >= 1
+    assert len(msgs) <= 2
     assert msgs[0]["role"] == "assistant"
+    if len(msgs) == 2:
+        assert msgs[1]["role"] == "system"