diff options
author | kqlio67 <kqlio67@users.noreply.github.com> | 2024-10-15 11:03:20 +0200 |
---|---|---|
committer | kqlio67 <kqlio67@users.noreply.github.com> | 2024-10-15 11:03:20 +0200 |
commit | e1e37b34a54168130ddf59ef9f15e074c811c47f (patch) | |
tree | 66d8b1519d3379f1e1eb6d086093b1a3fedf73d2 /etc/unittest/async_client.py | |
parent | refactor(docs/client.): Update G4F Client API documentation to reflect changes in API usage and add examples (diff) | |
download | gpt4free-e1e37b34a54168130ddf59ef9f15e074c811c47f.tar gpt4free-e1e37b34a54168130ddf59ef9f15e074c811c47f.tar.gz gpt4free-e1e37b34a54168130ddf59ef9f15e074c811c47f.tar.bz2 gpt4free-e1e37b34a54168130ddf59ef9f15e074c811c47f.tar.lz gpt4free-e1e37b34a54168130ddf59ef9f15e074c811c47f.tar.xz gpt4free-e1e37b34a54168130ddf59ef9f15e074c811c47f.tar.zst gpt4free-e1e37b34a54168130ddf59ef9f15e074c811c47f.zip |
Diffstat (limited to '')
-rw-r--r-- | etc/unittest/async_client.py | 56 |
1 files changed, 0 insertions, 56 deletions
diff --git a/etc/unittest/async_client.py b/etc/unittest/async_client.py deleted file mode 100644 index 06ea5cdb..00000000 --- a/etc/unittest/async_client.py +++ /dev/null @@ -1,56 +0,0 @@ -import unittest - -from g4f.client import AsyncClient, ChatCompletion, ChatCompletionChunk -from .mocks import AsyncGeneratorProviderMock, ModelProviderMock, YieldProviderMock - -DEFAULT_MESSAGES = [{'role': 'user', 'content': 'Hello'}] - -class AsyncTestPassModel(unittest.IsolatedAsyncioTestCase): - - async def test_response(self): - client = AsyncClient(provider=AsyncGeneratorProviderMock) - response = await client.chat.completions.create(DEFAULT_MESSAGES, "") - self.assertIsInstance(response, ChatCompletion) - self.assertEqual("Mock", response.choices[0].message.content) - - async def test_pass_model(self): - client = AsyncClient(provider=ModelProviderMock) - response = await client.chat.completions.create(DEFAULT_MESSAGES, "Hello") - self.assertIsInstance(response, ChatCompletion) - self.assertEqual("Hello", response.choices[0].message.content) - - async def test_max_tokens(self): - client = AsyncClient(provider=YieldProviderMock) - messages = [{'role': 'user', 'content': chunk} for chunk in ["How ", "are ", "you", "?"]] - response = await client.chat.completions.create(messages, "Hello", max_tokens=1) - self.assertIsInstance(response, ChatCompletion) - self.assertEqual("How ", response.choices[0].message.content) - response = await client.chat.completions.create(messages, "Hello", max_tokens=2) - self.assertIsInstance(response, ChatCompletion) - self.assertEqual("How are ", response.choices[0].message.content) - - async def test_max_stream(self): - client = AsyncClient(provider=YieldProviderMock) - messages = [{'role': 'user', 'content': chunk} for chunk in ["How ", "are ", "you", "?"]] - response = await client.chat.completions.create(messages, "Hello", stream=True) - async for chunk in response: - self.assertIsInstance(chunk, ChatCompletionChunk) - if chunk.choices[0].delta.content is not None: - self.assertIsInstance(chunk.choices[0].delta.content, str) - messages = [{'role': 'user', 'content': chunk} for chunk in ["You ", "You ", "Other", "?"]] - response = await client.chat.completions.create(messages, "Hello", stream=True, max_tokens=2) - response = [chunk async for chunk in response] - self.assertEqual(len(response), 3) - for chunk in response: - if chunk.choices[0].delta.content is not None: - self.assertEqual(chunk.choices[0].delta.content, "You ") - - async def test_stop(self): - client = AsyncClient(provider=YieldProviderMock) - messages = [{'role': 'user', 'content': chunk} for chunk in ["How ", "are ", "you", "?"]] - response = await client.chat.completions.create(messages, "Hello", stop=["and"]) - self.assertIsInstance(response, ChatCompletion) - self.assertEqual("How are you?", response.choices[0].message.content) - -if __name__ == '__main__': - unittest.main() |