gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

移除性能统计的打印输出

Browse Source 
- 注释掉非流式聊天中的性能统计打印
- 注释掉流式聊天中的性能统计打印
- 保持代码简洁，减少冗余输出
This commit is contained in:
yangdx
2025-01-16 01:16:53 +08:00
parent 350e080ec1
commit 7658f4cbf0
1 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
test_lightrag_ollama_chat.py
View File

@@ -261,15 +261,15 @@ def test_non_stream_chat():
"message": response_json["message"]
}, "响应内容")
# 打印性能统计
print_json_response({
"total_duration": response_json["total_duration"],
"load_duration": response_json["load_duration"],
"prompt_eval_count": response_json["prompt_eval_count"],
"prompt_eval_duration": response_json["prompt_eval_duration"],
"eval_count": response_json["eval_count"],
"eval_duration": response_json["eval_duration"]
}, "性能统计")
# # 打印性能统计
# print_json_response({
# "total_duration": response_json["total_duration"],
# "load_duration": response_json["load_duration"],
# "prompt_eval_count": response_json["prompt_eval_count"],
# "prompt_eval_duration": response_json["prompt_eval_duration"],
# "eval_count": response_json["eval_count"],
# "eval_duration": response_json["eval_duration"]
# }, "性能统计")
def test_stream_chat():
"""测试流式调用 /api/chat 接口
@@ -309,7 +309,7 @@ def test_stream_chat():
data = json.loads(line.decode('utf-8'))
if data.get("done", True): # 如果是完成标记
if "total_duration" in data: # 最终的性能统计消息
print_json_response(data, "性能统计")
# print_json_response(data, "性能统计")
break
else: # 正常的内容消息
message = data.get("message", {})