diff --git a/tests/unit/integrations/test_custom_guardrail.py b/tests/unit/integrations/test_custom_guardrail.py
index 8a8917fb59e..5b8946dc2a9 100644
--- a/tests/unit/integrations/test_custom_guardrail.py
+++ b/tests/unit/integrations/test_custom_guardrail.py
@@ -56,14 +56,21 @@ class TestGetGuardrailsMessagesForCallType:
         assert data == {"input": embedding_input}
 
     @pytest.mark.parametrize("call_type", [CallTypes.embedding, CallTypes.aembedding])
-    def test_embedding_batch_with_non_text_values_is_not_extracted(self, call_type: CallTypes) -> None:
+    @pytest.mark.parametrize(
+        "embedding_input",
+        [[1, 2], ["first", b"second"]],
+        ids=["numbers", "mixed-text-bytes"],
+    )
+    def test_embedding_batch_with_non_text_values_is_not_extracted(
+        self, call_type: CallTypes, embedding_input: list[object]
+    ) -> None:
         guardrail: Final = CustomGuardrail(guardrail_name="embedding-extractor")
-        data: Final = {"input": [1, 2]}
+        data: Final = {"input": embedding_input}
 
         messages: Final = guardrail.get_guardrails_messages_for_call_type(call_type=call_type, data=data)
 
         assert messages is None
-        assert data == {"input": [1, 2]}
+        assert data == {"input": embedding_input}
 
     @pytest.mark.parametrize("call_type", [CallTypes.embedding, CallTypes.aembedding])
     @pytest.mark.parametrize("embedding_input", [None, 123, {"text": "hello"}], ids=["none", "number", "object"])