fix

Author: jrobertboos

tests/e2e/features/steps/llm_query_response.py

@@ -93,7 +93,9 @@ def wait_for_complete_response(context: Context) -> None:
     """Wait for the response to be complete."""
     context.response_data = _parse_streaming_response(context.response.text)
     context.response.raise_for_status()
-    assert context.response_data["finished"] is True, f"Response is not finished: {context.response_data}"
+    assert (
+        context.response_data["finished"] is True
+    ), f"Response is not finished: {context.response_data}"
 
 
 @step('I use "{endpoint}" to ask question')
@@ -366,7 +368,6 @@ def _parse_streaming_response(response_text: str) -> dict:
     full_response = ""
     full_response_split = []
     finished = False
-    first_token = True
     stream_error = (
         None  # {"status_code": int, "response": str, "cause": str} if event "error"
     )
@@ -380,10 +381,6 @@ def _parse_streaming_response(response_text: str) -> dict:
                 if event == "start":
                     conversation_id = data["data"]["conversation_id"]
                 elif event == "token":
-                    # Skip the first token (shield status message)
-                    if first_token:
-                        first_token = False
-                        continue
                     full_response_split.append(data["data"]["token"])
                 elif event == "turn_complete":
                     full_response = data["data"]["token"]

tests/e2e/test_list.txt

@@ -1 +1,30 @@
+features/authorized_noop.feature
+features/health.feature
+features/info.feature
+features/models.feature
+features/rest_api.feature
+features/smoketests.feature
+features/authorized_noop_token.feature
+features/conversation_cache_v2.feature
+features/conversations.feature
+features/prompts.feature
+features/faiss.feature
+features/inline_rag.feature
+features/feedback.feature
+features/query.feature
+features/responses.feature
+features/responses_streaming.feature
+features/rlsapi_v1.feature
 features/streaming_query.feature
+features/http_401_unauthorized.feature
+features/authorized_rh_identity.feature
+features/rbac.feature
+features/rlsapi_v1_errors.feature
+features/llama_stack_disrupted.feature
+features/mcp.feature
+features/mcp_servers_api.feature
+features/mcp_servers_api_auth.feature
+features/mcp_servers_api_no_config.feature
+features/proxy.feature
+features/tls.feature
+features/opentelemetry.feature