bugfix: 修复在对话场景下压测输入长度数据不对的问题 (#835)

* 修复在对话场景下压测数据不对的问题 * 重命名 TotalQuery
baidubce · Oct 15, 2024 · 9c7d006 · 9c7d006
1 parent b9424ba
commit 9c7d006
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/python/qianfan/dataset/stress_test/load_statistics.py b/python/qianfan/dataset/stress_test/load_statistics.py
@@ -153,7 +153,7 @@ def gen_brief(
         + "SendQuery: %s\n" % round(total_count, 2)
         + "SuccessQuery: %s\n" % round(success_count, 2)
         + "FailureQuery: %s\n" % round(failure_count, 2)
-        + "TotalQuery: %s\n" % round(count, 2)
+        + "DataEntryCount: %s\n" % round(count, 2)
         + "TotalTime: %s\n" % round(time, 2)
         + "SuccessRate: %s%%" % success_rate
     )

diff --git a/python/qianfan/dataset/stress_test/qianfan_llm_load.py b/python/qianfan/dataset/stress_test/qianfan_llm_load.py
@@ -542,7 +542,7 @@ def _prepare_request_meta(self, context: Dict, **kwargs: Any) -> Dict:
 
         request_meta: Dict[str, Any] = {
             "response_length": 0,
-            "request_length": sum([len(msg) for msg in messages]),
+            "request_length": sum([len(msg["content"]) for msg in messages]),
         }
 
         return request_meta