Skip to content

Commit 31d639d

Browse files
committed
fix unit test
1 parent dd30110 commit 31d639d

File tree

2 files changed

+3
-0
lines changed

2 files changed

+3
-0
lines changed

tests/input/test_ernie_processor.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@ def setUp(self):
2020
self.processor.generation_config = MagicMock()
2121
self.processor.eos_token_ids = [1]
2222
self.processor.reasoning_parser = None
23+
self.processor.model_status_dict = {}
2324

2425
# 模拟 ids2tokens 方法
2526
def mock_ids2tokens(token_ids, task_id):
@@ -66,6 +67,7 @@ def test_process_response_dict_streaming_normal_case(self):
6667

6768
def test_process_request_dict(self):
6869
request_dict = {
70+
"request_id": "123",
6971
"messages": [{"role": "user", "content": "Hello!"}],
7072
"chat_template_kwargs": {"chat_template": "Hello!"},
7173
"eos_token_ids": [1],

tests/input/test_text_processor.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@ def setUp(self):
2020
self.processor.tool_parser_dict = {}
2121
self.processor.generation_config = MagicMock()
2222
self.processor.eos_token_ids = [1]
23+
self.processor.reasoning_parser = None
2324

2425
def mock_messages2ids(request, **kwargs):
2526
if "chat_template" in kwargs:

0 commit comments

Comments
 (0)