Skip to content

Commit c87e81c

Browse files
authored
[https://nvbugs/5685015][fix] Update invalid max_token test (NVIDIA#9435)
Signed-off-by: Junyi Xu <[email protected]>
1 parent 658d9fc commit c87e81c

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/integration/defs/examples/serve/test_serve_negative.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -117,7 +117,8 @@ async def test_invalid_max_tokens(async_client: openai.AsyncOpenAI,
117117
max_tokens=0,
118118
)
119119
error_msg = str(exc_info.value).lower()
120-
assert any(keyword in error_msg for keyword in ("mMaxNewTokens", "failed"))
120+
assert any(keyword in error_msg
121+
for keyword in ("max_tokens", "must be greater than 0"))
121122

122123

123124
@pytest.mark.asyncio

0 commit comments

Comments
 (0)