github/gpt4all
1
0
Fork 0
mirror of https://github.com/nomic-ai/gpt4all.git synced 2025-09-10 21:09:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

Remove binary state from high-level API and use Jinja templates (#3147)

Browse Source 
Signed-off-by: Jared Van Bortel <jared@nomic.ai>
Signed-off-by: Adam Treat <treat.adam@gmail.com>
Co-authored-by: Adam Treat <treat.adam@gmail.com>
This commit is contained in:
Jared Van Bortel
2024-11-25 10:04:17 -05:00
committed by GitHub
parent 3320094d29
commit 225bf6be93
54 changed files with 3423 additions and 2224 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
gpt4all-chat/tests/python/test_server_api.py
View File

@@ -203,11 +203,10 @@ EXPECTED_MODEL_INFO = {
EXPECTED_COMPLETIONS_RESPONSE = {
'choices': [
{
'finish_reason': 'stop',
'finish_reason': 'length',
'index': 0,
'logprobs': None,
'references': None,
'text': ' jumps over the lazy dog.',
'text': ' jumps over the lazy dog.\n',
},
],
'id': 'placeholder',
@@ -242,18 +241,14 @@ def test_with_models(chat_server_with_model: None) -> None:
'type': 'invalid_request_error',
}}
data = {
'model': 'Llama 3.2 1B Instruct',
'prompt': 'The quick brown fox',
'temperature': 0,
}
data = dict(
model = 'Llama 3.2 1B Instruct',
prompt = 'The quick brown fox',
temperature = 0,
max_tokens = 6,
)
response = request.post('completions', data=data)
assert len(response['choices']) == 1
assert response['choices'][0].keys() == {'text', 'index', 'logprobs', 'references', 'finish_reason'}
assert response['choices'][0]['text'] == ' jumps over the lazy dog.'
assert 'created' in response
response.pop('created') # Remove the dynamic field for comparison
del response['created'] # Remove the dynamic field for comparison
assert response == EXPECTED_COMPLETIONS_RESPONSE