diff --git a/llama_stack/providers/remote/safety/nvidia/nvidia.py b/llama_stack/providers/remote/safety/nvidia/nvidia.py index c0f76a8df..c0df8f095 100644 --- a/llama_stack/providers/remote/safety/nvidia/nvidia.py +++ b/llama_stack/providers/remote/safety/nvidia/nvidia.py @@ -133,7 +133,7 @@ class NeMoGuardrails: """ request_data = { "model": self.model, - "messages": messages, + "messages": [{"role": message.role, "content": message.content} for message in messages], "temperature": self.temperature, "top_p": 1, "frequency_penalty": 0,