diff --git a/llama_stack/apis/agents/openai_responses.py b/llama_stack/apis/agents/openai_responses.py
index e048cc664..1f28f0d16 100644
--- a/llama_stack/apis/agents/openai_responses.py
+++ b/llama_stack/apis/agents/openai_responses.py
@@ -134,6 +134,7 @@ class OpenAIResponseOutputMessageContentOutputText(BaseModel):
 @json_schema_type
 class OpenAIResponseContentPartRefusal(BaseModel):
     """Refusal content within a streamed response part.
+
     :param type: Content part type identifier, always "refusal"
     :param refusal: Refusal text supplied by the model
     """
diff --git a/llama_stack/providers/inline/agents/meta_reference/responses/utils.py b/llama_stack/providers/inline/agents/meta_reference/responses/utils.py
index df1feaebc..e55e7b1fb 100644
--- a/llama_stack/providers/inline/agents/meta_reference/responses/utils.py
+++ b/llama_stack/providers/inline/agents/meta_reference/responses/utils.py
@@ -312,6 +312,7 @@ def is_function_tool_call(
             return True
     return False
 
+
 async def run_multiple_shields(safety_api: Safety, messages: list[Message], shield_ids: list[str]) -> None:
     """Run multiple shields against messages and raise SafetyException for violations."""
     if not shield_ids or not messages:
@@ -340,7 +341,7 @@ def extract_shield_ids(shields: list | None) -> list[str]:
         elif isinstance(shield, ResponseShieldSpec):
             shield_ids.append(shield.type)
         else:
-            raise ValueError(f"Unsupported shield type: {type(shield)}")
+            raise ValueError(f"Unknown shield format: {shield}, expected str or ResponseShieldSpec")
 
     return shield_ids
 
diff --git a/tests/unit/providers/agents/meta_reference/test_responses_safety_utils.py b/tests/unit/providers/agents/meta_reference/test_responses_safety_utils.py
index 383216364..1df1d0d82 100644
--- a/tests/unit/providers/agents/meta_reference/test_responses_safety_utils.py
+++ b/tests/unit/providers/agents/meta_reference/test_responses_safety_utils.py
@@ -38,11 +38,6 @@ def responses_impl(mock_apis):
     return OpenAIResponsesImpl(**mock_apis)
 
 
-# ============================================================================
-# Shield ID Extraction Tests
-# ============================================================================
-
-
 def test_extract_shield_ids_from_strings(responses_impl):
     """Test extraction from simple string shield IDs."""
     shields = ["llama-guard", "content-filter", "nsfw-detector"]
@@ -92,11 +87,6 @@ def test_extract_shield_ids_unknown_format(responses_impl):
         extract_shield_ids(shields)
 
 
-# ============================================================================
-# Text Content Extraction Tests
-# ============================================================================
-
-
 def test_extract_text_content_string(responses_impl):
     """Test extraction from simple string content."""
     content = "Hello world"