[Bugfix] Add validation for seed (#4529)

This commit is contained in:
sasha0552
2024-05-01 19:31:22 +00:00
committed by GitHub
parent 24bb4fe432
commit c47ba4aaa9
2 changed files with 26 additions and 2 deletions

View File

@@ -13,6 +13,7 @@ import pytest
# and debugging.
import ray
import requests
import torch
# downloading lora to test lora requests
from huggingface_hub import snapshot_download
from openai import BadRequestError
@@ -870,5 +871,24 @@ async def test_echo_logprob_completion(server, client: openai.AsyncOpenAI,
assert len(logprobs.tokens) > 5
async def test_long_seed(server, client: openai.AsyncOpenAI):
for seed in [
torch.iinfo(torch.long).min - 1,
torch.iinfo(torch.long).max + 1
]:
with pytest.raises(BadRequestError) as exc_info:
await client.chat.completions.create(
model=MODEL_NAME,
messages=[{
"role": "system",
"content": "You are a helpful assistant.",
}],
temperature=0,
seed=seed)
assert ("greater_than_equal" in exc_info.value.message
or "less_than_equal" in exc_info.value.message)
if __name__ == "__main__":
pytest.main([__file__])