Skip to content

Commit 10a46a7

Browse files
committed
update TestOpenAIServingCompletion for merge
1 parent 7ce6ac8 commit 10a46a7

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

test/entrypoints/openai/test_serving_completion.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@ def test_calc_finish_reason_tool_calls(self):
1111
engine_client = Mock()
1212
engine_client.reasoning_parser = "ernie_x1"
1313
# 创建一个OpenAIServingCompletion实例
14-
serving_completion = OpenAIServingCompletion(engine_client, "pid", "ips")
14+
serving_completion = OpenAIServingCompletion(engine_client, "pid", "ips", 360)
1515
# 创建一个模拟的output,并设置finish_reason为"tool_calls"
1616
output = {"finish_reason": "tool_calls"}
1717
# 调用calc_finish_reason方法
@@ -24,7 +24,7 @@ def test_calc_finish_reason_stop(self):
2424
engine_client = Mock()
2525
engine_client.reasoning_parser = "ernie_x1"
2626
# 创建一个OpenAIServingCompletion实例
27-
serving_completion = OpenAIServingCompletion(engine_client, "pid", "ips")
27+
serving_completion = OpenAIServingCompletion(engine_client, "pid", "ips", 360)
2828
# 创建一个模拟的output,并设置finish_reason为其他值
2929
output = {"finish_reason": "other_reason"}
3030
# 调用calc_finish_reason方法
@@ -36,7 +36,7 @@ def test_calc_finish_reason_length(self):
3636
# 创建一个模拟的engine_client
3737
engine_client = Mock()
3838
# 创建一个OpenAIServingCompletion实例
39-
serving_completion = OpenAIServingCompletion(engine_client, "pid", "ips")
39+
serving_completion = OpenAIServingCompletion(engine_client, "pid", "ips", 360)
4040
# 创建一个模拟的output
4141
output = {}
4242
# 调用calc_finish_reason方法

0 commit comments

Comments
 (0)