From 4363025bde4593974fac4d51b23588d025e82b7e Mon Sep 17 00:00:00 2001
From: ZHANG Hao <zhanghao@4paradigm.com>
Date: Wed, 3 Sep 2025 12:06:44 +0800
Subject: [PATCH] add avg latency

---
 main.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/main.py b/main.py
index a5b06c9..b40f15a 100644
--- a/main.py
+++ b/main.py
@@ -74,6 +74,7 @@ def main():
 
     # 两两相似度（余弦）——已归一化则点积即余弦
     similarity = embeddings @ embeddings.T
+    avg_latency = encode_time / len(sentences) if sentences else 0
 
     # 组织输出
     result = {
@@ -81,7 +82,8 @@ def main():
         "device": device,
         "count": len(sentences),
         "dim": int(embeddings.shape[1]) if len(embeddings.shape) == 2 else None,
-        "encode_time_seconds": round(float(encode_time), 6),
+        "total_elapsed_seconds": round(float(encode_time), 6),
+        "avg_latency": avg_latency,
         "sentences": sentences,
         "embeddings": embeddings.tolist(),          # [N, D]
         "similarity": similarity.tolist()           # [N, N]