phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-07-21 03:59:42 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix unit tests

Browse source
This commit is contained in:
Jash Gulabrai 2025-04-30 10:57:40 -04:00
parent f8f59c8335
commit bfbaf09fa8
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/unit/providers/nvidia/test_safety.py
View file

@ -139,7 +139,7 @@ class TestNVIDIASafetyAdapter(unittest.TestCase):
"model": shield_id, "model": shield_id,
"messages": [ "messages": [
{"role": "user", "content": "Hello, how are you?"}, {"role": "user", "content": "Hello, how are you?"},
{"role": "assistant", "content": "I'm doing well, thank you for asking!", "tool_calls": None}, {"role": "assistant", "content": "I'm doing well, thank you for asking!"},
], ],
"temperature": 1.0, "temperature": 1.0,
"top_p": 1, "top_p": 1,
@ -193,7 +193,7 @@ class TestNVIDIASafetyAdapter(unittest.TestCase):
"model": shield_id, "model": shield_id,
"messages": [ "messages": [
{"role": "user", "content": "Hello, how are you?"}, {"role": "user", "content": "Hello, how are you?"},
{"role": "assistant", "content": "I'm doing well, thank you for asking!", "tool_calls": None}, {"role": "assistant", "content": "I'm doing well, thank you for asking!"},
], ],
"temperature": 1.0, "temperature": 1.0,
"top_p": 1, "top_p": 1,
@ -269,7 +269,7 @@ class TestNVIDIASafetyAdapter(unittest.TestCase):
"model": shield_id, "model": shield_id,
"messages": [ "messages": [
{"role": "user", "content": "Hello, how are you?"}, {"role": "user", "content": "Hello, how are you?"},
{"role": "assistant", "content": "I'm doing well, thank you for asking!", "tool_calls": None}, {"role": "assistant", "content": "I'm doing well, thank you for asking!"},
], ],
"temperature": 1.0, "temperature": 1.0,
"top_p": 1, "top_p": 1,