mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-10-10 02:50:19 +08:00
fix TestOpenAIServingCompletion fail (#3368)
This commit is contained in:
@@ -17,10 +17,10 @@ class TestOpenAIServingCompletion(unittest.TestCase):
|
|||||||
engine_client.reasoning_parser = "ernie_x1"
|
engine_client.reasoning_parser = "ernie_x1"
|
||||||
# 创建一个OpenAIServingCompletion实例
|
# 创建一个OpenAIServingCompletion实例
|
||||||
serving_completion = OpenAIServingCompletion(engine_client, "pid", "ips", 360)
|
serving_completion = OpenAIServingCompletion(engine_client, "pid", "ips", 360)
|
||||||
# 创建一个模拟的output,并设置finish_reason为"tool_calls"
|
# 创建一个模拟的output,并设置finish_reason为"tool_call"
|
||||||
output = {"finish_reason": "tool_calls"}
|
output = {"tool_call": "tool_call"}
|
||||||
# 调用calc_finish_reason方法
|
# 调用calc_finish_reason方法
|
||||||
result = serving_completion.calc_finish_reason(None, 100, output)
|
result = serving_completion.calc_finish_reason(None, 100, output, False)
|
||||||
# 断言结果为"tool_calls"
|
# 断言结果为"tool_calls"
|
||||||
assert result == "tool_calls"
|
assert result == "tool_calls"
|
||||||
|
|
||||||
@@ -33,7 +33,7 @@ class TestOpenAIServingCompletion(unittest.TestCase):
|
|||||||
# 创建一个模拟的output,并设置finish_reason为其他值
|
# 创建一个模拟的output,并设置finish_reason为其他值
|
||||||
output = {"finish_reason": "other_reason"}
|
output = {"finish_reason": "other_reason"}
|
||||||
# 调用calc_finish_reason方法
|
# 调用calc_finish_reason方法
|
||||||
result = serving_completion.calc_finish_reason(None, 100, output)
|
result = serving_completion.calc_finish_reason(None, 100, output, False)
|
||||||
# 断言结果为"stop"
|
# 断言结果为"stop"
|
||||||
assert result == "stop"
|
assert result == "stop"
|
||||||
|
|
||||||
@@ -45,7 +45,7 @@ class TestOpenAIServingCompletion(unittest.TestCase):
|
|||||||
# 创建一个模拟的output
|
# 创建一个模拟的output
|
||||||
output = {}
|
output = {}
|
||||||
# 调用calc_finish_reason方法
|
# 调用calc_finish_reason方法
|
||||||
result = serving_completion.calc_finish_reason(100, 100, output)
|
result = serving_completion.calc_finish_reason(100, 100, output, False)
|
||||||
# 断言结果为"length"
|
# 断言结果为"length"
|
||||||
assert result == "length"
|
assert result == "length"
|
||||||
|
|
||||||
@@ -95,6 +95,7 @@ class TestOpenAIServingCompletion(unittest.TestCase):
|
|||||||
model_name=model_name,
|
model_name=model_name,
|
||||||
prompt_batched_token_ids=prompt_batched_token_ids,
|
prompt_batched_token_ids=prompt_batched_token_ids,
|
||||||
completion_batched_token_ids=completion_batched_token_ids,
|
completion_batched_token_ids=completion_batched_token_ids,
|
||||||
|
text_after_process_list=["1", "1"],
|
||||||
)
|
)
|
||||||
|
|
||||||
assert completion_response.id == request_id
|
assert completion_response.id == request_id
|
||||||
|
Reference in New Issue
Block a user