Skip to content

Commit d8b3000

Browse files
committed
优化单测代码2,有针对性地缩小单测范围。
1 parent 23bd28f commit d8b3000

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

fastdeploy/entrypoints/openai/serving_completion.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -236,7 +236,7 @@ async def completion_full_generator(
236236

237237
async def _echo_back_prompt(self, request, res, idx):
238238
# 如果是第一个响应片段,拼接prompt到text前
239-
if res["outputs"].get("send_idx", -1) == 0 and request.prompt is not None and request.echo:
239+
if res["outputs"].get("send_idx", -1) == 0 and request.echo:
240240
if isinstance(request.prompt, list):
241241
prompt_text = request.prompt[idx]
242242
else:

test/entrypoints/openai/test_completion_echo.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -72,6 +72,7 @@ def mock_echo_side_effect(req, res, idx):
7272

7373
# 验证结果
7474
self.assertEqual(mock_response["outputs"]["text"], "test prompttest output")
75+
self.assertEqual(request.prompt, "test prompt")
7576

7677
def test_multi_prompt_non_streaming(self):
7778
"""测试多prompt非流式响应"""
@@ -142,8 +143,8 @@ def mock_echo_side_effect(req, res, idx):
142143
# 验证结果
143144
self.assertEqual(mock_responses[0]["outputs"]["text"], "prompt1 response1")
144145
self.assertEqual(mock_responses[1]["outputs"]["text"], "prompt2 response2")
146+
self.assertEqual(request.prompt, ["prompt1", "prompt2"])
145147

146148

147149
if __name__ == "__main__":
148150
unittest.main()
149-
# self.assertEqual(results[4], "data: [DONE]\n\n")

0 commit comments

Comments
 (0)