We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent b75f47b commit 6dd07d9Copy full SHA for 6dd07d9
1 file changed
tests/providers/test_nvidia_nim.py
@@ -87,10 +87,13 @@ async def test_init_uses_configurable_timeouts():
87
88
89
@pytest.mark.asyncio
90
-async def test_build_request_body(nim_provider):
+async def test_build_request_body(provider_config):
91
"""Test request body construction."""
92
+ from config.nim import NimSettings
93
+
94
+ provider = NvidiaNimProvider(provider_config, nim_settings=NimSettings(enable_thinking=True))
95
req = MockRequest()
- body = nim_provider._build_request_body(req)
96
+ body = provider._build_request_body(req)
97
98
assert body["model"] == "test-model"
99
assert body["temperature"] == 0.5
0 commit comments