mirrors/hermes-agent
1
0
Fork 0
mirror of https://github.com/NousResearch/hermes-agent.git synced 2026-07-01 12:02:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

test(auxiliary): cover NVIDIA NIM max_tokens in _build_call_kwargs

Browse source
This commit is contained in:
HexLab98 2026-06-29 16:56:09 +07:00 • committed by kshitij
parent 88e6f9b98c
commit f1345290ed
1 changed files with 12 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

12
tests/agent/test_auxiliary_client.py
View file

@ -163,6 +163,18 @@ class TestBuildCallKwargsMaxTokens:
assert kwargs["max_tokens"] == 1234
assert "max_completion_tokens" not in kwargs
def test_keeps_max_tokens_for_nvidia_nim(self):
from agent.auxiliary_client import _build_call_kwargs
kwargs = _build_call_kwargs(
provider="nvidia",
model="minimaxai/minimax-m3",
messages=[{"role": "user", "content": "hi"}],
max_tokens=4096,
base_url="https://integrate.api.nvidia.com/v1",
)
assert kwargs["max_tokens"] == 4096
class TestNousTagsScoping:
def test_tags_injected_when_provider_is_nous(self, monkeypatch):