From e78750412dcc079bf3d05b73f58c08803b7d866f Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Thu, 18 Apr 2024 11:33:42 +0000
Subject: refactor: adjust command revision prompt

---
 src/main.rs | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/main.rs b/src/main.rs
index 90e1136..4ab4cb8 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -198,7 +198,10 @@ fn execute(config: &GlobalConfig, mut input: Input) -> Result<()> {
                 }
                 "3" => {
                     let revision = Text::new("Enter your revision:").prompt()?;
-                    let text = format!("INPUT: {}\n{eval_str}\nINPUT: {revision}\n", input.text());
+                    let text = format!(
+                        "[INST] {} [/INST]\n{eval_str}\n[INST] {revision} [/INST]\n",
+                        input.text()
+                    );
                     input.set_text(text);
                     return execute(config, input);
                 }
-- 
cgit v1.2.3