From c59408d05865e2303e557ad860e15319917f92fc Mon Sep 17 00:00:00 2001
From: Insop <1240382+insop@users.noreply.github.com>
Date: Sun, 8 Jun 2025 18:24:15 -0700
Subject: [PATCH] Fix run_repl to use user code

---
 scripts/kernelllm.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/scripts/kernelllm.py b/scripts/kernelllm.py
index 68e58dc..85fcc6e 100644
--- a/scripts/kernelllm.py
+++ b/scripts/kernelllm.py
@@ -318,7 +318,7 @@ def run_repl(self):
             if not code:
                 print(f"Using default prompt:\n{DEFAULT_MODEL_CODE}\n")
                 code = DEFAULT_MODEL_CODE
-            prompt = PROMPT_TEMPLATE.format(DEFAULT_MODEL_CODE)
+            prompt = PROMPT_TEMPLATE.format(code)
 
             try:
                 self.stream_raw(prompt)