From e940724593593c12f20a5a865a65d355a190c2bb Mon Sep 17 00:00:00 2001
From: fly_dust <flydust328@gmail.com>
Date: Sat, 8 Feb 2025 00:29:37 -0800
Subject: [PATCH] fix make_raw_chat_prompt when prefill is disabled

---
 bigcodebench/provider/utility.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/bigcodebench/provider/utility.py b/bigcodebench/provider/utility.py
index 9c5b48f..ad09159 100644
--- a/bigcodebench/provider/utility.py
+++ b/bigcodebench/provider/utility.py
@@ -72,7 +72,7 @@ def make_raw_chat_prompt(
                 [
                     {"role": "user", "content": task_prompt},
                 ],
-                tokenize=False,
+                tokenize=False, add_generation_prompt=True
             ).split(_MAGIC_SPLITTER_)[0]
     return task_prompt