mirror of
https://github.com/huggingface/text-generation-inference.git
synced 2025-04-23 16:02:10 +00:00
2d56f106a6
# What does this PR do? Since ([#1097](https://github.com/huggingface/text-generation-inference/pull/1097)) the clients do not need to specify a max_length anymore. However, the python client in this repo had not yet been adapted to these changes. This PR makes it possible to use the python client and not provide max_new_tokens. <!-- Remove if not applicable --> ## Before submitting - [ ] This PR fixes a typo or improves the docs (you can dismiss the other checks if that's the case). - [x] Did you read the [contributor guideline](https://github.com/huggingface/transformers/blob/main/CONTRIBUTING.md#start-contributing-pull-requests), Pull Request section? - [ ] Was this discussed/approved via a Github issue or the [forum](https://discuss.huggingface.co/)? Please add a link to it if that's the case. - [ ] Did you make sure to update the documentation with your changes? Here are the [documentation guidelines](https://github.com/huggingface/transformers/tree/main/docs), and [here are tips on formatting docstrings](https://github.com/huggingface/transformers/tree/main/docs#writing-source-documentation). - [x] Did you write any new necessary tests? ## Who can review? Anyone in the community is free to review the PR once the tests have passed. Feel free to tag members/contributors who may be interested in your PR.
167 lines
6.0 KiB
Python
167 lines
6.0 KiB
Python
import pytest
|
|
|
|
from text_generation import Client, AsyncClient
|
|
from text_generation.errors import NotFoundError, ValidationError
|
|
from text_generation.types import FinishReason, InputToken
|
|
|
|
|
|
def test_generate(flan_t5_xxl_url, hf_headers):
|
|
client = Client(flan_t5_xxl_url, hf_headers)
|
|
response = client.generate("test", max_new_tokens=1, decoder_input_details=True)
|
|
|
|
assert response.generated_text == ""
|
|
assert response.details.finish_reason == FinishReason.Length
|
|
assert response.details.generated_tokens == 1
|
|
assert response.details.seed is None
|
|
assert len(response.details.prefill) == 1
|
|
assert response.details.prefill[0] == InputToken(id=0, text="<pad>", logprob=None)
|
|
assert len(response.details.tokens) == 1
|
|
assert response.details.tokens[0].id == 3
|
|
assert response.details.tokens[0].text == " "
|
|
assert not response.details.tokens[0].special
|
|
|
|
|
|
def test_generate_max_new_tokens_not_set(flan_t5_xxl_url, hf_headers):
|
|
client = Client(flan_t5_xxl_url, hf_headers)
|
|
response = client.generate("test", decoder_input_details=True)
|
|
|
|
assert response.generated_text != ""
|
|
assert response.details.finish_reason == FinishReason.EndOfSequenceToken
|
|
assert response.details.generated_tokens > 1
|
|
assert response.details.seed is None
|
|
assert len(response.details.prefill) == 1
|
|
assert response.details.prefill[0] == InputToken(id=0, text="<pad>", logprob=None)
|
|
assert len(response.details.tokens) > 1
|
|
assert response.details.tokens[0].id == 3
|
|
assert response.details.tokens[0].text == " "
|
|
assert not response.details.tokens[0].special
|
|
|
|
|
|
def test_generate_best_of(flan_t5_xxl_url, hf_headers):
|
|
client = Client(flan_t5_xxl_url, hf_headers)
|
|
response = client.generate(
|
|
"test", max_new_tokens=1, best_of=2, do_sample=True, decoder_input_details=True
|
|
)
|
|
|
|
assert response.details.seed is not None
|
|
assert response.details.best_of_sequences is not None
|
|
assert len(response.details.best_of_sequences) == 1
|
|
assert response.details.best_of_sequences[0].seed is not None
|
|
|
|
|
|
def test_generate_not_found(fake_url, hf_headers):
|
|
client = Client(fake_url, hf_headers)
|
|
with pytest.raises(NotFoundError):
|
|
client.generate("test")
|
|
|
|
|
|
def test_generate_validation_error(flan_t5_xxl_url, hf_headers):
|
|
client = Client(flan_t5_xxl_url, hf_headers)
|
|
with pytest.raises(ValidationError):
|
|
client.generate("test", max_new_tokens=10_000)
|
|
|
|
|
|
def test_generate_stream(flan_t5_xxl_url, hf_headers):
|
|
client = Client(flan_t5_xxl_url, hf_headers)
|
|
responses = [
|
|
response for response in client.generate_stream("test", max_new_tokens=1)
|
|
]
|
|
|
|
assert len(responses) == 1
|
|
response = responses[0]
|
|
|
|
assert response.generated_text == ""
|
|
assert response.details.finish_reason == FinishReason.Length
|
|
assert response.details.generated_tokens == 1
|
|
assert response.details.seed is None
|
|
|
|
|
|
def test_generate_stream_not_found(fake_url, hf_headers):
|
|
client = Client(fake_url, hf_headers)
|
|
with pytest.raises(NotFoundError):
|
|
list(client.generate_stream("test"))
|
|
|
|
|
|
def test_generate_stream_validation_error(flan_t5_xxl_url, hf_headers):
|
|
client = Client(flan_t5_xxl_url, hf_headers)
|
|
with pytest.raises(ValidationError):
|
|
list(client.generate_stream("test", max_new_tokens=10_000))
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_generate_async(flan_t5_xxl_url, hf_headers):
|
|
client = AsyncClient(flan_t5_xxl_url, hf_headers)
|
|
response = await client.generate(
|
|
"test", max_new_tokens=1, decoder_input_details=True
|
|
)
|
|
|
|
assert response.generated_text == ""
|
|
assert response.details.finish_reason == FinishReason.Length
|
|
assert response.details.generated_tokens == 1
|
|
assert response.details.seed is None
|
|
assert len(response.details.prefill) == 1
|
|
assert response.details.prefill[0] == InputToken(id=0, text="<pad>", logprob=None)
|
|
assert len(response.details.tokens) == 1
|
|
assert response.details.tokens[0].id == 3
|
|
assert response.details.tokens[0].text == " "
|
|
assert not response.details.tokens[0].special
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_generate_async_best_of(flan_t5_xxl_url, hf_headers):
|
|
client = AsyncClient(flan_t5_xxl_url, hf_headers)
|
|
response = await client.generate(
|
|
"test", max_new_tokens=1, best_of=2, do_sample=True, decoder_input_details=True
|
|
)
|
|
|
|
assert response.details.seed is not None
|
|
assert response.details.best_of_sequences is not None
|
|
assert len(response.details.best_of_sequences) == 1
|
|
assert response.details.best_of_sequences[0].seed is not None
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_generate_async_not_found(fake_url, hf_headers):
|
|
client = AsyncClient(fake_url, hf_headers)
|
|
with pytest.raises(NotFoundError):
|
|
await client.generate("test")
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_generate_async_validation_error(flan_t5_xxl_url, hf_headers):
|
|
client = AsyncClient(flan_t5_xxl_url, hf_headers)
|
|
with pytest.raises(ValidationError):
|
|
await client.generate("test", max_new_tokens=10_000)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_generate_stream_async(flan_t5_xxl_url, hf_headers):
|
|
client = AsyncClient(flan_t5_xxl_url, hf_headers)
|
|
responses = [
|
|
response async for response in client.generate_stream("test", max_new_tokens=1)
|
|
]
|
|
|
|
assert len(responses) == 1
|
|
response = responses[0]
|
|
|
|
assert response.generated_text == ""
|
|
assert response.details.finish_reason == FinishReason.Length
|
|
assert response.details.generated_tokens == 1
|
|
assert response.details.seed is None
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_generate_stream_async_not_found(fake_url, hf_headers):
|
|
client = AsyncClient(fake_url, hf_headers)
|
|
with pytest.raises(NotFoundError):
|
|
async for _ in client.generate_stream("test"):
|
|
pass
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_generate_stream_async_validation_error(flan_t5_xxl_url, hf_headers):
|
|
client = AsyncClient(flan_t5_xxl_url, hf_headers)
|
|
with pytest.raises(ValidationError):
|
|
async for _ in client.generate_stream("test", max_new_tokens=10_000):
|
|
pass
|