From c4d1b5aaf143aab9dca8f295290539f087e6ba46 Mon Sep 17 00:00:00 2001
From: Pierrick HYMBERT <pierrick.hymbert@gmail.com>
Date: Sat, 9 Mar 2024 11:04:27 +0100
Subject: [PATCH] server: bench: fix assistant message sent instead of user
 message

---
 examples/server/bench/script.js | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/examples/server/bench/script.js b/examples/server/bench/script.js
index 510a01cb5..49ed9db8e 100644
--- a/examples/server/bench/script.js
+++ b/examples/server/bench/script.js
@@ -34,10 +34,10 @@ const data = new SharedArray('conversations', function () {
     return JSON.parse(open(dataset_path))
         // Filter out the conversations with less than 2 turns.
         .filter(data => data["conversations"].length >= 2)
-        // Only keep the first two turns of each conversation.
+        .filter(data => data["conversations"][0]["from"] === "human")
         .map(data => {
             return {
-                prompt: data["conversations"][1]["value"],
+                prompt: data["conversations"][0]["value"],
                 n_prompt_tokens: tokenizer(data["conversations"][0]["value"]).length,
                 n_completion_tokens: tokenizer(data["conversations"][1]["value"]).length,
             }