fix tests

This commit is contained in:
Swapna Lekkala 2025-10-12 06:48:50 -07:00
parent bc5eeef6f3
commit 1954b60600

View file

@ -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,