diff --git a/examples/benchmark_ark_text.py b/examples/benchmark_ark_text.py index 6391071..681945c 100644 --- a/examples/benchmark_ark_text.py +++ b/examples/benchmark_ark_text.py @@ -71,9 +71,11 @@ async def main(): times = 100 text_latency = [] for i in range(times): - logid, text, latency = test_latency(ep, token, text) + logid, first_text, latency = test_latency(ep, token, text) text_latency.append(latency) - print(f"[latency.ark.text] {i}, latency: {cal_latency(latency, text_latency)}, log: {logid}, text: {text}") + print( + f"[latency.ark.text] {i}, latency: {cal_latency(latency, text_latency)}, log: {logid}, text: {first_text}" + ) if __name__ == "__main__": diff --git a/examples/benchmark_text_chat.py b/examples/benchmark_text_chat.py index 401ebd2..de846cf 100644 --- a/examples/benchmark_text_chat.py +++ b/examples/benchmark_text_chat.py @@ -115,9 +115,9 @@ async def main(): times = 100 text_latency = [] for i in range(times): - logid, text, latency = await test_latency(coze, bot_id, text) + logid, first_text, latency = await test_latency(coze, bot_id, text) text_latency.append(latency) - print(f"[latency.text] {i}, latency: {cal_latency(latency, text_latency)}, log: {logid}, text: {text}") + print(f"[latency.text] {i}, latency: {cal_latency(latency, text_latency)}, log: {logid}, text: {first_text}") if __name__ == "__main__":