Skip to content

Commit 0c171f5

Browse files
author
Olivier Chafik
committed
Update test_chat_completion.py
1 parent 06c4ca5 commit 0c171f5

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

examples/server/tests/unit/test_chat_completion.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -13,9 +13,9 @@ def create_server():
1313
@pytest.mark.parametrize(
1414
"model,system_prompt,user_prompt,max_tokens,re_content,n_prompt,n_predicted,finish_reason,jinja,chat_template",
1515
[
16-
(None, "Book", "What is the best book", 8, "\\{ \" Sarax.", 77, 8, "length", False, None),
17-
(None, "Book", "What is the best book", 8, "\\{ \" Sarax.", 77, 8, "length", True, None),
18-
(None, "Book", "What is the best book", 8, "I want to play with", 23, 8, "length", True, "This is not a chat template, it is"),
16+
(None, "Book", "What is the best book", 8, "(Suddenly)+|\\{ \" Sarax.", 77, 8, "length", False, None),
17+
(None, "Book", "What is the best book", 8, "(Suddenly)+|\\{ \" Sarax.", 77, 8, "length", True, None),
18+
(None, "Book", "What is the best book", 8, "^ blue|I want to play with", 23, 8, "length", True, "This is not a chat template, it is"),
1919
("codellama70b", "You are a coding assistant.", "Write the fibonacci function in c++.", 128, "(Aside|she|felter|alonger)+", 104, 64, "length", False, None),
2020
("codellama70b", "You are a coding assistant.", "Write the fibonacci function in c++.", 128, "(Aside|she|felter|alonger)+", 104, 64, "length", True, None),
2121
]

0 commit comments

Comments
 (0)